#!/bin/bash
# Test Archipelago ISO in QEMU
#
# Usage:
#   ./test-iso-qemu.sh [path-to-iso] [--bios] [--nographic]
#
# Options:
#   --bios       Force legacy BIOS mode (default: UEFI)
#   --nographic  No GUI window, serial console only (great for logging)
#
# Serial log is always written to /tmp/archipelago-qemu-serial.log

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
SERIAL_LOG="/tmp/archipelago-qemu-serial.log"
FORCE_BIOS=false
NOGRAPHIC=false
TIMEOUT=0
ISO=""

# Simple arg parsing. First non-flag positional is the ISO path. A bare
# numeric (e.g. `120`) is taken as a boot-test timeout in seconds so CI
# can call `test-iso-qemu.sh <iso> 120` without hanging the job. The
# pre-fix version used `case *) ISO=...`, which silently overwrote ISO
# with the timeout value and sent QEMU looking for a file literally
# named "120".
for arg in "$@"; do
    case "$arg" in
        --bios) FORCE_BIOS=true ;;
        --nographic) NOGRAPHIC=true ;;
        --timeout=*) TIMEOUT="${arg#--timeout=}" ;;
        [0-9]*) TIMEOUT="$arg" ;;
        *) ISO="$arg" ;;
    esac
done

# A positive TIMEOUT implies headless (no DISPLAY in CI anyway) and keeps
# the entire script wrapped in `timeout` to guarantee the job returns.
if [ "$TIMEOUT" -gt 0 ] 2>/dev/null; then
    NOGRAPHIC=true
fi

# Auto-detect ISO
if [ -z "$ISO" ]; then
    ISO=$(ls -t "$SCRIPT_DIR"/results/archipelago-installer-unbundled-*.iso 2>/dev/null | head -1)
fi
if [ -z "$ISO" ] || [ ! -f "$ISO" ]; then
    ISO=$(ls -t "$SCRIPT_DIR"/results/archipelago-*.iso 2>/dev/null | head -1)
fi
if [ -z "$ISO" ] || [ ! -f "$ISO" ]; then
    echo "ISO not found."
    echo ""
    echo "Usage: $0 [path-to-iso] [--bios] [--nographic]"
    echo ""
    echo "Or place an ISO in: $SCRIPT_DIR/results/"
    exit 1
fi

echo "Testing Archipelago ISO in QEMU"
echo "  ISO:    $ISO"
echo "  Size:   $(du -h "$ISO" | cut -f1)"
echo "  RAM:    4GB"
echo "  CPU:    2 cores"
echo "  Serial: $SERIAL_LOG"
echo ""

# Create test disk if it doesn't exist
DISK="/tmp/archipelago-test-disk.qcow2"
if [ ! -f "$DISK" ]; then
    echo "Creating 20GB test disk..."
    qemu-img create -f qcow2 "$DISK" 20G
fi

# Common QEMU args
QEMU_ARGS=(
    -m 4G
    -smp 2
    -boot d
    -cdrom "$ISO"
    -drive if=virtio,format=qcow2,file="$DISK"
    -net nic,model=virtio -net user,hostfwd=tcp::2222-:22,hostfwd=tcp::8100-:80
    -serial file:"$SERIAL_LOG"
)

# Display mode
if [ "$NOGRAPHIC" = true ]; then
    QEMU_ARGS+=(-nographic -append "console=ttyS0")
else
    QEMU_ARGS+=(-vga virtio -display default)
fi

echo "Starting VM..."
echo "(Serial console logging to $SERIAL_LOG)"
echo "(Press Ctrl+Alt+G to release mouse, Ctrl+C to stop VM)"
echo ""

# Detect UEFI firmware
OVMF=""
if [ "$FORCE_BIOS" = false ]; then
    if [ -f "/opt/homebrew/share/qemu/edk2-x86_64-code.fd" ]; then
        OVMF="/opt/homebrew/share/qemu/edk2-x86_64-code.fd"
    elif [ -f "/usr/share/OVMF/OVMF_CODE.fd" ]; then
        OVMF="/usr/share/OVMF/OVMF_CODE.fd"
    fi
fi

run_qemu() {
    if [ -n "$OVMF" ]; then
        echo "  Boot: UEFI ($OVMF)"
        qemu-system-x86_64 \
            -machine q35 \
            -drive if=pflash,format=raw,readonly=on,file="$OVMF" \
            "${QEMU_ARGS[@]}"
    else
        echo "  Boot: Legacy BIOS"
        qemu-system-x86_64 \
            -machine pc \
            "${QEMU_ARGS[@]}"
    fi
}

# Wrap the QEMU invocation in `timeout` when a CI caller passed one so
# the script always returns instead of hanging on a VM that never exits
# its boot loop. Exit 124 from coreutils' timeout is treated as "VM
# reached the timeout", which for a CI boot test is success as long as
# the serial log shows a kernel reaching userspace — we inspect that
# after the QEMU process ends.
if [ "$TIMEOUT" -gt 0 ] 2>/dev/null; then
    timeout --foreground --preserve-status "${TIMEOUT}s" bash -c "$(declare -f run_qemu); run_qemu"
    rc=$?
    if [ $rc -eq 124 ] || [ $rc -eq 137 ]; then
        echo "(QEMU terminated after ${TIMEOUT}s boot-test window)"
        rc=0
    fi
else
    run_qemu
    rc=$?
fi

echo ""
echo "VM stopped. Serial log: $SERIAL_LOG"
echo "Last 20 lines:"
tail -20 "$SERIAL_LOG" 2>/dev/null

# Boot-sanity check: the CI wrapper wants a non-zero exit only when the
# kernel never reached userspace. Look for a well-known marker emitted
# by live-boot/systemd early in the sequence. If the marker never
# appeared, surface the real failure; otherwise treat "timeout reached
# with a live kernel" as a pass.
if [ "$TIMEOUT" -gt 0 ] 2>/dev/null && [ -f "$SERIAL_LOG" ]; then
    if grep -qE "Welcome to Debian|Reached target|systemd\[1\]:" "$SERIAL_LOG"; then
        echo "  Boot sanity: OK (systemd reached in serial log)"
        exit 0
    fi
    echo "  Boot sanity: FAIL — no systemd markers in serial log within ${TIMEOUT}s"
    exit 1
fi
exit "${rc:-0}"