From 7b90c97e5fca0411f1926358193df7d2bf04042a Mon Sep 17 00:00:00 2001
From: mafro <github@mafro.net>
Date: Fri, 4 Apr 2014 01:51:32 +0000
Subject: [PATCH] enable display of dirty unmerged status in git prompt

---
 lib/git.zsh | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/lib/git.zsh b/lib/git.zsh
index 305a77aff..53bf94987 100644
--- a/lib/git.zsh
+++ b/lib/git.zsh
@@ -23,7 +23,16 @@ parse_git_dirty() {
         GIT_STATUS=$(command git status -s ${SUBMODULE_SYNTAX} 2> /dev/null | tail -n1)
     fi
     if [[ -n $GIT_STATUS ]]; then
-      echo "$ZSH_THEME_GIT_PROMPT_DIRTY"
+      # display dirty unmerged status
+      if [[ ! -z "$ZSH_THEME_GIT_PROMPT_DIRTY_UNMERGED" ]]; then
+        if $(echo "$(command git status --porcelain -b 2> /dev/null)" | grep -E '^(AA|UU) ' &> /dev/null); then
+          echo "$ZSH_THEME_GIT_PROMPT_DIRTY_UNMERGED"
+        else
+          echo "$ZSH_THEME_GIT_PROMPT_DIRTY"
+        fi
+      else
+        echo "$ZSH_THEME_GIT_PROMPT_DIRTY"
+      fi
     else
       echo "$ZSH_THEME_GIT_PROMPT_CLEAN"
     fi