.AbFab.sh

#!/bin/bash
#shellcheck shell=bash
#shellcheck disable=SC2016,SC1083,SC1090,SC2116,SC2154

export AbFab_SCREENSAVER_SELECT=0
export AbFab_SCREENSAVER_IDLE_TIME=45
export AbFab_SCREENSAVER_NOTICE_TIME=3
export AbFab_SCREENSAVER_TIMEOUT=3600
export AbFab_SCREEN_SAVER_DISABLE=false
export AbFab_SCREEN_SAVER_LIST=("cmatrix -bu 4 -a | lolcat -F 0.00015" 
"cmatrix -bu 4 -a | lolcat -F 0.0005" 
"cmatrix -bu 4 -a" 
"nyancat --frames 600" 
"asciiquarium" ) # "hollywood -s 6" )
export AbFab_SCREEN_SAVER_PRESHOW_LIST=("sl -Fal | lolcat" 
"sl -al | lolcat" 
"nyancat --frames 100" )  # randomize the Fal 0..8 -3 bits THis is also only a preexec option it does not loop

export AbFab_EXEC_START_PREFIX='🏳️‍🌈'
export AbFab_EXEC_END_PREFIX='🏁 '
export AbFab_EXEC_DURATION_PRECISION=4
export AbFab_RUN_DURATION_DISABLE=false
export AbFab_LOLCAT_SEED=1
export AbFab_LOLCAT_INCREMENT=2
export AbFab_LOLCAT_FREQ=0.3
export AbFab_PROMPT_COLOR='rainbow'
export AbFab_ANIMAL_FRIENDS=7  # The Number of friends to show in prompt -1 for random
export AbFab_ANIMAL_FRIENDS_MIN=2 
export AbFab_ANIMAL_FRIENDS_MAX=14
export AbFab_ANIMAL_HERD="false"
export AbFab_ANIMAL_PARADE="rotate"  # none || rotate || random 
export AbFab_ANIMAL_SELECT=0  # Set to 0 for random
export AbFab_ANIMALS=🐒🦍🐕🐩🐈🐅🐆🐴🐎🦄🦓🦌🐂🐃🐄🐖🐗🐏🐑🐐🐪🐫🦙🦒🐘🦏🦛🐁🐀🐇🦔🦇🦘🦡🦃🐔🐓🐣🐤🐥🐦🐧🦅🦆🦢🦉🦚🦜🐊🐢🦎🐍🐲🐉🦕🦖🐳🐋🐬🐟🐠🐡🦈🐙🐚🦀🦞🦐🦑🐌🦋🐛🐜🐝🐞🦗🦂

#🕊🐿🕷  # These had kerning issues with font need monospace emojis
#🌈🏳️‍🌈
# 🤑👹☠️
# 😺😸😹😻😼😽🙀😿😾
# 🙈🙉🙊🐵🐽
# 🐺🦊🦝🐱🦁🐯🐸🐻🐨🐼🐭🐮🐷🐹🐰🐶
# 🐒🦍🐕🐩🐈
# 🐅🐆🐴🐎🦄🦓🦌🐂🐃🐄
# 🐖🐗🐏🐑🐐🐪🐫🦙🦒🐘🦏🦛
# 🐁🐀🐇🐿🦔🦘🦡
# 🦇
# 🦜🦃🐔🐓🐣🐤🐥🐦🐧🕊🦅🦆🦢🦉🦚
# 🐢🐊🦎🐍🐲🐉🦕🦖
# 🐳🐋🐬🦈🐟🐠🐡🐙🦀🦞🦐🦑
# 🐌🦋🐛🐜🐝🐞🦗🕷🦂
# 🦟🦠
# 🕸🐾🐚
# 🦄🦚🦜
#😀😁😂😃😄😅😆😉😊😋😎😍😘😗😙😚☺🙂😐😑😶😏😣😥😮😯😪😫😴😌😛😜😝😒😓😔😕😲☹🙁😖😞😟😤😢😭😦😧😨😩😬😰😱😳🤪😵😡😠😷😇😈👿💀💩😺😸😹😻😼😽🙀😿😾👨👩👴👵👌👍🤛🏻🤛🏼🤛🏽🤛🏾🤛🏿🤜🏻🤜🏼🤜🏽🤜🏾🤜🏿💦🕶🐵🐒🐶🐺🐱🐯🐴🐮🐷🐭🐹🐰🐻🐨🐼🐥🐸🐍🐲🍜🌚🌛🌜🌝🌞🌬🎃♥🎵💡📄🏁🚩🏳️‍🌈🏳🏴
#🌿🌽🌶🍞🍟🍕🍲🍚🍜☕🔪

