blob: 18e5cc4c4de2583bc41156688b23dc221c3d4cb0 (
plain)
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
|
autoload -Uz promptinit && promptinit
function tool_versions() {
if [[ -s ".tool-versions" ]]; then
echo '.tool-versions'
return
fi
echo "${HOME}/.tool-versions"
}
if [[ "$TERM" == (dumb|linux|*bsd*) ]] || (( $#prompt_argv < 1 )); then
prompt 'off'
fi
#
# Functions
#
function env_info_provider() { echo "%{%F{green}%}n/a"; }
function env_info {
echo "[%{%F{cyan}%}ENV%{%F{black}%} $(env_info_provider)%{%F{black}%}]"
}
function python_info {
pythoninfo=`cat "$(tool_versions)" | rg python | cut -d ' ' -f 2`
pythoninfo=${ASDF_PYTHON_VERSION:-$pythoninfo}
venv="${VIRTUAL_ENV##*/}"
echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}"
}
function rust_info {
rustinfo=`cat "$(tool_versions)" | rg rust | cut -d ' ' -f 2`
rustinfo=${ASDF_RUST_VERSION:-$rustinfo}
echo "%{%F{red}%}rs%{%F{black}%} ${rustinfo:-no}"
}
function node_info {
nodeinfo=`cat "$(tool_versions)" | rg nodejs | cut -d ' ' -f 2`
nodeinfo=${ASDF_NODEJS_VERSION:-$nodeinfo}
echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}"
}
function git_info {
branch=$(git branch --show-current 2> /dev/null)
if [ ! -z $branch ]; then
echo -n "@%F{cyan}$branch%f"
status_output=$(git status --short)
if echo "$status_output" | grep -q '^??'; then
echo -n "%{%F{magenta}%}?%f"
fi
if echo "$status_output" | grep -q '^ M'; then
echo -n "%{%F{magenta}%}!%f"
fi
if echo "$status_output" | grep -q '^ D'; then
echo -n "%{%F{magenta}%}×%f"
fi
if echo "$status_output" | grep -q '^A '; then
echo -n "%{%F{green}%}+%f"
fi
fi
}
function box_name {
[ -f ~/.box-name ] && cat ~/.box-name || hostname -s
}
#
# ZLE widgets
#
function zle-line-init zle-keymap-select {
# Couldn't figure out how to get tmux and iTerm to agree on how to render
# these characters... so there you go.
local normal_symbol="N"
local insert_symbol="I"
local error_symbol="X"
# Show vi status / return status on the right side
local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}"
if [ $KEYMAP = 'vicmd' ]; then
local edit_status="%{%F{cyan}%}$normal_symbol"
else
local edit_status="%{%F{green}%}$insert_symbol"
fi
RPROMPT="${edit_status}${return_status}%{$reset_color%}"
zle reset-prompt
}
zle -N zle-line-init
zle -N zle-keymap-select
#
# Setup Prompt
#
function prompt_setup {
setopt LOCAL_OPTIONS
unsetopt XTRACE KSH_ARRAYS
prompt_opts=(cr percent subst)
# Load required functions.
autoload -Uz add-zsh-hook
# Define prompts.
PROMPT='
%{%F{black}%}$(env_info) $(node_info) $(rust_info) $(python_info)
%{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}$(git_info)%{%f%}
%(?,,%{${%B%F{white}%}[%?]%{%f%b%} )%% '
}
prompt_setup "$@"
setopt PROMPT_SUBST
setopt PROMPT_PERCENT
setopt TRANSIENT_RPROMPT
|