#!/usr/bin/env bash
set -euo pipefail

if [[ "${EUID}" -ne 0 ]]; then
  echo "Kjør som root: sudo bash $0"
  exit 1
fi

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"

SRC_PS1="$REPO_ROOT/ps1/ps1.sh"
SRC_BANNER="$REPO_ROOT/banner/banner.sh"

DST_PS1="/etc/profile.d/ps1.sh"
DST_BANNER="/etc/profile.d/banner.sh"

BASH_BASHRC="/etc/bash.bashrc"

MARKER_START="# >>> dotfiles (managed) >>>"
MARKER_END="# <<< dotfiles (managed) <<<"

# Select install mode (PS1, banner, or both)
install_ps1=1
install_banner=1
if [[ -t 0 ]]; then
  echo "Install options:"
  echo "  1) PS1 only"
  echo "  2) Banner only"
  echo "  3) Both (PS1 + banner)"
  read -r -p "Choose [3]: " install_choice
  case "${install_choice}" in
    1) install_ps1=1; install_banner=0 ;;
    2) install_ps1=0; install_banner=1 ;;
    ""|3) install_ps1=1; install_banner=1 ;;
    *) echo "Invalid choice, using both."; install_ps1=1; install_banner=1 ;;
  esac
fi

if [[ "$install_banner" -eq 1 ]]; then
  # Ensure required packages are installed (no prompt)
  missing_pkgs=()
  if ! command -v figlet >/dev/null 2>&1; then
    missing_pkgs+=("figlet")
  fi
  if ! command -v lolcat >/dev/null 2>&1 && [[ ! -x /usr/games/lolcat ]]; then
    missing_pkgs+=("lolcat")
  fi
  if ! command -v landscape-sysinfo >/dev/null 2>&1; then
    missing_pkgs+=("landscape-common")
  fi

  if ((${#missing_pkgs[@]} > 0)); then
    if command -v apt-get >/dev/null 2>&1; then
      apt-get update
      DEBIAN_FRONTEND=noninteractive apt-get install -y "${missing_pkgs[@]}"
    elif command -v dnf >/dev/null 2>&1; then
      dnf install -y "${missing_pkgs[@]}"
    elif command -v yum >/dev/null 2>&1; then
      yum install -y "${missing_pkgs[@]}"
    elif command -v pacman >/dev/null 2>&1; then
      pacman -Sy --noconfirm "${missing_pkgs[@]}"
    elif command -v zypper >/dev/null 2>&1; then
      zypper -n in "${missing_pkgs[@]}"
    else
      echo "Fant ingen støttet pakkehåndterer for å installere: ${missing_pkgs[*]}"
      exit 1
    fi
  fi

  # Prompt for punchline (stored globally)
  current_punchline="Hello"
  if [[ -r /etc/ps1-punchline ]]; then
    current_punchline="$(head -n 1 /etc/ps1-punchline 2>/dev/null || echo "Hello")"
    current_punchline="${current_punchline%%$'\r'}"
  fi

  if [[ -t 0 ]]; then
    echo "Punchline (enter for å beholde eksisterende):"
    read -r -p "> " new_punchline
  else
    new_punchline=""
  fi

  if [[ -n "${new_punchline}" ]]; then
    printf '%s\n' "${new_punchline}" > /etc/ps1-punchline
  else
    printf '%s\n' "${current_punchline}" > /etc/ps1-punchline
  fi
  chmod 0644 /etc/ps1-punchline
fi

# Copy modules
if [[ "$install_ps1" -eq 1 ]]; then
  cp -f "$SRC_PS1" "$DST_PS1"
  chmod 0644 "$DST_PS1"
else
  rm -f "$DST_PS1"
fi

if [[ "$install_banner" -eq 1 ]]; then
  cp -f "$SRC_BANNER" "$DST_BANNER"
  chmod 0644 "$DST_BANNER"
else
  rm -f "$DST_BANNER"
fi

# Ensure /etc/bash.bashrc sources both (idempotent)
tmp="$(mktemp)"
orig_mode="$(stat -c '%a' "$BASH_BASHRC")"
orig_owner="$(stat -c '%u' "$BASH_BASHRC")"
orig_group="$(stat -c '%g' "$BASH_BASHRC")"
awk -v s="$MARKER_START" -v e="$MARKER_END" '
  $0==s {inside=1; next}
  $0==e {inside=0; next}
  !inside {print}
' "$BASH_BASHRC" > "$tmp"

cat >> "$tmp" <<EOF

$MARKER_START
# Load PS1 + Banner (managed)
$(if [[ "$install_ps1" -eq 1 ]]; then cat <<'INNER'
if [ -r "/etc/profile.d/ps1.sh" ]; then
  . "/etc/profile.d/ps1.sh"
fi
INNER
fi)
$(if [[ "$install_banner" -eq 1 ]]; then cat <<'INNER'
if [ -r "/etc/profile.d/banner.sh" ]; then
  . "/etc/profile.d/banner.sh"
fi
INNER
fi)
$MARKER_END
EOF

install -m "$orig_mode" -o "$orig_owner" -g "$orig_group" "$tmp" "$BASH_BASHRC"
if command -v restorecon >/dev/null 2>&1; then
  restorecon "$BASH_BASHRC" >/dev/null 2>&1 || true
fi
rm -f "$tmp"

echo "✅ Installert globalt (Linux/WSL):"
if [[ "$install_ps1" -eq 1 ]]; then
  echo "  - $DST_PS1"
fi
if [[ "$install_banner" -eq 1 ]]; then
  echo "  - $DST_BANNER"
fi
echo "  - sourcet fra $BASH_BASHRC"
echo
echo "Åpne et nytt shell for å se endringene."
echo
echo "Disable per bruker:"
echo "  export DISABLE_GLOBAL_PS1=1"
echo "  export DISABLE_GLOBAL_BANNER=1"
echo "  eller:"
echo "  mkdir -p ~/.config/ps1 && touch ~/.config/ps1/disable  # disable PS1"
echo "  mkdir -p ~/.config/ps1 && touch ~/.config/ps1/disable-banner  # disable banner"
