#!/bin/bash
set -euo pipefail

##########################
# Color helpers (printf-safe, no tput)
##########################
RED="\033[0;31m"
GREEN="\033[0;32m"
YELLOW="\033[0;33m"
CYAN="\033[0;36m"
GRAY="\033[0;37m"
BOLD="\033[1m"
RESET="\033[0m"

# Disable colors if output is not a terminal
if [ ! -t 1 ]; then
    RED="" GREEN="" YELLOW="" CYAN="" GRAY="" BOLD="" RESET=""
fi

say_red()    { printf "\n"; printf "%b%s%b\n" "$RED" "$*" "$RESET"; }
say_green()    { printf "\n"; printf "%b%s%b\n" "$GREEN" "$*" "$RESET"; }
say_yellow()   { printf "\n"; printf "%b%s%b\n" "$YELLOW" "$*" "$RESET"; }
say_cyan()     { printf "\n"; printf "%b%s%b\n" "$CYAN" "$*" "$RESET"; }
say_gray()     { printf "\n"; printf "%b%s%b\n" "$GRAY" "$*" "$RESET"; }
say_bold()     { printf "\n"; printf "%b%s%b\n" "$BOLD" "$*" "$RESET"; }

##########################
# Use exported variables from main detection script
##########################
OS="${DETECTED_OS}"
DDE="${DETECTED_DE}"
DE="${SELECTED_DE:-none}"
TWM="${SELECTED_TWM:-none}"
INSTALL_LEVEL="${INSTALL_LEVEL:-minimal}"
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

echo
say_cyan "Starting Debian setup..."
echo
say_gray "DE: $DE, TWM: $TWM, Install Level: $INSTALL_LEVEL"

##########################
# Helper functions
##########################
install_packages() { sudo pacman -S --noconfirm --needed "$@"; }
is_package_installed() { pacman -Qi "$@" &>/dev/null; }
remove_packages() {
    local packages_to_remove=()
    
    # 1. Identify which packages are actually installed
    for pkg in "$@"; do
        if is_package_installed "$pkg"; then
            packages_to_remove+=("$pkg")
        else
            say_gray "Package '$pkg' not found. Skipping removal."
        fi
    done

    # 2. Only run pacman if there are packages to remove
    if [ ${#packages_to_remove[@]} -gt 0 ]; then
        say_cyan "Removing packages: ${packages_to_remove[*]}"
        sudo pacman -Rs --noconfirm "${packages_to_remove[@]}" || true
    else
        say_gray "No packages to remove from the list."
    fi
}

##########################
# 0. Ensure base is ready for installation
##########################
installed_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
pkg_dir="packages"

if [[ ! -d "$pkg_dir" ]]; then
    say_red "Directory not found: $pkg_dir"
    exit 1
fi

if ! command -v curl >/dev/null 2>&1; then
    say_yellow "curl is not installed. Installing..."
    sudo pacman -Sy
    install_packages curl
fi

# Install all local packages using pacman
find "$pkg_dir" -maxdepth 1 -name '*.pkg.tar.zst' -print0 | sudo xargs -0 pacman -U --noconfirm

##########################
# 1. Add repositories if missing and update system
##########################
say_yellow "Checking pacman.conf and sources..."
sudo cp /etc/pacman.conf /etc/pacman.conf.bak.$(date +%s)
sudo cp -v "$installed_dir/config-files/pacman.conf" /etc/pacman.conf

# Update the system
say_gray "Updating the system - sudo pacman -Syyu"
echo
sudo pacman -Syyu --noconfirm

##########################
# 2. Fix missing console font
##########################
say_yellow "Fix missing console font"

if grep -q FONT= /etc/vconsole.conf; then
    say_green "Font is already set in /etc/vconsole.conf"
else
    say_green "Setting console font in /etc/vconsole.conf"
    echo 'FONT=gr737c-8x14' | sudo tee -a /etc/vconsole.conf
fi

##########################
# 3. Detected OS / DE / TWM info
##########################
say_yellow "Detected OS / Desktop Environment / Tiling Window Manager"
echo
say_gray "Installing OS-specific packages..."
say_gray "Selected DE: $DE"
say_gray "Selected TWM: $TWM"
say_gray "Installation Level: $INSTALL_LEVEL"

##########################
# 4. Desktop Environment installation
##########################
case "$DE" in
    xfce|plasma|gnome)
        say_yellow "Preparing to install $DE..."
        SCRIPT_NAME="$SCRIPT_DIR/${OS}-${DE}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            say_cyan "Running $SCRIPT_NAME..."
            bash "$SCRIPT_NAME"
        else
            say_red "Error: $SCRIPT_NAME not found!"
            exit 1
        fi
        ;;
    none)
        say_gray "No Desktop Environment selected, skipping DE installation."
        ;;
esac

##########################
# 5. Tiling Window Manager installation
##########################
case "$TWM" in
    chadwm|hyprland)
        say_yellow "Installing $TWM..."
        SCRIPT_NAME="$SCRIPT_DIR/${OS}-${TWM}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            say_cyan "Running $SCRIPT_NAME..."
            bash "$SCRIPT_NAME"
        else
            say_red "Error: $SCRIPT_NAME not found!"
            exit 1
        fi
        ;;
    none)
        say_gray "No tiling window manager selected."
        ;;
esac

##########################
# 6. Installation level handling
##########################
case "$INSTALL_LEVEL" in
    minimal|full|workstation|server)
        say_cyan "Installation level: $INSTALL_LEVEL"
        SCRIPT_NAME="$SCRIPT_DIR/${OS}-${INSTALL_LEVEL}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            say_cyan "Running $SCRIPT_NAME..."
            bash "$SCRIPT_NAME"
        else
            say_red "Error: $SCRIPT_NAME not found!"
            exit 1
        fi
        ;;
esac

say_green "Arch setup complete."
say_red "The installation has finished. A reboot is recommended before continuing."
read -rp "Reboot now? [y/N]: " reboot_choice
case "${reboot_choice,,}" in
    y|yes)
        say_red "Rebooting now."
        sudo reboot
        ;;
    *)
        say_yellow "Skipping reboot. Make sure to reboot manually before continuing upgrades."
        exit 0
        ;;
esac