admone.zsh

   1 # Admone Zsh prompt
   2 # Copyright Armaan Bhojwani 2020 <code@armaanb.net>
   3 # MIT licensed, see the LICENSE file for more information.
   4
   5 #######################################################################
   6 # USER CONFIGURATIONS
   7 #######################################################################
   8
   9 PROMPT_CHAR="➤"
  10   # PROMPT_CHAR="❯❯❯"
  11   # PROMPT_CHAR="$"
  12   # PROMPT_CHAR="%"
  13   # PROMPT_CHAR=">"
  14   # PROMPT_CHAR="➢"
  15   # PROMPT_CHAR="➪"
  16 PRE_UPPER="╭"
  17 PRE_LOWER="╰"
  18   # PRE_UPPER="┌"
  19   # PRE_LOWER="└"
  20 BLOCK_SEPARATOR="─"
  21   # BLOCK_SEPARATOR="∙"
  22 BLOCK_LEFT="["
  23 BLOCK_RIGHT="]"
  24   # BLOCK_LEFT="("
  25   # BLOCK_RIGHT=")"
  26 SHOW_0_EXIT_CODE="0"
  27 INCLUDE="admone-pwd;admone-vcs;admone-exit-code"
  28 FORMAT_LOWER="%B%F{%(?.cyan.red)}"
  29
  30 TRANSIENT_PROMPT="0" # This is buggy
  31 SHORT_PROMPT="➤"
  32 FORMAT_SHORT="%B%F{%(?.cyan.red)}"
  33
  34 zstyle ':vcs_info:git:*' formats '%b%u%c'
  35 zstyle ':vcs_info:git:*' actionformats '%b|%a%u%c'
  36 zstyle ':vcs_info:*' unstagedstr ' *'
  37 zstyle ':vcs_info:*' stagedstr ' +'
  38
  39 #######################################################################
  40 # Functions
  41 #######################################################################
  42
  43 admone-pwd() {
  44   echo "%F{blue}%~%f"
  45 }
  46 admone-vcs() {
  47   echo "%F{green}${vcs_info_msg_0_}%f"
  48 }
  49 admone-exit-code() {
  50   ([[ $SHOW_0_EXIT_CODE == "1" ]] && echo "%(?,%F{cyan}%?%f,%F{red}%?%f)") ||
  51     echo "%(?,,%F{red}%?%f)"
  52 }
  53 admone-date() {
  54   echo $(date -I)
  55 }
  56 admone-12hr() {
  57   echo $(date +%r)
  58 }
  59 admone-24hr() {
  60   echo $(date +%T)
  61 }
  62
  63 #######################################################################
  64 # BACKEND
  65 #######################################################################
  66
  67 # Cursor shape depends on vi mode
  68 function zle-keymap-select zle-line-init zle-line-finish {
  69   case $KEYMAP in
  70     vicmd)
  71       # Block
  72       print -n -- "\E]50;CursorShape=0\C-G"
  73       ;;
  74     viins|main)
  75       # Beam
  76       print -n -- "\E]50;CursorShape=1\C-G"
  77       ;;
  78   esac
  79   zle reset-prompt
  80   zle -R
  81 }
  82
  83 set-prompt() {
  84   # Reset prompts
  85   upper="${PRE_UPPER}"
  86   lower="${PRE_LOWER}${FORMAT_LOWER}${PROMPT_CHAR}%f%b "
  87
  88   # Add in functions as defined above
  89   NUM_FUNCS="$(echo $INCLUDE | grep -o ';' | wc -l)"
  90
  91   THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f 1)"
  92   upper+="$BLOCK_LEFT$($THE_COMMAND)$BLOCK_RIGHT"
  93
  94   for (( i = 0; i < $NUM_FUNCS; i++ )); do
  95     THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f $(($i + 2)))"
  96     upper+="$BLOCK_SEPARATOR$BLOCK_LEFT$($THE_COMMAND)$BLOCK_RIGHT"
  97   done
  98
  99   # Add line above
 100   [[ -v PS1_NL ]] && echo || PS1_NL=
 101
 102   PROMPT=$upper$'\n'$lower
 103 }
 104
 105
 106 # Set everything!
 107 if [[ $TRANSIENT_PROMPT == "1" ]]; then
 108   SHORT_PROMPT=$FORMAT_SHORT$SHORT_PROMPT' '%f%b
 109
 110   set-transient-prompt() {
 111     if [[ $PROMPT != $SHORT_PROMPT ]]; then
 112       PROMPT=$SHORT_PROMPT
 113       zle .reset-prompt
 114     fi
 115   }
 116
 117   zle-line-finish() {
 118     set-transient-prompt
 119   }
 120
 121   trap 'set-transient-prompt; return 130' INT
 122 fi
 123
 124 if [[ ! $(echo $INCLUDE | grep -q "vcs") ]]; then
 125   zstyle ':vcs_info:*' check-for-changes true
 126   autoload -Uz add-zsh-hook vcs-info
 127   add-zsh-hook precmd vcs_info
 128 fi
 129
 130 autoload -Uz add-zsh-hook
 131 add-zsh-hook precmd set-prompt
 132
 133 zle -N zle-line-init
 134 zle -N zle-line-finish
 135 zle -N zle-keymap-select