.gitprompt

# ============================================================================ #
#                                                                              #
# ---------                         .gitpromt                          ------- #
#                                                                          1.0 #
# ============================================================================ #

# Adapted from code found at <https://gist.github.com/1712320>.

setopt prompt_subst
autoload -U colors && colors

GIT_PROMPT_SYMBOL="%{$fg[blue]%}"
GIT_PROMPT_PREFIX="%{$fg[green]%}%{$reset_color%}"
GIT_PROMPT_SUFFIX="%{$fg[green]%} %{$reset_color%}"
GIT_PROMPT_AHEAD="%{$fg[red]%}ANUM %{$reset_color%}"
GIT_PROMPT_BEHIND="%{$fg[cyan]%}BNUM %{$reset_color%}"
GIT_PROMPT_MERGING="%{$fg_bold[magenta]%}⚡︎%{$reset_color%}"
GIT_PROMPT_UNTRACKED="%{$fg_bold[red]%}●%{$reset_color%}"
GIT_PROMPT_MODIFIED="%{$fg_bold[yellow]%}●%{$reset_color%}"
GIT_PROMPT_STAGED="%{$fg_bold[green]%}●%{$reset_color%}"

parse_git_branch() {
	(git symbolic-ref -q HEAD || git name-rev --name-only --no-undefined --always HEAD) 2> /dev/null
}

parse_git_state() {

	local GIT_STATE=""

		local NUM_AHEAD="$(git log --oneline @{u}.. 2> /dev/null | wc -l | tr -d ' ')"
		if [ "$NUM_AHEAD" -gt 0 ]; then
			GIT_STATE=$GIT_STATE${GIT_PROMPT_AHEAD//NUM/$NUM_AHEAD}
	fi

		local NUM_BEHIND="$(git log --oneline ..@{u} 2> /dev/null | wc -l | tr -d ' ')"
		if [ "$NUM_BEHIND" -gt 0 ]; then
			GIT_STATE=$GIT_STATE${GIT_PROMPT_BEHIND//NUM/$NUM_BEHIND}
	fi

		local GIT_DIR="$(git rev-parse --git-dir 2> /dev/null)"
		if [ -n $GIT_DIR ] && test -r $GIT_DIR/MERGE_HEAD; then
			GIT_STATE=$GIT_STATE$GIT_PROMPT_MERGING
				fi

				if [[ -n $(git ls-files --other --exclude-standard 2> /dev/null) ]]; then
					GIT_STATE=$GIT_STATE$GIT_PROMPT_UNTRACKED
						fi

						if ! git diff --quiet 2> /dev/null; then
							GIT_STATE=$GIT_STATE$GIT_PROMPT_MODIFIED
								fi

								if ! git diff --cached --quiet 2> /dev/null; then
									GIT_STATE=$GIT_STATE$GIT_PROMPT_STAGED
										fi

										if [[ -n $GIT_STATE ]]; then
											echo "$GIT_PROMPT_PREFIX$GIT_STATE$GIT_PROMPT_SUFFIX"
												fi

}

git_prompt_string() {
	local git_where="$(parse_git_branch)"
		[ -n "$git_where" ] && echo "$GIT_PROMPT_SYMBOL$(parse_git_state)$GIT_PROMPT_PREFIX%{$fg[yellow]%}${git_where#(refs/heads/|tags/)}$GIT_PROMPT_SUFFIX"
}

RPS1='$(git_prompt_string)'