source /etc/os-release
if [[ "$ID" != "ubuntu" ]]; then
    warn "This script is optimized for Ubuntu, but detected: $ID"
    if ! confirm "Continue anyway?"; then
        exit 1
    fi
fi

local version_id=${VERSION_ID%.*}
if [[ $version_id -lt 20 ]]; then
    error "Ubuntu 20.04 LTS or newer required (detected: $VERSION_ID)"
    exit 1
fi

info "Ubuntu $VERSION_ID detected"

# Add Charm repository
sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://repo.charm.sh/apt/gpg.key | sudo gpg --dearmor -o /etc/apt/keyrings/charm.gpg
echo "deb [signed-by=/etc/apt/keyrings/charm.gpg] https://repo.charm.sh/apt/ * *" | sudo tee /etc/apt/sources.list.d/charm.list >/dev/null

sudo apt update
sudo apt install -y gum

info "Gum installed successfully"

# Update package list
spin "Updating package list..." sudo apt update

# Install required packages
local packages=(
    "sddm"
    "qt6-base-dev"
    "qt6-declarative-dev"
    "qml6-module-qtquick"
    "qml6-module-qtquick-controls"
    "qml6-module-qtquick-layouts"
    "qml6-module-qt-labs-platform"
    "qml6-module-qtquick-virtualkeyboard"
    "qt6-multimedia-dev"
    "libqt6svg6-dev"
    "git"
)

spin "Installing SDDM and Qt6 packages..." sudo apt install -y "${packages[@]}"

info "Dependencies installed successfully"

# Check if we're in a desktop environment that might conflict
if [[ -n "${DESKTOP_SESSION:-}" ]]; then
    warn "Desktop session detected: $DESKTOP_SESSION"
    warn "SDDM will replace your current display manager"
fi

spin "Cloning theme repository..." git clone -b master --depth 1 "$THEME_REPO" "$PATH_TO_GIT_CLONE"
info "Repository cloned to $PATH_TO_GIT_CLONE"

if [[ ! -d "$src" ]]; then
    error "Theme repository not found. Please clone repository first."
    return 1
fi

# Backup existing theme
if [[ -d "$dst" ]]; then
    warn "Existing theme installation found"
    sudo mv "$dst" "${dst}_backup_$DATE"
    info "Previous theme backed up"
fi

