admone.zsh

   1 #######################################################################
   2 # Admone ZSH prompt. Copyright Armaan Bhojwani, palb91, romkatv.      #
   3 # MIT licensed, see the LICENSE file or                               #
   4 # https://www.opensource.org/licenses/MIT for more information.       #
   5 # Git repo at https://codeberg.org/armaan/admone                      #
   6 #######################################################################
   7
   8 #
   9 # USER CONFIGURATIONS
  10 #
  11
  12 ADMONE_PROMPT_CHAR="❯❯❯"
  13 ADMONE_PRE_UPPER="╭ "
  14 ADMONE_PRE_LOWER="╰ "
  15 ADMONE_BLOCK_SEPARATOR="-"
  16 ADMONE_BLOCK_LEFT="["
  17 ADMONE_BLOCK_RIGHT="]"
  18 ADMONE_INCLUDE="admone-pwd;admone-git;admone-exit-code"
  19
  20 function admone-pwd {
  21   echo "%F{blue}%~%f%F{240}"
  22 }
  23 function admone-git {
  24   local git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
  25   echo "%F{green}${git_branch//\%/%%}%f"
  26 }
  27 function admone-exit-code {
  28   echo "%(?,,%F{1}%?%F{240})"
  29 }
  30
  31 #######################################################################
  32
  33 #
  34 # BACKEND
  35 #
  36
  37 # Cursor shape depends on vi mode
  38 function zle-keymap-select zle-line-init zle-line-finish {
  39   case $KEYMAP in
  40     vicmd)
  41       print -n -- "\E]50;CursorShape=0\C-G"
  42       ;; # block cursor
  43     viins|main)
  44       print -n -- "\E]50;CursorShape=1\C-G"
  45       ;; # line cursor
  46     *)
  47       ;;
  48   esac
  49
  50   zle reset-prompt
  51   zle -R
  52 }
  53
  54 function set-prompt() {
  55   upper="%B${ADMONE_PRE_UPPER}"
  56   lower="%B${ADMONE_PRE_LOWER}%B%F{%(?.cyan.red)}${ADMONE_PROMPT_CHAR}%f%b "
  57
  58   for (( i = 0; i <= $(echo $ADMONE_INCLUDE | grep -o ';' | wc -l); i++ )); do
  59     (( val = $i + 1 ))
  60     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
  61
  62     if [[ $((val%2)) == 0 ]]; then
  63         upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
  64       else
  65         upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  66       fi
  67   done
  68
  69   PROMPT=$upper$'\n'$lower
  70 }
  71
  72 zle -N zle-line-init
  73 zle -N zle-line-finish
  74 zle -N zle-keymap-select
  75 autoload -Uz add-zsh-hook
  76 add-zsh-hook precmd set-prompt