#!/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 "🌅"      # tidlig
  elif (( h >= 9 && h <= 10 )); then
    echo "☕"      # formiddag
  elif (( h == 11 && m < 30 )); then
    echo "🥪"      # lunsj
  elif (( (h == 11 && m >= 30) || (h >= 12 && h <= 15) )); then
    echo "💻"      # dag
  elif (( h == 16 )); then
    echo "🍲"      # middag
  elif (( h >= 17 && h <= 22 )); then
    echo "🌆"      # kveld
  else
    echo "🌙"      # natt
  fi
}

# -------------------------------------------
# Path shortening (keeps /home/user visible)
# /home/steffen/a/b/c/d -> /home/steffen/…/c/d
# -------------------------------------------
_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]}"
}

# -------------------------------
# Dynamic vars updated 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"

  # Reset
  local RST="\[\e[0m\]"

  # Foregrounds
  local FG_TXT="\[\e[38;5;255m\]"   # crisp white
  local FG_DIM="\[\e[38;5;250m\]"   # soft white/gray

  # Backgrounds (pastel/dark, high contrast)
  local BG_DATE="\[\e[48;5;60m\]"   # muted purple-blue
  local BG_TIME="\[\e[48;5;66m\]"   # teal-ish
  local BG_USER="\[\e[48;5;24m\]"   # deep blue
  local BG_HOST="\[\e[48;5;94m\]"   # warm brown/orange
  local BG_PATH="\[\e[48;5;238m\]"  # graphite

  # Segment separators (Powerline glyphs)
  local SEP=""
  local LEFT=""
  local RIGHT=""

  # Separator color = next segment background (classic powerline trick)
  local SEP_TO_TIME="\[\e[38;5;66m\]"
  local SEP_TO_USER="\[\e[38;5;24m\]"
  local SEP_TO_HOST="\[\e[38;5;94m\]"
  local SEP_TO_PATH="\[\e[38;5;238m\]"
  local SEP_TO_RST="\[\e[38;5;0m\]"

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

  # Keep exact order: date -> time -> user@host -> path, then newline, then prompt
  PS1="\
\[\e[38;5;60m\]╭─${RST}\
${BG_DATE}${FG_TXT}${LEFT} ${FG_TXT}\d ${RST}${BG_DATE}${SEP_TO_TIME}${SEP}\
${BG_TIME}${FG_TXT} \A ${RST}${BG_TIME}${SEP_TO_USER}${SEP}\
${BG_USER}${FG_TXT} \u ${RST}${BG_USER}${SEP_TO_HOST}${SEP}\
${BG_HOST}${FG_TXT} @\h ${RST}${BG_HOST}${SEP_TO_PATH}${SEP}\
${BG_PATH}${FG_DIM} \${__PS1_PATH} ${RST}${BG_PATH}${FG_DIM}${RIGHT}${RST}\
\n\[\e[38;5;60m\]╰── ${RST}${PROMPT_SYM} ${FG_DIM}\$${RST} \${__PS1_SYM} "
}

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}