# Create theme directory and copy files
sudo mkdir -p "$dst"
spin "Installing theme files..." sudo cp -r "$src"/* "$dst"/

# Install fonts if available
if [[ -d "$dst/Fonts" ]]; then
    spin "Installing fonts..." sudo cp -r "$dst/Fonts"/* /usr/share/fonts/
    sudo fc-cache -fv >/dev/null 2>&1
    info "Fonts installed and cache updated"
fi

# Configure SDDM
sudo mkdir -p /etc/sddm.conf.d

# Main SDDM configuration
if [[ ! -f /etc/sddm.conf ]]; then
    echo "[Theme]

# Virtual keyboard configuration
echo "[General]

info "Theme installed and configured"

info "Available theme variants:"
local theme
if command -v gum &>/dev/null; then
    theme=$(gum choose --header="Select a theme variant:" "${THEMES[@]}")
else
    echo "Please select a theme variant:"
    select theme in "${THEMES[@]}"; do
        [[ -n "$theme" ]] && break
    done
fi

[[ -z "$theme" ]] && theme="astronaut"

sudo sed -i "s|^ConfigFile=.*|ConfigFile=Themes/${theme}.conf|" "$METADATA"
info "Selected theme variant: $theme"

# Disable other display managers
local current_dm
if systemctl is-enabled gdm3 &>/dev/null; then
    current_dm="gdm3"
elif systemctl is-enabled lightdm &>/dev/null; then
    current_dm="lightdm"
elif systemctl is-enabled kdm &>/dev/null; then
    current_dm="kdm"
fi

if [[ -n "${current_dm:-}" ]]; then
    warn "Current display manager: $current_dm"
    if confirm "Disable $current_dm and enable SDDM?"; then
        sudo systemctl disable "$current_dm"
        info "Disabled $current_dm"
    fi
fi

sudo systemctl disable display-manager.service 2>/dev/null || true
sudo systemctl enable sddm.service

info "SDDM service enabled"
warn "Reboot required to see the new login screen"

if ! command -v sddm-greeter-qt6 &>/dev/null; then
    error "sddm-greeter-qt6 not found. Please install SDDM first."
    return 1
fi

local log_file="/tmp/${THEME_NAME}_preview_$DATE.log"

info "Starting theme preview (will close automatically in 10 seconds)..."
info "Use Ctrl+C to close preview manually"

# Start preview in background
sddm-greeter-qt6 --test-mode --theme "$THEMES_DIR/$THEME_NAME/" > "$log_file" 2>&1 &
local greeter_pid=$!

# Wait for 10 seconds or until process dies
for i in {1..10}; do
    if ! kill -0 "$greeter_pid" 2>/dev/null; then
        break
    fi
    sleep 1
    if command -v gum &>/dev/null; then
        echo -ne "\r$(gum style --foreground 12 "Preview running... ${i}/10 seconds")"
    else
        echo -ne "\rPreview running... ${i}/10 seconds"
    fi
done

# Kill preview if still running
if kill -0 "$greeter_pid" 2>/dev/null; then
    kill "$greeter_pid" 2>/dev/null || true
fi

echo # New line after progress indicator

# Show current theme
local current_theme
current_theme=$(sed -n 's|^ConfigFile=Themes/\(.*\)\.conf|\1|p' "$METADATA" 2>/dev/null || echo "unknown")
info "Preview closed (current theme: $current_theme)"
info "Preview log saved to: $log_file"

# Remove theme directory
if [[ -d "$THEMES_DIR/$THEME_NAME" ]]; then
    sudo rm -rf "$THEMES_DIR/$THEME_NAME"
    info "Theme files removed"
fi

# Reset SDDM config
if [[ -f /etc/sddm.conf ]]; then
    sudo sed -i "s|^Current=$THEME_NAME|Current=|" /etc/sddm.conf
    info "SDDM configuration reset"
fi

# Remove local clone
if [[ -d "$PATH_TO_GIT_CLONE" ]]; then
    rm -rf "$PATH_TO_GIT_CLONE"
    info "Local repository removed"
fi

info "Theme uninstalled successfully"

# Check dependencies
if ! command -v git &>/dev/null; then
    error "git is required but not installed"
    error "Install with: sudo apt install git"
    exit 1
fi

check_ubuntu
check_gum

while true; do
    clear
    if command -v gum &>/dev/null; then
        gum style --bold --padding "1 2" --border double --border-foreground 12 --align center \
            "🚀 SDDM Astronaut Theme Installer" \
            "Ubuntu Edition" \
            "" \
            "Choose an option below:"
    else
        echo -e "\e[36m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\e[0m"
        echo -e "\e[36m🚀 SDDM Astronaut Theme Installer - Ubuntu Edition\e[0m"
        echo -e "\e[36m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\e[0m"
        echo
    fi

    local choice
    choice=$(choose \
        "🚀 Complete Installation (Recommended)" \
        "📦 Install Dependencies Only" \
        "📥 Clone Theme Repository" \
        "🎨 Install Theme Files" \
        "🔧 Enable SDDM Service" \
        "✨ Select Theme Variant" \
        "👀 Preview Current Theme" \
        "🗑️  Uninstall Theme" \
        "ℹ️  System Information" \
        "❌ Exit")

    case "$choice" in
        "🚀 Complete Installation (Recommended)")
            install_deps && clone_repo && install_theme && select_theme && enable_sddm
            info "🎉 Complete installation finished!"
            info "💡 Reboot your system to see the new login screen"
            exit 0
            ;;
        "📦 Install Dependencies Only") install_deps ;;
        "📥 Clone Theme Repository") clone_repo ;;
        "🎨 Install Theme Files") install_theme ;;
        "🔧 Enable SDDM Service") enable_sddm ;;
        "✨ Select Theme Variant") select_theme ;;
        "👀 Preview Current Theme") preview_theme ;;
        "🗑️  Uninstall Theme") uninstall_theme ;;
        "ℹ️  System Information")
            echo
            info "System Information:"
            echo "  OS: $(lsb_release -d | cut -f2)"
            echo "  Kernel: $(uname -r)"
            echo "  Desktop: ${DESKTOP_SESSION:-Unknown}"
            echo "  Display Server: ${XDG_SESSION_TYPE:-Unknown}"
            echo "  SDDM Installed: $(command -v sddm &>/dev/null && echo "Yes" || echo "No")"
            echo "  Theme Installed: $([[ -d "$THEMES_DIR/$THEME_NAME" ]] && echo "Yes" || echo "No")"
            ;;
        "❌ Exit") 
            info "👋 Thanks for using SDDM Astronaut Theme!"
            exit 0 
            ;;
    esac

    echo
    if command -v gum &>/dev/null; then
        gum input --placeholder="Press Enter to continue..." >/dev/null
    else
        echo -n "Press Enter to continue..."
        read -r
    fi
done