AbFab_ANIMAL_ARRAY=($(eval echo "$AbFab_ANIMALS" | sed "s/./& /g"))
# mapfile -t AbFab_ANIMAL_ARRAY < <($AbFab_ANIMALS)

setopt promptsubst
alias AbFab-SHOW_SETTINGS='export | rg AbFab'

AbFab_fn_DEPENDENCY_LIST() {
  export AbFab_DEPENDENCIES=("coreutils" 
  "tmux" 
  "cmatrix" 
  "cowsay" 
  "lolcat" 
  "fortune" 
  "hollywood" 
  "asciiquarium" 
  "nyancat" "sl" )
}

AbFab_fn_ANIMAL() {
  if [[ $1 -gt 0 ]]; then
    export AbFab_ANIMAL_SELECT=$1
  else
   export AbFab_ANIMAL_SELECT=0
  fi
  echo "AbFab_ANIMAL_SELECT: $AbFab_ANIMAL_SELECT  (Reminder: 0 is random)"
}

AbFab_fn_ANIMAL_FRIENDS_SET_NUM() {
  if [[ $1 -ge $AbFab_ANIMAL_FRIENDS_MIN && $1 -le $AbFab_ANIMAL_FRIENDS_MAX ]]; then
    AbFab_ANIMAL_FRIENDS=$1
  elif [[ $1 -le $AbFab_ANIMAL_FRIENDS_MIN ]]; then
    AbFab_ANIMAL_FRIENDS=$AbFab_ANIMAL_FRIENDS_MIN
  else
    AbFab_ANIMAL_FRIENDS=$AbFab_ANIMAL_FRIENDS_MAX
  fi
  if [[ $1 -le -1 ]]; then
    lo=$AbFab_ANIMAL_FRIENDS_MIN
    hi=$AbFab_ANIMAL_FRIENDS_MAX
    AbFab_ANIMAL_FRIENDS=$(shuf -i $lo-$hi -n 1)
    echo "Randomly Picked : $AbFab_ANIMAL_FRIENDS"
  fi
  export AbFab_ANIMAL_FRIENDS
}

