git.r.bdr.sh - rbdr/dotfiles/blob - 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