#!/usr/bin/env bash
set -euo pipefail

ROOT="$(cd "$(dirname "$0")" && pwd)"
FIXTURES="$ROOT/e2e/fixtures"

LOADER_PORT=8080
ANNOTATIONS_PORT=8081
DETECTIONS_PORT=8000
PIDS=()

cleanup() {
    for pid in "${PIDS[@]+"${PIDS[@]}"}"; do
        kill "$pid" 2>/dev/null || true
    done
    wait 2>/dev/null
}
trap cleanup EXIT

PY="$(command -v python3 2>/dev/null || command -v python 2>/dev/null || echo python)"

echo "Installing dependencies ..."
"$PY" -m pip install -q -r "$ROOT/requirements.txt" -r "$ROOT/e2e/requirements.txt" 2>/dev/null || echo "  (some deps skipped — verify manually if tests fail with import errors)"

echo "Building Cython extensions ..."
"$PY" setup.py build_ext --inplace

for port in $LOADER_PORT $ANNOTATIONS_PORT $DETECTIONS_PORT; do
    if lsof -ti :"$port" >/dev/null 2>&1; then
        echo "Killing stale process on port $port ..."
        lsof -ti :"$port" | xargs kill -9 2>/dev/null || true
        sleep 1
    fi
done

echo "Starting mock-loader on :$LOADER_PORT ..."
cd "$ROOT"
MODELS_ROOT="$FIXTURES" PYTHONPATH="$ROOT" \
    "$PY" -m gunicorn --bind "0.0.0.0:$LOADER_PORT" --workers 1 --timeout 120 \
    'e2e.mocks.loader.app:app' >/dev/null 2>&1 &
PIDS+=($!)

echo "Starting mock-annotations on :$ANNOTATIONS_PORT ..."
MEDIA_DIR="$FIXTURES" PYTHONPATH="$ROOT" \
    "$PY" -m gunicorn --bind "0.0.0.0:$ANNOTATIONS_PORT" --workers 1 --timeout 120 \
    'e2e.mocks.annotations.app:app' >/dev/null 2>&1 &
PIDS+=($!)

echo "Starting detections service on :$DETECTIONS_PORT ..."
LOADER_URL="http://localhost:$LOADER_PORT" \
ANNOTATIONS_URL="http://localhost:$ANNOTATIONS_PORT" \
PYTHONPATH="$ROOT/src" \
    "$PY" -m uvicorn main:app --host 0.0.0.0 --port "$DETECTIONS_PORT" \
    --log-level warning >/dev/null 2>&1 &
PIDS+=($!)

echo "Waiting for services ..."
for url in \
    "http://localhost:$DETECTIONS_PORT/health" \
    "http://localhost:$LOADER_PORT/mock/status" \
    "http://localhost:$ANNOTATIONS_PORT/mock/status"; do
    for i in $(seq 1 30); do
        if curl -sf "$url" >/dev/null 2>&1; then break; fi
        if [ "$i" -eq 30 ]; then echo "ERROR: $url not ready" >&2; exit 1; fi
        sleep 1
    done
done

echo "All services ready. Running tests ..."
echo ""

BASE_URL="http://localhost:$DETECTIONS_PORT" \
MOCK_LOADER_URL="http://localhost:$LOADER_PORT" \
MOCK_ANNOTATIONS_URL="http://localhost:$ANNOTATIONS_PORT" \
MEDIA_DIR="$FIXTURES" \
PYTHONPATH="$ROOT/src" \
    "$PY" -m pytest e2e/tests/ tests/ -v --tb=short --durations=0 "$@"