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

# -------------------------------------------------
# Emoji / symbol based on time (Europe/Oslo)
# -------------------------------------------------
_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 (( 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
}

# -------------------------------------------------
# Build PS1
# -------------------------------------------------
_ps1_build() {
  local sym path
  sym="$(_ps1_symbol)"
  path="$(pwd)"

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

  # NOTE:
  # - PS1 escapes must be \\d \\A \\u \\h \\n
  # - NO $variables inside PS1 (PS1 re-parses $)
  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"
}

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

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

ps1_off() {
  PROMPT_COMMAND="$__PS1_PREV_PROMPT_COMMAND"
}