#!/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

# -------------------------------------------------
# Nerd Font detection + fallback controls
#
# Env overrides:
#   PS1_FORCE_ASCII=1   -> always use ASCII/Unicode fallback separators
#   PS1_FORCE_NF=1      -> always use Nerd Font separators
# -------------------------------------------------
_ps1_has_nf() {
  # Explicit overrides win
  if [[ "${PS1_FORCE_ASCII:-0}" == "1" ]]; then
    return 1
  fi
  if [[ "${PS1_FORCE_NF:-0}" == "1" ]]; then
    return 0
  fi

  # Heuristic: if fc-list exists and shows a Nerd Font family, assume NF works.
  if command -v fc-list >/dev/null 2>&1; then
    # Keep it generic: any "Nerd Font" is good enough
    if fc-list 2>/dev/null | grep -qi "Nerd Font"; then
      return 0
    fi
  fi

  # Otherwise: safe fallback
  return 1
}

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

  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)
# Compatible with bash 3.2+ (no negative array indexes)
# -------------------------------------------------
_ps1_path() {
  local p="$PWD"
  local parts=()
  IFS='/' read -ra parts <<< "$p"

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

  # / + first two comps + … + last two comps
  local n=${#parts[@]}
  local a="${parts[1]}"
  local b="${parts[2]}"
  local c="${parts[$((n-2))]}"
  local d="${parts[$((n-1))]}"

  echo "/${a}/${b}/…/${c}/${d}"
}

# -------------------------------------------------
# Dynamic vars updated before each prompt
# -------------------------------------------------
__PS1_SYM=""
__PS1_PATH=""
__PS1_STATUS=0
__PS1_USE_NF=0

_ps1_update() {
  __PS1_STATUS=$?
  __PS1_SYM="$(_ps1_symbol)"
  __PS1_PATH="$(_ps1_path)"
  if _ps1_has_nf; then __PS1_USE_NF=1; else __PS1_USE_NF=0; fi
}

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

ps1_on() {
  PROMPT_COMMAND="_ps1_update"

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

  # Two-zone palette (as requested)
  # Zone 1: gray-blue pastel (date/time/user)
  local Z1_BG="\[\e[48;5;61m\]"
  local Z1_FG="\[\e[38;5;255m\]"

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

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

  # Status colors
  local OK="\[\e[38;5;76m\]"
  local BAD="\[\e[38;5;203m\]"

  # Build separators based on NF availability (evaluated at prompt time via $__PS1_USE_NF)
  # We avoid embedding raw variables that bash re-parses weirdly by using $(...) inside PS1 only for
  # selecting literal separators (safe), while keeping all color escapes static and balanced.
  local SEP_EXPR='\$( [ "$__PS1_USE_NF" -eq 1 ] && printf "" || printf "▶" )'
  local LEFT_EXPR='\$( [ "$__PS1_USE_NF" -eq 1 ] && printf "" || printf "[" )'
  local RIGHT_EXPR='\$( [ "$__PS1_USE_NF" -eq 1 ] && printf "" || printf "]" )'

  local PROMPT_SYM="\$( [ \$__PS1_STATUS -eq 0 ] && printf '${OK}' || printf '${BAD}' )➜${RST}"

  # Keep exact order: date -> time -> user -> host -> path, then newline, then prompt
  # Note: \d, \A, \u, \h are PS1 escapes.
  PS1="\
${FRAME}╭─${RST}\
${Z1_BG}${Z1_FG}${LEFT_EXPR} \d \A \u ${RST}${Z1_BG}${Z2_BG}${Z2_FG}${SEP_EXPR}${RST}\
${Z2_BG}${Z2_FG} @\h ${PATH_FG}\${__PS1_PATH} ${RST}${Z2_BG}${Z2_FG}${RIGHT_EXPR}${RST}\
\n${FRAME}╰── ${RST}${PROMPT_SYM} \${__PS1_SYM} "
}

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}