From 2d6e5c545ecc6d35a2a4e781395a56a2dbfd2cf8 Mon Sep 17 00:00:00 2001
From: lightlfyan <lightlfyan@gmail.com>
Date: Wed, 29 Oct 2014 12:38:53 +0800
Subject: [PATCH] Update git.zsh

show the current branch upstream
---
 lib/git.zsh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/git.zsh b/lib/git.zsh
index 7aa5a0ea2..a1c839c45 100644
--- a/lib/git.zsh
+++ b/lib/git.zsh
@@ -3,7 +3,10 @@ function git_prompt_info() {
   if [[ "$(command git config --get oh-my-zsh.hide-status 2>/dev/null)" != "1" ]]; then
     ref=$(command git symbolic-ref HEAD 2> /dev/null) || \
     ref=$(command git rev-parse --short HEAD 2> /dev/null) || return 0
-    echo "$ZSH_THEME_GIT_PROMPT_PREFIX${ref#refs/heads/}$(parse_git_dirty)$ZSH_THEME_GIT_PROMPT_SUFFIX"
+    upstream=$(command git rev-parse --abbrev-ref --symbolic-full-name @{u} 2> /dev/null) || \
+    upstream="empty"
+    symbol=" => "
+    echo "$ZSH_THEME_GIT_PROMPT_PREFIX${ref#refs/heads/}$(parse_git_dirty)${symbol}${upstream}$ZSH_THEME_GIT_PROMPT_SUFFIX"
   fi
 }