git branch in ZSH prompt

14Apr/083

git branch in ZSH prompt

Today, I came across a little trick on how to output the current git branch on your bash prompt. This is very useful, but not as much for me as I'm using ZSH. Of course, I wanted to adapt the method (and to use fewer backslashes ).

Also, in my setup, I'm making use of ZSH's prompt themes feature of which I've chosen the theme "adam1". So let's use that as a starting point.

First, create a copy of the prompt theme into a directory of your control where you intend to store private ZSH functions (~/zshfuncs in my case).
^?View Code BASH
cp /usr/share/zsh/4.3.4/functions/prompt_adam1_setup ~/zshfuncs/prompt_pilif_setup
Tweak the file. I've adapted the prompt from the original article, but I've managed to get rid of all the backslashes (to actually make the regex readable) and to place it nicely in the adam1 prompt framework.
Advise ZSH about the new ZSH function directory (if you haven't already done so).
^?View Code BASH
fpath=(~/zshfunc $fpath)
Load your new prompt theme.
^?View Code BASH
prompt pilif

And here's the adapted adam1 prompt theme:

^?View Code BASH

# pilif prompt theme
 
prompt_pilif_help () {
  cat <<'EOF'
This prompt is color-scheme-able.  You can invoke it thus:
 
  prompt pilif [<color1> [<color2> [<color3>]]]
 
This is heavily based on adam1 which is distributed with ZSH. In fact, 
the only change from adam1 is support for displaying the current branch
of your git repository (if you are in one)
EOF
}
 
prompt_pilif_setup () {
  prompt_adam1_color1=${1:-'blue'}
  prompt_adam1_color2=${2:-'cyan'}
  prompt_adam1_color3=${3:-'green'}
 
  base_prompt="%{$bg_no_bold[$prompt_adam1_color1]%}%n@%m%{$reset_color%} "
  post_prompt="%{$reset_color%}"
 
  base_prompt_no_color=$(echo "$base_prompt" | perl -pe "s/%{.*?%}//g")
  post_prompt_no_color=$(echo "$post_prompt" | perl -pe "s/%{.*?%}//g")
 
  precmd  () { prompt_pilif_precmd }
  preexec () { }
}
 
prompt_pilif_precmd () {
  setopt noxtrace localoptions
  local base_prompt_expanded_no_color base_prompt_etc
  local prompt_length space_left
  local git_branch
 
  git_branch=`git branch 2>/dev/null | grep -e '^*' | sed -E 's/^\* (.+)$/(\1) /'`
  base_prompt_expanded_no_color=$(print -P "$base_prompt_no_color")
  base_prompt_etc=$(print -P "$base_prompt%(4~|...|)%3~")
  prompt_length=${#base_prompt_etc}
  if [[ $prompt_length -lt 40 ]]; then
    path_prompt="%{$fg_bold[$prompt_adam1_color2]%}%(4~|...|)%3~%{$fg_bold[white]%}$git_branch"
  else
    space_left=$(( $COLUMNS - $#base_prompt_expanded_no_color - 2 ))
    path_prompt="%{$fg_bold[$prompt_adam1_color3]%}%${space_left}<...<%~ %{$reset_color%}$git_branch%{$fg_bold[$prompt_adam1_color3]%} $prompt_newline%{$fg_bold_white%}"
  fi
 
  PS1="$base_prompt$path_prompt %# $post_prompt"
  PS2="$base_prompt$path_prompt %_&gt; $post_prompt"
  PS3="$base_prompt$path_prompt ?# $post_prompt"
}
 
prompt_pilif_setup "$@"

The theme file can be downloaded here

Tagged as: git, solution, zsh Leave a comment

Comments (3) Trackbacks (1) ( subscribe to comments on this post )

Michel Peterson
May 6th, 2008 - 16:02

I suggest using this command instead:

git branch 2>/dev/null | grep -e ‘^*’ | awk ‘/^\* (.+)$/ { print “(“$2″)” }’

since sed -E doesn’t seem too practical for portability

( REPLY )
Michel Peterson
May 6th, 2008 - 17:45

Or even this, which has a better performance:

git symbolic-ref HEAD|awk ‘{sub(/^refs\/heads\//, “”, $1); print ” (“$1″)”}’

Regards,
Michel

( REPLY )
bartman
October 14th, 2008 - 16:32

Maybe you can steel some stuff from my prompt…

http://www.jukie.net/~bart/blog/zsh-git-prompt

Features:
- highlight the part of the path that is in the git repo
- show the current mode of rebase, am, merge, bisect, etc
- avoid reading rereading state unless the PWD changed or a git command was ran.

( REPLY )