Colorize master and main branches

prompt.sh

@@ -6,15 +6,26 @@ PROMPT_HOST_COLOR="255;255;255"
 PROMPT_DIR_COLOR="100;149;237"
 PROMPT_VPN_NAME=""
 PROMPT_VPN_COLOR="255;255;255"
+PROMPT_HIGHLIGHTED_BRANCH_COLOR="255;84;84"
 
 _prompt_dir="${BASH_SOURCE[0]%/*}"
 [[ -f "$_prompt_dir/.env" ]] && . "$_prompt_dir/.env"
 
 __prompt_git() {
+  local color
+  local color_reset
   local branch
+  color=""
+  color_reset=""
   branch=$(git branch --show-current 2>/dev/null) || return
   [[ -z $branch ]] && return
-  echo -n " (${branch})"
+
+  if [[ $branch =~ main|master ]]&& [[ -n $PROMPT_HIGHLIGHTED_BRANCH_COLOR ]]; then
+    color="\[\e[38;2;${PROMPT_HIGHLIGHTED_BRANCH_COLOR}m\]"
+    color_reset="\[\e[0m\]"
+  fi
+
+  echo -n " (${color}${branch}${color_reset})"
 }
 
 __prompt_vpn() {