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_VI_MODE_CURSOR="1"
  19 ADMONE_INCLUDE="admone-pwd;admone-git;admone-exit-code;admone-12hr;admone-time"
  20
  21 function admone-pwd {
  22   echo "%F{blue}%~%f%F{240}"
  23 }
  24 function admone-git {
  25   local git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
  26   echo "%F{green}${git_branch//\%/%%}%f"
  27 }
  28 function admone-exit-code {
  29   echo "%(?,,%F{1}%?%F{240})"
  30 }
  31 function admone-date {
  32   echo $(date -I)
  33 }
  34 function admone-12hr {
  35   echo $(date +%r)
  36 }
  37 function admone-24hr {
  38   echo $(date +%x)
  39 }
  40 function admone-time {
  41   # Intergration with https://codeberg.org/armaan/zsh-command-time
  42   echo %F{cyan}$timer_show%f
  43 }
  44
  45 #######################################################################
  46
  47 #
  48 # BACKEND
  49 #
  50
  51 # Cursor shape depends on vi mode
  52 function zle-keymap-select zle-line-init zle-line-finish {
  53   if [[ $ADMONE_VI_MODE_CURSOR == 1 ]]; then
  54     case $KEYMAP in
  55       vicmd)
  56         print -n -- "\E]50;CursorShape=0\C-G"
  57         ;;
  58       viins|main)
  59         print -n -- "\E]50;CursorShape=1\C-G"
  60         ;;
  61       *)
  62         ;;
  63     esac
  64
  65     zle reset-prompt
  66     zle -R
  67   fi
  68 }
  69
  70 function set-prompt() {
  71   upper="${ADMONE_PRE_UPPER}"
  72   lower="${ADMONE_PRE_LOWER}%B%F{%(?.cyan.red)}${ADMONE_PROMPT_CHAR}%f%b "
  73
  74   NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
  75   REAL_NUM_FUNCS=$((NUM_FUNCS+1))
  76
  77   for (( i = 0; i <= $NUM_FUNCS; i++ )); do
  78     (( val = $i + 1 ))
  79     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
  80
  81     if [[ $val == 1 ]]; then
  82       upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  83     else
  84       upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  85     fi
  86   done
  87
  88   [[ -v PS1_NL ]] && echo || PS1_NL=
  89   PROMPT=$upper$'\n'$lower
  90 }
  91
  92 zle -N zle-line-init
  93 zle -N zle-line-finish
  94 zle -N zle-keymap-select
  95 autoload -Uz add-zsh-hook
  96 add-zsh-hook precmd set-prompt