AbFab_fn_SHOW_ANIMAL_FRIENDS_HERD() {
  animal_friends=""
  i=0
  if [[ "$AbFab_ANIMAL_HERD" = "true" ]]; then
    herd_index=$((RANDOM % ${#AbFab_ANIMAL_ARRAY[@]}))
  fi
  until [[ $i -ge $AbFab_ANIMAL_FRIENDS ]]; do
    if [[ "$AbFab_ANIMAL_SELECT" -eq 0 ]]; then
      if [[ "$AbFab_ANIMAL_HERD" = "true" ]]; then
        idx=$herd_index
      else
        idx=$((RANDOM % ${#AbFab_ANIMAL_ARRAY[@]}))
      fi
      animal_friends+=${AbFab_ANIMAL_ARRAY[$idx+1]}
    else
      animal_friends+=${AbFab_ANIMAL_ARRAY[$AbFab_ANIMAL_SELECT]}
    fi
    ((i++))
  done
  export animal_friends
  echo "$animal_friends"
}

AbFab_fn_GEN_ANIMAL_FRIENDS() {
  if [[ "$AbFab_ANIMAL_PARADE" != "rotate" ]]; then
    animal_friends=""
    i=0
    until [[ $i -ge $AbFab_ANIMAL_FRIENDS ]]; do
      if [[ "$AbFab_ANIMAL_SELECT" = "0" ]]; then
        animal_index=$((RANDOM % ${#AbFab_ANIMAL_ARRAY[@]}))
        animal_friends+=${AbFab_ANIMAL_ARRAY[$animal_index+1]}
      else
        animal_friends+=${AbFab_ANIMAL_ARRAY[$AbFab_ANIMAL_SELECT]}
      fi
      ((i++))
    done
  else
    while [[ ${#animal_friends} -gt $AbFab_ANIMAL_FRIENDS ]]; do
      animal_friends=${animal_friends:1}
    done
    while [[ ${#animal_friends} -lt $AbFab_ANIMAL_FRIENDS ]]; do
      animal_index=$((RANDOM % ${#AbFab_ANIMAL_ARRAY[@]}))
      animal_friends+=${AbFab_ANIMAL_ARRAY[$animal_index+1]}
    done
  fi
  export animal_friends
}

AbFab_fn_ANIMALS() {
  if [[ "$AbFab_ANIMAL_HERD" = "true" ]]; then
    AbFab_fn_SHOW_ANIMAL_FRIENDS_HERD "$@"
  else
    AbFab_fn_GEN_ANIMAL_FRIENDS
    echo "$animal_friends"
  fi
}

ABFab_fn_rotate() {
  animal_friends=${animal_friends:1}
  animal_index=$((RANDOM % ${#AbFab_ANIMAL_ARRAY[@]}))
  animal_friends+=${AbFab_ANIMAL_ARRAY[$animal_index+1]}
}

AbFab_fn_RAINBOW() {
  lolcat -f -F $AbFab_LOLCAT_FREQ -S $AbFab_LOLCAT_SEED
}

AbFab_fn_COLOR() {  #TODO:This needs to be completed, single color prompt not working
  if [[ "$1" != "" ]]; then
    echo '%{$fg['"$1"']%}'
  else
    echo '%{$fg[$AbFab_PROMPT_COLOR]%}'
  fi
}

AbFab_fn_RAINBOW_INC() {
  AbFab_fn_RAINBOW
  ((AbFab_LOLCAT_SEED += AbFab_LOLCAT_INCREMENT))
}

AbFab_fn_SHOW_ALL_ANIMAL_NUMBERS() {
  out='Animals with Numbers:\n'
  for (( x = 1; x <= $#AbFab_ANIMAL_ARRAY; x++ ))
  do
    out+="$x ${AbFab_ANIMAL_ARRAY[$x]} "
  done
  echo "$out" | AbFab_fn_RAINBOW_INC
}

AbFab_fn_SHOW_ALL_ANIMALS() {
  echo $AbFab_ANIMALS
}

AbFab_fn_SHOW_ANIMAL() {
  if [[ $1 -gt 0 ]]; then
    echo "$1 is ${AbFab_ANIMAL_ARRAY[$1]}" | AbFab_fn_RAINBOW_INC
  else
    echo "Please enter a number to see that animal, but here they all are."
    AbFab_SHOW_ALL_ANIMAL_NUMBERS "$@"
  fi
}

__prev_cmd() {
  cmd_hist=$HISTCMD && ((cmd_hist-=1))
  cmd="${history[$cmd_hist]}"
  return "$cmd"
}

AbFab_fn_DURATION() {
  if [[ -v start_cmd && ! -v AbFab_RUN_DURATION_DISABLE ]]; then
    end_cmd=$(AbFab_fn_EXEC_TS)
    cmd_dur="$(bc <<<"$end_cmd-$start_cmd")"
    printf -v cmd_dur "%.${AbFab_EXEC_DURATION_PRECISION}f" "$cmd_dur"
    # echo ${TERM_BLUE}"Duration: ${cmd_dur}"${TERM_DEFAULT} | AbFab_fn_RAINBOW_INC
    prev_hist=$HISTCMD && ((prev_hist-=1))
    prev=$(echo "${history[$prev_hist]}")
    fin=$(echo "$AbFab_EXEC_END_PREFIX" "$(get_date_blink_colon)" Duration "$cmd_dur" - "$prev" | AbFab_fn_RAINBOW_INC )
    echo "$fin"
    done=""
    if [ "$__exit" -ne 0 ]; then
      done+="🚩"
    fi
    done+="${cmd_dur}[$__exit]🏁${prev}🏁(${prev_hist})"
    AbFab_fn_TERM_TITLE "$done"
    unset start_cmd; unset end_cmd; unset cmd_dur
  fi
}

AbFab_fn_EXEC_TS() {
  date -u +%s.%N
}

AbFab_fn_START_EXEC() {
  if [[ ! -v AbFab_RUN_DURATION_DISABLE ]]; then
    start_cmd=$(AbFab_fn_EXEC_TS)
    #echo "$AbFab_EXEC_START_PREFIX $(get_date_blink_colon)-Run Start from: $(pwd)" | AbFab_fn_RAINBOW_INC
  fi
}

AbFab_fn_TERM_TITLE() { printf "\033]0;%s\007" "$*"; }

AbFab_fn_START_TITLE() {
  cmd=""
  if [[ "$1" != "" ]]; then
   cmd=$1
  else
    cmd_hist=$HISTCMD && ((prev_hist-=1))
    cmd=$(echo "${history[$cmd_hist]}")
  fi
  start_run=$(echo "🏳️‍🌈 $(get_date_blink_colon) Started: $cmd" | AbFab_fn_RAINBOW_INC )
  echo "$start_run"
  AbFab_fn_TERM_TITLE "🏳️‍🌈 $cmd_hist Started Running : $cmd"
}

preexec() {
  AbFab_fn_START_TITLE
  AbFab_fn_START_EXEC
}

AbFab_fn_COUNTDOWN () {
  i="$1"
  while [ "$i" -ge 0 ]; do
    printf "\rStarting in %s seconds." "$i" | AbFab_fn_RAINBOW
    ((i -= 1))
    sleep 1
  done
  printf '\nStarting now.\n' | AbFab_fn_RAINBOW_INC
}

AbFab_fn_SCREENSAVER() {
  prev_scr=124
  while [[ "$prev_scr" -eq 124 ]]; do
    screen_saver_on="true"
    ind=$1
    timeout=$2
    if [[ "$ind" -eq 0 ]]; then
      printf "\nRandomly selecting screen saver. Iteractively, provide a number to choose specific screensaver.\n" | AbFab_fn_RAINBOW_INC
      ind=$((RANDOM % ${#AbFab_SCREEN_SAVER_LIST[@]} + 1))
    fi
    chosen=${AbFab_SCREEN_SAVER_LIST[$ind]}
    echo "$(get_date) Starting the chosen one: ${chosen}" | AbFab_fn_RAINBOW_INC
    AbFab_fn_COUNTDOWN $AbFab_SCREENSAVER_NOTICE_TIME
    AbFab_fn_START_TITLE "$chosen"
    AbFab_fn_START_EXEC
    #TODO Add pre show screen saver option 
    if [[ "$timeout" -eq 0 ]]; then
      eval "$chosen"
    else
      timeout_cmd="timeout ${timeout}s $chosen"
      eval "$timeout_cmd"
      prev_scr=$?
    fi
    AbFab_fn_DURATION
    START=$SECONDS+$AbFab_SCREENSAVER_IDLE_TIME
  done
}

if [ -f ~/scripts/.git_prompt.sh ]; then
    . ~/scripts/.git_prompt.sh
fi 

function parse_git_branch() {
  __git_ps1 "(%s)"
}

function parse_git_user() {
  git config user.name
}

function parse_git_set_info() {
  GIT_PS1_SHOWDIRTYSTATE=1
  export GIT_PS1_SHOWDIRTYSTATE
  gb=$(parse_git_branch)
  if [ "" = "$gb" ]; then
    echo ""  # No git branch
  else
    echo " $(parse_git_user)-$gb"
  fi
}

function get_work_dir() {
  wdir='('
  wdir+=$(print -P '%~')
  wdir+=')'
  echo "$wdir"
}

function get_date() {
  date +"%a %F %T.%3N"
}

function get_date_blink_colon() {
  bcl=$(tput blink):$(tput sgr0)
  get_date | sed "s/:/$bcl/g"
}

function get_prev_result() {
  pr=$__DEFAULT_LITERAL
  pr+='['
  pr+='$('
  pr+='if [ "$__exit" -eq 0 ]; then '
  pr+="echo '$__GREEN_LITERAL'; "
  pr+='else '
  pr+="echo '$__RED_LITERAL'; fi"
  pr+=')'
  pr+='$__exit'
  pr+=$__DEFAULT_LITERAL
  pr+='] '
  echo "$pr"
}

function get_random() {
  echo $((RANDOM))
}

function get_abfab_animals() {
  AbFab_fn_SHOW_ANIMAL_FRIENDS_RETURN "$(get_random)"
}

function get_animals_rev() {
  AbFab_fn_ANIMALS "$@" | rev
}

function get_animals_fwd() {
  AbFab_fn_ANIMALS "$@"
}

precmd_functions+=('__prompt_command')
# precmd_functions=('__prompt_command' "${precmd_functions[@]}")  # this is to prepend
AbFab_fn_COLORIZE='__abfab'

__prompt_command() {
  __exit=$? # Save last exit code
  AbFab_fn_DURATION
  AbFab_fn_GEN_ANIMAL_FRIENDS
  ((AbFab_LOLCAT_SEED += AbFab_LOLCAT_INCREMENT))
  START=$SECONDS
}

TERM_DEFAULT=$(tput sgr0)
TERM_RED=$(tput setaf 1)
TERM_GREEN=$(tput setaf 2)

ESC=$'\033'
SOH=%{
STX=%}

__DEFAULT_LITERAL=$SOH$TERM_DEFAULT$STX
__RED_LITERAL=$SOH$TERM_RED$STX
__GREEN_LITERAL=$SOH$TERM_GREEN$STX

PS1=$(get_prev_result)

#Seperate Rainbow per section ( date, path, git, etc )
# PS1+='$(eval "$AbFab_fn_COLORIZE" <<<$(get_date))'
# PS1+=' $(eval "$AbFab_fn_COLORIZE" <<<"(\w)")'
# PS1+=' $(eval "$AbFab_fn_COLORIZE" <<<$(parse_git_set_info))'

#Single Rainbow with blinking colons
PS1+='$(eval "$AbFab_fn_COLORIZE" <<< "$(get_animals_fwd)$(get_date_blink_colon)$(get_animals_rev)$(parse_git_set_info)($(print -P "%~"))!>")'

__remove_newline() {
  tr -d '\n'
}
__color_wrap_non_printing() {
  local REGEX=$ESC'\[[[:digit:];]*m'
  local REPLACE=$SOH'&'$STX
  sed "s/$REGEX/$REPLACE/g"
}

__collapse_non_printing() {
  sed -E "s/$SOH/%{/g" -E "s/$STX/%}/g"
}

__abfab() {
  if [[ "$AbFab_PROMPT_COLOR" = "rainbow" ]]; then
    AbFab_fn_RAINBOW | __color_wrap_non_printing 
  else 
    AbFab_fn_COLOR $AbFab_PROMPT_COLOR | __color_wrap_non_printing
  fi
}

TRAPALRM () {
  if [ $((SECONDS-START)) -ge $AbFab_SCREENSAVER_IDLE_TIME ]; then
    if [[ ! -v AbFab_SCREEN_SAVER_DISABLE && ! -v screen_saver_on ]]; then
      AbFab_fn_SCREENSAVER $AbFab_SCREENSAVER_SELECT $AbFab_SCREENSAVER_TIMEOUT  # randomly selects by default
    fi
  else
    unset screen_saver_on
  fi
  if [[ "$AbFab_ANIMAL_HERD" = "false" ]]; then
    if [[ "$AbFab_ANIMAL_PARADE" = "random" ]]; then
      get_animals_fwd "" > /dev/null 2>&1  # |& /dev/null
    else
      if [[ "$AbFab_ANIMAL_PARADE" = "rotate" ]]; then
        ABFab_fn_rotate
      fi
    fi
  fi
  zle reset-prompt
}
TMOUT=1