PROMPT_COMMAND=(__prompt_command "${PROMPT_COMMAND[@]}")

# custom terminal prompt format.
__prompt_command()
{
  local last_status="${?}"
  local is_error=false
  local is_root=false

  if [[ $last_status != 0 && $last_status != 130 ]]; then
    is_error=true
  fi
  if [[ "$UID" -eq 0 ]]; then
    is_root=true
  fi

  # add newline
  PS1="\n"

  # set error red
  if $is_error; then
    PS1+="\[${color_bred}\]"
    PS1+="["
  else
    PS1+="\[${color_default}\]"
    PS1+="["
  fi

  # add time
  PS1+="\[${color_white}\]"
  PS1+="$(date +%H:%M) "

  # set root red
  if $is_root; then
    PS1+="\[${color_red}\]"
  else
    PS1+="\[${color_cyan}\]"
  fi

  # add user, host and working dir
  PS1+="\u@\h "
  PS1+="\[${color_blue}\]"
  PS1+="\w"
  # PS1+="\${PWD}"

  # Add git branch if available.
  local git_branch="$(_git_current_branch)"
  if [[ "${git_branch}" != "" ]]; then
    [[ "${git_branch}" =~ (master|main) ]] || PS1+=" ${color_bblue}@${git_branch}"
  fi

  # set error red
  if $is_error; then
    PS1+="\[${color_bred}\]"
    PS1+="] "
  else
    PS1+="\[${color_default}\]"
    PS1+="] "
  fi

  # If error, show code.
  if ${is_error}; then
    PS1+="${color_red}("
    PS1+="${last_status}"
    local error_type="$(_ps1error ${last_status})"
    [[ "${error_type}" != "" ]] && PS1+=" ${error_type}"
    PS1+=")${color_default} "
  fi

  # command on new line
  PS1+="\n"
  PS1+="\[${color_default}\]"

  # set user tag
  if $is_root; then
    PS1+="# "
  else
    PS1+="$ "
  fi
}

_ps1error()
{
  local type
  case ${1} in
    1)   type="GENERAL" ;;
    2)   type="MISUSE" ;;
    126) type="CANTEXEC" ;;
    127) type="CMDNF" ;;
    129) type="SIGHUP" ;;
    130) type="SIGINT" ;;
    131) type="SIGQUIT" ;;
    132) type="SIGILL" ;;
    133) type="SIGTRAP" ;;
    134) type="SIGABRT" ;;
    135) type="SIGBUS" ;;
    136) type="SIGFPE" ;;
    137) type="SIGKILL" ;;
    138) type="SIGUSR1" ;;
    139) type="SIGSEGV" ;;
    140) type="SIGUSR2" ;;
    141) type="SIGPIPE" ;;
    142) type="SIGALRM" ;;
    143) type="SIGTERM" ;;
    144) type="" ;;
    145) type="SIGCHLD" ;;
    146) type="SIGCONT" ;;
    147) type="SIGSTOP" ;;
    148) type="SIGTSTP" ;;
    149) type="SIGTTIN" ;;
    150) type="SIGTTOU" ;;
    151) type="SIGURG" ;;
    152) type="SIGXCPU" ;;
    153) type="SIGXFSZ" ;;
    154) type="SIGVTALRM" ;;
    155) type="SIGPROF" ;;
    156) type="SIGWINCH" ;;
    157) type="SIGIO" ;;
    158) type="SIGPWR" ;;
    159) type="SIGSYS" ;;
    *)   type="" ;;
  esac

  echo -n "${type}"
}