#!/bin/bash
set -euo pipefail

##########################
# Color helpers (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

color_red()    { printf '%b' "$RED"; }
color_green()  { printf '%b' "$GREEN"; }
color_yellow() { printf '%b' "$YELLOW"; }
color_cyan()   { printf '%b' "$CYAN"; }
color_gray()   { printf '%b' "$GRAY"; }
color_reset()  { printf '%b' "$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}"

color_cyan
echo
echo "Starting Debian setup..."
color_reset
echo
echo "DE: $DE, TWM: $TWM, Install Level: $INSTALL_LEVEL"

##########################
# 0. Ensure curl is installed
##########################
if ! command -v curl >/dev/null 2>&1; then
    color_yellow
    echo
    echo "curl is not installed. Installing..."
    color_reset
    sudo apt update
    sudo apt -y install curl
fi

##########################
# 1. Add contrib and non-free if missing
##########################
color_yellow
echo
echo "Checking /etc/apt/sources.list for contrib/non-free..."
color_reset
sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak.$(date +%s)
sudo sed -i -r 's/^(deb\s+\S+\s+\S+)\s+(main)$/\1 main contrib non-free/' /etc/apt/sources.list
color_green
echo
echo "Updated sources.list to include contrib/non-free where needed."
color_reset

##########################
# 1a. Update archive.debian.org if needed
##########################
CURRENT_CODENAME=$(grep -Po 'deb\s+\S+\s+\K\S+' /etc/apt/sources.list | grep -E '^(buster|bullseye|bookworm|trixie)$' | head -n1)
DEBIAN_ORDER=(buster bullseye bookworm trixie)

codename_index() {
    local code="$1"
    for i in "${!DEBIAN_ORDER[@]}"; do
        [[ "${DEBIAN_ORDER[$i]}" == "$code" ]] && echo "$i" && return
    done
    echo -1
}

CURRENT_INDEX=$(codename_index "$CURRENT_CODENAME")
BULLSEYE_INDEX=$(codename_index "bullseye")

if [[ "$CURRENT_INDEX" -ge "$BULLSEYE_INDEX" ]] && grep -q "archive.debian.org" /etc/apt/sources.list; then
    color_yellow
    echo
    echo "Found archive.debian.org in sources.list and system is Bullseye or newer, updating to deb.debian.org..."
    color_reset
    sudo sed -i -r 's|archive\.debian\.org|deb.debian.org|g' /etc/apt/sources.list
    color_green
    echo
    echo "Updated sources.list to use deb.debian.org."
    color_reset
fi

##########################
# 2. Full update and upgrade
##########################
color_yellow
echo
echo "Updating package lists..."
color_reset
sudo apt update

# Autoremove before full-upgrade
AUTOREMOVE_PENDING=$(apt -s autoremove | grep -E 'Remv' || true)
if [[ -n "$AUTOREMOVE_PENDING" ]]; then
    color_yellow
    echo
    echo "Removing packages that are no longer required before upgrade..."
    color_reset
    sudo apt -y autoremove
fi

color_yellow
echo
echo "Upgrading installed packages..."
color_reset
sudo apt -y full-upgrade

# Autoremove after full-upgrade
AUTOREMOVE_PENDING=$(apt -s autoremove | grep -E 'Remv' || true)
if [[ -n "$AUTOREMOVE_PENDING" ]]; then
    color_yellow
    echo
    echo "Removing packages that are no longer required after upgrade..."
    color_reset
    sudo apt -y autoremove
fi

UPGRADE_PENDING=$(apt list --upgradable 2>/dev/null | grep -v Listing || true)
if [[ -n "$UPGRADE_PENDING" ]]; then
    color_red
    echo
    echo "Some packages were upgraded. A reboot is recommended before continuing."
    color_reset
    read -rp "Reboot now? [y/N]: " reboot_choice
    case "${reboot_choice,,}" in
        y|yes)
            color_red
            echo
            echo "Rebooting now. After reboot, please restart this script to continue..."
            color_reset
            sudo reboot
            ;;
        *)
            color_yellow
            echo
            echo "Skipping reboot. Make sure to reboot manually before continuing upgrades."
            color_reset
            exit 0
            ;;
    esac
else
    color_green
    echo
    echo "All packages are up to date. Continuing to Debian major version check..."
    color_reset
fi

##########################
# 3. Stepwise major version upgrade
##########################
DEBIAN_SEQUENCE=(buster bullseye bookworm trixie)
CURRENT_CODENAME=$(grep -Po 'deb\s+\S+\s+\K\S+' /etc/apt/sources.list | grep -E '^(buster|bullseye|bookworm|trixie)$' | head -n1)
LATEST_CODENAME=${DEBIAN_SEQUENCE[-1]}

