#!/usr/bin/env bash
#
# start-emulator.sh — boot the JTube TV test emulator on the HOST (not inside the
# agent container), with GPU acceleration and a window. Use this when the app should
# be tested over scrcpy (ATD/system quirks make scrcpy fail against headless
# in-container emulators) or when you simply want to watch it.
#
# Boots on console :5600 (adbd :5601); the agent container bridges
# rutherford:5556 -> 127.0.0.1:5601 with socat, so after this script the emulator is
# reachable from anywhere as:
#
#     adb connect rutherford:5556 && scrcpy -s rutherford:5556
#
# An RSA authorization popup appears on the emulator window on first connect —
# click "Always allow"/OK.
set -euo pipefail

SDK="${ANDROID_SDK_ROOT:-$HOME/Android/Sdk}"
EMU="$SDK/emulator/emulator"
ADB="$SDK/platform-tools/adb"
SDKM="$SDK/cmdline-tools/latest/bin/sdkmanager"
AVDM="$SDK/cmdline-tools/latest/bin/avdmanager"
AVD="${1:-jtubesc}"
PORT=5600
IMAGE="system-images;android-36;android-tv;x86_64"

[ -x "$EMU" ] || { echo "emulator not found at $EMU — install Android SDK cmdline-tools first"; exit 1; }

# --- bootstrap: java, system image, AVD -------------------------------------
if ! command -v java >/dev/null; then
  echo "no java on PATH — required by sdkmanager/AVD tools"
  echo "  install it (Arch):  sudo pacman -S --noconfirm jdk21-openjdk"
  echo "  then re-run this script"
  exit 1
fi

if ! "$SDKM" --list_installed 2>/dev/null | grep -q "$IMAGE"; then
  echo "installing $IMAGE (~1.5 GB, once)…"
  yes | "$SDKM" "$IMAGE" > /dev/null
  echo "installed"
fi

if [ ! -f "$HOME/.android/avd/$AVD.ini" ]; then
  echo "creating AVD '$AVD'…"
  echo no | "$AVDM" create avd -n "$AVD" -k "$IMAGE" -d tv_1080p > /dev/null
fi

# --- free stale emulator instances holding our port --------------------------
for pid in $(ps aux | awk '/[q]emu-system-x86_64/ {print $2}' 2>/dev/null); do
  if tr '\0' ' ' < "/proc/$pid/cmdline" 2>/dev/null | grep -q "\-port $PORT"; then
    echo "Killing stale emulator pid $pid"
    kill -9 "$pid" 2>/dev/null || true
  fi
done
sleep 1

ACCEL=()
WINDOWED=1
if [ -n "${WAYLAND_DISPLAY:-}${DISPLAY:-}" ]; then
  echo "desktop session detected — emulator window + GPU host"
else
  WINDOWED=0
  ACCEL+=(-no-window)
  echo "headless (no display) — running without a window, swiftshader rendering"
fi
if [ -e /dev/kvm ]; then
  [ $WINDOWED -eq 1 ] && ACCEL+=(-accel kvm)
  echo "/dev/kvm found — hardware acceleration on"
fi

echo "Booting AVD '$AVD' …"
setsid nohup "$EMU" -avd "$AVD" -port "$PORT" -gpu host "${ACCEL[@]:-}" \
  -no-snapshot > /tmp/jtube-emu.log 2>&1 < /dev/null &
echo "emulator PID $! — log: /tmp/jtube-emu.log"

for i in $(seq 1 90); do
  ok=$("$ADB" -s "emulator-$PORT" shell getprop sys.boot_completed 2>/dev/null | tr -d '\r')
  [ "$ok" = "1" ] && { echo "BOOTED"; break; }
  sleep 5
done

# --- authorize remote adb clients headlessly ---------------------------------
# The host console transport is always trusted; use it to whitelist client keys
# so `adb connect rutherford:5556` from other machines works without a popup.
# Drop extra public keys into /tmp/jtube-adbkeys/ (one per file), e.g. from
# Chester:  cat ~/.android/adbkey.pub > /tmp/jtube-adbkeys/chester.pub
"$ADB" -s "emulator-$PORT" root > /dev/null 2>&1 || true
for key_file in "$HOME/.android/adbkey.pub" /tmp/jtube-adbkeys/*.pub; do
  [ -f "$key_file" ] || continue
  "$ADB" -s "emulator-$PORT" push "$key_file" "/data/misc/adb/adb_keys" > /dev/null 2>&1 || \
    "$ADB" -s "emulator-$PORT" shell "cat > /data/misc/adb/adb_keys" < "$key_file" || { 
      echo "could not whitelist $key_file"; 
    }
done
"$ADB" -s "emulator-$PORT" shell "setprop service.adb.tcp.port 5601; stop adbd; start adbd" > /dev/null 2>&1 || true

# --- the bridge: advertise adbd on the LAN as :5556 ---------------------------
if ! ss -tln 2>/dev/null | grep -q ":5556"; then
  if command -v socat > /dev/null; then
    setsid nohup socat TCP-LISTEN:5556,bind=0.0.0.0,fork,reuseaddr TCP:127.0.0.1:$((PORT+1)) > /tmp/jtube-socat.log 2>&1 < /dev/null &
    echo "socat bridge listening on :5556 (pid $!)"
  else
    echo "install socat for the remote bridge:  sudo pacman -S --noconfirm socat"
  fi
fi

# Keep the JTube TV app current + running so scrcpy lands on it.
APK="$HOME/Projects/jtube-tv/app/build/outputs/apk/debug/app-debug.apk"
if [ -f "$APK" ]; then
  echo "Installing JTube TV…"
  "$ADB" -s "emulator-$PORT" install -r "$APK" >/dev/null && echo "installed"
  "$ADB" -s "emulator-$PORT" shell am start -n net.jeena.jtubetv/.MainActivity && echo "app started"
fi

cat <<EOF

Ready. From any machine that can reach rutherford:

    adb connect rutherford:5556
    scrcpy -s rutherford:5556

If scrcpy says "unauthorized", click the RSA dialog on the emulator window.
EOF