From 7d597e8f4ac3318a6e89837e257ef5086e0fdce0 Mon Sep 17 00:00:00 2001
From: xero <x@xero.nu>
Date: Fri, 25 Apr 2014 11:43:04 -0400
Subject: add zsh git prompt functions

---
 .zsh/functions/chpwd_update_git_vars   |  1 +
 .zsh/functions/precmd_update_git_vars  |  4 ++++
 .zsh/functions/preexec_update_git_vars |  5 +++++
 .zsh/functions/prompt_git_info         | 21 +++++++++++++++++++++
 .zsh/functions/update_current_git_vars | 31 +++++++++++++++++++++++++++++++
 5 files changed, 62 insertions(+)
 create mode 100644 .zsh/functions/chpwd_update_git_vars
 create mode 100644 .zsh/functions/precmd_update_git_vars
 create mode 100644 .zsh/functions/preexec_update_git_vars
 create mode 100644 .zsh/functions/prompt_git_info
 create mode 100644 .zsh/functions/update_current_git_vars

diff --git a/.zsh/functions/chpwd_update_git_vars b/.zsh/functions/chpwd_update_git_vars
new file mode 100644
index 0000000..2cc7ffa
--- /dev/null
+++ b/.zsh/functions/chpwd_update_git_vars
@@ -0,0 +1 @@
+update_current_git_vars
\ No newline at end of file
diff --git a/.zsh/functions/precmd_update_git_vars b/.zsh/functions/precmd_update_git_vars
new file mode 100644
index 0000000..7fa32df
--- /dev/null
+++ b/.zsh/functions/precmd_update_git_vars
@@ -0,0 +1,4 @@
+if [ -n "$__EXECUTED_GIT_COMMAND" ]; then
+    update_current_git_vars
+    unset __EXECUTED_GIT_COMMAND
+fi
\ No newline at end of file
diff --git a/.zsh/functions/preexec_update_git_vars b/.zsh/functions/preexec_update_git_vars
new file mode 100644
index 0000000..998de21
--- /dev/null
+++ b/.zsh/functions/preexec_update_git_vars
@@ -0,0 +1,5 @@
+case "$1" in 
+    git*)
+    __EXECUTED_GIT_COMMAND=1
+    ;;
+esac
\ No newline at end of file
diff --git a/.zsh/functions/prompt_git_info b/.zsh/functions/prompt_git_info
new file mode 100644
index 0000000..d492ac8
--- /dev/null
+++ b/.zsh/functions/prompt_git_info
@@ -0,0 +1,21 @@
+if [ -n "$__CURRENT_GIT_BRANCH" ]; then
+    local s="%F{cyan}-[%F{white}"
+    s+="$__CURRENT_GIT_BRANCH"
+    case "$__CURRENT_GIT_BRANCH_STATUS" in
+        ahead)
+        s+="↑"
+        ;;
+        diverged)
+        s+="↕"
+        ;;
+        behind)
+        s+="↓"
+        ;;
+    esac
+    if [ -n "$__CURRENT_GIT_BRANCH_IS_DIRTY" ]; then
+        s+="⚡"
+    fi
+    s+="%F{cyan}]"
+ 
+    printf "%s%s" $s
+fi
\ No newline at end of file
diff --git a/.zsh/functions/update_current_git_vars b/.zsh/functions/update_current_git_vars
new file mode 100644
index 0000000..716d984
--- /dev/null
+++ b/.zsh/functions/update_current_git_vars
@@ -0,0 +1,31 @@
+unset __CURRENT_GIT_BRANCH
+unset __CURRENT_GIT_BRANCH_STATUS
+unset __CURRENT_GIT_BRANCH_IS_DIRTY
+
+local st="$(git status 2>/dev/null)"
+local st2="$(git log -1 --pretty=%h 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="$st2"
+    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'
+        else
+            __CURRENT_GIT_BRANCH_STATUS='behind'
+        fi
+    fi
+
+    if [[ ! $st =~ 'nothing to commit' ]]; then
+        __CURRENT_GIT_BRANCH_IS_DIRTY='1'
+    fi
+fi
\ No newline at end of file
-- 
cgit v1.2.1