#!/bin/bash
source ~/.bashrc
# Fix Boot Configuration - Ensure VMs can boot from ISO

set -e

# 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_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

log_step() {
    echo -e "${BLUE}[STEP]${NC} $1"
}

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

PVE_USERNAME="${PVE_USERNAME:-root@pam}"
PVE_PASSWORD="${PVE_ROOT_PASS:-}"
PROXMOX_HOST="${1:-192.168.1.206}"
PROXMOX_URL="https://${PROXMOX_HOST}:8006"
PROXMOX_NODE="${2:-pve}"
ISO_FILE="${ISO_FILE:-ubuntu-24.04.3-live-server-amd64.iso}"

# Get authentication ticket
get_ticket() {
    local response=$(curl -k -s -d "username=$PVE_USERNAME&password=$PVE_PASSWORD" \
        "$PROXMOX_URL/api2/json/access/ticket")
    
    local ticket=$(echo "$response" | grep -o '"ticket":"[^"]*' | cut -d'"' -f4)
    local csrf=$(echo "$response" | grep -o '"CSRFPreventionToken":"[^"]*' | cut -d'"' -f4)
    
    if [ -z "$ticket" ] || [ -z "$csrf" ]; then
        log_error "Failed to authenticate with Proxmox"
        return 1
    fi
    
    echo "$ticket|$csrf"
}

# Fix boot configuration for a VM
fix_boot() {
    local auth=$1
    local vmid=$2
    local name=$3
    
    local ticket=$(echo "$auth" | cut -d'|' -f1)
    local csrf=$(echo "$auth" | cut -d'|' -f2)
    
    log_step "Fixing boot configuration for $name (ID: $vmid)..."
    
    # Stop VM
    log_info "Stopping VM..."
    curl -k -s -X POST \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/status/stop" > /dev/null 2>&1
    sleep 2
    
    # First, ensure we have a network device
    log_info "Ensuring network device exists..."
    curl -k -s -X POST \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        -d "net0=virtio,bridge=vmbr0" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null 2>&1 || true
    
    # Ensure ISO is attached - try multiple formats
    log_info "Attaching ISO..."
    local iso_volid="local:iso/${ISO_FILE}"
    
    # Try format 1: ide2=storage:iso/file.iso,media=cdrom
    local iso1=$(curl -k -s -X POST \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        -d "ide2=${iso_volid},media=cdrom" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" 2>&1)
    
    if echo "$iso1" | grep -q '"errors"'; then
        # Try format 2: ide2=storage:iso/file.iso
        curl -k -s -X POST \
            -H "Cookie: PVEAuthCookie=$ticket" \
            -H "CSRFPreventionToken: $csrf" \
            -d "ide2=${iso_volid}" \
            "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null 2>&1 || true
    fi
    
    # Set boot order - CD-ROM first, then disk
    log_info "Setting boot order to CD-ROM first..."
    curl -k -s -X PUT \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        -d "boot=order=ide2;scsi0" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null 2>&1 || \
    curl -k -s -X PUT \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        -d "boot=order=ide2" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null 2>&1 || true
    
    # Also try setting bootdisk
    log_info "Setting boot disk..."
    curl -k -s -X PUT \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        -d "bootdisk=ide2" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null 2>&1 || true
    
    # Start VM
    log_info "Starting VM..."
    curl -k -s -X POST \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/status/start" > /dev/null 2>&1
    
    log_info "✓ Boot configuration fixed for $name"
    echo ""
}

main() {
    echo "========================================="
    echo "Fix Boot Configuration"
    echo "========================================="
    echo ""
    
    if [ -z "$PVE_PASSWORD" ]; then
        log_error "PVE_ROOT_PASS not set in .env"
        exit 1
    fi
    
    # Authenticate
    auth=$(get_ticket)
    if [ $? -ne 0 ]; then
        exit 1
    fi
    
    # Fix all VMs
    for vmid in 100 101 102 103; do
        case $vmid in
            100) name="cloudflare-tunnel" ;;
            101) name="k3s-master" ;;
            102) name="git-server" ;;
            103) name="observability" ;;
        esac
        fix_boot "$auth" "$vmid" "$name"
    done
    
    log_info "========================================="
    log_info "Boot Configuration Fixed"
    log_info "========================================="
    echo ""
    log_info "VMs should now boot from ISO."
    log_info "If still having issues, verify via Proxmox Web UI:"
    log_info "  - Hardware tab: Check CD/DVD drive has ISO"
    log_info "  - Options tab: Boot order should be CD-ROM first"
}

main "$@"