admone.zsh

   1 #######################################################################
   2 # Admone ZSH prompt. Copyright Armaan Bhojwani. MIT licensed, see the #
   3 # LICENSE file or https://www.opensource.org/licenses/MIT for more    #
   4 # information. Git repo at https://codeberg.org/armaan/admone         #
   5 #######################################################################
   6
   7 #
   8 # USER CONFIGURATIONS
   9 #
  10
  11 ADMONE_PROMPT_CHAR="➤"
  12   # ADMONE_PROMPT_CHAR="❯❯❯"
  13   # ADMONE_PROMPT_CHAR="$"
  14   # ADMONE_PROMPT_CHAR="%"
  15   # ADMONE_PROMPT_CHAR=">"
  16   # ADMONE_PROMPT_CHAR="➢"
  17   # ADMONE_PROMPT_CHAR="➪"
  18 ADMONE_PRE_UPPER="╭"
  19 ADMONE_PRE_LOWER="╰"
  20   # ADMONE_PRE_UPPER="┌"
  21   # ADMONE_PRE_LOWER="└"
  22 ADMONE_BLOCK_SEPARATOR="─"
  23   # ADMONE_BLOCK_SEPARATOR="∙"
  24 ADMONE_BLOCK_LEFT="["
  25 ADMONE_BLOCK_RIGHT="]"
  26   # ADMONE_BLOCK_LEFT="{"
  27   # ADMONE_BLOCK_RIGHT="}"
  28   # ADMONE_BLOCK_LEFT="("
  29   # ADMONE_BLOCK_RIGHT=")"
  30 ADMONE_VI_MODE_CURSOR="1"
  31   # ADMONE_VI_MODE_CURSOR="1"
  32 ADMONE_SHOW_0_EXIT_CODE="0"
  33 ADMONE_GIT_SHOW_BLANK="1"
  34 ADMONE_INCLUDE="admone-pwd;admone-git;admone-exit-code;admone-time"
  35 ADMONE_FORMAT_LOWER="%B%F{%(?.cyan.red)}"
  36
  37 function admone-pwd {
  38   echo "%F{blue}%~%f"
  39 }
  40 function admone-git {
  41   git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
  42   if [[ $ADMONE_GIT_SHOW_BLANK == "0" ]]; then
  43     if [[ ${git_branch//\%/%%} == "" ]]; then
  44       echo "%F{blue}n/a%f"
  45     else
  46       echo "%F{green}${git_branch//\%/%%}%f"
  47     fi
  48   else
  49     echo "%F{green}${git_branch//\%/%%}%f"
  50   fi
  51 }
  52 function admone-exit-code {
  53   if [[ $ADMONE_SHOW_0_EXIT_CODE == "1" ]]; then
  54     echo "%(?,%F{cyan}%?%f,%F{red}%?%f)"
  55   elif [[ $ADMONE_SHOW_0_EXIT_CODE == "0" ]]; then
  56     echo "%(?,,%F{red}%?%f)"
  57   fi
  58 }
  59 function admone-date {
  60   echo $(date -I)
  61 }
  62 function admone-12hr {
  63   echo $(date +%r)
  64 }
  65 function admone-24hr {
  66   echo $(date +%T)
  67 }
  68 # Intergration with https://codeberg.org/armaan/zsh-command-time
  69 function admone-time {
  70   echo "%F{cyan}$timer_show%f"
  71 }
  72
  73 #######################################################################
  74
  75 #
  76 # BACKEND
  77 #
  78
  79 # Cursor shape depends on vi mode
  80 function zle-keymap-select zle-line-init zle-line-finish {
  81   if [[ $ADMONE_VI_MODE_CURSOR == "1" ]]; then
  82     case $KEYMAP in
  83       vicmd)
  84         # Block
  85         print -n -- "\E]50;CursorShape=0\C-G"
  86         ;;
  87       viins|main)
  88         # Beam
  89         print -n -- "\E]50;CursorShape=1\C-G"
  90         ;;
  91     esac
  92     zle reset-prompt
  93     zle -R
  94   fi
  95 }
  96
  97 function set-prompt() {
  98   # Reset prompts
  99   upper="${ADMONE_PRE_UPPER}"
 100   lower="${ADMONE_PRE_LOWER}${ADMONE_FORMAT_LOWER}${ADMONE_PROMPT_CHAR}%f%b "
 101
 102   # Add in functions as defined above
 103   NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
 104   for (( i = 0; i <= $NUM_FUNCS; i++ )); do
 105     (( val = $i + 1 ))
 106     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
 107    if [[ $i == 0 ]]; then
 108       upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
 109     else
 110       upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
 111     fi
 112   done
 113
 114   # Add line above
 115   [[ -v PS1_NL ]] && echo || PS1_NL=
 116
 117   PROMPT=$upper$'\n'$lower
 118 }
 119
 120 # Set everything!
 121 zle -N zle-line-init
 122 zle -N zle-line-finish
 123 zle -N zle-keymap-select
 124 autoload -Uz add-zsh-hook
 125 add-zsh-hook precmd set-prompt