]> git.r.bdr.sh - rbdr/dotfiles/blob - zsh/modules/prompt.zsh
Simplify prompt functions
[rbdr/dotfiles] / zsh / modules / prompt.zsh
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