#!/usr/bin/env bash
# EDGE bench / field bridge launcher.
#
# Single-hop (legacy pos4↔pos5):
#   POG_EDGE_ROLE=office|pos4  FIXTURE_POS=…  sudo -E bash run_bridge_edge_bench.sh
#
# Hub / spoke wlan star:
#   EDGE_TOPOLOGY=hub   POG_EDGE_ROLE=hub   EDGE_BANK=0 EDGE_BANK_WLAN=1 …
#   EDGE_TOPOLOGY=spoke POG_EDGE_ROLE=spoke EDGE_SPOKE_ID=2 …
#
# Software tip still via update.sh / SCP — this launcher only brings tunnels + bridge.

set -euo pipefail

EDGE_ROOT="$(cd "$(dirname "$0")" && pwd)"
TIMING_ROOT="$(cd "${EDGE_ROOT}/.." && pwd)"

# Allow run_bridge_jun11_good.sh to run under EDGE without re-dispatching.
export ETHERGOLD_ENTRY=1

# shellcheck disable=SC1091
source "${EDGE_ROOT}/pog_edge_env.sh"

if [[ -z "${POG_EDGE_ROLE:-}" ]]; then
  echo "FATAL: copy bench.env.example to bench.env and set POG_EDGE_ROLE=office|pos4|hub|spoke" >&2
  exit 1
fi

FIXTURE_POS="${FIXTURE_POS:-}"
if [[ -z "${FIXTURE_POS}" ]]; then
  case "${POG_EDGE_ROLE}" in
    office|hub) FIXTURE_POS=5 ;;
    pos4) FIXTURE_POS=4 ;;
    spoke) FIXTURE_POS="${EDGE_SPOKE_ID:-4}" ;;
  esac
fi

# Seat label only — protocol-agnostic. Allow 1..99 (hub-spoke scale).
if ! [[ "${FIXTURE_POS}" =~ ^[1-9][0-9]?$ ]]; then
  echo "FATAL: FIXTURE_POS=1..99 required (got ${FIXTURE_POS})" >&2
  exit 1
fi

case "${POG_EDGE_ROLE}:${FIXTURE_POS}" in
  office:5|pos4:4|hub:*|spoke:*) ;;
  *)
    echo "WARN: unexpected role/pos pairing (got ${POG_EDGE_ROLE} pos${FIXTURE_POS})" >&2
    ;;
esac

echo "=== POG-EDGE bridge (role=${POG_EDGE_ROLE} topo=${EDGE_TOPOLOGY:-single-hop} pos=${FIXTURE_POS} bank=${EDGE_BANK}) ==="
echo "  eth_lan=${POG_EDGE_LAN_IF} (WireGuard only — not in bridge)"

if [[ "${EUID}" -ne 0 ]]; then
  echo "FATAL: run with sudo -E so tunnel_up can configure wg/gretap" >&2
  exit 1
fi

bash "${EDGE_ROOT}/tunnel_up.sh"
# tunnel_up may select gretap1+ / eg-sNN — re-read active name(s).
if [[ -f "${EDGE_ROOT}/.active_gretap" ]]; then
  GRETAP_IF="$(tr -d '\r\n' < "${EDGE_ROOT}/.active_gretap")"
  export GRETAP_IF
fi

bash "${EDGE_ROOT}/prep_eth_game_for_fixture.sh"

