#######################################################################
# 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
#

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}"
}
function admone-git {
  local git_branch="$(git rev-parse --abbrev-ref HEAD 2>/dev/null)"
  echo "%F{green}${git_branch//\%/%%}%f"
}
function admone-exit-code {
  echo "%(?,,%F{1}%?%F{240})"
}

function admone-date {
  echo $(date -I)
}
function admone-time {
  echo $(date -Iseconds)
}
#######################################################################

#
# BACKEND
#

# Cursor shape depends on vi mode
function zle-keymap-select zle-line-init zle-line-finish {
  case $KEYMAP in
    vicmd)
      print -n -- "\E]50;CursorShape=0\C-G"
      ;;
    viins|main)
      print -n -- "\E]50;CursorShape=1\C-G"
      ;;
    *)
      ;;
  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 $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)"

    REAL_NUM_FUNCS=$((NUM_FUNCS+1))
      if [[ $val == $REAL_NUM_FUNCS ]]; then
        upper+="$ADMONE_BLOCK_SEPARATOR$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT"
      elif [[ $((REAL_NUM_FUNCS%2)) == 0 ]]; then
        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
      elif [[ $((REAL_NUM_FUNCS%2)) != 0 ]]; then
        if [[ $val == 1 ]]; then
          upper+="$ADMONE_BLOCK_LEFT$($THE_COMMAND)$ADMONE_BLOCK_RIGHT$ADMONE_BLOCK_SEPARATOR"
        elif [[ $((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
      fi
  done

  [[ -v PS1_NL ]] && echo || PS1_NL=
  PROMPT=$upper$'\n'$lower
}

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