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   pythoninfo=${ASDF_PYTHON_VERSION:-$pythoninfo}
  27   venv="${VIRTUAL_ENV##*/}"
  28   echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}"
  29 }
  30
  31 function ruby_info {
  32   rubyinfo=`cat ~/.tool-versions | rg ruby | cut -d ' ' -f 2`
  33   rubyinfo=${ASDF_RUBY_VERSION:-$rubyinfo}
  34   echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo:-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 box_name {
  44     [ -f ~/.box-name ] && cat ~/.box-name || hostname -s
  45 }
  46
  47 #
  48 # ZLE widgets
  49 #
  50
  51 function zle-line-init zle-keymap-select {
  52
  53     # Couldn't figure out how to get tmux and iTerm to agree on how to render
  54     # these characters... so there you go.
  55
  56     local normal_symbol="N"
  57     local insert_symbol="I"
  58     local error_symbol="X"
  59
  60     # Show vi status / return status on the right side
  61
  62     local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}"
  63     if [ $KEYMAP = 'vicmd' ]; then
  64       local edit_status="%{%F{cyan}%}$normal_symbol"
  65     else
  66       local edit_status="%{%F{green}%}$insert_symbol"
  67     fi
  68
  69     RPROMPT="${edit_status}${return_status}%{$reset_color%}"
  70     zle reset-prompt
  71 }
  72 zle -N zle-line-init
  73 zle -N zle-keymap-select
  74
  75 #
  76 # Setup Prompt
  77 #
  78
  79 function prompt_setup {
  80   setopt LOCAL_OPTIONS
  81   unsetopt XTRACE KSH_ARRAYS
  82   prompt_opts=(cr percent subst)
  83
  84   # Load required functions.
  85   autoload -Uz add-zsh-hook
  86
  87   # Add hook for calling git-info before each command.
  88   add-zsh-hook precmd prompt_precmd
  89
  90   # Set git-info parameters.
  91   zstyle ':git:info' verbose 'no'
  92   zstyle ':git:info:branch' format '@%F{cyan}%b%f'
  93   zstyle ':git:info:clean' format ''
  94   zstyle ':git:info:added' format '%{%F{green}%}!'
  95   zstyle ':git:info:modified' format '%{%F{green}%}?'
  96   zstyle ':git:info:untracked' format '%{%F{magenta}%}?'
  97   zstyle ':git:info:keys' format \
  98     'prompt' '%b%a%m%u'
  99
 100   # Define prompts.
 101   PROMPT='
 102 %{%F{black}%}$(env_info) $(node_info) $(ruby_info) $(python_info)
 103 %{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}${git_info[prompt]} %{%f%}
 104 %(?,,%{${%B%F{white}%}[%?]%{%f%b%} )$ '
 105 }
 106
 107 prompt_setup "$@"
 108 setopt PROMPT_SUBST
 109 setopt PROMPT_PERCENT
 110 setopt TRANSIENT_RPROMPT