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 virtualenv_info {
  20   if [[ -z "$VIRTUAL_ENV" ]]; then
  21     venv="default"
  22   else
  23     venv="${VIRTUAL_ENV##*/}"
  24   fi
  25   echo "🐍 ${venv}"
  26 }
  27
  28 function rvm_info {
  29   if [[ -z "$RUBY_VERSION" ]]; then
  30     rvminfo="default"
  31   else
  32     rvminfo=`rvm-prompt`
  33   fi
  34   echo "💎 ${rvminfo}"
  35 }
  36
  37 function nvm_info {
  38   if [[ -z "$NVM_BIN" ]]; then
  39     nvminfo="default"
  40   else
  41     nvminfo=`nvm current`
  42   fi
  43   echo "💛 ${nvminfo}"
  44 }
  45
  46 function box_name {
  47     [ -f ~/.box-name ] && cat ~/.box-name || hostname -s
  48 }
  49
  50 #
  51 # ZLE widgets
  52 #
  53
  54 function zle-line-init zle-keymap-select {
  55
  56     # Couldn't figure out how to get tmux and iTerm to agree on how to render
  57     # these characters... so there you go.
  58
  59     local normal_symbol="🅽"
  60     local insert_symbol="🅸"
  61     local error_symbol="🆇"
  62
  63     [ ! -z "$TMUX" ] && normal_symbol="%{$normal_symbol%2G%}"
  64     [ ! -z "$TMUX" ] && insert_symbol="%{$insert_symbol%2G%}"
  65     [ ! -z "$TMUX" ] && error_symbol="%{$error_symbol%2G%}"
  66
  67     # Show vi status / return status on the right side
  68
  69     local return_status="%{%F{red}%}%(?.. $error_symbol)%{$reset_color%}"
  70     if [ $KEYMAP = 'vicmd' ]; then
  71       local edit_status="%{%F{cyan}%}$normal_symbol"
  72     else
  73       local edit_status="%{%F{green}%}$insert_symbol"
  74     fi
  75
  76     RPROMPT="${edit_status}${return_status}%{$reset_color%}"
  77     zle reset-prompt
  78 }
  79 zle -N zle-line-init
  80 zle -N zle-keymap-select
  81
  82 #
  83 # Setup Prompt
  84 #
  85
  86 function prompt_setup {
  87   setopt LOCAL_OPTIONS
  88   unsetopt XTRACE KSH_ARRAYS
  89   prompt_opts=(cr percent subst)
  90
  91   # Load required functions.
  92   autoload -Uz add-zsh-hook
  93
  94   # Add hook for calling git-info before each command.
  95   add-zsh-hook precmd prompt_precmd
  96
  97   # Set git-info parameters.
  98   zstyle ':git:info' verbose 'no'
  99   zstyle ':git:info:branch' format '@%F{cyan}%b%f'
 100   zstyle ':git:info:clean' format ''
 101   zstyle ':git:info:added' format '%{%F{green}%}!'
 102   zstyle ':git:info:modified' format '%{%F{green}%}?'
 103   zstyle ':git:info:untracked' format '%{%F{magenta}%}?'
 104   zstyle ':git:info:keys' format \
 105     'prompt' '%b%a%m%u'
 106
 107   # Define prompts.
 108   PROMPT='
 109 %{%F{black}%}$(nvm_info) $(rvm_info) $(virtualenv_info)
 110 %{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}${git_info[prompt]} %{%f%}
 111 %(?,,%{${%B%F{white}%}[%?]%{%f%b%} )$ '
 112 }
 113
 114 prompt_setup "$@"
 115 setopt PROMPT_SUBST
 116 setopt PROMPT_PERCENT
 117 setopt TRANSIENT_RPROMPT