1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
# Frontend to controlling prompt
prompt() {
# What's done next depends on the first argument to the function
case $1 in
# Turn complex, colored PS1 and debugging PS4 prompts on
on)
setopt promptsubst promptpercent
# Basic prompt shape depends on whether we're in SSH or not
PS1=
if [[ -n $SSH_CLIENT ]] || [[ -n $SSH_CONNECTION ]] ; then
PS1=$PS1'%n@%m:'
fi
PS1=$PS1'%~'
# Add sub-commands; VCS, job, and return status checks
PS1=$PS1'$(ret=$?;prompt vcs;prompt job;prompt ret)'
# Add prefix and suffix
PS1='${PROMPT_PREFIX}'$PS1'${PROMPT_SUFFIX}'
# Add terminating "$" or "#" sign
PS1=$PS1'%#'
# Add > symbols to show nested shells
local shlvl
for ((shlvl = 1; shlvl < SHLVL; shlvl++)) ; do
PS1='>'$PS1
done
# Bold and color the prompt if it looks like we can
if (( $({ tput colors || tput Co ; } 2>/dev/null) >= 8 )) ; then
PS1='%B%F{cyan}'$PS1'%f%b'
fi
# Add a space and define the rest of the prompts
PS1=$PS1' '
PS2='> '
PS3='? '
PS4='+<$?> $LINENO:'
;;
# Revert to simple inexpensive prompts
off)
PS1='\$ '
PS2='> '
PS3='? '
PS4='+ '
;;
# Git prompt function
git)
# Wrap as compound command; we don't want to see output from any of
# these git(1) calls
{
# Bail if we're not in a work tree--or, implicitly, if we don't
# have git(1).
[[ $(git rev-parse --is-inside-work-tree) == true ]] ||
return
# Refresh index so e.g. git-diff-files(1) is accurate
git update-index --refresh
# Find a local branch, remote branch, or tag (annotated or
# not), or failing all of that just show the short commit ID,
# in that order of preference; if none of that works, bail out
local name
name=$(
git symbolic-ref --quiet HEAD ||
git describe --tags --exact-match HEAD ||
git rev-parse --short HEAD
) || return
name=${name##*/}
[[ -n $name ]] || return
# Check various files in .git to flag processes
local proc
[[ -d .git/rebase-merge || -d .git/rebase-apply ]] &&
proc=${proc:+"$proc",}'REBASE'
[[ -f .git/MERGE_HEAD ]] &&
proc=${proc:+"$proc",}'MERGE'
[[ -f .git/CHERRY_PICK_HEAD ]] &&
proc=${proc:+"$proc",}'PICK'
[[ -f .git/REVERT_HEAD ]] &&
proc=${proc:+"$proc",}'REVERT'
[[ -f .git/BISECT_LOG ]] &&
proc=${proc:+"$proc",}'BISECT'
# Collect symbols representing repository state
local state
# Upstream HEAD has commits after local HEAD; we're "behind"
(($(git rev-list --count 'HEAD..@{u}'))) &&
state=${state}'<'
# Local HEAD has commits after upstream HEAD; we're "ahead"
(($(git rev-list --count '@{u}..HEAD'))) &&
state=${state}'>'
# Tracked files are modified
git diff-files --no-ext-diff --quiet ||
state=${state}'!'
# Changes are staged
git diff-index --cached --no-ext-diff --quiet HEAD ||
state=${state}'+'
# There are some untracked and unignored files
git ls-files --directory --error-unmatch --exclude-standard \
--no-empty-directory --others -- ':/*' &&
state=${state}'?'
# There are stashed changes
git rev-parse --quiet --verify refs/stash &&
state=${state}'^'
} >/dev/null 2>&1
# Print the status in brackets; add a git: prefix only if there
# might be another VCS prompt (because PROMPT_VCS is set)
printf '(%s%s%s%s)' \
"${PROMPT_VCS:+git:}" "$name" "${proc:+:"$proc"}" "$state"
;;
# Subversion prompt function
svn)
# Determine the repository URL and root directory
local key value url root
while [[ -z $url || -z $root ]] && IFS=: read -r key value ; do
case $key in
'URL') url=${value## } ;;
'Repository Root') root=${value## } ;;
esac
done < <(svn info 2>/dev/null)
# Exit if we couldn't get either--or, implicitly, if we don't have
# svn(1).
[[ -n $url && -n $root ]] || return
# Remove the root from the URL to get what's hopefully the branch
# name, removing leading slashes and the 'branches' prefix, and any
# trailing content after a slash
local branch
branch=${url/"$root"}
branch=${branch#/}
branch=${branch#branches/}
branch=${branch%%/*}
# Parse the output of svn status to determine working copy state
local symbol
local -i modified untracked
while ((!modified || !untracked)) && read -r symbol _ ; do
case $symbol in
*\?*) untracked=1 ;;
*) modified=1 ;;
esac
done < <(svn status 2>/dev/null)
# Add appropriate state flags
local state
((modified)) && state=${state}'!'
((untracked)) && state=${state}'?'
# Print the state in brackets with an svn: prefix
printf '(svn:%s%s)' "${branch:-unknown}" "$state"
;;
# VCS wrapper prompt function; print the first relevant prompt, if any
vcs)
local vcs
for vcs in "${PROMPT_VCS[@]:-git}" ; do
prompt "$vcs" && return
done
;;
# Show return status of previous command in angle brackets, if not zero
ret)
((ret)) && printf '<%u>' "$ret"
;;
# Show the count of background jobs in curly brackets, if not zero
job)
local -i jobc
while read -r ; do
((jobc++))
done < <(jobs -p)
((jobc)) && printf '{%u}' "$jobc"
;;
# No argument given, print prompt strings and vars
'')
declare -p PS1 PS2 PS3 PS4
;;
# Print error
*)
printf 'prompt: Unknown command %s\n' "$1" >&2
return 2
;;
esac
}
# Start with full-fledged prompt
prompt on
|