#!/bin/bash
# Round01 Quick Switch entry QA. It launches the real Quick Switch session path
# through the app-level controller, verifies one full-screen overlay on the
# mouse screen, then verifies clean auto-close.

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
OUTPUT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
# shellcheck source=build-output-paths.sh
source "$SCRIPT_DIR/build-output-paths.sh"
APP="$BUILD_CURRENT_APP"
OPEN_WAIT="${ALIGNER_ROUND1_QUICK_SWITCH_OPEN_WAIT:-2.0}"
HIDE_AFTER="${ALIGNER_ROUND1_QUICK_SWITCH_HIDE_AFTER:-5.0}"
QUIT_AFTER="${ALIGNER_ROUND1_QUICK_SWITCH_QUIT_AFTER:-7.0}"
AFTER_HIDE_WAIT="${ALIGNER_ROUND1_QUICK_SWITCH_AFTER_HIDE_WAIT:-0.8}"
QUIT_TIMEOUT="${ALIGNER_ROUND1_QUICK_SWITCH_QUIT_TIMEOUT:-5.0}"

fail() {
  echo "Round01 Quick Switch entry QA failed: $*" >&2
  exit 1
}

python_float_check() {
  /usr/bin/python3 - "$@" <<'PY'
import sys

hide_after, quit_after, after_hide_wait = map(float, sys.argv[1:4])
if quit_after <= hide_after + after_hide_wait:
    print(
        f"QUIT_AFTER must be greater than HIDE_AFTER + AFTER_HIDE_WAIT: "
        f"{quit_after} <= {hide_after} + {after_hide_wait}",
        file=sys.stderr,
    )
    sys.exit(1)
PY
}

seconds_after_open_until_hidden() {
  /usr/bin/python3 - "$OPEN_WAIT" "$HIDE_AFTER" "$AFTER_HIDE_WAIT" <<'PY'
import sys

open_wait, hide_after, after_hide_wait = map(float, sys.argv[1:4])
print(max(0.0, hide_after + after_hide_wait - open_wait))
PY
}

aligner_pids_for_current_app() {
  ps -axo pid=,args= | while read -r pid command; do
    case "$command" in
      "$APP/Contents/MacOS/Aligner"*) echo "$pid" ;;
    esac
  done
}

stop_current_aligner() {
  for pid in $(aligner_pids_for_current_app); do
    kill "$pid" 2>/dev/null || true
  done

  for _ in {1..30}; do
    [ -z "$(aligner_pids_for_current_app)" ] && return
    sleep 0.1
  done

  fail "current Aligner app did not exit before QA"
}

wait_for_app_exit() {
  local timeout="$1"
  /usr/bin/python3 - "$APP_PID" "$timeout" <<'PY'
import os
import sys
import time

pid = int(sys.argv[1])
timeout = float(sys.argv[2])
deadline = time.monotonic() + timeout

while time.monotonic() < deadline:
    stat = os.popen(f"ps -p {pid} -o stat=").read().strip()
    if not stat or "Z" in stat:
        sys.exit(0)
    time.sleep(0.1)

print(f"Aligner app did not quit within {timeout:.1f}s", file=sys.stderr)
sys.exit(1)
PY
}

wait_for_no_quick_switch() {
  local timeout="$1"
  /usr/bin/python3 - "$SCRIPT_DIR" "$timeout" <<'PY'
import subprocess
import sys
import time

script_dir = sys.argv[1]
timeout = float(sys.argv[2])
deadline = time.monotonic() + timeout
last_output = ""

while time.monotonic() < deadline:
    result = subprocess.run(
        ["swift", f"{script_dir}/window-logic-qa.swift", "--expect-no-quick-switch"],
        text=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
    )
    last_output = result.stdout
    if result.returncode == 0:
        print(last_output, end="")
        sys.exit(0)
    time.sleep(0.2)

print(last_output, end="")
sys.exit(1)
PY
}

python_float_check "$HIDE_AFTER" "$QUIT_AFTER" "$AFTER_HIDE_WAIT"
POST_OPEN_HIDE_WAIT="$(seconds_after_open_until_hidden)"
stop_current_aligner
"$SCRIPT_DIR/package-app.sh" >&2

"$APP/Contents/MacOS/Aligner" \
  --round0-skip-permissions \
  --round01-open-quick-switch \
  --round01-quick-switch-auto-hide-after="$HIDE_AFTER" \
  --round01-quick-switch-quit-after="$QUIT_AFTER" &

APP_PID=$!

cleanup() {
  if kill -0 "$APP_PID" 2>/dev/null; then
    kill "$APP_PID" 2>/dev/null || true
    wait "$APP_PID" 2>/dev/null || true
  fi
}
trap cleanup EXIT

sleep "$OPEN_WAIT"

swift "$SCRIPT_DIR/window-logic-qa.swift" \
  --expect-quick-switch \
  --expect-single-aligner-overlay \
  --expect-overlay-on-mouse-screen \
  --expect-overlay-uses-screen-frame

sleep "$POST_OPEN_HIDE_WAIT"
wait_for_no_quick_switch "$QUIT_TIMEOUT"

wait_for_app_exit "$QUIT_TIMEOUT" || fail "app did not quit after Quick Switch hide"
wait "$APP_PID"
trap - EXIT

wait_for_no_quick_switch "$QUIT_TIMEOUT"