zsh/lib/git.zsh

   1 # get the name of the branch we are on
   2 function git_prompt_info() {
   3   ref=$(git symbolic-ref HEAD 2> /dev/null) || \
   4   ref=$(git rev-parse --short HEAD 2> /dev/null) || return
   5   echo "$ZSH_THEME_GIT_PROMPT_PREFIX${ref#refs/heads/}$(parse_git_dirty)$ZSH_THEME_GIT_PROMPT_SUFFIX"
   6 }
   7
   8
   9 # Checks if working tree is dirty
  10 parse_git_dirty() {
  11   if [[ -n $(git status -s 2> /dev/null) ]]; then
  12     echo "$ZSH_THEME_GIT_PROMPT_DIRTY"
  13   else
  14     echo "$ZSH_THEME_GIT_PROMPT_CLEAN"
  15   fi
  16 }
  17
  18 # get the difference between the local and remote branches
  19 git_remote_status() {
  20     remote=${$(git rev-parse --verify ${hook_com[branch]}@{upstream} --symbolic-full-name 2>/dev/null)/refs\/remotes\/}
  21     if [[ -n ${remote} ]] ; then
  22         ahead=$(git rev-list ${hook_com[branch]}@{upstream}..HEAD 2>/dev/null | wc -l)
  23         behind=$(git rev-list HEAD..${hook_com[branch]}@{upstream} 2>/dev/null | wc -l)
  24
  25         if [ $ahead -eq 0 ] && [ $behind -gt 0 ]
  26         then
  27             echo "$ZSH_THEME_GIT_PROMPT_BEHIND_REMOTE"
  28         elif [ $ahead -gt 0 ] && [ $behind -eq 0 ]
  29         then
  30             echo "$ZSH_THEME_GIT_PROMPT_AHEAD_REMOTE"
  31         elif [ $ahead -gt 0 ] && [ $behind -gt 0 ]
  32         then
  33             echo "$ZSH_THEME_GIT_PROMPT_DIVERGED_REMOTE"
  34         fi
  35     fi
  36 }
  37
  38 # Checks if there are commits ahead from remote
  39 function git_prompt_ahead() {
  40   if $(echo "$(git log origin/$(current_branch)..HEAD 2> /dev/null)" | grep '^commit' &> /dev/null); then
  41     echo "$ZSH_THEME_GIT_PROMPT_AHEAD"
  42   fi
  43 }
  44
  45 # Formats prompt string for current git commit short SHA
  46 function git_prompt_short_sha() {
  47   SHA=$(git rev-parse --short HEAD 2> /dev/null) && echo "$ZSH_THEME_GIT_PROMPT_SHA_BEFORE$SHA$ZSH_THEME_GIT_PROMPT_SHA_AFTER"
  48 }
  49
  50 # Formats prompt string for current git commit long SHA
  51 function git_prompt_long_sha() {
  52   SHA=$(git rev-parse HEAD 2> /dev/null) && echo "$ZSH_THEME_GIT_PROMPT_SHA_BEFORE$SHA$ZSH_THEME_GIT_PROMPT_SHA_AFTER"
  53 }
  54
  55 # Get the status of the working tree
  56 git_prompt_status() {
  57   INDEX=$(git status --porcelain -b 2> /dev/null)
  58   STATUS=""
  59   if $(echo "$INDEX" | grep '^?? ' &> /dev/null); then
  60     STATUS="$ZSH_THEME_GIT_PROMPT_UNTRACKED$STATUS"
  61   fi
  62   if $(echo "$INDEX" | grep '^A  ' &> /dev/null); then
  63     STATUS="$ZSH_THEME_GIT_PROMPT_ADDED$STATUS"
  64   elif $(echo "$INDEX" | grep '^M  ' &> /dev/null); then
  65     STATUS="$ZSH_THEME_GIT_PROMPT_ADDED$STATUS"
  66   fi
  67   if $(echo "$INDEX" | grep '^ M ' &> /dev/null); then
  68     STATUS="$ZSH_THEME_GIT_PROMPT_MODIFIED$STATUS"
  69   elif $(echo "$INDEX" | grep '^AM ' &> /dev/null); then
  70     STATUS="$ZSH_THEME_GIT_PROMPT_MODIFIED$STATUS"
  71   elif $(echo "$INDEX" | grep '^ T ' &> /dev/null); then
  72     STATUS="$ZSH_THEME_GIT_PROMPT_MODIFIED$STATUS"
  73   fi
  74   if $(echo "$INDEX" | grep '^R  ' &> /dev/null); then
  75     STATUS="$ZSH_THEME_GIT_PROMPT_RENAMED$STATUS"
  76   fi
  77   if $(echo "$INDEX" | grep '^ D ' &> /dev/null); then
  78     STATUS="$ZSH_THEME_GIT_PROMPT_DELETED$STATUS"
  79   elif $(echo "$INDEX" | grep '^D  ' &> /dev/null); then
  80     STATUS="$ZSH_THEME_GIT_PROMPT_DELETED$STATUS"
  81   elif $(echo "$INDEX" | grep '^AD ' &> /dev/null); then
  82     STATUS="$ZSH_THEME_GIT_PROMPT_DELETED$STATUS"
  83   fi
  84   if $(git rev-parse --verify refs/stash >/dev/null 2>&1); then
  85     STATUS="$ZSH_THEME_GIT_PROMPT_STASHED$STATUS"
  86   fi
  87   if $(echo "$INDEX" | grep '^UU ' &> /dev/null); then
  88     STATUS="$ZSH_THEME_GIT_PROMPT_UNMERGED$STATUS"
  89   fi
  90   if $(echo "$INDEX" | grep '^## .*ahead' &> /dev/null); then
  91     STATUS="$ZSH_THEME_GIT_PROMPT_AHEAD$STATUS"
  92   fi
  93   if $(echo "$INDEX" | grep '^## .*behind' &> /dev/null); then
  94     STATUS="$ZSH_THEME_GIT_PROMPT_BEHIND$STATUS"
  95   fi
  96   if $(echo "$INDEX" | grep '^## .*diverged' &> /dev/null); then
  97     STATUS="$ZSH_THEME_GIT_PROMPT_DIVERGED$STATUS"
  98   fi
  99   echo $STATUS
 100 }