#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
HYPR_DIR="$(cd -- "$SCRIPT_DIR/.." && pwd)"
HOMELAB_CONFIG="${HOMELAB_CONFIG:-$HOME/.config/homelab/config.yaml}"

notify() {
    notify-send "Homelab" "$1" >/dev/null 2>&1 || true
}

if ! command -v ags >/dev/null 2>&1; then
    notify "ags ist nicht installiert."
    exit 1
fi

if ! command -v sshpass >/dev/null 2>&1; then
    if command -v zenity >/dev/null 2>&1; then
        zenity --question --title="sshpass" --text="sshpass wird benötigt. Installieren?" --width=300 && sudo pacman -S --noconfirm sshpass || true
    elif [[ -t 0 ]] && command -v gum >/dev/null 2>&1; then
        gum confirm "sshpass ist nicht installiert. Installieren?" && sudo pacman -S --noconfirm sshpass || true
    fi
    if ! command -v sshpass >/dev/null 2>&1; then
        notify "sshpass wird benötigt. Bitte installieren: sudo pacman -S sshpass"
        exit 1
    fi
fi

setup_homelab() {
    local server_address server_username server_port

    if command -v zenity >/dev/null 2>&1; then
        local data
        data="$(zenity --forms --title="Homelab Control Center" \
            --text="Server-Zugangsdaten eingeben" \
            --add-entry="Server-Adresse" \
            --add-entry="SSH-Benutzer" \
            --add-entry="SSH-Port")" || return 1
        server_address="$(printf '%s\n' "$data" | cut -d'|' -f1)"
        server_username="$(printf '%s\n' "$data" | cut -d'|' -f2)"
        server_port="$(printf '%s\n' "$data" | cut -d'|' -f3)"
        [[ -z "$server_address" ]] && { notify "Keine Adresse angegeben."; return 1; }
        server_username="${server_username:-root}"
        server_port="${server_port:-22}"

    elif [[ -t 0 ]] && command -v gum >/dev/null 2>&1; then
        gum style --foreground 212 "Homelab Control Center Setup"
        echo
        server_address="$(gum input --prompt "Server address: " --placeholder "192.168.1.100")"
        [[ -z "$server_address" ]] && { notify "Keine Adresse angegeben."; return 1; }
        server_username="$(gum input --prompt "SSH username: " --value "root")"
        server_port="$(gum input --prompt "SSH port: " --value "22")"

    elif [[ -t 0 ]] && command -v whiptail >/dev/null 2>&1; then
        server_address="$(whiptail --inputbox "Server address" 10 60 3>&1 1>&2 2>&3)" || return 1
        [[ -z "$server_address" ]] && { notify "Keine Adresse angegeben."; return 1; }
        server_username="$(whiptail --inputbox "SSH username" 10 60 "root" 3>&1 1>&2 2>&3)" || return 1
        server_port="$(whiptail --inputbox "SSH port" 10 60 "22" 3>&1 1>&2 2>&3)" || return 1

    elif [[ -t 0 ]]; then
        echo "=== Homelab Control Center Setup ==="
        echo
        read -r -p "Server address (IP or domain): " server_address
        [[ -z "$server_address" ]] && { echo "Keine Adresse eingegeben."; return 1; }
        read -r -p "SSH username [root]: " server_username
        server_username="${server_username:-root}"
        read -r -p "SSH port [22]: " server_port
        server_port="${server_port:-22}"

    else
        notify "Kein grafischer Dialog (zenity) verfügbar. Bitte installieren: sudo pacman -S zenity"
        return 1
    fi

    mkdir -p "$(dirname "$HOMELAB_CONFIG")"
    cat > "$HOMELAB_CONFIG" <<CONFIG
# Homelab Configuration
server:
  address: "${server_address}"
  username: "${server_username}"
  port: ${server_port:-22}
control_center:
  refresh_interval: 5
  theme: "dark"
features:
  docker: true
  services: true
  storage: true
  network: true
  monitoring: true
CONFIG

    notify-send "Homelab" "Konfiguration gespeichert" >/dev/null 2>&1 || true
}

if [[ -f "$HOMELAB_CONFIG" ]]; then
    export HOMELAB_CONFIG
else
    notify "Keine Homelab-Konfiguration gefunden. Setup startet..."
    if ! setup_homelab; then
        exit 1
    fi
    export HOMELAB_CONFIG
fi

cd "$HYPR_DIR"
ags quit --instance homelab-control >/dev/null 2>&1 || true
exec ags run "$HYPR_DIR/ags/homelab.tsx"