zsh: clean/improve brethil theme

themes/brethil.zsh-theme

@@ -1,58 +1,40 @@
 # vim:ft=zsh ts=2 sw=2 sts=2
-# color chart available at https://www.calmar.ws/vim/256-xterm-24bit-rgb-color-chart.html
+# brethil oh-my-zsh theme
-# %F{202}: orange
+# See "EXPANSION OF PROMPT SEQUENCES" in `man zshmisc`
-# %K{202}: background orange
+# Note: Most of the expansions/color sequences can be tested using `print -P`
-# 255: white
-# See the prompt expansion section in `info zsh` for more info
 
 function user_prompt(){
-  # prints info about username/hostname
+  # prints hostname on an orange background if on an ssh connection, adds username if root
-  # for ssh connections
+
-  local p
+  local user='%(!.%K{160}%F{255}%B%n%b%k%f.)' # prints 'root' on a red background if user is root
-  local host
+  local host='%K{202}%B%F{255}%m%f%k%b' # bold host name on an orange background
-  if [[ $UID -eq 0 ]]; then
+  local open_bracket='%(!.[.)'
-      # root
+  local close_bracket='%(!.].)'
-      echo -n "%F{255}[%f" # open square bracket
+  local at='%(!.@.)'
-      echo -n "%K{160}%F{255}%Broot%b%k%f"
+  if [[ -n $SSH_CONNECTION ]]; then
-      if [[ $SSH_CONNECTION ]]; then
+    echo "${open_bracket}${user}${at}${host}${close_bracket}"
-        # add the white @
+  else
-        # echo -n '%F{255} at %f'
+    echo "${open_bracket}${user}${close_bracket}"
-        echo -n "%F{255}@%f"
-      fi
   fi
-  if [[ $SSH_CONNECTION ]]; then
-    if [[ $UID -ne 0 ]]; then
-      echo -n "%F{255}[%f" # open square bracket
-    fi
-    # underlined orange hostname (%m)
-    echo -n "%K{202}%B%F{255}%m%f%k%b"
-    if [[ $UID -ne 0 ]]; then
-      echo -n "%F{255}]%f" # close square bracket
-    fi
-  fi
-  # close square bracket
-  if [[ $UID -eq 0 ]]; then echo "%F{255}]%f"; fi
 }
 
 function job_prompt() {
-  local JOB_PROMPT="%(1j.%B%K{202}%F{220} %j %k%b.)" # Shows jobs number on an orange background if there are background jobs
+  # Shows jobs number on an orange background if there are background jobs
-  echo "$JOB_PROMPT"
+  echo "%(1j.%B%K{202}%F{220} %j %k%b.)"
 }
 
 function shlvl_prompt() {
-  local SHLVL_PROMPT
+  # Shows SHLVL on a magenta background if SHLVL > 1 (2 if in a tmux session)
-  if [[ -n "$TMUX" ]]; then
+  if [[ -z "$TMUX" ]]; then
-    SHLVL_PROMPT="%(3L.%K{161}%F{255}%B $((SHLVL-1)) %f%b%k.)" # Shows SHLVL on a magenta background if SHLVL > 1
+    echo "%(2L.%K{161}%F{255}%B %L %f%b%k.)"
   else
-    SHLVL_PROMPT="%(2L.%K{161}%F{255}%B %L %f%b%k.)" # Shows SHLVL on a magenta background if SHLVL > 1
+    echo "%(3L.%K{161}%F{255}%B $((SHLVL-1)) %f%b%k.)"
   fi
-  echo "$SHLVL_PROMPT"
 }
 
 function path_prompt(){
   # Shows last 3 items in path if there are more than 4
-  local PATH_PROMPT="%F{white}[%F{green}%(4~:…/%3~:%~)%F{white}]%f"
+  echo "[%F{green}%(4~:…/%3~:%~)%f]"
-  echo "$PATH_PROMPT"
 }
 
 # PREFIX/SUFFIX are added before/after `git_prompt_info`
@@ -75,9 +57,9 @@ ZSH_THEME_GIT_PROMPT_UNTRACKED="%B%F{red}?%b%f" # red
 
 # The following are part of `git_remote_status`
 # ZSH_THEME_GIT_PROMPT_EQUAL_REMOTE="%{$fg[green]%}="
