Make host based color a generic function.

Author: crasx

functions/geometry_status

@@ -6,24 +6,9 @@ geometry_status() {
 
   local color=${GEOMETRY_STATUS_COLOR:-default}
 
-  if ( $error ); then color=${GEOMETRY_STATUS_COLOR_ERROR:-red}; else
-    ( ${GEOMETRY_STATUS_SYMBOL_COLOR_HASH:=false} ) && {
-      local colors=({1..9})
-
-      (($(echotc Co) == 256)) && colors+=({17..230})
-
-      if (( ${+GEOMETRY_STATUS_SYMBOL_COLOR_HASH_COLORS} )); then
-        colors=(${GEOMETRY_STATUS_SYMBOL_COLOR_HASH_COLORS})
-      fi
-
-      local sum=0; for c in ${(s::)^HOST}; do ((sum += $(print -f '%d' "'$c"))); done
-
-      local index="$(($sum % ${#colors}))"
-
-      [[ "$index" -eq 0 ]] && index=1
-
-      color=${colors[${index}]}
-    }
+  if ( $error );
+  then
+    color=${GEOMETRY_STATUS_COLOR_ERROR:-red};
   fi
 
   ( $root && $error ) && ansi $color ${GEOMETRY_STATUS_SYMBOL_ROOT_ERROR:=▽} && return

geometry.zsh

@@ -37,6 +37,29 @@ geometry::time() {
   ${2:-false} && ansi $color ${(j: :)human} || ansi $color $human[1]
 }
 
+# Generate a color based on hostname.
+geometry::hostcolor() {
+
+      if (( ${+GEOMETRY_HOST_COLOR} )); then
+        echo ${GEOMETRY_HOST_COLOR}
+        return
+      fi
+
+      if (( ${+GEOMETRY_HOST_COLORS} )); then
+        local colors=(${GEOMETRY_HOST_COLORS})
+      else
+        local colors=({1..9})
+        (($(echotc Co) == 256)) && colors+=({17..230})
+      fi
+
+      local sum=0; for c in ${(s::)^HOST}; do ((sum += $(print -f '%d' "'$c"))); done
+      local index="$(($sum % ${#colors}))"
+
+      [[ "$index" -eq 0 ]] && index = 1
+
+      echo ${colors[${index}]}
+}
+
 # set title to COMMAND @ CURRENT_DIRECTORY
 geometry::set_title() { print -n "\e]0;${2} @ ${PWD##*/}\a"; }
 add-zsh-hook preexec geometry::set_title

images/screenshots/colorize.png

@@ -84,11 +84,8 @@ GEOMETRY_STATUS_SYMBOL_ERROR="△"       # displayed when exit value is != 0
 GEOMETRY_STATUS_COLOR_ERROR="magenta"  # prompt symbol color when exit value is != 0
 GEOMETRY_STATUS_COLOR="default"        # prompt symbol color
 GEOMETRY_STATUS_COLOR_ROOT="red"       # root prompt symbol color
-GEOMETRY_STATUS_COLOR_HASH=true        # color status symbol based on hostname
 ```
 
-![colorize](/images/screenshots/colorize.png)
-
 ### geometry_exitcode
 This renders the exit code of the previous function if it is not success.
 
@@ -113,6 +110,19 @@ GEOMETRY_GIT_TIME_DETAILED=true     # show full time (e.g. `12h 30m 53s`) instea
 
 ![git_conflicts](/images/screenshots/git_conflicts.png)
 
+## Hostname based colors
+Geometry provides a generic function (`geometry::hostcolor`) for setting any color based on hostname. The color is calculated as the sum of each character converted to an integer. For example the hostname 'abc' generates a value of 294.
+
+By default the colors 1-9 and 17-230 are used as colors depending on what the environment supports. The calculated value is then modded by the size of the color list to choose a value. This value will be consistent given the same hostname.
+
+These colors can be overridden by setting variables.
+
+```shell
+GEOMETRY_HOST_COLORS=({1..7})  # Only use the colors 1-7
+GEOMETRY_HOST_COLOR=4          # Override the color for a specific host. 
+``` 
+![colorize](/images/screenshots/colorize.png)
+
 ## Thanks
 
 geometry is maintained by [frm](https://github.com/frm), [desyncr](https://github.com/desyncr) and [jedahan](https://github.com/jedahan).