#!/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 gum >/dev/null 2>&1; then
        gum confirm "sshpass ist nicht installiert. Installieren?" && sudo pacman -S --noconfirm sshpass || true
    else
        read -r -p "sshpass ist nicht installiert. Installieren? [Y/n] " response
        response="${response:-Y}"
        if [[ "$response" =~ ^[yY](es)?$ ]]; then
            sudo pacman -S --noconfirm sshpass
        fi
    fi
    if ! command -v sshpass >/dev/null 2>&1; then
        notify "sshpass wird benötigt."
        exit 1
    fi
fi

setup_homelab() {
    local server_address server_username server_port

    if 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 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
    else
        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}"
    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"