]> git.armaanb.net Git - admone.git/blobdiff - admone.zsh
disable transient prompt by default
[admone.git] / admone.zsh
index dfbb9c398c939b4cc7da5db1a12f595254ba8a9d..77fd9875fd83e97ed9540f30c4027edadd6bde53 100644 (file)
+# Admone Zsh prompt
+# Copyright Armaan Bhojwani 2020 <code@armaanb.net>
+# MIT licensed, see the LICENSE file for more information.
+# Git repo at <https://codeberg.org/armaan/admone>
+
 #######################################################################
-# Admone ZSH prompt. Copyright Armaan Bhojwani. MIT licensed, see the #
-# LICENSE file or https://www.opensource.org/licenses/MIT for more    #
-# information. Git repo at https://codeberg.org/armaan/admone         #
+# USER CONFIGURATIONS
 #######################################################################
 
-#
-# USER CONFIGURATIONS
-#
-
-ADMONE_PROMPT_CHAR="➤"
-  # ADMONE_PROMPT_CHAR="❯❯❯"
-  # ADMONE_PROMPT_CHAR="$"
-  # ADMONE_PROMPT_CHAR="%"
-  # ADMONE_PROMPT_CHAR=">"
-  # ADMONE_PROMPT_CHAR="➢"
-  # ADMONE_PROMPT_CHAR="➪"
-ADMONE_PRE_UPPER="╭"
-ADMONE_PRE_LOWER="╰"
-  # ADMONE_PRE_UPPER="┌"
-  # ADMONE_PRE_LOWER="└"
-ADMONE_BLOCK_SEPARATOR="─"
-  # ADMONE_BLOCK_SEPARATOR="∙"
-ADMONE_BLOCK_LEFT="["
-ADMONE_BLOCK_RIGHT="]"
-  # ADMONE_BLOCK_LEFT="{"
-  # ADMONE_BLOCK_RIGHT="}"
-  # ADMONE_BLOCK_LEFT="("
-  # ADMONE_BLOCK_RIGHT=")"
-ADMONE_VI_MODE_CURSOR="1"
-  # ADMONE_VI_MODE_CURSOR="1"
-ADMONE_SHOW_0_EXIT_CODE="0"
-ADMONE_GIT_SHOW_BLANK="1"
-ADMONE_INCLUDE="admone-pwd;admone-git;admone-exit-code;admone-time"
-ADMONE_FORMAT_LOWER="%B%F{%(?.cyan.red)}"
-
-function admone-pwd {
+PROMPT_CHAR="➤"
+  # PROMPT_CHAR="❯❯❯"
+  # PROMPT_CHAR="$"
+  # PROMPT_CHAR="%"
+  # PROMPT_CHAR=">"
+  # PROMPT_CHAR="➢"
+  # PROMPT_CHAR="➪"
+PRE_UPPER="╭"
+PRE_LOWER="╰"
+  # PRE_UPPER="┌"
+  # PRE_LOWER="└"
+BLOCK_SEPARATOR="─"
+  # BLOCK_SEPARATOR="∙"
+BLOCK_LEFT="["
+BLOCK_RIGHT="]"
+  # BLOCK_LEFT="("
+  # BLOCK_RIGHT=")"
+SHOW_0_EXIT_CODE="0"
+INCLUDE="admone-pwd;admone-vcs;admone-exit-code"
+FORMAT_LOWER="%B%F{%(?.cyan.red)}"
+
+TRANSIENT_PROMPT="0" # This is buggy
+SHORT_PROMPT="➤"
+FORMAT_SHORT="%B%F{%(?.cyan.red)}"
+
+zstyle ':vcs_info:git:*' formats '%b%u%c'
+zstyle ':vcs_info:git:*' actionformats '%b|%a%u%c'
+zstyle ':vcs_info:*' unstagedstr ' *'
+zstyle ':vcs_info:*' stagedstr ' +'
+
+#######################################################################
+# Functions
+#######################################################################
+
+admone-pwd() {
   echo "%F{blue}%~%f"
 }
-function admone-git {
-  git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
-  if [[ $ADMONE_GIT_SHOW_BLANK == "0" ]]; then
-    if [[ ${git_branch//\%/%%} == "" ]]; then
-      echo "%F{blue}n/a%f"
-    else
-      echo "%F{green}${git_branch//\%/%%}%f"
-    fi
-  else
-    echo "%F{green}${git_branch//\%/%%}%f"
-  fi
+admone-vcs() {
+  echo "%F{green}${vcs_info_msg_0_}%f"
+  VCS="1"
 }
-function admone-exit-code {
-  if [[ $ADMONE_SHOW_0_EXIT_CODE == "1" ]]; then
-    echo "%(?,%F{cyan}%?%f,%F{red}%?%f)"
-  elif [[ $ADMONE_SHOW_0_EXIT_CODE == "0" ]]; then
+admone-exit-code() {
+  ([[ $SHOW_0_EXIT_CODE == "1" ]] && echo "%(?,%F{cyan}%?%f,%F{red}%?%f)") ||
     echo "%(?,,%F{red}%?%f)"
-  fi
 }
-function admone-date {
+admone-date() {
   echo $(date -I)
 }
-function admone-12hr {
+admone-12hr() {
   echo $(date +%r)
 }
-function admone-24hr {
+admone-24hr() {
   echo $(date +%T)
 }
 # Intergration with https://codeberg.org/armaan/zsh-command-time
-function admone-time {
+admone-time() {
   echo "%F{cyan}$timer_show%f"
 }
 
 #######################################################################
-
-#
 # BACKEND
-#
+#######################################################################
 
 # Cursor shape depends on vi mode
 function zle-keymap-select zle-line-init zle-line-finish {
-  if [[ $ADMONE_VI_MODE_CURSOR == "1" ]]; then
-    case $KEYMAP in
-      vicmd)
-        # Block
-        print -n -- "\E]50;CursorShape=0\C-G"
-        ;;
-      viins|main)
-        # Beam
-        print -n -- "\E]50;CursorShape=1\C-G"
-        ;;
-    esac
-    zle reset-prompt
-    zle -R
-  fi
+  case $KEYMAP in
+    vicmd)
+      # Block
+      print -n -- "\E]50;CursorShape=0\C-G"
+      ;;
+    viins|main)
+      # Beam
+      print -n -- "\E]50;CursorShape=1\C-G"
+      ;;
+  esac
+  zle reset-prompt
+  zle -R
 }
 
-function set-prompt() {
+set-prompt() {
   # Reset prompts
-  upper="${ADMONE_PRE_UPPER}"
-  lower="${ADMONE_PRE_LOWER}${ADMONE_FORMAT_LOWER}${ADMONE_PROMPT_CHAR}%f%b "
+  upper="${PRE_UPPER}"
+  lower="${PRE_LOWER}${FORMAT_LOWER}${PROMPT_CHAR}%f%b "
 
   # Add in functions as defined above
-  NUM_FUNCS="$(echo $ADMONE_INCLUDE | grep -o ';' | wc -l)"
-  for (( i = 0; i <= $NUM_FUNCS; i++ )); do
-    (( val = $i + 1 ))
-    THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
-   if [[ $i == 0 ]]; then
-      upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
-    else
-      upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
-    fi
+  NUM_FUNCS="$(echo $INCLUDE | grep -o ';' | wc -l)"
+
+  THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f 1)"
+  upper+="$BLOCK_LEFT$($THE_COMMAND)$BLOCK_RIGHT"
+
+  for (( i = 0; i < $NUM_FUNCS; i++ )); do
+    THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f $(($i + 2)))"
+    upper+="$BLOCK_SEPARATOR$BLOCK_LEFT$($THE_COMMAND)$BLOCK_RIGHT"
   done
 
   # Add line above
@@ -117,9 +108,34 @@ function set-prompt() {
   PROMPT=$upper$'\n'$lower
 }
 
+
 # Set everything!
+if [[ $TRANSIENT_PROMPT == "1" ]]; then
+  SHORT_PROMPT=$FORMAT_SHORT$SHORT_PROMPT' '%f%b
+
+  set-transient-prompt() {
+    if [[ $PROMPT != $SHORT_PROMPT ]]; then
+      PROMPT=$SHORT_PROMPT
+      zle .reset-prompt
+    fi
+  }
+
+  zle-line-finish() {
+    set-transient-prompt
+  }
+
+  trap 'set-transient-prompt; return 130' INT
+fi
+
+if [[ $VCS = "1" ]]; then
+  zstyle ':vcs_info:*' check-for-changes true
+  autoload -Uz add-zsh-hook vcs-info
+  add-zsh-hook precmd vcs_info
+fi
+
+autoload -Uz add-zsh-hook
+add-zsh-hook precmd set-prompt
+
 zle -N zle-line-init
 zle -N zle-line-finish
 zle -N zle-keymap-select
-autoload -Uz add-zsh-hook
-add-zsh-hook precmd set-prompt