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