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 function admone-date {
  32   echo $(date -I)
  33 }
  34 function admone-time {
  35   echo $(date -Iseconds)
  36 }
  37 #######################################################################
  38
  39 #
  40 # BACKEND
  41 #
  42
  43 # Cursor shape depends on vi mode
  44 function zle-keymap-select zle-line-init zle-line-finish {
  45   case $KEYMAP in
  46     vicmd)
  47       print -n -- "\E]50;CursorShape=0\C-G"
  48       ;;
  49     viins|main)
  50       print -n -- "\E]50;CursorShape=1\C-G"
  51       ;;
  52     *)
  53       ;;
  54   esac
  55
  56   zle reset-prompt
  57   zle -R
  58 }
  59
  60 function set-prompt() {
  61   upper="%B${ADMONE_PRE_UPPER}"
  62   lower="%B${ADMONE_PRE_LOWER}%B%F{%(?.cyan.red)}${ADMONE_PROMPT_CHAR}%f%b "
  63
  64   NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
  65   for (( i = 0; i <= $NUM_FUNCS; i++ )); do
  66     (( val = $i + 1 ))
  67     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
  68
  69     REAL_NUM_FUNCS=$((NUM_FUNCS+1))
  70       if [[ $val == $REAL_NUM_FUNCS ]]; then
  71         upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  72       elif [[ $((REAL_NUM_FUNCS%2)) == 0 ]]; then
  73         if [[ $((val%2)) == 0 ]]; then
  74           upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
  75         else
  76           upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  77         fi
  78       elif [[ $((REAL_NUM_FUNCS%2)) != 0 ]]; then
  79         if [[ $val == 1 ]]; then
  80           upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
  81         elif [[ $((val%2)) != 0 ]]; then
  82           upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
  83         else
  84           upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  85       fi
  86       fi
  87   done
  88
  89   [[ -v PS1_NL ]] && echo || PS1_NL=
  90   PROMPT=$upper$'\n'$lower
  91 }
  92
  93 zle -N zle-line-init
  94 zle -N zle-line-finish
  95 zle -N zle-keymap-select
  96 autoload -Uz add-zsh-hook
  97 add-zsh-hook precmd set-prompt