dotfiles/zsh/prompt_hyacinthe_setup

# hyacinthe prompt theme with git support
# Derivated from <http://sebastiancelis.com/2009/11/16/zsh-prompt-git-users/>

prompt_hyacinthe_help() {
  cat <<'EOF'

  This prompt is color-scheme-able.  You can invoke it thus:

  prompt hyacinthe [<color1> [<color2> [<color3> [<color4> [<color5>]]]]]

  The defaults colors are red, cyan, green, white, and yellow, respectively.

EOF
}

prompt_hyacinthe_setup() {

  local p_date p_userpwd p_end p_win
  local -A pc

  setopt prompt_subst

  pcc[1]=${1:-${${SSH_CLIENT+'yellow'}:-'red'}}
  pcc[2]=${2:-'cyan'}

  if [[ $USERNAME == "root" ]]; then
    pcc[3]=${3:-'red'}
  else
    pcc[3]=${3:-'green'}
  fi

  pcc[4]=${4:-'white'}
  pcc[5]=${5:-'yellow'}

  pc['\[']="%F{$pcc[1]}["
  pc['\]']="%F{$pcc[1]}]"
  pc['<']="%F{$pcc[1]}<"
  pc['>']="%F{$pcc[1]}>"
  pc['\(']="%F{$pcc[1]}("
  pc['\)']="%F{$pcc[1]})"

  p_date="$pc['\[']%F{$pcc[2]}%D{%R}$pc['\]']"

  [[ -n $WINDOW ]] && p_win="$pc['\(']%F{$pcc[4]}$WINDOW$pc['\)']"

  p_userpwd="$pc['<']%F{$pcc[3]}%n@%m$p_win%F{$pcc[5]}:%F{$pcc[4]}%~$pc['>']"

  p_end="%f%B%#%b "

  zle_highlight[(r)default:*]=default:$pcc[2]
  prompt="$p_date $p_userpwd $p_end"
  PS2='%(4_.\.)%3_> %E'

  add-zsh-hook precmd update_git_vars
  add-zsh-hook preexec preexec_update_git_vars
  add-zsh-hook chpwd update_git_vars
}

update_current_git_vars() {
  unset __CURRENT_GIT_BRANCH
  unset __CURRENT_GIT_BRANCH_STATUS
  unset __CURRENT_GIT_BRANCH_IS_DIRTY

  local st="$(git status 2>/dev/null)"
  if [[ -n $st ]]; then
    local -a arr
    arr=(${(f)st})

    if [[ $arr[1] =~ 'Not currently on any branch.' ]]; then
      __CURRENT_GIT_BRANCH='no-branch'
    else
      __CURRENT_GIT_BRANCH="${arr[1][(w)4]}";
    fi

    if [[ $arr[2] =~ 'Your branch is' ]]; then
      if [[ $arr[2] =~ 'ahead' ]]; then
        __CURRENT_GIT_BRANCH_STATUS='ahead'
      elif [[ $arr[2] =~ 'diverged' ]]; then
        __CURRENT_GIT_BRANCH_STATUS='diverged'
      elif [[ $arr[2] =~ 'up-to-date' || $arr[2] =~ 'up to date' ]]; then
        __CURRENT_GIT_BRANCH_STATUS=''
      else
        __CURRENT_GIT_BRANCH_STATUS='behind'
      fi
    fi

    if [[ ! $st =~ 'nothing to commit' ]]; then
      __CURRENT_GIT_BRANCH_IS_DIRTY='1'
    fi
  fi
}

prompt_git_info() {
  if [ -n "$__CURRENT_GIT_BRANCH" ]; then
    local s="("
    case "$__CURRENT_GIT_BRANCH_STATUS" in
    ahead)
      s+="↑"
      ;;
    diverged)
      s+="↕"
      ;;
    behind)
      s+="↓"
      ;;
    esac
    if [ -n "$__CURRENT_GIT_BRANCH_IS_DIRTY" ]; then
      s+="✗"
    fi
    s+="$__CURRENT_GIT_BRANCH"
    s+=")"

    printf " %s%s" "%{${fg[yellow]}%}" $s
  fi
}

preexec_update_git_vars() {
  case "$1" in
  git*)
    __EXECUTED_GIT_COMMAND=1
    ;;
  esac
}

update_git_vars() {
  update_current_git_vars

  if [ "$__CURRENT_GIT_BRANCH" != "no-branch" ]; then
    RPROMPT="%F{$pcc[3]}$(prompt_git_info)"
  fi
}

prompt_hyacinthe_setup "$@"