#!/usr/bin/env bash
###############################################################################
#                           GEEKCADE MANAGEMENT MENU
#                         Cabinet Boot/Admin Console
#                           Version: v2.0 (build 38)
#
# Boot behavior:
#   - tty1 auto-login runs this script via .bash_profile
#   - Boot gate: 5s countdown; any key -> menu, otherwise -> AM+
#   - When AM+ exits, getty@tty1 is restarted to clear stuck KMSDRM
#     keyboard grabs from MAME's SDL2 backend.
#   - On the next login, .bash_profile runs this script again, but the
#     boot gate is suppressed (via /tmp/.geekcade_skip_gate) so we land
#     directly in the menu instead of relaunching AM+.
#
# Assumptions:
#   - Ubuntu is already installed
#   - User "geekcade" exists
#   - GeekCade installer has created:
#       /home/geekcade/mame
#       /home/geekcade/attractplus
#       /home/geekcade/mame.sh
#       /home/geekcade/attractplus.sh
###############################################################################

# ==============================
# [0] CONFIG & PATHS
# ==============================
BASE_DIR="$HOME"

INSTALL_SCRIPT_PATH="$BASE_DIR/geekcade_install.sh"
MENU_SCRIPT_PATH="$BASE_DIR/geekcade_menu.sh"

# All GeekCade-hosted files live at this base URL. Override with:
#   GEEKCADE_BASE_URL=http://other.host/path ./geekcade_menu.sh
GEEKCADE_BASE_URL="${GEEKCADE_BASE_URL:-https://install.geekcade.com/deploy}"
INSTALL_SCRIPT_URL="${GEEKCADE_BASE_URL}/geekcade_install.sh"
MENU_SCRIPT_URL="${GEEKCADE_BASE_URL}/geekcade_menu.sh"

MAME_DIR="$BASE_DIR/mame"
ATTRACT_DIR="$BASE_DIR/attractplus"
LOG_DIR="$BASE_DIR/logs"

MAME_LAUNCHER="$BASE_DIR/mame.sh"
ATTRACT_LAUNCHER="$BASE_DIR/attractplus.sh"

SUCCESS_FLAG_FILE="$BASE_DIR/.geekcade_install_success"

# Used to signal "skip the boot gate, go straight to the menu" - set
# right before we restart getty@tty1, so the next login lands in the
# menu instead of bouncing right back into AM+.
SKIP_GATE_FLAG="/tmp/.geekcade_skip_gate"

# Boot admin-gate timeout in seconds
AUTO_MENU_TIMEOUT="${AUTO_MENU_TIMEOUT:-5}"

# Stable PATH
export PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games"

# Scrollable logs
export LESS='-R'

# ==============================
# [1] COLORS
# ==============================
GREEN=$'\033[0;32m'
RED=$'\033[0;31m'
YELLOW=$'\033[1;33m'
CYAN=$'\033[0;36m'
NC=$'\033[0m'