color_cyan
echo
echo "Current codename: $CURRENT_CODENAME"
echo "Latest stable codename: $LATEST_CODENAME"
color_reset

while [[ "$CURRENT_CODENAME" != "$LATEST_CODENAME" ]]; do
    NEXT_CODENAME=""
    for i in "${!DEBIAN_SEQUENCE[@]}"; do
        if [[ "${DEBIAN_SEQUENCE[$i]}" == "$CURRENT_CODENAME" ]]; then
            NEXT_CODENAME="${DEBIAN_SEQUENCE[$((i+1))]}"
            break
        fi
    done

    if [[ -z "$NEXT_CODENAME" ]]; then
        color_red
        echo
        echo "Error: Cannot determine next codename after $CURRENT_CODENAME"
        color_reset
        exit 1
    fi

    color_yellow
    echo
    echo "Detected codename $CURRENT_CODENAME, next stable version: $NEXT_CODENAME"
    color_reset
    read -rp "Do you want to upgrade to $NEXT_CODENAME? [y/N]: " choice
    case "${choice,,}" in
        y|yes)
            color_yellow
            echo
            echo "Updating sources.list to $NEXT_CODENAME..."
            color_reset
            sudo sed -i -r "s/\b$CURRENT_CODENAME\b/$NEXT_CODENAME/g" /etc/apt/sources.list

            color_yellow
            echo
            echo "Updating packages..."
            color_reset
            sudo apt update

            AUTOREMOVE_PENDING=$(apt -s autoremove | grep -E 'Remv' || true)
            if [[ -n "$AUTOREMOVE_PENDING" ]]; then
                color_yellow
                echo
                echo "Removing packages that are no longer required before upgrade..."
                color_reset
                sudo apt -y autoremove
            fi

            sudo apt -y full-upgrade

            AUTOREMOVE_PENDING=$(apt -s autoremove | grep -E 'Remv' || true)
            if [[ -n "$AUTOREMOVE_PENDING" ]]; then
                color_yellow
                echo
                echo "Removing packages that are no longer required after upgrade..."
                color_reset
                sudo apt -y autoremove
            fi

            color_green
            echo
            echo "Upgrade to $NEXT_CODENAME complete. A reboot is recommended."
            color_reset
            read -rp "Press Enter to reboot..." _
            sudo reboot
            ;;
        *)
            color_yellow
            echo
            echo "Skipping upgrade to $NEXT_CODENAME. Continuing with current version."
            color_reset
            break
            ;;
    esac

    CURRENT_CODENAME=$(grep -Po 'deb\s+\S+\s+\K\S+' /etc/apt/sources.list | grep -E '^(buster|bullseye|bookworm|trixie)$' | head -n1)
done

color_green
echo
echo "Debian is now at codename $CURRENT_CODENAME. Continuing with DE/TWM installation..."
color_reset

##########################
# 4. Desktop Environment installation
##########################
case "$DE" in
    xfce|plasma|gnome)
        color_yellow
        echo
        echo "Preparing to install $DE..."
        color_reset

        SCRIPT_NAME="${OS}-${DE}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            color_cyan
            echo
            echo "Running $SCRIPT_NAME..."
            color_reset
            bash "$SCRIPT_NAME"
        else
            color_red
            echo
            echo "Error: $SCRIPT_NAME not found!"
            color_reset
            exit 1
        fi
        ;;
    none)
        color_gray
        echo
        echo "No Desktop Environment selected, skipping DE installation."
        color_reset
        ;;
esac

##########################
# 5. Tiling Window Manager installation
##########################
case "$TWM" in
    chadwm|hyprland)
        color_yellow
        echo
        echo "Installing $TWM..."
        color_reset

        SCRIPT_NAME="${OS}-${TWM}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            color_cyan
            echo
            echo "Running $SCRIPT_NAME..."
            color_reset
            bash "$SCRIPT_NAME"
        else
            color_red
            echo
            echo "Error: $SCRIPT_NAME not found!"
            color_reset
            exit 1
        fi
        ;;
    none)
        color_gray
        echo
        echo "No tiling window manager selected."
        color_reset
        ;;
esac

##########################
# 6. Installation level handling
##########################
case "$INSTALL_LEVEL" in
    minimal|full|workstation|server)
        color_cyan
        echo
        echo "Installation level: $INSTALL_LEVEL"
        color_reset

        SCRIPT_NAME="${OS}-${INSTALL_LEVEL}.sh"
        if [[ -f "$SCRIPT_NAME" ]]; then
            color_cyan
            echo
            echo "Running $SCRIPT_NAME..."
            color_reset
            bash "$SCRIPT_NAME"
        else
            color_red
            echo
            echo "Error: $SCRIPT_NAME not found!"
            color_reset
            exit 1
        fi
        ;;
esac

color_green
echo
echo "Debian setup complete."
color_reset