#!/usr/bin/env bash
# Boot the normal (non-headless) kernel with no window, "type" some keys into
# it through the QEMU monitor's `sendkey` command, then dump the VGA screen to
# a PNG. Doubles as an automated keyboard + user-mode test: the default types
# a line (with a typo fixed by Backspace) and presses Enter; the `echo` user
# program reads it with read(0) and prints it back, all visible on serial.
#
# Usage: ./scripts/screenshot.sh [out.png] [text-to-type] [expected-result]
# In text-to-type, "<" means Backspace and "|" means Enter.
# Exits 1 if the expected text doesn't show up on serial.
set -euo pipefail
cd "$(dirname "$0")/.."

cargo build
# Force bootimage to repack the disk image (see scripts/runner.sh for why).
touch target/x86_64-rust_kernel/debug/rust_kernel
cargo bootimage
IMG=target/x86_64-rust_kernel/debug/bootimage-rust_kernel.bin
OUT="${1:-$PWD/boot.png}"
TEXT="${2:-Hello, kernex<l!|}"
EXPECT="${3:-[echo] you typed: Hello, kernel!}"
SOCK=$(mktemp -u /tmp/qemu-mon.XXXXXX)
SERIAL_LOG=$(mktemp /tmp/serial.XXXXXX)

qemu-system-x86_64 -drive format=raw,file="$IMG" \
    -drive file=target/disk.img,format=raw,if=virtio,snapshot=on \
    -serial file:"$SERIAL_LOG" -display none -monitor unix:"$SOCK",server,nowait &
QPID=$!
sleep 6   # boot, and let the demo threads finish their serial output

# Talk to the QEMU monitor over its UNIX socket (python avoids needing socat).
python3 - "$SOCK" "$OUT" "$TEXT" <<'PY'
import socket, sys, time
sock, out, text = sys.argv[1:]
# QEMU key names for the characters we might type.
names = {'<': 'backspace', ' ': 'spc', ',': 'comma', '.': 'dot', '!': 'shift-1', '?': 'shift-slash',
         '-': 'minus', '|': 'ret'}
def key(c):
    if c in names: return names[c]
    if c.isupper(): return 'shift-' + c.lower()
    return c
s = socket.socket(socket.AF_UNIX); s.connect(sock)
for c in text:
    s.sendall(f"sendkey {key(c)}\n".encode()); time.sleep(0.1)
time.sleep(0.5)
s.sendall(f"screendump {out} -f png\n".encode()); time.sleep(1)
s.sendall(b"quit\n"); time.sleep(0.5)
PY
wait $QPID || true
echo "---- serial output ----"
cat -v "$SERIAL_LOG"; echo   # cat -v shows Backspace as ^H
echo "-----------------------"
# Replay the serial output like a terminal would ("\b \b" = erase one
# character) and check the expected text is there.
if python3 - "$SERIAL_LOG" "$EXPECT" <<'PY'
import sys
data = open(sys.argv[1], 'rb').read().decode(errors='replace').replace('\x08 \x08', '\x08')
out = []
for ch in data:
    if ch == '\x08':
        if out: out.pop()
    else:
        out.append(ch)
sys.exit(0 if sys.argv[2] in ''.join(out) else 1)
PY
then
    echo "[screenshot] keyboard echo OK: typed '$TEXT' -> '$EXPECT'"
else
    echo "[screenshot] keyboard echo NOT found (expected '$EXPECT')"
    rm -f "$SERIAL_LOG"
    exit 1
fi
rm -f "$SERIAL_LOG"
echo "[screenshot] saved $OUT"