# Build ETH_PORTS from game if + gretap(s). Hub: all eg-sNN from .active_spoke_gretaps.
build_eth_ports() {
  local ports=()
  if [[ -n "${POG_EDGE_GAME_IF}" ]]; then
    ports+=("${POG_EDGE_GAME_IF}")
  fi
  if [[ "${POG_EDGE_ROLE}" == "hub" && -f "${EDGE_ROOT}/.active_spoke_gretaps" ]]; then
    while IFS= read -r g; do
      g="$(echo "${g}" | tr -d '\r\n')"
      [[ -n "${g}" ]] && ports+=("${g}")
    done < "${EDGE_ROOT}/.active_spoke_gretaps"
  fi
  if [[ ${#ports[@]} -eq 0 ]] || [[ "${POG_EDGE_ROLE}" != "hub" ]]; then
    # Single-hop / spoke: one gretap leg.
    if [[ -n "${GRETAP_IF}" ]]; then
      # Avoid duplicate if already listed.
      local found=0
      for p in "${ports[@]+"${ports[@]}"}"; do
        [[ "${p}" == "${GRETAP_IF}" ]] && found=1
      done
      if [[ "${found}" -eq 0 ]]; then
        ports+=("${GRETAP_IF}")
      fi
    fi
  fi
  local IFS=,
  echo "${ports[*]}"
}

ETH_PORTS="$(build_eth_ports)"
if [[ -z "${ETH_PORTS}" ]]; then
  echo "FATAL: no ETH_PORTS (need game if and/or gretap)" >&2
  exit 1
fi
export ETH_PORTS
export LINK_WATCH_INTERFACE="${POG_EDGE_GAME_IF:-${GRETAP_IF}}"
export LINK_DIAG_DEVICE="${LINK_WATCH_INTERFACE}"
export FIXTURE_LINK_ARP=0
export POG_EDGE_BENCH=1
export EDGE_TID_UNICAST="${EDGE_TID_UNICAST:-1}"
export EDGE_BANK_WLAN="${EDGE_BANK_WLAN:-0}"
export EDGE_TOPOLOGY="${EDGE_TOPOLOGY:-}"
export EDGE_BANK="${EDGE_BANK:-0}"

echo "  eth_ports=${ETH_PORTS}"

REAL_USER="${SUDO_USER:-topbox}"

cd "${TIMING_ROOT}"
sudo -u "${REAL_USER}" env \
  ETHERGOLD_ENTRY=1 \
  FIXTURE_POS="${FIXTURE_POS}" \
  CHANNEL="${CHANNEL:-both}" \
  ETH_PORTS="${ETH_PORTS}" \
  FIXTURE_LINK_ARP="${FIXTURE_LINK_ARP}" \
  LINK_WATCH_INTERFACE="${LINK_WATCH_INTERFACE}" \
  LINK_DIAG_DEVICE="${LINK_DIAG_DEVICE}" \
  POG422_UARTD="${POG422_UARTD:-0}" \
  POG_EDGE_BENCH=1 \
  EDGE_MODE="${EDGE_MODE:-1}" \
  EDGE_TOPOLOGY="${EDGE_TOPOLOGY:-}" \
  EDGE_BANK="${EDGE_BANK}" \
  EDGE_SPOKE_ID="${EDGE_SPOKE_ID:-}" \
  EDGE_BANK_WLAN="${EDGE_BANK_WLAN}" \
  EDGE_TID_UNICAST="${EDGE_TID_UNICAST}" \
  EDGE_UPLINK="${EDGE_UPLINK:-}" \
  EDGE_UPLINK_IP="${EDGE_UPLINK_IP:-}" \
  EDGE_FLEET_HMAC_KEY_FILE="${EDGE_FLEET_HMAC_KEY_FILE:-}" \
  EDGE_HUB_WG_PUB="${EDGE_HUB_WG_PUB:-}" \
  VERBOSE="${VERBOSE:-1}" \
  CLEAR_CAPTURE="${CLEAR_CAPTURE:-1}" \
  FIXTURE_LAB_WITHHOLD_STATUS_0D="${FIXTURE_LAB_WITHHOLD_STATUS_0D:-${LAB_WITHHOLD_STATUS_0D:-0}}" \
  FIXTURE_LAB_WITHHOLD_STATUS_0D_COUNT="${FIXTURE_LAB_WITHHOLD_STATUS_0D_COUNT:-${LAB_WITHHOLD_STATUS_0D_COUNT:-1}}" \
  LAB_WITHHOLD_STATUS_0D="${LAB_WITHHOLD_STATUS_0D:-${FIXTURE_LAB_WITHHOLD_STATUS_0D:-0}}" \
  LAB_WITHHOLD_STATUS_0D_COUNT="${LAB_WITHHOLD_STATUS_0D_COUNT:-${FIXTURE_LAB_WITHHOLD_STATUS_0D_COUNT:-1}}" \
  bash "${TIMING_ROOT}/run_bridge_jun11_good.sh"