From 54c334962c0bf09fb6c75649fb9043b66681a855 Mon Sep 17 00:00:00 2001
From: Maksym Bilyk <maxym.bilyk@gmail.com>
Date: Fri, 28 Jul 2023 17:16:39 +0300
Subject: [PATCH] feature_58: Add support for detecting the stdout (#63)

Resolve whether decoration (color) is needed based on flags --color_{auto,never,always} (auto is default).
If --color_auto is passed, text decoration will ocurr only to the stdout.
If --color_{never,always} is passed, will or will not occur in any case.
---
 has | 41 +++++++++++++++++++++++++++++++++++++----
 1 file changed, 37 insertions(+), 4 deletions(-)

diff --git a/has b/has
index 9301cfe..3ec98e7 100755
--- a/has
+++ b/has
@@ -34,6 +34,14 @@ readonly checkmark='\342\234\223'
 readonly PASS="${txtbold}${txtgreen}${checkmark}${txtreset}"
 readonly FAIL="${txtbold}${txtred}${fancyx}${txtreset}"
 
+## These variables are used to control decoration of output
+COLOR_AUTO="auto"
+COLOR_NEVER="never"
+COLOR_ALWAYS="always"
+COLOR_OPTS=(${COLOR_AUTO} ${COLOR_NEVER} ${COLOR_ALWAYS})
+COLOR="${COLOR_AUTO}"
+COLOR_PREFIX="--color"
+
 ## These variables are used to keep track of passed and failed commands
 OK=0
 KO=0
@@ -62,6 +70,15 @@ _version() {
   printf '%s\n' "${VERSION}"
 }
 
+# function to parse color option
+_set_color() {
+  local found=0;
+  for opt in "${COLOR_OPTS[@]}"; do
+    [ "${1}" == "${COLOR_PREFIX}_${opt}" ] && COLOR="${opt}" && found=1 && break
+  done
+  [ ${found} -eq 1 ] || >&2 echo "Error: wrong flag ${1}"
+}
+
 # try to extract version by executing "${1}" with "${2}" arg
 # command name to be executed is dynamically passed to this function as ${1}
 # arg to ${1} is passed in ${2}
@@ -101,6 +118,12 @@ __dynamic_detect-arg_version(){
 __detect(){
   name="${1}"
 
+  # default values (for case when decoration applied)
+  fail=${FAIL}
+  pass=${PASS}
+  wrapper_beg="${txtbold}${txtyellow}"
+  wrapper_end="${txtreset}"
+
   # setup aliases - maps commonly used name to exact command name
   case ${name} in
     rust                  ) command="rustc"         ;;
@@ -276,20 +299,27 @@ __detect(){
       ;;
   esac
 
+  if [ "${COLOR}" == "${COLOR_NEVER}" ] || [[ ! -t 1 && "${COLOR}" == "${COLOR_AUTO}" ]]; then
+    pass=${checkmark}
+    fail=${fancyx}
+    wrapper_beg=""
+    wrapper_end=""
+  fi
+
   if [ "$status" -eq "-1" ]; then     ## When unsafe processing is not allowed, the -1 signifies
-    printf '%b %s not understood\n' "${FAIL}" "${command}" > "$OUTPUT"
+    printf '%b %s not understood\n' "${fail}" "${command}" > "$OUTPUT"
     KO=$(( KO+1 ))
 
   elif [ ${status} -eq 127 ]; then    ## command not installed
-    printf '%b %s\n' "${FAIL}" "${command}" > "$OUTPUT"
+    printf '%b %s\n' "${fail}" "${command}" > "$OUTPUT"
     KO=$(( KO+1 ))
 
   elif [ ${status} -eq 0 ] || [ ${status} -eq 141 ]; then      ## successfully executed
-    printf "%b %s %b\n" "${PASS}" "${command}" "${txtbold}${txtyellow}${version}${txtreset}" > "$OUTPUT"
+    printf "%b %s %b\n" "${pass}" "${command}" "${wrapper_beg}${version}${wrapper_end}" > "$OUTPUT"
     OK=$(( OK+1 ))
 
   else  ## as long as its not 127, command is there, but we might not have been able to extract version
-    printf '%b %s\n' "${PASS}" "${command}" > "$OUTPUT"
+    printf '%b %s\n' "${pass}" "${command}" > "$OUTPUT"
     OK=$(( OK+1 ))
   fi
 } #end __detect
@@ -325,6 +355,9 @@ while getopts ":qhV-" OPTION; do
       _usage
       exit 0
       ;;
+    ${COLOR_PREFIX}*)
+      _set_color "${OPTION}"
+      ;;
     *)
       printf '%s: unrecognized option '%s'\n' "${BINARY_NAME}" "${OPTARG}"
       _usage