Remove transient prompt option

[admone.git] / admone.zsh

diff --git a/admone.zsh b/admone.zsh
index 9e32ac802bf5d1689a5c5805c4f6b68c0eb2ff08..7f01fd4f5ff7e97d5c0fe8727a3b321a5843dd4d 100644 (file)
--- a/admone.zsh
+++ b/admone.zsh
@@ -1,76 +1,133 @@
+# Admone Zsh prompt
+# Copyright Armaan Bhojwani 2021 <me@armaanb.net>
+# MIT licensed, see the LICENSE file for more information.
+
 #######################################################################
-# Admone ZSH prompt. Copyright Armaan Bhojwani, palb91, romkatv.      #
-# 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_PRE_UPPER="╭ "
-ADMONE_PRE_LOWER="╰ "
-ADMONE_BLOCK_SEPARATOR="-"
-ADMONE_BLOCK_LEFT="["
-ADMONE_BLOCK_RIGHT="]"
-ADMONE_INCLUDE="admone-pwd;admone-git;admone-exit-code"
-
-function admone-pwd {
-  echo "%F{blue}%~%f%F{240}"
+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="1"
+INCLUDE="admone-pwd;admone-vcs;admone-exit-code"
+FORMAT_LOWER="%B%F{%(?.cyan.red)}"
+
+SHORT_PROMPT="➤"
+FORMAT_SHORT="%B%F{%(?.cyan.red)}"
+TWO_LINE="0"
+
+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"
+}
+admone-vcs() {
+  echo "%F{green}${vcs_info_msg_0_}%f"
+}
+admone-exit-code() {
+  ([[ $SHOW_0_EXIT_CODE == "1" ]] && echo "%(?,%F{cyan}%?%f,%F{red}%?%f)") ||
+    echo "%(?,,%F{red}%?%f)"
 }
-function admone-git {
-  local git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
-  echo "%F{green}${git_branch//\%/%%}%f"
+admone-date() {
+  echo $(date -I)
 }
-function admone-exit-code {
-  echo "%(?,,%F{1}%?%F{240})"
+admone-12hr() {
+  echo $(date +%r)
+}
+admone-24hr() {
+  echo $(date +%T)
 }
 
 #######################################################################
-
-#
 # BACKEND
-#
+#######################################################################
 
 # Cursor shape depends on vi mode
 function zle-keymap-select zle-line-init zle-line-finish {
   case $KEYMAP in
     vicmd)
+      # Block
       print -n -- "\E]50;CursorShape=0\C-G"
-      ;; # block cursor
+      ;;
     viins|main)
+      # Beam
       print -n -- "\E]50;CursorShape=1\C-G"
-      ;; # line cursor
-    *)
       ;;
   esac
-
   zle reset-prompt
   zle -R
 }
 
-function set-prompt() {
-  upper="%B${ADMONE_PRE_UPPER}"
-  lower="%B${ADMONE_PRE_LOWER}%B%F{%(?.cyan.red)}${ADMONE_PROMPT_CHAR}%f%b "
+NUM_FUNCS="$(echo $INCLUDE | grep -o ';' | wc -l)"
+THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f 1)"
+[[ -v PS1_NL ]] && echo || PS1_NL=
 
-  for (( i = 0; i <= $(echo $ADMONE_INCLUDE | grep -o ';' | wc -l); i++ )); do
-    (( val = $i + 1 ))
-    THE_COMMAND="$(echo $ADMONE_INCLUDE | cut -d ';' -f $val)"
+set-twoline-prompt() {
+  # Reset prompts
+  upper="${PRE_UPPER}"
+  lower="${PRE_LOWER}${FORMAT_LOWER}${PROMPT_CHAR}%f%b "
 
-    if [[ $((val%2)) == 0 ]]; then
-        upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
-      else
-        upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
-      fi
+  # Add in functions as defined above
+  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
   PROMPT=$upper$'\n'$lower
 }
 
+set-oneline-prompt() {
+  PROMPT="${FORMAT_LOWER}${PROMPT_CHAR}%f%b "
+  RPROMPT=""
+
+  for (( i = 0; i <= $NUM_FUNCS; i++ )); do
+    THE_COMMAND="$(echo $INCLUDE | cut -d ';' -f $(($i + 1)))"
+    RPROMPT+="$BLOCK_LEFT$($THE_COMMAND)$BLOCK_RIGHT"
+  done
+}
+# Set everything!
+set-prompt() {
+  if [[ $TWO_LINE == "1" ]]; then
+    set-twoline-prompt
+  else
+    set-oneline-prompt
+  fi
+}
+
+if [[ ! $(echo $INCLUDE | grep -q "vcs") ]]; 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