git-prompt.plugin.zsh
4067 bytes
1# Handle $0 according to the standard:
2# https://zdharma-continuum.github.io/Zsh-100-Commits-Club/Zsh-Plugin-Standard.html
30="${${ZERO:-${0:#$ZSH_ARGZERO}}:-${(%):-%N}}"
40="${${(M)0:#/*}:-$PWD/$0}"
5
6__GIT_PROMPT_DIR="${0:A:h}"
7
8## Hook function definitions
9function chpwd_update_git_vars() {
10 update_current_git_vars
11}
12
13function preexec_update_git_vars() {
14 case "$2" in
15 git*|hub*|gh*|stg*)
16 __EXECUTED_GIT_COMMAND=1
17 ;;
18 esac
19}
20
21function precmd_update_git_vars() {
22 if [ -n "$__EXECUTED_GIT_COMMAND" ] || [ ! -n "$ZSH_THEME_GIT_PROMPT_CACHE" ]; then
23 update_current_git_vars
24 unset __EXECUTED_GIT_COMMAND
25 fi
26}
27
28autoload -U add-zsh-hook
29add-zsh-hook chpwd chpwd_update_git_vars
30add-zsh-hook precmd precmd_update_git_vars
31add-zsh-hook preexec preexec_update_git_vars
32
33
34## Function definitions
35function update_current_git_vars() {
36 unset __CURRENT_GIT_STATUS
37
38 local gitstatus="$__GIT_PROMPT_DIR/gitstatus.py"
39 _GIT_STATUS=$(python3 ${gitstatus} 2>/dev/null)
40 __CURRENT_GIT_STATUS=("${(@s: :)_GIT_STATUS}")
41 GIT_BRANCH=$__CURRENT_GIT_STATUS[1]
42 GIT_AHEAD=$__CURRENT_GIT_STATUS[2]
43 GIT_BEHIND=$__CURRENT_GIT_STATUS[3]
44 GIT_STAGED=$__CURRENT_GIT_STATUS[4]
45 GIT_CONFLICTS=$__CURRENT_GIT_STATUS[5]
46 GIT_CHANGED=$__CURRENT_GIT_STATUS[6]
47 GIT_UNTRACKED=$__CURRENT_GIT_STATUS[7]
48 GIT_STASHED=$__CURRENT_GIT_STATUS[8]
49 GIT_CLEAN=$__CURRENT_GIT_STATUS[9]
50 GIT_DELETED=$__CURRENT_GIT_STATUS[10]
51
52 if [ -z ${ZSH_THEME_GIT_SHOW_UPSTREAM+x} ]; then
53 GIT_UPSTREAM=
54 else
55 GIT_UPSTREAM=$(git rev-parse --abbrev-ref --symbolic-full-name "@{upstream}" 2>/dev/null) && GIT_UPSTREAM="${ZSH_THEME_GIT_PROMPT_UPSTREAM_SEPARATOR}${GIT_UPSTREAM}"
56 fi
57}
58
59git_super_status() {
60 precmd_update_git_vars
61 if [ -n "$__CURRENT_GIT_STATUS" ]; then
62 STATUS="$ZSH_THEME_GIT_PROMPT_PREFIX$ZSH_THEME_GIT_PROMPT_BRANCH$GIT_BRANCH$GIT_UPSTREAM%{${reset_color}%}"
63 if [ "$GIT_BEHIND" -ne "0" ]; then
64 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_BEHIND$GIT_BEHIND%{${reset_color}%}"
65 fi
66 if [ "$GIT_AHEAD" -ne "0" ]; then
67 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_AHEAD$GIT_AHEAD%{${reset_color}%}"
68 fi
69 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_SEPARATOR"
70 if [ "$GIT_STAGED" -ne "0" ]; then
71 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_STAGED$GIT_STAGED%{${reset_color}%}"
72 fi
73 if [ "$GIT_CONFLICTS" -ne "0" ]; then
74 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CONFLICTS$GIT_CONFLICTS%{${reset_color}%}"
75 fi
76 if [ "$GIT_CHANGED" -ne "0" ]; then
77 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CHANGED$GIT_CHANGED%{${reset_color}%}"
78 fi
79 if [ "$GIT_DELETED" -ne "0" ]; then
80 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_DELETED$GIT_DELETED%{${reset_color}%}"
81 fi
82 if [ "$GIT_UNTRACKED" -ne "0" ]; then
83 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_UNTRACKED$GIT_UNTRACKED%{${reset_color}%}"
84 fi
85 if [ "$GIT_STASHED" -ne "0" ]; then
86 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_STASHED$GIT_STASHED%{${reset_color}%}"
87 fi
88 if [ "$GIT_CLEAN" -eq "1" ]; then
89 STATUS="$STATUS$ZSH_THEME_GIT_PROMPT_CLEAN"
90 fi
91 STATUS="$STATUS%{${reset_color}%}$ZSH_THEME_GIT_PROMPT_SUFFIX"
92 echo "$STATUS"
93 fi
94}
95
96# Default values for the appearance of the prompt.
97ZSH_THEME_GIT_PROMPT_PREFIX="("
98ZSH_THEME_GIT_PROMPT_SUFFIX=")"
99ZSH_THEME_GIT_PROMPT_SEPARATOR="|"
100ZSH_THEME_GIT_PROMPT_BRANCH="%{$fg_bold[magenta]%}"
101ZSH_THEME_GIT_PROMPT_STAGED="%{$fg[red]%}%{●%G%}"
102ZSH_THEME_GIT_PROMPT_CONFLICTS="%{$fg[red]%}%{✖%G%}"
103ZSH_THEME_GIT_PROMPT_CHANGED="%{$fg[blue]%}%{✚%G%}"
104ZSH_THEME_GIT_PROMPT_DELETED="%{$fg[blue]%}%{-%G%}"
105ZSH_THEME_GIT_PROMPT_BEHIND="%{↓%G%}"
106ZSH_THEME_GIT_PROMPT_AHEAD="%{↑%G%}"
107ZSH_THEME_GIT_PROMPT_UNTRACKED="%{$fg[cyan]%}%{…%G%}"
108ZSH_THEME_GIT_PROMPT_STASHED="%{$fg_bold[blue]%}%{⚑%G%}"
109ZSH_THEME_GIT_PROMPT_CLEAN="%{$fg_bold[green]%}%{✔%G%}"
110ZSH_THEME_GIT_PROMPT_UPSTREAM_SEPARATOR="->"
111
112# Set the prompt.
113RPROMPT='$(git_super_status)'