Merge branch 'master' of ssh://git.madduck.net/~/git/pub/etc/zsh

[etc/zsh.git] / .zsh / zshrc / 60_vcsprompt

diff --git a/.zsh/zshrc/60_vcsprompt b/.zsh/zshrc/60_vcsprompt
index d72e205eb3589e12cf84b95fae03715afdc32353..d8fdaeacdd05c4c2a3178e66d614244a77941c47 100644 (file)
--- a/.zsh/zshrc/60_vcsprompt
+++ b/.zsh/zshrc/60_vcsprompt
@@ -32,6 +32,21 @@ __git_get_branch()
   echo "${ref#refs/heads/}"
 }
 
   echo "${ref#refs/heads/}"
 }
 

+__git_print_preprompt()
+{
+  local output
+  output=(${(f):-"$(git diff --stat --relative)"})
+  if [[ ${#output} -gt 1 ]]; then
+    echo changes on filesystem:
+    print "${(F)output[1,-2]}"
+  fi
+  output=(${(f):-"$(git diff --cached --stat --relative)"})
+  if [[ ${#output} -gt 1 ]]; then
+    echo cached/staged changes:
+    print "${(F)output[1,-2]}"
+  fi
+}
+

 __hg_get_reporoot()
 {
   hg root
 __hg_get_reporoot()
 {
   hg root


@@ -70,6 +85,7 @@ __vcs_get_repo_type()
   local dir
   while true; do
     [ -d ${dir}.git ] && echo git && break
   local dir
   while true; do
     [ -d ${dir}.git ] && echo git && break

+    [ -d "$GIT_DIR" ] && echo git && break

     [ -d ${dir}.bzr ] && echo bzr && break
     [ -d ${dir}.hg ] && echo hg && break
     [ "$(readlink -f ${dir:-.})" = / ] && echo NONE && break
     [ -d ${dir}.bzr ] && echo bzr && break
     [ -d ${dir}.hg ] && echo hg && break
     [ "$(readlink -f ${dir:-.})" = / ] && echo NONE && break


@@ -143,9 +159,8 @@ __vcs_set_prompt_variables()
       if [ -d "$GIT_DIR" ]; then
         # poor man's replace until I find out how to do named dirs properly
         # here:
       if [ -d "$GIT_DIR" ]; then
         # poor man's replace until I find out how to do named dirs properly
         # here:

-        _D="${GIT_DIR/$HOME/~}"
+        local _D="${GIT_DIR/$HOME/~}"

         set -- "$_D" "$2" "${${1#$_D}%/}"
         set -- "$_D" "$2" "${${1#$_D}%/}"

-        unset _D

       fi
       ;;
     hg)
       fi
       ;;
     hg)


@@ -178,6 +193,18 @@ __vcs_set_prompt_variables()
   psvar[1,3]=($1 $2 $3)
 }
 
   psvar[1,3]=($1 $2 $3)
 }
 

+__vcs_print_preprompt()
+{
+  local reporoot
+  repotype="${1:-$(__vcs_get_repo_type)}"
+
+  case "$repotype" in
+    git)
+      __git_print_preprompt
+      ;;
+  esac
+}
+

 if ! is_root; then
   # too dangerous to be run as root
 
 if ! is_root; then
   # too dangerous to be run as root
 


@@ -195,6 +222,11 @@ if ! is_root; then
   }
   chpwd_functions+=_update_vcs_prompt_vars
 
   }
   chpwd_functions+=_update_vcs_prompt_vars
 

+  _print_preprompt() {
+    [[ $? -eq 0 ]] && __vcs_print_preprompt
+  }
+  precmd_functions+=_print_preprompt
+

   # call it once
   _update_vcs_prompt_vars
 fi
   # call it once
   _update_vcs_prompt_vars
 fi