#!/usr/bin/env bash

# Must be sourced
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
  echo "Bruk: source ps1.sh"
  exit 1
fi

case "$-" in
  *i*) ;;
  *) return 0 ;;
esac

# -------------------------------------------------
# Time-based symbol (Europe/Oslo)
# -------------------------------------------------
_ps1_symbol() {
  local h m
  h=$(TZ=Europe/Oslo date +%H)
  m=$(TZ=Europe/Oslo date +%M)
  h=$((10#$h))
  m=$((10#$m))

  if (( h >= 5 && h <= 8 )); then
    echo "🌅"
  elif (( h >= 9 && h <= 10 )); then
    echo "☕"
  elif (( h == 11 && m < 30 )); then
    echo "🥪"
  elif (( (h == 11 && m >= 30) || (h >= 12 && h <= 15) )); then
    echo "💻"
  elif (( h == 16 )); then
    echo "🍲"
  elif (( h >= 17 && h <= 22 )); then
    echo "🌆"
  else
    echo "🌙"
  fi
}

# -------------------------------------------------
# Path shortening (keeps /home/user)
# -------------------------------------------------
_ps1_path() {
  local p="$PWD"
  local parts=()
  IFS='/' read -ra parts <<< "$p"

  if ((${#parts[@]} < 6)); then
    echo "$p"
    return
  fi

  echo "/${parts[1]}/${parts[2]}/…/${parts[-2]}/${parts[-1]}"
}

# -------------------------------------------------
# Variables updated before each prompt
# -------------------------------------------------
__PS1_SYM=""
__PS1_PATH=""
__PS1_STATUS=0

_ps1_update() {
  __PS1_STATUS=$?
  __PS1_SYM="$(_ps1_symbol)"
  __PS1_PATH="$(_ps1_path)"
}

# -------------------------------------------------
# Enable / disable
# -------------------------------------------------
__PS1_PREV_PROMPT_COMMAND="${PROMPT_COMMAND-}"

ps1_on() {
  PROMPT_COMMAND="_ps1_update"

  local BOX="\[\e[38;5;183m\]"
  local VAL="\[\e[38;5;117m\]"
  local ACC="\[\e[38;5;229m\]"
  local OK="\[\e[38;5;76m\]"
  local BAD="\[\e[38;5;203m\]"
  local RST="\[\e[0m\]"

  PS1="${BOX}╭─(${ACC}\d${BOX} ${ACC}\A${BOX})-(${VAL}\u${BOX}@${VAL}\h${BOX})-(${VAL}\${__PS1_PATH}${BOX})${RST}\n${BOX}╰──\$( [ \$__PS1_STATUS -eq 0 ] && printf '${OK}' || printf '${BAD}' )➜ ${RST}\$ \${__PS1_SYM} "
}

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}