loc_az_hci/scripts/vm-management/configure/fix-guest-agent.sh

#!/bin/bash
source ~/.bashrc
# Fix Guest Agent Issues - Troubleshooting Script
# Use this if guest agent is not working properly

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

log_warn() {
    echo -e "${YELLOW}[WARN]${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_URL="https://192.168.1.206:8006"
PROXMOX_NODE="pve"

declare -A VMS=(
    [100]="cloudflare-tunnel:192.168.1.60"
    [101]="k3s-master:192.168.1.188"
    [102]="git-server:192.168.1.121"
    [103]="observability:192.168.1.82"
)

SSH_USER="${SSH_USER:-ubuntu}"
SSH_KEY="${SSH_KEY:-~/.ssh/id_rsa}"

# 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 guest agent on a VM
fix_guest_agent() {
    local vmid=$1
    local name=$2
    local ip=$3
    
    log_step "Fixing guest agent on VM $vmid: $name"
    
    # Check SSH
    if ! ssh -o ConnectTimeout=5 -o StrictHostKeyChecking=no -i "$SSH_KEY" "${SSH_USER}@${ip}" "echo 'SSH OK'" > /dev/null 2>&1; then
        log_error "SSH not available, skipping..."
        return 1
    fi
    
    log_info "Reinstalling and restarting guest agent..."
    
    ssh -o StrictHostKeyChecking=no -i "$SSH_KEY" "${SSH_USER}@${ip}" <<EOF
sudo systemctl stop qemu-guest-agent || true
sudo apt-get update
sudo apt-get install --reinstall -y qemu-guest-agent
sudo systemctl daemon-reload
sudo systemctl enable qemu-guest-agent
sudo systemctl restart qemu-guest-agent
sleep 2
sudo systemctl status qemu-guest-agent --no-pager | head -10
EOF
    
    if [ $? -eq 0 ]; then
        log_info "✓ Guest agent fixed on VM $vmid"
        return 0
    else
        log_error "✗ Failed to fix guest agent"
        return 1
    fi
}

# Verify Proxmox configuration
verify_proxmox_config() {
    local auth=$1
    local vmid=$2
    
    local ticket=$(echo "$auth" | cut -d'|' -f1)
    local csrf=$(echo "$auth" | cut -d'|' -f2)
    
    log_step "Checking Proxmox configuration for VM $vmid"
    
    local config=$(curl -k -s \
        -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config")
    
    local agent=$(echo "$config" | grep -o '"agent":"[^"]*' | cut -d'"' -f4)
    
    if [ "$agent" = "1" ]; then
        log_info "✓ Agent enabled in Proxmox (agent=1)"
    else
        log_warn "⚠ Agent not enabled in Proxmox"
        log_info "Enabling agent..."
        
        curl -k -s -X PUT \
            -H "Cookie: PVEAuthCookie=$ticket" \
            -H "CSRFPreventionToken: $csrf" \
            -d "agent=1" \
            "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/$vmid/config" > /dev/null
        
        log_info "✓ Agent enabled"
    fi
}

main() {
    echo "========================================="
    echo "Fix Guest Agent Issues"
    echo "========================================="
    echo ""
    
    if [ -z "$PVE_PASSWORD" ]; then
        log_error "PVE_ROOT_PASS not set in .env"
        exit 1
    fi
    
    if [ ! -f "$SSH_KEY" ]; then
        log_error "SSH key not found: $SSH_KEY"
        exit 1
    fi
    
    # Authenticate
    auth=$(get_ticket)
    if [ $? -ne 0 ]; then
        exit 1
    fi
    
    # Fix each VM
    for vmid in 100 101 102 103; do
        IFS=':' read -r name ip <<< "${VMS[$vmid]}"
        
        echo "----------------------------------------"
        log_step "Fixing VM $vmid: $name"
        echo ""
        
        # Fix guest agent on VM
        fix_guest_agent "$vmid" "$name" "$ip"
        
        # Verify Proxmox config
        verify_proxmox_config "$auth" "$vmid"
        
        echo ""
    done
    
    log_info "========================================="
    log_info "Guest Agent Fix Complete"
    log_info "========================================="
    echo ""
    log_info "Wait a few minutes, then verify in Proxmox:"
    echo "  VM → Monitor → QEMU Guest Agent"
}

main "$@"