From 84259876ebf817174de97330c9771e68e485ce8b Mon Sep 17 00:00:00 2001
From: "martin f. krafft" <madduck@madduck.net>
Date: Mon, 5 May 2008 23:56:04 +0100
Subject: [PATCH] do the git prompt thing proper

---
 .zsh/zshrc/85_git_prompt | 89 ++++++++++++++++++++++++++++++++++------
 .zsh/zshrc/90_prompt     |  4 +-
 2 files changed, 78 insertions(+), 15 deletions(-)

diff --git a/.zsh/zshrc/85_git_prompt b/.zsh/zshrc/85_git_prompt
index 626b4c3..32d0e4f 100644
--- a/.zsh/zshrc/85_git_prompt
+++ b/.zsh/zshrc/85_git_prompt
@@ -7,36 +7,99 @@
 #
 # Source repository: http://git.madduck.net/v/etc/zsh.git
 #
-# Shamelessly adapted from http://www.jukie.net/~bart/conf/zsh.d/S55_git
+# Shamelessly based on http://glandium.org/blog/?p=170
 #
 
-_get_git_cur_branch() {
-  git branch --no-color | sed -rne 's,^\* ,,p'
+__git_get_repo_root()
+{
+  local relroot
+  relroot="$(git rev-parse --show-cdup 2>/dev/null)" || return 1
+  if [ -n "$relroot" ]; then
+    readlink -f "$relroot"
+  else
+    echo $PWD
+  fi
+}
+
+__git_get_branch()
+{
+  local ref
+  ref=$(git symbolic-ref -q HEAD 2>/dev/null \
+     || git-name-rev --name-only HEAD 2>/dev/null)
+  echo "${ref#refs/heads/}"
 }
 
-_is_git_repo() {
-  [ "$(git rev-parse --is-inside-work-tree 2>/dev/null)" = true ]
+__get_root_offsets()
+{
+  local pwda reporoot loc
+  pwda=(${(s:/:)PWD})
+  reporoot=(${(s:/:)1})
+  echo $((1 - $#reporoot)) $(($#pwda - $#reporoot))
+}
+
+__get_prompt_path_components()
+{
+  local reporoot
+  reporoot="$1"
+
+  set -- $(__get_root_offsets "$reporoot")
+  if [ "$2" -le 0 ]; then
+    echo %~
+  else
+    echo "%${1}~" "%${2}~"
+  fi
 }
 
-_set_git_psvar() {
-  if _is_git_repo; then
-    psvar[1]="$(_get_git_cur_branch)"
+__vcs_get_repo_type()
+{
+  if __git_get_repo_root >/dev/null; then
+    echo git
   else
-    [ -n "$psvar[1]" ] && unset "psvar[1]"
+    echo NONE
   fi
 }
 
+__vcs_set_prompt_variables()
+{
+  local pre branch post
+  local MAXLEN=25
+
+  case "${1:-$(__vcs_get_repo_type)}" in
+    git)
+      local reporoot="$(__git_get_repo_root)"
+      set -- $(__get_prompt_path_components "$reporoot")
+      branch="$(__git_get_branch)"
+      post="${(%)2}"
+      local prelen="$((MAXLEN - $#post - $#branch))"
+      [ $prelen -lt 10 ] && prelen=10
+      pre="%${prelen}<..<${1}%<<"
+      pre="${(%)pre}"
+      ;;
+    *)
+      local p="%${MAXLEN}<..<%~%<<"
+      #TODO find a better way so we don't have to nuke $psvar, but since the
+      #     %(nv.true.false) check for prompts checks element count, not
+      #     content, that's all we get for now
+      psvar=("${(%)p}")
+      return
+  esac
+
+  psvar[1]="$pre"
+  psvar[2]="$branch"
+  psvar[3]="$post"
+}
+
 update_git_vars_if_git_ran() {
-  # if the last command included the word git, the branch may have changed.
-  # checkout should be alright, but you never know...
+  local vcs="$(__vcs_get_repo_type)"
   case "$(history $(($HISTCMD - 1)))" in
-    *git*) _set_git_psvar;;
+    # $vcs appeared in last command, so be sure to update
+    *${vcs}*) __vcs_set_prompt_variables
   esac
 }
 precmd_functions+=update_git_vars_if_git_ran
 
 update_git_vars() {
-  _set_git_psvar
+  __vcs_set_prompt_variables
 }
 chpwd_functions+=update_git_vars
 
diff --git a/.zsh/zshrc/90_prompt b/.zsh/zshrc/90_prompt
index 6d5f635..fcb97f4 100644
--- a/.zsh/zshrc/90_prompt
+++ b/.zsh/zshrc/90_prompt
@@ -9,8 +9,8 @@
 #
 
 is_root && PS1_HL=U
-PS1="%${PS1_HL:=B}%m%${(L)PS1_HL}:%25<..<%~%# "
+PS1="%${PS1_HL:=B}%m%${(L)PS1_HL}:%1v%(2v.|%B%2v%b|.)%(3v.%3v.)%# "
 unset PS1_HL
-RPS1="#%(0?..%?,)%(1v.%1v,.)%!"
+RPS1="#%(0?..%?,)%!"
 
 # vim:ft=zsh
-- 
2.39.5