#!/bin/bash

# Unified theme switching script for WSL and macOS
# This script handles theme switching for both platforms

set -e

# Helper functions
show_usage() {
  echo "Usage: theme {dark|light|toggle|status}"
  echo ""
  echo "Commands:"
  echo "  dark    - Switch to dark theme"
  echo "  light   - Switch to light theme"
  echo "  toggle  - Toggle between light and dark themes"
  echo "  status  - Show current theme status"
  echo ""
  echo "This command switches themes for:"
  echo "  - Neovim (via nvim_color_scheme file)"
  echo "  - Alacritty on macOS follows System Settings (LaunchAgent sync)"
  echo "  - Windows Terminal (via settings.json on WSL)"
  echo "  - Windows system theme (on WSL)"
}

show_status() {
  echo "Current theme status:"
  echo ""

  # Check Neovim theme
  nvim_color_theme_path=~/.local/share/nvim_color_scheme
  if [ -f "$nvim_color_theme_path" ]; then
    nvim_theme=$(cat "$nvim_color_theme_path" | tr -d '\n')
    echo "  Neovim: $nvim_theme"
  else
    echo "  Neovim: no theme file found"
  fi

  # Check platform-specific themes
  if [[ -n "$WSL_DISTRO_NAME" ]]; then
    echo "  Platform: WSL"
    echo "  Windows Terminal: configured via settings.json"
    echo "  Windows system theme: managed by theme command"
  elif [[ "$OSTYPE" == "darwin"* ]]; then
    echo "  Platform: macOS"

    marker="$HOME/.config/alacritty/.last-system-theme"
    if [ -f "$marker" ]; then
      echo "  Alacritty: follows system (active palette: $(tr -d '\n' <"$marker"))"
    else
      echo "  Alacritty: follows system (sync after next login or run alacritty-sync-system-theme)"
    fi
  else
    echo "  Platform: other"
  fi
}

toggle_theme() {
  # Get current theme - prefer platform-specific detection
  current_theme=""

  if [[ "$OSTYPE" == "darwin"* ]]; then
    if [[ "$(defaults read -g AppleInterfaceStyle 2>/dev/null)" == "Dark" ]]; then
      current_theme="dark"
    else
      current_theme="light"
    fi
  fi

  # Fallback to Neovim file if platform-specific detection didn't work
  if [ -z "$current_theme" ]; then
    nvim_color_theme_path=~/.local/share/nvim_color_scheme
    if [ -f "$nvim_color_theme_path" ]; then
      current_theme=$(cat "$nvim_color_theme_path" | tr -d '\n')
    else
      current_theme="light"  # Default to light if no theme file exists
    fi
  fi

  # Determine new theme
  if [ "$current_theme" = "light" ]; then
    new_theme="dark"
  else
    new_theme="light"
  fi

  echo "Toggling theme from $current_theme to $new_theme"

  # Call the main script with the new theme
  exec "$0" "$new_theme"
}

color_scheme=$1

# Handle special commands
case "$color_scheme" in
  "status")
    show_status
    exit 0
    ;;
  "toggle")
    toggle_theme
    exit 0
    ;;
  "dark"|"light")
    # Valid theme, continue with normal flow
    ;;
  *)
    show_usage
    exit 1
    ;;
esac

# Get the directory where this script is located
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
DOTFILES_DIR="$(dirname "$SCRIPT_DIR")"

# Write theme to nvim color scheme file (works on both platforms)
nvim_color_theme_path=~/.local/share/nvim_color_scheme
mkdir -p "$(dirname "$nvim_color_theme_path")"
echo "$color_scheme" > "$nvim_color_theme_path"
echo "Updated Neovim theme: $color_scheme"

# Detect platform and handle platform-specific theme switching
if [[ -n "$WSL_DISTRO_NAME" ]]; then
  # WSL platform - handle Windows Terminal and system theme
  echo "Detected WSL platform"

  # Check that all relevant files exist
  windows_username=$(powershell.exe '$env:UserName' | tr -d '\r\n')
  windows_terminal_settings_path="/mnt/c/Users/${windows_username}/AppData/Local/Packages/Microsoft.WindowsTerminal_8wekyb3d8bbwe/LocalState/settings.json"
  dark_mode_config_path="$DOTFILES_DIR/assets/windows_terminal/dark.settings.json"
  light_mode_config_path="$DOTFILES_DIR/assets/windows_terminal/light.settings.json"

  if [ ! -f "$light_mode_config_path" ]; then
    echo "error: light_mode_config_path missing"
    echo "expected: $light_mode_config_path"
    exit 1
  fi

  if [ ! -f "$dark_mode_config_path" ]; then
    echo "error: dark_mode_config_path missing"
    echo "expected: $dark_mode_config_path"
    exit 1
  fi

  if [ ! -f "$windows_terminal_settings_path" ]; then
    echo "error: windows terminal settings path missing"
    echo "expected: $windows_terminal_settings_path"
    exit 1
  fi

  # Update Windows Terminal settings
  if [ "$color_scheme" = 'dark' ]; then
    cp "$dark_mode_config_path" "$windows_terminal_settings_path"
    echo "Updated Windows Terminal: dark theme"
    echo "Switching Windows system theme to dark..."
    powershell.exe -Command "start C:\Windows\Resources\Themes\dark.theme"
    powershell.exe "timeout /t 3; taskkill /im systemsettings.exe /f"
  else
    cp "$light_mode_config_path" "$windows_terminal_settings_path"
    echo "Updated Windows Terminal: light theme"
    echo "Switching Windows system theme to light..."
    powershell.exe -Command "start C:\Windows\Resources\Themes\aero.theme"
    powershell.exe "timeout /t 3; taskkill /im systemsettings.exe /f"
  fi

  echo "Setting Sound Schema to None"
  powershell.exe -Command "Set-ItemProperty -Path HKCU:\AppEvents\Schemes -Name '(Default)' -Value '.None'"

elif [[ "$OSTYPE" == "darwin"* ]]; then
  echo "Detected macOS platform"
  echo "Alacritty follows System Settings → Appearance (no rebuild). Neovim theme file updated above."

else
  # Other platforms - just update the theme file
  echo "Detected other platform - only updating Neovim theme file"
fi

echo "Theme switching complete: $color_scheme"