#!/bin/bash
source ~/.bashrc
# SSH Test All VMs - Using Guest Agent IP Discovery
# Tests SSH access to all VMs using dynamically discovered IPs

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"

# Load environment variables
if [ -f "$PROJECT_ROOT/.env" ]; then
    set -a
    source <(grep -v '^#' "$PROJECT_ROOT/.env" | grep -v '^$' | sed 's/#.*$//' | grep '=')
    set +a
fi

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

VM_USER="${VM_USER:-ubuntu}"
SSH_KEY="${SSH_KEY:-$HOME/.ssh/id_ed25519_proxmox}"
PROXMOX_HOST="${PROXMOX_ML110_IP:-192.168.1.206}"

# VMID NAME (no IP here)
VMS=(
  "100 cloudflare-tunnel"
  "101 k3s-master"
  "102 git-server"
  "103 observability"
)

# Import helper (must be run on Proxmox host or via SSH)
if [ -f "$PROJECT_ROOT/scripts/lib/proxmox_vm_helpers.sh" ]; then
    source "$PROJECT_ROOT/scripts/lib/proxmox_vm_helpers.sh"
else
    log_error "Helper library not found: $PROJECT_ROOT/scripts/lib/proxmox_vm_helpers.sh"
    exit 1
fi

main() {
    log_info "Testing SSH access to all VMs using guest-agent IP discovery"
    echo ""

    for vm_spec in "${VMS[@]}"; do
        read -r vmid name <<< "$vm_spec"
        echo "=== VM $vmid: $name ==="

        # Ensure guest agent is enabled in VM config
        ensure_guest_agent_enabled "$vmid" || true

        # Get IP from guest agent
        ip="$(get_vm_ip_or_warn "$vmid" "$name" || true)"
        if [[ -z "$ip" ]]; then
            echo "  Skipping VM $vmid ($name) – no IP from guest agent."
            echo
            continue
        fi

        echo "  Using IP: $ip"
        echo "  Running 'hostname' via SSH..."

        if ssh -i "$SSH_KEY" -o StrictHostKeyChecking=accept-new -o ConnectTimeout=5 "${VM_USER}@${ip}" hostname 2>/dev/null; then
            log_info "  ✓ SSH working"
        else
            log_error "  ✗ SSH failed for ${VM_USER}@${ip}"
        fi
        echo
    done

    log_info "SSH test complete"
}

main "$@"