]>
Commit | Line | Data |
---|---|---|
1 | autoload -Uz promptinit && promptinit | |
2 | ||
3 | if [[ "$TERM" == (dumb|linux|*bsd*) ]] || (( $#prompt_argv < 1 )); then | |
4 | prompt 'off' | |
5 | fi | |
6 | ||
7 | # | |
8 | # Precmd | |
9 | # | |
10 | ||
11 | function prompt_precmd { | |
12 | git-info | |
13 | } | |
14 | ||
15 | # | |
16 | # Functions | |
17 | # | |
18 | ||
19 | function env_info_provider() { echo "%{%F{green}%}n/a"; } | |
20 | function env_info { | |
21 | echo "[%{%F{cyan}%}ENV%{%F{black}%} $(env_info_provider)%{%F{black}%}]" | |
22 | } | |
23 | ||
24 | function python_info { | |
25 | pythoninfo=`cat ~/.tool-versions | rg python | cut -d ' ' -f 2` | |
26 | venv="${VIRTUAL_ENV##*/}" | |
27 | echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}" | |
28 | } | |
29 | ||
30 | function ruby_info { | |
31 | rubyinfo=`cat ~/.tool-versions | rg ruby | cut -d ' ' -f 2` | |
32 | echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo:-no}" | |
33 | } | |
34 | ||
35 | function node_info { | |
36 | nodeinfo=`cat ~/.tool-versions | rg nodejs | cut -d ' ' -f 2` | |
37 | echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}" | |
38 | } | |
39 | ||
40 | function box_name { | |
41 | [ -f ~/.box-name ] && cat ~/.box-name || hostname -s | |
42 | } | |
43 | ||
44 | # | |
45 | # ZLE widgets | |
46 | # | |
47 | ||
48 | function zle-line-init zle-keymap-select { | |
49 | ||
50 | # Couldn't figure out how to get tmux and iTerm to agree on how to render | |
51 | # these characters... so there you go. | |
52 | ||
53 | local normal_symbol="N" | |
54 | local insert_symbol="I" | |
55 | local error_symbol="X" | |
56 | ||
57 | # Show vi status / return status on the right side | |
58 | ||
59 | local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}" | |
60 | if [ $KEYMAP = 'vicmd' ]; then | |
61 | local edit_status="%{%F{cyan}%}$normal_symbol" | |
62 | else | |
63 | local edit_status="%{%F{green}%}$insert_symbol" | |
64 | fi | |
65 | ||
66 | RPROMPT="${edit_status}${return_status}%{$reset_color%}" | |
67 | zle reset-prompt | |
68 | } | |
69 | zle -N zle-line-init | |
70 | zle -N zle-keymap-select | |
71 | ||
72 | # | |
73 | # Setup Prompt | |
74 | # | |
75 | ||
76 | function prompt_setup { | |
77 | setopt LOCAL_OPTIONS | |
78 | unsetopt XTRACE KSH_ARRAYS | |
79 | prompt_opts=(cr percent subst) | |
80 | ||
81 | # Load required functions. | |
82 | autoload -Uz add-zsh-hook | |
83 | ||
84 | # Add hook for calling git-info before each command. | |
85 | add-zsh-hook precmd prompt_precmd | |
86 | ||
87 | # Set git-info parameters. | |
88 | zstyle ':git:info' verbose 'no' | |
89 | zstyle ':git:info:branch' format '@%F{cyan}%b%f' | |
90 | zstyle ':git:info:clean' format '' | |
91 | zstyle ':git:info:added' format '%{%F{green}%}!' | |
92 | zstyle ':git:info:modified' format '%{%F{green}%}?' | |
93 | zstyle ':git:info:untracked' format '%{%F{magenta}%}?' | |
94 | zstyle ':git:info:keys' format \ | |
95 | 'prompt' '%b%a%m%u' | |
96 | ||
97 | # Define prompts. | |
98 | PROMPT=' | |
99 | %{%F{black}%}$(env_info) $(node_info) $(ruby_info) $(python_info) | |
100 | %{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}${git_info[prompt]} %{%f%} | |
101 | %(?,,%{${%B%F{white}%}[%?]%{%f%b%} )$ ' | |
102 | } | |
103 | ||
104 | prompt_setup "$@" | |
105 | setopt PROMPT_SUBST | |
106 | setopt PROMPT_PERCENT | |
107 | setopt TRANSIENT_RPROMPT |