#!/usr/bin/env bash

# Function to retrieve module configs and populate an array
# Arguments:
#   1. Variable name to store result
#   2. jq query
#   3. Module config content
get_json_array() {
  local -n arr="${1}"
  local jq_query="${2}"
  local module_config="${3}"

  if [[ -z "${jq_query}" || -z "${module_config}" ]]; then
    echo "Usage: get_json_array VARIABLE_TO_STORE_RESULTS JQ_QUERY MODULE_CONFIG" >&2
    return 1
  fi
  
  readarray -t arr < <(echo "${module_config}" | jq -c -r "${jq_query}")
}

color_string() {
  local string="${1}"
  local color_code="${2}"
  local reset_code="\033[0m"

  # ANSI color codes: https://en.wikipedia.org/wiki/ANSI_escape_code#Colors
  # Example color codes: 31=red, 32=green, 33=yellow, 34=blue, 35=magenta, 36=cyan, 37=white

  # Check if color code is provided, otherwise default to white (37)
  if [[ -z "${color_code}" ]]; then
    color_code="37"
  fi

  # Determine if we should force color
  if [ -n "${FORCE_COLOR:-}" ] || [ -n "${CLICOLOR_FORCE:-}" ]; then
    # Force color: Apply color codes regardless of whether output is a TTY
    echo -e "\033[${color_code}m${string}${reset_code}"
  elif [ -t 1 ]; then
    # Output is a TTY and color is not forced: Apply color codes
    echo -e "\033[${color_code}m${string}${reset_code}"
  else
    # Output is not a TTY: Do not apply color codes
    echo "${string}"
  fi
}

feature_enabled() {
    # Ensure the function is called with exactly one argument
    if [ "$#" -ne 1 ]; then
        echo "Usage: feature_enabled <feature_name>" >&2
        return 1
    fi

    local feature="$1"
    local -a features

    # Split BB_BUILD_FEATURES by commas and read into an array
    IFS=,
    read -r -a features <<< "$BB_BUILD_FEATURES"

    # Loop through the array and check for a match
    for f in "${features[@]}"; do
        # Trim leading and trailing whitespace
        local trimmed_f="${f## }"
        trimmed_f="${trimmed_f%% }"

        if [[ "$trimmed_f" == "$feature" ]]; then
            return 0
        fi
    done

    # Feature not found
    return 1
}

# Parse OS version and export it
export OS_VERSION="$(awk -F= '/^VERSION_ID=/ {gsub(/"/, "", $2); print $2}' /usr/lib/os-release)"
case "$TARGETARCH" in
  "amd64")
    OS_ARCH="x86_64"
    ;;
  "arm64")
    OS_ARCH="aarch64"
    ;;
  *)
    OS_ARCH="$TARGETARCH"
    ;;
esac
export OS_ARCH

# Export functions for use in sub-shells or sourced scripts
export -f get_json_array

mkdir -p /var/roothome /var/opt /var/lib/alternatives /var/opt /var/usrlocal