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
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
|
##
## zsh prompt themes extension
## by Adam Spiers <adam@spiers.net>
##
## Load with `autoload -Uz promptinit; promptinit'.
## Type `prompt -h' for help.
##
typeset -gaU prompt_themes
typeset -ga prompt_theme
typeset -g prompt_newline
prompt_themes=()
promptinit () {
emulate -L zsh
setopt extendedglob
autoload -Uz add-zsh-hook add-zle-hook-widget
local ppath='' name theme
local -a match mbegin mend
# Autoload all prompt_*_setup functions in fpath
for theme in $^fpath/prompt_*_setup(N); do
if [[ $theme == */prompt_(#b)(*)_setup ]]; then
name="$match[1]"
if [[ -r "$theme" ]]; then
prompt_themes=($prompt_themes $name)
autoload -Uz prompt_${name}_setup
else
print "Couldn't read file $theme containing theme $name."
fi
else
print "Eh? Mismatch between glob patterns in promptinit."
fi
done
# Variables common to all prompt styles
prompt_newline=$'\n%{\r%}'
}
prompt_preview_safely() {
emulate -L zsh
print -P "%b%f%k"
if [[ -z "$prompt_themes[(r)$1]" ]]; then
print "Unknown theme: $1"
return
fi
# Run this in a subshell, so we don't need to clean up afterwards.
(
# Execute current theme's cleanup sequence, if any.
zstyle -t :prompt-theme cleanup
# If we can't find a _preview function, run the _setup function to see if
# it will create one.
typeset +f prompt_${1}_preview >&/dev/null ||
prompt_${1}_setup
# ...then try again.
if typeset +f prompt_${1}_preview >&/dev/null; then
prompt_${1}_preview "$@[2,-1]"
else
prompt_preview_theme "$@"
fi
)
}
set_prompt() {
emulate -L zsh
local opt preview theme usage old_theme
usage='Usage: prompt <options>
Options:
-c Show currently selected theme and parameters
-l List currently available prompt themes
-p [<themes>] Preview given themes (defaults to all except current theme)
-h [<theme>] Display help (for given theme)
-s <theme> Set and save theme
<theme> Switch to new theme immediately (changes not saved)
Use prompt -h <theme> for help on specific themes.'
getopts "chlps:" opt
case "$opt" in
c) if [[ -n $prompt_theme ]]; then
print -n "Current prompt theme"
(( $#prompt_theme > 1 )) && print -n " with parameters"
print " is:\n $prompt_theme"
else
print "Current prompt is not a theme."
fi
return
;;
h) if [[ -n "$2" && -n $prompt_themes[(r)$2] ]]; then
# Run this in a subshell, so we don't need to clean up afterwards.
(
# Execute current theme's cleanup sequence, if any.
zstyle -t :prompt-theme cleanup
# If we can't find a _help function, run the _setup function to see
# if it will create one.
typeset +f prompt_$2_help >/dev/null ||
prompt_$2_setup
# ...then try again.
if typeset +f prompt_$2_help >/dev/null; then
print "Help for $2 theme:\n"
prompt_$2_help
else
print "No help available for $2 theme."
fi
print "\nType \`prompt -p $2' to preview the theme, \`prompt $2'"
print "to try it out, and \`prompt -s $2' to use it in future sessions."
)
else
print "$usage"
fi
;;
l) print Currently available prompt themes:
print $prompt_themes
return
;;
p) preview=( ${prompt_themes:#$prompt_theme} )
(( $#* > 1 )) && preview=( "$@[2,-1]" )
for theme in $preview; do
prompt_preview_safely "$=theme"
done
print -P "%b%f%k"
;;
s) print "Set and save not yet implemented. Please ensure your ~/.zshrc"
print "contains something similar to the following:\n"
print " autoload -Uz promptinit"
print " promptinit"
print " prompt $*[2,-1]"
shift
;&
*) if [[ "$1" == 'random' ]]; then
local random_themes
if (( $#* == 1 )); then
random_themes=( $prompt_themes )
else
random_themes=( "$@[2,-1]" )
fi
local i=$(( ( $RANDOM % $#random_themes ) + 1 ))
argv=( "${=random_themes[$i]}" )
fi
if [[ -z "$1" || -z $prompt_themes[(r)$1] ]]; then
print "$usage"
return
fi
# Reset some commonly altered bits to the default
local hook
for hook in chpwd precmd preexec periodic zshaddhistory zshexit \
zsh_directory_name; do
add-zsh-hook -D "$hook" "prompt_*_$hook"
done
for hook in isearch-exit isearch-update line-pre-redraw line-init \
line-finish history-line-set keymap-select; do
add-zle-hook-widget -D "$hook" "prompt_*_$hook"
done
typeset -ga zle_highlight=( ${zle_highlight:#default:*} )
(( ${#zle_highlight} )) || unset zle_highlight
zstyle -t :prompt-theme cleanup
prompt_$1_setup "$@[2,-1]" && prompt_theme=( "$@" )
;;
esac
}
prompt_cleanup () {
local -a cleanup_hooks theme_active
if ! zstyle -g cleanup_hooks :prompt-theme cleanup; then
if ! zstyle -g theme_active :prompt-theme restore; then
print -u2 "prompt_cleanup: no prompt theme active"
return 1
fi
# Set the cleanup sequence up.
zstyle -e :prompt-theme cleanup \
'zstyle -d :prompt-theme cleanup;' \
'reply=(yes)'
zstyle -g cleanup_hooks :prompt-theme cleanup
fi
cleanup_hooks+=(';' "$@")
zstyle -e :prompt-theme cleanup "${cleanup_hooks[@]}"
}
prompt () {
local -a prompt_opts theme_active
zstyle -g theme_active :prompt-theme restore || {
# This is done here rather than in set_prompt so that it
# is safe and sane for set_prompt to setopt localoptions,
# which will be cleared before we arrive back here again.
# This is also why we pass around the prompt_opts array.
[[ -o promptbang ]] && prompt_opts+=(bang)
[[ -o promptcr ]] && prompt_opts+=(cr)
[[ -o promptpercent ]] && prompt_opts+=(percent)
[[ -o promptsp ]] && prompt_opts+=(sp)
[[ -o promptsubst ]] && prompt_opts+=(subst)
zstyle -e :prompt-theme restore "
zstyle -d :prompt-theme restore
prompt_default_setup
${PS1+PS1=${(q+)PS1}}
${PS2+PS2=${(q+)PS2}}
${PS3+PS3=${(q+)PS3}}
${PS4+PS4=${(q+)PS4}}
${RPS1+RPS1=${(q+)RPS1}}
${RPS2+RPS2=${(q+)RPS2}}
${RPROMPT+RPROMPT=${(q+)RPROMPT}}
${RPROMPT2+RPROMPT2=${(q+)RPROMPT2}}
${PSVAR+PSVAR=${(q+)PSVAR}}
prompt_opts=( $prompt_opts[*] )
reply=( yes )
"
}
set_prompt "$@"
(( ${#prompt_opts} )) &&
setopt noprompt{bang,cr,percent,sp,subst} "prompt${^prompt_opts[@]}"
true
}
prompt_preview_theme () {
emulate -L zsh
# Minimal preview for prompts that don't supply one
local -a prompt_opts
print -n "$1 theme"
(( $#* > 1 )) && print -n " with parameters \`$*[2,-1]'"
print ":"
zstyle -t :prompt-theme cleanup
prompt_${1}_setup "$@[2,-1]"
(( ${#prompt_opts} )) &&
setopt noprompt{bang,cr,percent,sp,subst} "prompt${^prompt_opts[@]}"
[[ -n ${chpwd_functions[(r)prompt_${1}_chpwd]} ]] &&
prompt_${1}_chpwd
[[ -n ${precmd_functions[(r)prompt_${1}_precmd]} ]] &&
prompt_${1}_precmd
# We'd like to call zle-line-init/finish hooks, too, but that's not possible
# while the ZLE is not active.
[[ -o promptcr ]] && print -n $'\r'
:; print -P -- "${PS1}command arg1 arg2 ... argn"
[[ -n ${preexec_functions[(r)prompt_${1}_preexec]} ]] &&
prompt_${1}_preexec
}
[[ -o kshautoload ]] || promptinit "$@"
|