-ZSH_THEME_GIT_PROMPT_AHEAD_REMOTE="%F{255}>%f"
+ZSH_THEME_GIT_PROMPT_AHEAD_REMOTE=">"
-ZSH_THEME_GIT_PROMPT_BEHIND_REMOTE="%F{255}<%f"
+ZSH_THEME_GIT_PROMPT_BEHIND_REMOTE="<"
-ZSH_THEME_GIT_PROMPT_DIVERGED_REMOTE="%F{255}<>%f"
+ZSH_THEME_GIT_PROMPT_DIVERGED_REMOTE="<>"
 
 function virtualenv_info {
     [ -z "$VIRTUAL_ENV" ] && return
@@ -90,70 +72,42 @@ function virtualenv_info {
       venv="$(basename $(dirname "$VIRTUAL_ENV") )"
     fi
 
-    echo -n "$BOLD_PURPLE"
+    echo -n " $BOLD_PURPLE<$venv>$CLEAR "
-    echo -n " <$venv> "
-    echo -n "$CLEAR"
 }
 
 function git_prompt() {
-    if [[ -n "$(git status 2>/dev/null)" ]]; then
+    # prints git info if in a git repo
-        local GIT_PROMPT=" %F{255}$(git_prompt_info)%B%F{255} |%f%b$(git_prompt_status)$(git_remote_status)"
+    if ! git rev-parse --git-dir &>/dev/null; then
-        echo "$GIT_PROMPT"
+      return
-    else
-        echo ""
     fi
+
+    echo " $(git_prompt_info) |$(git_prompt_status)$(git_remote_status)"
 }
 
-function check_last_exit_code() {
-  local LAST_EXIT_CODE=$?
-  if [[ $LAST_EXIT_CODE -ne 0 ]]; then
-    local EXIT_CODE_PROMPT
-    EXIT_CODE_PROMPT+="%B%F{88}-$LAST_EXIT_CODE-%f%b"
-    echo "$EXIT_CODE_PROMPT"
-  fi
-}
 
 function return_code_RPS1() {
-  # For right-side prompt
+  # red status code if last exit code is non-zero
-  local return_code="%(?..$(check_last_exit_code))" # for RPS1
+  echo "%(?..%B%F{88}-$?-%f%b)"
-  echo "$return_code"
 }
 
-function ret_status() {
+function prompt_with_previous_return_status() {
-  local ret_status="%(?:%B%F{28} →%f%b:%{$fg[red]%} x%{$reset_color%})"
+  local success="%B%F{28} →%f%b"
-  echo "$ret_status"
+  local failure="%B%{$fg[red]%} x%b%{$reset_color%}"
+  echo "%(?:$success:$failure)"
 }
 
-export MODE_INDICATOR="[%B%K{88}nav%k%b]"
-export INSERT_MODE_INDICATOR="[%B%K{28}ins%k%b]"
-
-_PROMPT_PROTO='$(virtualenv_info)$(user_prompt)$(path_prompt)$(git_prompt)$(job_prompt)$(shlvl_prompt)'
 
 function prompt_too_long(){
-  local zero='%([BSUbfksu]|([FK]|){*})'
+  # prints a newline if more than $COLUMNS/2 characters have been printed
-  if [[ -x $commands[gsed] ]]; then # fuck darwin
+  local newline=$'\n'
-    sed='gsed'
+  echo "%-$((COLUMNS/2))(l..${newline})"
-  elif [[ "$(uname)" == "Darwin" ]]; then
-    local msg="error: multi-line prompt requires GNU sed"
-    msg+="install with \"brew install gnu-sed\""
-    echo "$msg"
-    return
-  else
-    sed='sed'
-  fi
-
-  local stripped="$(echo ${(S%%)_PROMPT_PROTO//$~zero/} | $sed 's|\x1B\[[0-9;]*[a-zA-Z]||g')"
-  local prompt_len=${#stripped}
-  local max_len=$(($COLUMNS/2))
-  if [[ $prompt_len -ge $max_len ]]; then
-    echo true
-  fi
 }
 
-PROMPT="$(echo "$_PROMPT_PROTO")"
+## oh-my-zsh vi-mode plugins indicators:
-NEWLINE=$'\n'
+export MODE_INDICATOR="[%B%K{red}nav%k%b]" # red background
-PROMPT+='${$(prompt_too_long)/true/${NEWLINE}}'
+export INSERT_MODE_INDICATOR="[%B%K{28}ins%k%b]" # green background
-PROMPT+='$(ret_status) '
+
+PROMPT='$(virtualenv_info)$(user_prompt)$(path_prompt)$(git_prompt)$(job_prompt)$(shlvl_prompt)$(prompt_too_long)$(prompt_with_previous_return_status) '
 
 # Right prompt is just return code and time
-RPS1='$(return_code_RPS1)$(vi_mode_prompt_info)%F{255}[%*]%f'
+RPS1='$(return_code_RPS1)$(vi_mode_prompt_info)[%*]'