#!/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 emoji (Europe/Oslo)
# -------------------------------------------------
_ps1_symbol() {
  local h m
  h=$(TZ=Europe/Oslo date +%H); h=$((10#$h))
  m=$(TZ=Europe/Oslo date +%M); 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
# -------------------------------------------------
_ps1_path() {
  local p="$PWD"
  local parts=()
  IFS='/' read -ra parts <<< "$p"

  if ((${#parts[@]} < 6)); then
    echo "$p"
  else
    echo "/${parts[1]}/${parts[2]}/…/${parts[-2]}/${parts[-1]}"
  fi
}

# -------------------------------------------------
# Update vars
# -------------------------------------------------
__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 RST="\[\e[0m\]"

  # ===== ZONE 1: date / time / user =====
  local Z1_BG="\[\e[48;5;61m\]"     # gray-blue pastel
  local Z1_FG="\[\e[38;5;255m\]"

  # ===== ZONE 2: host / path =====
  local Z2_BG="\[\e[48;5;37m\]"     # turquoise / cool green
  local Z2_FG="\[\e[38;5;255m\]"
  local PATH_FG="\[\e[38;5;194m\]"

  # Frame
  local FRAME="\[\e[38;5;60m\]"

  # Separators
  local SEP=""
  local LEFT=""
  local RIGHT=""
  local SEP_Z1="\[\e[38;5;61m\]"
  local SEP_Z2="\[\e[38;5;37m\]"

  # Status
  local OK="\[\e[38;5;76m\]"
  local BAD="\[\e[38;5;203m\]"
  local PROMPT_SYM="\$( [ \$__PS1_STATUS -eq 0 ] && printf '${OK}' || printf '${BAD}' )➜${RST}"

  PS1="\
${FRAME}╭─${RST}\
${Z1_BG}${Z1_FG}${LEFT} \d \A \u ${RST}${Z1_BG}${SEP_Z2}${SEP}\
${Z2_BG}${Z2_FG} @\h ${PATH_FG}\${__PS1_PATH} ${RST}${Z2_BG}${RIGHT}${RST}\
\n${FRAME}╰── ${RST}${PROMPT_SYM} \${__PS1_SYM} "
}

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}