Provide example of local commit difference

1 files changed, 26 insertions, 0 deletions

diff --git a/git-base.sh b/git-base.sh
index e60f9fc..aca271b 100755
--- a/git-base.sh
+++ b/git-base.sh
@@ -369,3 +369,29 @@ zsh_color_changes_status() {
   fi
   echo $changes
 }
+
+zsh_color_local_commits() {
+  local ahead_arrow="%{$fg_bold[green]%}↑%{$reset_color%}"
+  local behind_arrow="%{$fg_bold[red]%}↓%{$reset_color%}"
+  local diverged_arrow="%{$fg_bold[yellow]%}⇵%{$reset_color%}"
+
+  local local_commits=""
+  if is_repo; then
+
+    if remote_branch="$(remote_branch_name)"; then
+      local_ahead="$(commits_ahead_of_remote "$remote_branch")"
+      local_behind="$(commits_behind_of_remote "$remote_branch")"
+      remote_ahead="$(remote_ahead_of_master "$remote_branch")"
+      remote_behind="$(remote_behind_of_master "$remote_branch")"
+
+      if [[ "$local_behind" -gt "0" && "$local_ahead" -gt "0" ]]; then
+        local_commits=" $local_behind$diverged_arrow$local_ahead"
+      elif [[ "$local_behind" -gt "0" ]]; then
+        local_commits=" $local_behind$behind_arrow"
+      elif [[ "$local_ahead" -gt "0" ]]; then
+        local_commits=" $local_ahead$ahead_arrow"
+      fi
+    fi
+  fi
+  echo $local_commits
+}