#!/bin/bash
source ~/.bashrc
# Fix Template from Cloud Image
# Recreates template VM 9000 from Ubuntu cloud image

set -e

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

log_step() {
    echo -e "\n${BLUE}=== $1 ===${NC}"
}

PVE_USERNAME="${PVE_USERNAME:-root@pam}"
PVE_PASSWORD="${PVE_ROOT_PASS:-}"
PROXMOX_URL="${PROXMOX_ML110_URL:-https://192.168.1.206:8006}"
PROXMOX_NODE="${PROXMOX_NODE:-pve}"
CLOUD_IMAGE="local:iso/ubuntu-24.04-server-cloudimg-amd64.img"

get_api_token() {
    local response=$(curl -s -k --connect-timeout 10 --max-time 15 \
        -d "username=$PVE_USERNAME&password=$PVE_PASSWORD" \
        "$PROXMOX_URL/api2/json/access/ticket" 2>&1)
    
    if echo "$response" | grep -q '"data"'; then
        local ticket=$(echo "$response" | grep -o '"ticket":"[^"]*' | cut -d'"' -f4)
        local csrf_token=$(echo "$response" | grep -o '"CSRFPreventionToken":"[^"]*' | cut -d'"' -f4)
        echo "$ticket|$csrf_token"
    else
        echo ""
    fi
}

recreate_template_from_cloud_image() {
    log_step "Recreating Template from Cloud Image"
    
    log_warn "This will DELETE template VM 9000 and recreate it from cloud image"
    log_warn "All VMs cloned from this template will need to be recreated"
    echo ""
    read -p "Continue? (yes/no): " confirm
    
    if [ "$confirm" != "yes" ]; then
        log_info "Cancelled"
        return 1
    fi
    
    local tokens=$(get_api_token)
    local ticket=$(echo "$tokens" | cut -d'|' -f1)
    local csrf_token=$(echo "$tokens" | cut -d'|' -f2)
    
    # Stop and delete template
    log_info "Stopping template VM 9000..."
    curl -s -k -X POST -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf_token" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/9000/status/stop" > /dev/null 2>&1
    sleep 5
    
    log_info "Deleting template VM 9000..."
    curl -s -k -X DELETE -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf_token" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu/9000" > /dev/null 2>&1
    sleep 3
    
    # Create new VM from cloud image
    log_info "Creating new VM 9000 from cloud image..."
    
    # Step 1: Create VM shell
    curl -s -k -X POST -H "Cookie: PVEAuthCookie=$ticket" \
        -H "CSRFPreventionToken: $csrf_token" \
        -d "vmid=9000" \
        -d "name=ubuntu-24.04-cloudinit" \
        -d "memory=2048" \
        -d "cores=2" \
        -d "net0=virtio,bridge=vmbr0" \
        "$PROXMOX_URL/api2/json/nodes/$PROXMOX_NODE/qemu" > /dev/null 2>&1
    
    sleep 2
    
    # Step 2: Import cloud image disk
    log_info "Importing cloud image disk..."
    log_warn "This requires SSH access to Proxmox host"
    log_info "To complete via SSH:"
    echo "  ssh root@192.168.1.206"
    echo "  qm importdisk 9000 /var/lib/vz/template/iso/ubuntu-24.04-server-cloudimg-amd64.img local-lvm"
    echo "  qm set 9000 --scsihw virtio-scsi-pci --scsi0 local-lvm:vm-9000-disk-0"
    echo "  qm set 9000 --boot order=scsi0"
    echo "  qm set 9000 --bios ovmf --efidisk0 local-lvm:1"
    echo "  qm set 9000 --agent 1"
    echo "  qm set 9000 --template 1"
    
    log_info "Or use Proxmox Web UI to import disk"
}

main() {
    log_info "Fix Template from Cloud Image"
    recreate_template_from_cloud_image
}

main "$@"