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