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_SHOW_0_EXIT_CODE="0"
  29 ADMONE_INCLUDE="admone-pwd;admone-vcs;admone-exit-code"
  30 ADMONE_FORMAT_LOWER="%B%F{%(?.cyan.red)}"
  31
  32 zstyle ':vcs_info:git:*' formats '%b%u%c'
  33 zstyle ':vcs_info:git:*' actionformats '%b|%a%u%c'
  34 zstyle ':vcs_info:*' unstagedstr ' *'
  35 zstyle ':vcs_info:*' stagedstr ' +'
  36
  37 admone-pwd() {
  38   echo "%F{blue}%~%f"
  39 }
  40 admone-vcs() {
  41   echo "%F{green}${vcs_info_msg_0_}%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   case $KEYMAP in
  70     vicmd)
  71       # Block
  72       print -n -- "\E]50;CursorShape=0\C-G"
  73       ;;
  74     viins|main)
  75       # Beam
  76       print -n -- "\E]50;CursorShape=1\C-G"
  77       ;;
  78   esac
  79   zle reset-prompt
  80   zle -R
  81 }
  82
  83 set-prompt() {
  84   # Reset prompts
  85   upper="${ADMONE_PRE_UPPER}"
  86   lower="${ADMONE_PRE_LOWER}${ADMONE_FORMAT_LOWER}${ADMONE_PROMPT_CHAR}%f%b "
  87
  88   # Add in functions as defined above
  89   NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
  90
  91   THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f 1)"
  92   upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  93
  94   for (( i = 0; i < $NUM_FUNCS; i++ )); do
  95     THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $(($i + 2)))"
  96     upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
  97   done
  98
  99   # Add line above
 100   [[ -v PS1_NL ]] && echo || PS1_NL=
 101
 102   PROMPT=$upper$'\n'$lower
 103 }
 104
 105 # Set everything!
 106 if [[ ! $(echo $ADMONE_INCLUDE | grep -q "vcs") ]]; then
 107   zstyle ':vcs_info:*' check-for-changes true
 108   autoload -Uz add-zsh-hook vcs-info
 109   add-zsh-hook precmd vcs_info
 110   add-zsh-hook precmd set-prompt
 111 else
 112   autoload -Uz add-zsh-hook
 113   add-zsh-hook precmd set-prompt
 114 fi
 115
 116 zle -N zle-line-init
 117 zle -N zle-line-finish
 118 zle -N zle-keymap-select