setopt prompt_subst
autoload colors; colors;
-hostcolor=green
+hostcolor=cyan
[[ $(hostname) == "tardis" ]] && hostcolor=red
precmd() {
if [[ "$line" == \#\ * ]]; then
[[ "$line" =~ '# branch.oid ([0-9a-f]+)' ]] && oid=$match[1]
[[ "$line" =~ '# branch.head (.*)' ]] && head=$match[1]
+ [[ "$line" =~ '# branch.upstream (.*)' ]] && upstream=$match[1]
[[ "$line" =~ '# branch.ab \+([0-9]+) -([0-9]+)' ]] && ahead=$match[1] && behind=$match[2]
else
# Count added, deleted, modified, renamed, unmerged, untracked, dirty.
[[ "$line" == 2\ R[.MTD]\ * ]] && (( renamed++ ))
# work tree
+ [[ "$line" == 1\ [.MARCT]A\ * ]] && (( added++ , modified++ )) # intend-to-add
[[ "$line" == 1\ [.MARCT]M\ * ]] && (( modified++ ))
[[ "$line" == 1\ [.MARCT]D\ * ]] && (( deleted_wt++ ))
[[ "$line" == \?\ * ]] && (( untracked++ ))
# merge conflicts
[[ "$line" == u\ (AA|DD|U?|?U)\ * ]] && (( unmerged++ ))
+
+ #TODO more info for submodules?
fi
done < <(${(z)status_cmd} 2> /dev/null)
# Format upstream
local upstream_str
+ [[ -z "$upstream" ]] && upstream_str+="%{$fg[blue]%} Ɇ"
(( ahead > 0 )) && upstream_str+="%{$fg[blue]%} >$ahead"
(( behind > 0 )) && upstream_str+="%{$fg[blue]%} <$behind"
git_status+="$upstream_str"