73 lines
2.1 KiB
Bash
73 lines
2.1 KiB
Bash
#!/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_yaml_array() {
|
|
local -n arr="${1}"
|
|
local jq_query="${2}"
|
|
local module_config="${3}"
|
|
|
|
if [[ -z "${jq_query}" || -z "${module_config}" ]]; then
|
|
echo "Usage: get_yaml_array VARIABLE_TO_STORE_RESULTS JQ_QUERY MODULE_CONFIG" >&2
|
|
return 1
|
|
fi
|
|
|
|
readarray -t arr < <(echo "${module_config}" | yq -I=0 "${jq_query}")
|
|
}
|
|
|
|
# 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
|
|
}
|
|
|
|
# Parse OS version and export it
|
|
export OS_VERSION="$(grep -Po "(?<=VERSION_ID=)\d+" /usr/lib/os-release)"
|
|
export OS_ARCH="$(uname -m)"
|
|
|
|
# Export functions for use in sub-shells or sourced scripts
|
|
export -f get_yaml_array
|
|
export -f get_json_array
|
|
|
|
mkdir -p /var/roothome /var/opt /var/lib/alternatives /var/opt /var/usrlocal
|