#!/usr/bin/env bash
# Launch or validate the Phase 6 KDE runtime surface in QEMU.

set -euo pipefail

find_uefi_firmware() {
    local candidates=(
        "/usr/share/ovmf/x64/OVMF.4m.fd"
        "/usr/share/OVMF/x64/OVMF.4m.fd"
        "/usr/share/ovmf/x64/OVMF_CODE.4m.fd"
        "/usr/share/OVMF/x64/OVMF_CODE.4m.fd"
        "/usr/share/ovmf/OVMF.fd"
        "/usr/share/OVMF/OVMF_CODE.fd"
        "/usr/share/qemu/edk2-x86_64-code.fd"
    )
    local path
    for path in "${candidates[@]}"; do
        if [[ -f "$path" ]]; then
            printf '%s\n' "$path"
            return 0
        fi
    done
    return 1
}

usage() {
    cat <<'USAGE'
Usage: test-phase6-kde-qemu.sh [--check] [extra qemu args...]

Boot or validate the Red Bear OS Phase 6 KDE session surface on redbear-kde.
USAGE
}

report_solid_recipe_blockers() {
    local recipe="local/recipes/kde/kf6-solid/recipe.toml"
    local blockers=0

    if [[ ! -f "$recipe" ]]; then
        echo "PHASE6_SOLID_RECIPE=missing:$recipe"
        return 0
    fi

    echo "PHASE6_SOLID_RECIPE=$recipe"

    if grep -Fq -- '-DUSE_DBUS=OFF' "$recipe"; then
        echo "PHASE6_SOLID_RECIPE_BLOCKER=USE_DBUS_off"
        blockers=1
    fi

    if grep -Fq -- '-DBUILD_DEVICE_BACKEND_upower=OFF' "$recipe"; then
        echo "PHASE6_SOLID_RECIPE_BLOCKER=upower_backend_off"
        blockers=1
    fi

    if grep -Fq -- '-DBUILD_DEVICE_BACKEND_udisks2=OFF' "$recipe"; then
        echo "PHASE6_SOLID_RECIPE_BLOCKER=udisks2_backend_off"
        blockers=1
    fi

    if [[ "$blockers" -eq 0 ]]; then
        echo "PHASE6_SOLID_RECIPE=ready_for_runtime_probe"
    fi
}

check_mode=0
filtered_args=()
for arg in "$@"; do
    case "$arg" in
        --help|-h|help)
            usage
            exit 0
            ;;
        --check)
            check_mode=1
            ;;
        *)
            filtered_args+=("$arg")
            ;;
    esac
done

firmware="$(find_uefi_firmware)" || {
    echo "ERROR: no usable x86_64 UEFI firmware found" >&2
    exit 1
}

arch="${ARCH:-$(uname -m)}"
image="build/$arch/redbear-kde/harddrive.img"
extra="build/$arch/redbear-kde/extra.img"
extra_qemu_args="${filtered_args[*]:-}"

if [[ ! -f "$image" ]]; then
    echo "ERROR: missing image $image" >&2
    echo "Build it first with: ./local/scripts/build-redbear.sh redbear-kde" >&2
    exit 1
fi

if [[ ! -f "$extra" ]]; then
    truncate -s 1g "$extra"
fi

if [[ "$check_mode" -eq 1 ]]; then
    report_solid_recipe_blockers
    expect <<EOF
log_user 1
set timeout 240
spawn qemu-system-x86_64 -name {Red Bear OS x86_64} -device qemu-xhci -smp 4 -m 2048 -bios $firmware -chardev stdio,id=debug,signal=off,mux=on -serial chardev:debug -mon chardev=debug -machine q35 -device ich9-intel-hda -device hda-output -device virtio-net,netdev=net0 -netdev user,id=net0 -object filter-dump,id=f1,netdev=net0,file=build/$arch/redbear-kde/network.pcap -vga std -drive file=$image,format=raw,if=none,id=drv0 -device nvme,drive=drv0,serial=NVME_SERIAL -drive file=$extra,format=raw,if=none,id=drv1 -device nvme,drive=drv1,serial=NVME_EXTRA -enable-kvm -cpu host $extra_qemu_args
expect "login:"
send "root\r"
expect "assword:"
send "password\r"
expect "Type 'help' for available commands."
send "redbear-phase6-kde-check\r"
expect "Red Bear OS Phase 6 KDE Runtime Check"
expect "orbital-kde"
expect "kwin_wayland"
expect {
    "PHASE6_UPOWER_ENUMERATE=ok" {}
    "PHASE6_UPOWER_ENUMERATE=skipped_missing_dbus_send" {}
}
expect {
    "PHASE6_UDISKS2_OBJECTS=ok" {}
    "PHASE6_UDISKS2_OBJECTS=skipped_missing_dbus_send" {}
}
expect {
    "PHASE6_SOLID_RUNTIME=checked" {}
    "PHASE6_SOLID_RUNTIME=blocked_missing_tool" {}
    "PHASE6_SOLID_RUNTIME=blocked_missing_storage_or_power_surface" {}
}
expect "virtio_net_present"
send "shutdown\r"
expect eof
EOF
    exit 0
fi

exec qemu-system-x86_64 \
  -name "Red Bear OS x86_64" \
  -device qemu-xhci \
  -smp 4 \
  -m 2048 \
  -bios "$firmware" \
  -chardev stdio,id=debug,signal=off,mux=on \
  -serial chardev:debug \
  -mon chardev=debug \
  -machine q35 \
  -device ich9-intel-hda -device hda-output \
  -device virtio-net,netdev=net0 \
  -netdev user,id=net0 \
  -object filter-dump,id=f1,netdev=net0,file="build/$arch/redbear-kde/network.pcap" \
  -vga std \
  -drive file="$image",format=raw,if=none,id=drv0 \
  -device nvme,drive=drv0,serial=NVME_SERIAL \
  -drive file="$extra",format=raw,if=none,id=drv1 \
  -device nvme,drive=drv1,serial=NVME_EXTRA \
  -enable-kvm -cpu host \
  $extra_qemu_args