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_INCLUDE="admone-pwd;admone-git;admone-exit-code"
  34 ADMONE_FORMAT_LOWER="%B%F{%(?.cyan.red)}"
  35
  36 admone-pwd() {
  37   echo "%F{blue}%~%f"
  38 }
  39 admone-git() {
  40   git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
  41   echo "%F{green}${git_branch}%f"
  42 }
  43 admone-exit-code() {
  44   ([[ $ADMONE_SHOW_0_EXIT_CODE == "1" ]] && echo "%(?,%F{cyan}%?%f,%F{red}%?%f)") ||
  45   echo "%(?,,%F{red}%?%f)"
  46 }
  47 admone-date() {
  48   echo $(date -I)
  49 }
  50 admone-12hr() {
  51   echo $(date +%r)
  52 }
  53 admone-24hr() {
  54   echo $(date +%T)
  55 }
  56 # Intergration with https://codeberg.org/armaan/zsh-command-time
  57 admone-time() {
  58   echo "%F{cyan}$timer_show%f"
  59 }
  60
  61 #######################################################################
  62
  63 #
  64 # BACKEND
  65 #
  66
  67 # Cursor shape depends on vi mode
  68 function zle-keymap-select zle-line-init zle-line-finish {
  69   if [[ $ADMONE_VI_MODE_CURSOR == "1" ]]; then
  70     case $KEYMAP in
  71       vicmd)
  72         # Block
  73         print -n -- "\E]50;CursorShape=0\C-G"
  74         ;;
  75       viins|main)
  76         # Beam
  77         print -n -- "\E]50;CursorShape=1\C-G"
  78         ;;
  79     esac
  80     zle reset-prompt
  81     zle -R
  82   fi
  83 }
  84
  85 set-prompt() {
  86   # Reset prompts
  87   upper="${ADMONE_PRE_UPPER}"
  88   lower="${ADMONE_PRE_LOWER}${ADMONE_FORMAT_LOWER}${ADMONE_PROMPT_CHAR}%f%b "
  89
  90   # Add in functions as defined above
  91   NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
  92   for (( i = 0; i <= $NUM_FUNCS; i++ )); do
  93     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $(($i + 1)))"
  94     [[ $i == 0 ]] && upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  95     [[ $i == 0 ]] || upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  96   done
  97
  98   # Add line above
  99   [[ -v PS1_NL ]] && echo || PS1_NL=
 100
 101   PROMPT=$upper$'\n'$lower
 102 }
 103
 104 # Set everything!
 105 zle -N zle-line-init
 106 zle -N zle-line-finish
 107 zle -N zle-keymap-select
 108 autoload -Uz add-zsh-hook
 109 add-zsh-hook precmd set-prompt