feat: Color output in terminal if running in TTY

scripts/exports.sh

@@ -18,6 +18,32 @@ get_yaml_array() {
   readarray -t arr < <(echo "$module_config" | yq -I=0 "$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)
 

scripts/run_module.sh

@@ -6,7 +6,7 @@ source /tmp/scripts/exports.sh
 
 # Function to print a centered text banner within a specified width
 print_banner() {
-  local term_width=120
+  local term_width=80
 
   local text=" ${1} "        # Text to print
   local padding="$(printf '%0.1s' '='{1..600})"
@@ -23,7 +23,17 @@ module="$1"
 params="$2"
 script_path="/tmp/modules/${module}/${module}.sh"
 
-print_banner "Start '${module}' Module"
+color_string "$(print_banner "Start '${module}' Module")" "33"
 chmod +x ${script_path}
+
+set +e
 ${script_path} "${params}"
-print_banner "End '${module}' Module"
+RETVAL=$?
+set -e
+
+if [ $RETVAL ]; then
+  color_string "$(print_banner  "End '${module}' Module")" "32"
+else
+  color_string "$(print_banner "Failed '${module}' Module")" "31"
+  exit 1
+fi