]> git.r.bdr.sh - rbdr/dotfiles/blame - zsh/modules/prompt.zsh
Update readme to be more accurate
[rbdr/dotfiles] / zsh / modules / prompt.zsh
CommitLineData
fa250235
BB
1autoload -Uz promptinit && promptinit
2
3if [[ "$TERM" == (dumb|linux|*bsd*) ]] || (( $#prompt_argv < 1 )); then
4 prompt 'off'
5fi
6
7#
8# Precmd
9#
10
11function prompt_precmd {
a40a5ec5
BB
12 git-info
13}
14
fa250235
BB
15#
16# Functions
17#
18
fab93d61
RBR
19function env_info_provider() { echo "%{%F{green}%}n/a"; }
20function env_info {
21 echo "[%{%F{cyan}%}ENV%{%F{black}%} $(env_info_provider)%{%F{black}%}]"
22}
23
24function python_info {
3cfb2402
RBR
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}"
01e20652
BB
28}
29
fab93d61 30function ruby_info {
3cfb2402
RBR
31 rubyinfo=`cat ~/.tool-versions | rg ruby | cut -d ' ' -f 2`
32 echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo:-no}"
01e20652
BB
33}
34
fab93d61 35function node_info {
3cfb2402
RBR
36 nodeinfo=`cat ~/.tool-versions | rg nodejs | cut -d ' ' -f 2`
37 echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}"
a40a5ec5
BB
38}
39
40function box_name {
41 [ -f ~/.box-name ] && cat ~/.box-name || hostname -s
42}
43
fa250235
BB
44#
45# ZLE widgets
46#
47
a40a5ec5 48function zle-line-init zle-keymap-select {
ec48932f
BB
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
64c29a52
RBR
53 local normal_symbol="N"
54 local insert_symbol="I"
55 local error_symbol="X"
ec48932f
BB
56
57 # Show vi status / return status on the right side
58
59 local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}"
fa250235 60 if [ $KEYMAP = 'vicmd' ]; then
ec48932f 61 local edit_status="%{%F{cyan}%}$normal_symbol"
fa250235 62 else
ec48932f 63 local edit_status="%{%F{green}%}$insert_symbol"
fa250235
BB
64 fi
65
ec48932f 66 RPROMPT="${edit_status}${return_status}%{$reset_color%}"
a40a5ec5
BB
67 zle reset-prompt
68}
69zle -N zle-line-init
70zle -N zle-keymap-select
71
fa250235
BB
72#
73# Setup Prompt
74#
75
76function prompt_setup {
a40a5ec5
BB
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.
fa250235 85 add-zsh-hook precmd prompt_precmd
a40a5ec5
BB
86
87 # Set git-info parameters.
fa250235
BB
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 \
a40a5ec5
BB
95 'prompt' '%b%a%m%u'
96
97 # Define prompts.
98 PROMPT='
fab93d61 99%{%F{black}%}$(env_info) $(node_info) $(ruby_info) $(python_info)
e8cfa78c 100%{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}${git_info[prompt]} %{%f%}
a40a5ec5
BB
101%(?,,%{${%B%F{white}%}[%?]%{%f%b%} )$ '
102}
103
fa250235
BB
104prompt_setup "$@"
105setopt PROMPT_SUBST
ec48932f
BB
106setopt PROMPT_PERCENT
107setopt TRANSIENT_RPROMPT