]> git.r.bdr.sh - rbdr/dotfiles/blob - zsh/modules/prompt.zsh
Use tree-sitter-cli
[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 # 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 pythoninfo=${ASDF_PYTHON_VERSION:-$pythoninfo}
27 venv="${VIRTUAL_ENV##*/}"
28 echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}"
29 }
30
31 function rust_info {
32 rustinfo=`cat "$(tool_versions)" | rg rust | cut -d ' ' -f 2`
33 rustinfo=${ASDF_RUST_VERSION:-$rustinfo}
34 echo "%{%F{red}%}rs%{%F{black}%} ${rustinfo:-no}"
35 }
36
37 function node_info {
38 nodeinfo=`cat "$(tool_versions)" | rg nodejs | cut -d ' ' -f 2`
39 nodeinfo=${ASDF_NODEJS_VERSION:-$nodeinfo}
40 echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}"
41 }
42
43 function git_info {
44 branch=$(git branch --show-current 2> /dev/null)
45
46 if [ ! -z $branch ]; then
47 echo -n "@%F{cyan}$branch%f"
48
49 status_output=$(git status --short)
50
51 if echo "$status_output" | grep -q '^??'; then
52 echo -n "%{%F{magenta}%}?%f"
53 fi
54
55 if echo "$status_output" | grep -q '^ M'; then
56 echo -n "%{%F{magenta}%}!%f"
57 fi
58
59 if echo "$status_output" | grep -q '^ D'; then
60 echo -n "%{%F{magenta}%}×%f"
61 fi
62
63 if echo "$status_output" | grep -q '^A '; then
64 echo -n "%{%F{green}%}+%f"
65 fi
66 fi
67 }
68
69 function box_name {
70 [ -f ~/.box-name ] && cat ~/.box-name || hostname -s
71 }
72
73 #
74 # ZLE widgets
75 #
76
77 function zle-line-init zle-keymap-select {
78
79 # Couldn't figure out how to get tmux and iTerm to agree on how to render
80 # these characters... so there you go.
81
82 local normal_symbol="N"
83 local insert_symbol="I"
84 local error_symbol="X"
85
86 # Show vi status / return status on the right side
87
88 local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}"
89 if [ $KEYMAP = 'vicmd' ]; then
90 local edit_status="%{%F{cyan}%}$normal_symbol"
91 else
92 local edit_status="%{%F{green}%}$insert_symbol"
93 fi
94
95 RPROMPT="${edit_status}${return_status}%{$reset_color%}"
96 zle reset-prompt
97 }
98 zle -N zle-line-init
99 zle -N zle-keymap-select
100
101 #
102 # Setup Prompt
103 #
104
105 function prompt_setup {
106 setopt LOCAL_OPTIONS
107 unsetopt XTRACE KSH_ARRAYS
108 prompt_opts=(cr percent subst)
109
110 # Load required functions.
111 autoload -Uz add-zsh-hook
112
113 # Define prompts.
114 PROMPT='
115 %{%F{black}%}$(env_info) $(node_info) $(rust_info) $(python_info)
116 %{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}$(git_info)%{%f%}
117 %(?,,%{${%B%F{white}%}[%?]%{%f%b%} )%% '
118 }
119
120 prompt_setup "$@"
121 setopt PROMPT_SUBST
122 setopt PROMPT_PERCENT
123 setopt TRANSIENT_RPROMPT