#!/usr/bin/env bash

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

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

# Returns 0 if current shell locale is UTF-8 capable
_ps1_is_utf8() {
  # locale charmap outputs e.g. UTF-8
  local cm
  cm="$(LC_ALL=${LC_ALL-} LANG=${LANG-} locale charmap 2>/dev/null || true)"
  [[ "$cm" == "UTF-8" ]]
}

# Emoji based on time (Europe/Oslo) with ASCII fallback if not UTF-8
_ps1_symbol() {
  local hh mm h m
  hh="$(TZ=Europe/Oslo date +%H)"
  mm="$(TZ=Europe/Oslo date +%M)"
  h=$((10#$hh))
  m=$((10#$mm))

  if _ps1_is_utf8; then
    if (( h >= 5 && h <= 8 )); then
      printf '%s' "🌅"
    elif (( h >= 9 && h <= 10 )); then
      printf '%s' "☕"
    elif (( h == 11 && m < 30 )); then
      printf '%s' "🥪"
    elif (( (h == 11 && m >= 30) || (h >= 12 && h <= 15) )); then
      printf '%s' "💻"
    elif (( h == 16 )); then
      printf '%s' "🍲"
    elif (( h >= 17 && h <= 22 )); then
      printf '%s' "🌆"
    else
      printf '%s' "🌙"
    fi
  else
    # ASCII fallbacks
    if (( h >= 5 && h <= 8 )); then
      printf '%s' "^"     # early
    elif (( h >= 9 && h <= 10 )); then
      printf '%s' "o"     # morning
    elif (( h == 11 && m < 30 )); then
      printf '%s' "="     # lunch
    elif (( (h == 11 && m >= 30) || (h >= 12 && h <= 15) )); then
      printf '%s' "*"     # day
    elif (( h == 16 )); then
      printf '%s' "#"     # dinner
    elif (( h >= 17 && h <= 22 )); then
      printf '%s' "~"     # evening
    else
      printf '%s' "."     # night
    fi
  fi
}

_ps1_build() {
  local sym path
  sym="$(_ps1_symbol)"
  path="$(pwd)"

  # Pastel-ish 256 colors
  local BOX="\[\e[38;5;183m\]"
  local VAL="\[\e[38;5;117m\]"
  local ACC="\[\e[38;5;229m\]"
  local RST="\[\e[0m\]"

  # IMPORTANT: keep PS1 escapes as \\d \\A \\u \\h \\n
  local ps1
  ps1="${BOX}╭─(${ACC}\\d${BOX} ${ACC}\\A${BOX})-(${VAL}\\u${BOX}@${VAL}\\h${BOX})-(${VAL}${path}${BOX})${RST}\\n${BOX}╰──➜ ${RST}\\$${sym} "

  printf '%s' "$ps1"
}

__PS1_PREV_PROMPT_COMMAND="${PROMPT_COMMAND-}"

ps1_on() {
  PROMPT_COMMAND='_PS1="$(_ps1_build)"; export PS1="$_PS1"'
  _PS1="$(_ps1_build)"; export PS1="$_PS1"
}

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}