#!/usr/bin/env bash
# Headless Google-TV emulator for test runs.
#
# The AVD keeps a "default_boot" snapshot, so a start restores a booted
# system in seconds instead of cold-booting for minutes. "stop" refreshes
# that snapshot before killing, so the next start stays fast.
#
#   tests/helpers/emulator.sh start        # boot (or restore) and wait
#   tests/helpers/emulator.sh stop         # snapshot, then kill
#   tests/helpers/emulator.sh status
#   tests/helpers/emulator.sh install <apk>
#   tests/helpers/emulator.sh shot <name>  # screenshot into tests/runs/screenshots
set -euo pipefail

AVD="${CASTARR_AVD:-castarr-googletv}"
SERIAL="${CASTARR_SERIAL:-emulator-5554}"
SDK="${ANDROID_SDK_ROOT:-$HOME/Android/Sdk}"
ADB="$SDK/platform-tools/adb"
EMULATOR="$SDK/emulator/emulator"

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
RUNS="$ROOT/tests/runs"
LOG="$RUNS/emulator.log"

booted() { [ "$("$ADB" -s "$SERIAL" shell getprop sys.boot_completed 2>/dev/null | tr -d '\r')" = "1" ]; }

start() {
  if booted; then echo "already up: $SERIAL"; return 0; fi
  mkdir -p "$RUNS"
  # No -no-snapshot here: that is what makes a start slow.
  nohup "$EMULATOR" -avd "$AVD" \
    -no-window -no-audio -no-boot-anim \
    -gpu swiftshader_indirect \
    >"$LOG" 2>&1 &
  local start_ts=$SECONDS
  "$ADB" -s "$SERIAL" wait-for-device
  for _ in $(seq 1 180); do
    booted && { echo "up after $((SECONDS - start_ts))s"; return 0; }
    sleep 1
  done
  echo "timeout waiting for boot; see $LOG" >&2
  return 1
}

stop() {
  if ! pgrep -f "qemu-system-.*-avd $AVD" >/dev/null 2>&1 && ! booted; then
    echo "not running"; return 0
  fi
  # Refresh the snapshot so the next start is a restore, not a cold boot.
  "$ADB" -s "$SERIAL" emu avd snapshot save default_boot
  "$ADB" -s "$SERIAL" emu kill
  echo "stopped, snapshot saved"
}

case "${1:-status}" in
  start)  start ;;
  stop)   stop ;;
  status) booted && echo "up: $SERIAL" || echo "down" ;;
  install)
    [ $# -ge 2 ] || { echo "usage: $0 install <apk>" >&2; exit 2; }
    "$ADB" -s "$SERIAL" install -r "$2"
    ;;
  shot)
    name="${2:-shot}"
    mkdir -p "$RUNS/screenshots"
    "$ADB" -s "$SERIAL" exec-out screencap -p > "$RUNS/screenshots/$name.png"
    echo "$RUNS/screenshots/$name.png"
    ;;
  *) echo "usage: $0 {start|stop|status|install <apk>|shot <name>}" >&2; exit 2 ;;
esac