# ==============================
# [2] UI HELPERS
# ==============================
display_banner() {
  clear
  echo -e "${GREEN}"
  cat << "EOF"
  ____           _     ____          _
 / ___| ___  ___| | __/ ___|__ _  __| | ___
| |  _ / _ \/ _ \ |/ / |   / _` |/ _` |/ _ \
| |_| |  __/  __/   <| |__| (_| | (_| |  __/
 \____|\___|\___|_|\_\\____\__,_|\__,_|\___|
EOF
  echo -e "             Cabinet Menu v2.0 (build 38)${NC}"
  echo
}

pause_menu() {
  echo
  read -n 1 -s -r -p "Press any key to continue..."
  echo
}

display_stats() {
  local mame_version="Not Installed"
  local attract_version="Not Installed"
  local rom_count="0"
  local install_status="${YELLOW}Incomplete${NC}"
  local layout_name="Unknown"

  # MAME version - call binary directly to avoid any launcher side effects
  if command -v /usr/games/mame >/dev/null 2>&1; then
    mame_version="$(/usr/games/mame -version 2>&1 | head -n1 | sed 's/ ([^)]*)//')"
  elif command -v mame >/dev/null 2>&1; then
    mame_version="$(command mame -version 2>&1 | head -n1 | sed 's/ ([^)]*)//')"
  else
    mame_version="${RED}Not Installed${NC}"
  fi

  # Attract-Mode Plus version - call binary directly (NOT through launcher
  # or alias) so we just print version and exit
  if command -v attractplus >/dev/null 2>&1; then
    attract_version="$(command attractplus --version 2>/dev/null | head -n1 || true)"
    [ -z "$attract_version" ] && attract_version="${GREEN}Installed${NC}"
  elif command -v attract >/dev/null 2>&1; then
    attract_version="$(command attract --version 2>/dev/null | head -n1 || true)"
    [ -z "$attract_version" ] && attract_version="${GREEN}Installed${NC}"
  else
    attract_version="${RED}Not Installed${NC}"
  fi

  # ROM count
  if [ -d "$MAME_DIR/roms" ]; then
    rom_count="$(find "$MAME_DIR/roms" -type f 2>/dev/null | wc -l | xargs)"
  fi

  # Install status
  if [ -f "$SUCCESS_FLAG_FILE" ]; then
    install_status="${GREEN}Complete${NC}"
  fi

  # Layout
  if [ -f "$ATTRACT_DIR/attract.cfg" ]; then
    layout_name="$(grep -E '^[[:space:]]*layout[[:space:]]+' "$ATTRACT_DIR/attract.cfg" | awk '{print $2}' | head -n1)"
    [ -z "$layout_name" ] && layout_name="Not set"
  fi

  echo -e "${CYAN}----------------------------------------------------${NC}"
  echo -e "                 ${YELLOW}System Status${NC}"
  echo -e "${CYAN}----------------------------------------------------${NC}"
  printf " %-16s %b\n" "MAME Version:" "$mame_version"
  printf " %-16s %b\n" "Attract-Mode:" "$attract_version"
  printf " %-16s %s\n" "MAME ROMs:" "$rom_count"
  printf " %-16s %b\n" "Setup Status:" "$install_status"
  printf " %-16s %s\n" "Layout:" "$layout_name"
  printf " %-16s %s\n" "MAME Dir:" "$MAME_DIR"
  printf " %-16s %s\n" "Attract Dir:" "$ATTRACT_DIR"
  echo -e "${CYAN}----------------------------------------------------${NC}"
  echo
}

display_menu() {
  echo -e "${YELLOW}Please choose an option:${NC}"
  echo "   1) Launch Attract-Mode"
  echo "   2) Sync Assets from USB"
  echo "   3) Rebuild Attract-Mode MAME Romlist"
  echo "   4) View Latest Installer Log"
  echo "   5) Check for System Updates (apt)"
  echo "   6) Download Latest GeekCade Installer"
  echo "   7) Download Latest GeekCade Menu"
  echo "   8) Run Installer - COMPILE Attract-Mode Plus"
  echo "   9) Run Installer - RELEASE Attract-Mode Plus"
  echo "  10) Run Installer - SKIP Attract-Mode Plus"
  echo "  11) Reboot System"
  echo "  12) Shutdown System"
  echo
  echo "   q) Exit to Shell"
  echo
}

# ==============================
# [3] CABINET INPUT RECOVERY
# ==============================
# After AM+ (or anything that ran MAME) exits, SDL2's KMSDRM backend can
# leave systemd-logind holding /dev/input/event* exclusively, freezing
# keyboard input on tty1. The reliable cure is to restart getty@tty1
# which kills our shell session and respawns a clean login. The next
# login will hit .bash_profile -> this script again, and we want it to
# land directly in the menu (not bounce back into AM+), so we touch a
# flag first that the boot gate honors.
recover_tty_and_relaunch_menu() {
  touch "$SKIP_GATE_FLAG"
  echo
  echo -e "${YELLOW}Resetting tty1 input (returning to menu)...${NC}"
  sleep 1
  sudo /bin/systemctl restart getty@tty1
  # systemctl will kill this shell; if for some reason it returns,
  # fall through and exit so .bash_profile reruns naturally.
  exit 0
}

# ==============================
# [4] ACTIONS
# ==============================

# Shared download helper. Args: <url> <destination> <label>
_download_to() {
  local url="$1"
  local dest="$2"
  local label="$3"

  echo -e "${YELLOW}======> Downloading ${label}...${NC}"
  echo "        $url"

  if command -v curl >/dev/null 2>&1; then
    if ! curl -fsSL "$url" -o "$dest"; then
      echo -e "${RED}FAILED: Could not download $label via curl.${NC}"
      return 1
    fi
  else
    if ! wget -qO "$dest" "$url"; then
      echo -e "${RED}FAILED: Could not download $label via wget.${NC}"
      return 1
    fi
  fi

  if [ ! -s "$dest" ]; then
    echo -e "${RED}FAILED: $label downloaded but file is empty.${NC}"
    rm -f "$dest"
    return 1
  fi

  chmod +x "$dest"
  echo -e "${GREEN}DONE. $label saved to $dest${NC}"
  return 0
}

download_installer() {
  _download_to "$INSTALL_SCRIPT_URL" "$INSTALL_SCRIPT_PATH" "GeekCade installer"
  pause_menu
}

download_menu() {
  _download_to "$MENU_SCRIPT_URL" "$MENU_SCRIPT_PATH" "GeekCade menu"
  echo
  echo -e "${YELLOW}Note:${NC} the new menu will take effect on next tty1 login."
  echo "      To use it now, exit (q) and the next login will pick it up."
  pause_menu
}

run_installer_with_flag() {
  local flag="$1"
  local label="$2"

  if [ ! -f "$INSTALL_SCRIPT_PATH" ]; then
    echo -e "${RED}Installer script not found at $INSTALL_SCRIPT_PATH${NC}"
    echo "Please download it first using option 5."
    pause_menu
    return
  fi

  chmod +x "$INSTALL_SCRIPT_PATH"

  echo -e "${YELLOW}Starting GeekCade installer: $label${NC}"
  "$INSTALL_SCRIPT_PATH" "$flag"
  echo -e "${GREEN}Installer finished: $label${NC}"
  pause_menu
}

sync_assets_from_usb() {
  local ASSETS_SCRIPT="$BASE_DIR/geekcade_assets.sh"

  if [ ! -x "$ASSETS_SCRIPT" ]; then
    echo -e "${RED}Asset sync script not found at $ASSETS_SCRIPT${NC}"
    echo "You can download it via Option 6 (Latest Installer) which fetches it,"
    echo "or grab it directly from $GEEKCADE_BASE_URL/geekcade_assets.sh"
    pause_menu
    return
  fi

  echo -e "${YELLOW}Running asset sync...${NC}"
  echo
  "$ASSETS_SCRIPT" || true
  pause_menu
}

rebuild_romlist() {
  if [ ! -x "$ATTRACT_LAUNCHER" ]; then
    echo -e "${RED}Attract launcher not found or not executable:${NC} $ATTRACT_LAUNCHER"
    pause_menu
    return
  fi

  echo -e "${YELLOW}Rebuilding MAME romlist...${NC}"
  mkdir -p "$ATTRACT_DIR/romlists"

  rm -f "$ATTRACT_DIR/romlists/mame.txt"
  "$ATTRACT_LAUNCHER" --build-romlist mame || true

  if [ -f "$ATTRACT_DIR/romlists/mame.txt" ]; then
    echo -e "${GREEN}Romlist rebuilt successfully:${NC} $ATTRACT_DIR/romlists/mame.txt"
  else
    echo -e "${YELLOW}Romlist command finished, but mame.txt was not created.${NC}"
  fi

  pause_menu
}

view_installer_log() {
  local log
  log="$(ls -1t "$LOG_DIR"/geekcade_install_*.log 2>/dev/null | head -n1 || true)"

  if [ -n "$log" ]; then
    echo -e "${YELLOW}Opening latest installer log:${NC} $log"
    sleep 0.5
    less "$log"
  else
    echo -e "${RED}No installer logs found in $LOG_DIR.${NC}"
    pause_menu
  fi
}

update_system() {
  echo -e "${YELLOW}Checking for system updates...${NC}"
  sudo apt-get update && sudo apt-get upgrade -y
  echo -e "${GREEN}System update complete.${NC}"
  pause_menu
}

launch_attract_now() {
  if [ ! -x "$ATTRACT_LAUNCHER" ]; then
    echo -e "${RED}Attract launcher not found or not executable:${NC} $ATTRACT_LAUNCHER"
    pause_menu
    return
  fi

  echo -e "${YELLOW}Launching Attract-Mode...${NC}"
  sleep 1
  "$ATTRACT_LAUNCHER"

  # AM+ has now exited. Reset tty1 to clear any stuck input grab from
  # MAME's SDL2 backend, then come back into this menu.
  recover_tty_and_relaunch_menu
}

confirm_reboot() {
  read -r -p "Are you sure you want to reboot? (y/N) " confirm
  if [[ "$confirm" == [yY] ]]; then
    sudo reboot
  fi
}

confirm_shutdown() {
  read -r -p "Are you sure you want to shut down? (y/N) " confirm
  if [[ "$confirm" == [yY] ]]; then
    sudo shutdown now
  fi
}

# ==============================
# [5] BOOT GATE
# ==============================
launch_attract_or_menu_gate() {
  # Force menu if explicitly requested:
  #   GEEKCADE_MENU=1 ./geekcade_menu.sh
  if [ -n "${GEEKCADE_MENU:-}" ]; then
    return 0
  fi

  # Skip the gate if we just came back from AM+ via the tty1 reset path.
  # We consume the flag so subsequent boots still go through the gate.
  if [ -f "$SKIP_GATE_FLAG" ]; then
    rm -f "$SKIP_GATE_FLAG"
    return 0
  fi

  display_banner
  echo -e "${YELLOW}Press any key in the next ${AUTO_MENU_TIMEOUT}s for ADMIN MENU.${NC}"
  echo "Otherwise, launching Attract-Mode..."
  echo

  local pressed=0

  for ((i=AUTO_MENU_TIMEOUT; i>0; i--)); do
    printf "\rStarting Attract-Mode in %2ds... " "$i"
    if read -r -s -n 1 -t 1; then
      pressed=1
      break
    fi
  done

  echo -e "\r                                      \r"

  if [ "$pressed" -eq 0 ]; then
    if [ -x "$ATTRACT_LAUNCHER" ]; then
      "$ATTRACT_LAUNCHER"
      # AM+ exited. Reset tty1, come back via fresh login -> menu.
      recover_tty_and_relaunch_menu
    else
      echo -e "${RED}Attract launcher not found. Showing admin menu instead.${NC}"
      sleep 2
    fi
  fi

  return 0
}

# ==============================
# [6] MAIN LOOP
# ==============================
launch_attract_or_menu_gate

while true; do
  display_banner
  display_stats
  display_menu

  read -r -p "Enter your choice [1-12, q]: " choice
  choice="${choice//[[:space:]]/}"

  case "$choice" in
    1)
      launch_attract_now
      ;;
    2)
      sync_assets_from_usb
      ;;
    3)
      rebuild_romlist
      ;;
    4)
      view_installer_log
      ;;
    5)
      update_system
      ;;
    6)
      download_installer
      ;;
    7)
      download_menu
      ;;
    8)
      run_installer_with_flag "--compile" "COMPILE"
      ;;
    9)
      run_installer_with_flag "--release" "RELEASE"
      ;;
    10)
      run_installer_with_flag "--skip-attract" "SKIP Attract-Mode"
      ;;
    11)
      confirm_reboot
      ;;
    12)
      confirm_shutdown
      ;;
    q|Q)
      echo "Exiting to shell."
      break
      ;;
    *)
      echo -e "${RED}Invalid option. Please try again.${NC}"
      sleep 1
      ;;
  esac
done