#!/bin/bash
# shellcheck disable=SC2139

# TODO: needs some refactoring
function __miaou_prompt_command {
  local EXIT="$?" # This needs to be first

  local R='\[\e[0m\]'
  local Gre='\[\e[32m\]'
  local Yel='\[\e[93m\]'
  local Cya='\[\e[36m\]'
  local Mag='\[\e[95m\]'
  local Gra='\[\e[90m\]'

  local DIVERGENT_GIT_SYMBOL="${Yel}➾${R}"
  local SYNCHRONIZED_GIT_SYMBOL="${Gre}✔${R}"
  local PUSHED_NEEDED_SYMBOL='⥱'
  local TAG_NEEDED_SYMBOL='⤽'

  local previous_IFS="$IFS"

  # set xterm title
  echo -ne "\033]0;${USER}@${HOSTNAME}: ${PWD}\007"

  PS1=''
  IFS=' '

  if [[ "$UID" -eq 0 ]]; then
    PS1+="$Yel\u$R"
    PROMPT='$'
  else
    PS1+="$Gre\u$R"
    PROMPT='#'
  fi

  PS1+="${Gra}@${R}${HOSTNAME}"

  local pwd='~'
  [ "$PWD" != "$HOME" ] && pwd=${PWD/#$HOME\//\~\/}
  PS1+=" ${Cya}${pwd}${R}"

  if hash git 2>&-; then
    # git command exists

    if git rev-parse --git-dir >/dev/null 2>&1; then
      # current dir is version controlled

      local branch tag_release dirty
      branch=$(git branch 2>/dev/null | grep -e ^* | cut -d ' ' -f2)
      PS1+=" ${R}[${Mag}${branch}${R}|"

      tag_release=$(git describe --tags 2>/dev/null | cut -d'-' -f1)
      dirty=$(git status -s | wc -l)
      if [[ $dirty -ne 0 ]]; then
        PS1+="${Yel}${DIVERGENT_GIT_SYMBOL}${R}|${tag_release}${Yel}…$dirty"
      else
        local ahead
        ahead=$(git rev-list --branches --not --remotes | wc -l)

        if [[ $ahead -ne 0 ]]; then
          PS1+="${DIVERGENT_GIT_SYMBOL}|${tag_release}${Yel}${PUSHED_NEEDED_SYMBOL}${ahead}"
        else
          if [[ -z $tag_release ]]; then
            PS1+="${SYNCHRONIZED_GIT_SYMBOL}"
          else
            local commit_ahead
            commit_ahead=$(git log "$tag_release"..HEAD --oneline | wc -l)

            if [[ $commit_ahead -gt 0 ]]; then
              PS1+="${DIVERGENT_GIT_SYMBOL}|${tag_release}${Cya}${TAG_NEEDED_SYMBOL}${commit_ahead}"
            else
              PS1+="${SYNCHRONIZED_GIT_SYMBOL}|${Cya}${tag_release}"
            fi
          fi
        fi
      fi
      PS1+="${R}]"
    fi
  fi

  if [ $EXIT != 0 ]; then
    PS1+="$Mag" # Add red if exit code non 0
  else
    PS1+="$Gra"
  fi

  PS1+=" $PROMPT ${R}"
  IFS="$previous_IFS"
}

###------
### MAIN
###------

# If not running interactively, don't do anything
[[ $- != *i* ]] && return

if [[ ! -v MIAOU_BASH_DIR ]]; then
  source /etc/profile.d/zz-miaou-bash.sh
else
  PROMPT_COMMAND=__miaou_prompt_command
fi

HOSTNAME=$(hostname -f)
HISTCONTROL=ignoreboth
HISTSIZE=10000
HISTFILESIZE=20000
BLOCK_SIZE=si
export BLOCK_SIZE

export BLOCK_SIZE=si

# append to the history file, don't overwrite it
shopt -s histappend

# check the window size after each command and, if necessary,
# update the values of LINES and COLUMNS.
shopt -s checkwinsize

# set a fancy prompt (non-color, unless we know we "want" color)
# One-liner: Do we have colors?
if [[ -t 1 ]] && [[ -v COLORTERM && "$COLORTERM" == "truecolor" || "$TERM" == *"256color"* || "$TERM" == 'xterm-ghostty' ]]; then
  export LS_COLORS="rs=0:di=01;34:ln=01;36:mh=00:pi=40;33:so=01;35:do=01;35:bd=40;33;01:cd=40;33;01:or=40;31;01:mi=00:su=37;41:sg=30;43:ca=30;41:tw=30;42:ow=34;42:st=37;44:ex=01;32:*.tar=01;31:*.tgz=01;31:*.arc=01;31:*.arj=01;31:*.taz=01;31:*.lha=01;31:*.lz4=01;31:*.lzh=01;31:*.lzma=01;31:*.tlz=01;31:*.txz=01;31:*.tzo=01;31:*.t7z=01;31:*.zip=01;31:*.z=01;31:*.Z=01;31:*.dz=01;31:*.gz=01;31:*.sql.gz=01;35:*.mariadb.gz=01;35:*.postgres.gz=01;35:*.lrz=01;31:*.lz=01;31:*.lzo=01;31:*.xz=01;31:*.zst=01;31:*.tzst=01;31:*.bz2=01;31:*.bz=01;31:*.tbz=01;31:*.tbz2=01;31:*.tz=01;31:*.deb=01;31:*.rpm=01;31:*.jar=01;31:*.war=01;31:*.ear=01;31:*.sar=01;31:*.rar=01;31:*.alz=01;31:*.ace=01;31:*.zoo=01;31:*.cpio=01;31:*.7z=01;31:*.rz=01;31:*.cab=01;31:*.jpg=01;35:*.jpeg=01;35:*.mjpg=01;35:*.mjpeg=01;35:*.gif=01;35:*.bmp=01;35:*.pbm=01;35:*.pgm=01;35:*.ppm=01;35:*.tga=01;35:*.xbm=01;35:*.xpm=01;35:*.tif=01;35:*.tiff=01;35:*.png=01;35:*.svg=01;35:*.svgz=01;35:*.mng=01;35:*.pcx=01;35:*.mov=01;35:*.mpg=01;35:*.mpeg=01;35:*.m2v=01;35:*.mkv=01;35:*.webm=01;35:*.ogm=01;35:*.mp4=01;35:*.m4v=01;35:*.mp4v=01;35:*.vob=01;35:*.qt=01;35:*.nuv=01;35:*.wmv=01;35:*.asf=01;35:*.rm=01;35:*.rmvb=01;35:*.flc=01;35:*.avi=01;35:*.fli=01;35:*.flv=01;35:*.gl=01;35:*.dl=01;35:*.xcf=01;35:*.xwd=01;35:*.yuv=01;35:*.cgm=01;35:*.emf=01;35:*.ogv=01;35:*.ogx=01;35:*.aac=00;36:*.au=00;36:*.flac=00;36:*.m4a=00;36:*.mid=00;36:*.midi=00;36:*.mka=00;36:*.mp3=00;36:*.mpc=00;36:*.ogg=00;36:*.ra=00;36:*.wav=00;36:*.oga=00;36:*.opus=00;36:*.spx=00;36:*.xspf=00;36:"
fi

if [[ -t 0 ]]; then
  stty -ixon      # prevent CTRL+S to freeze in vim, CTRL+Q to unfreeze!
  stty kill undef # release CTRL+U for the Readline library => undo
fi

export LESS="r"
# make less more friendly for non text input files, see lesspipe(1)
if [[ -x /usr/bin/lesspipe ]]; then eval "$(SHELL=/bin/sh lesspipe)"; fi

# ALIASES

COLOR_OPTIONS=('--color=auto')
alias ls="ls ${COLOR_OPTIONS[*]}"
alias ll="ls ${COLOR_OPTIONS[*]} -l"
alias la="ls ${COLOR_OPTIONS[*]} -la"
alias lt="ls ${COLOR_OPTIONS[*]} -lt"
alias ltr="ls ${COLOR_OPTIONS[*]} -ltr"
alias l.="ls ${COLOR_OPTIONS[*]} -ld .*"
alias grep="grep ${COLOR_OPTIONS[*]}"
alias fgrep="fgrep ${COLOR_OPTIONS[*]}"
alias egrep="egrep ${COLOR_OPTIONS[*]}"
unset COLOR_OPTIONS

alias l=ls
alias sudo='sudo ' # the smart sudo alias trick!
alias cd..='cd ..'
alias ..=cd..
alias transfer_cp='rsync -a --info=progress2 --no-inc-recursive'
alias ssu='ss -tlnp4'

# OVERRIDDEN ALIASES
# shellcheck source=/dev/null
[[ -f "/etc/bash.aliases" ]] && source "/etc/bash.aliases"
# shellcheck source=/dev/null
[[ -f "$HOME/.bash_aliases" ]] && source "$HOME/.bash_aliases"

# the miaou-bash helper function
function miaou-bash {
  "$MIAOU_BASH_DIR"/bin/miaou-bash "$@"
  local code=$?
  if [[ "$code" == 0 ]]; then
    if [[ "${1:-}" =~ --self-upgrade|--uninstall ]]; then
      echo 'reloading session now!'
      exec bash -l
    fi
  fi
  return $code
}

# BASH COMPLETIONS
if ! shopt -oq posix; then
  if [ -f /usr/share/bash-completion/bash_completion ]; then
    . /usr/share/bash-completion/bash_completion
  elif [ -f /etc/bash_completion ]; then
    . /etc/bash_completion
  fi
fi

## Add path for TOOLBOX if any
[[ -d "/TOOLBOX" ]] && PATH=$PATH:/TOOLBOX || true
# Add path to any tools folder in /opt/miaou-*
for i in {/opt,$HOME}/miaou-*/tools; do
  [[ -d "$i" ]] && PATH=$PATH:$i
done || true