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

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"

cleanup() {
    docker compose -f "$PROJECT_ROOT/docker-compose.yml" -f "$PROJECT_ROOT/docker-compose.tests.yml" down --remove-orphans || true
}
trap cleanup EXIT

usage() {
    cat <<EOF
Usage: $(basename "$0") [--unit-only | --smoke | --full]

Modes:
  --unit-only   Run only the .NET unit test project (no Docker Compose, no integration tests)
  --smoke       Run unit tests + a fast integration subset (~2 min target, tightened timeouts)
  --full        Run unit tests + the full integration suite (default if no flag is provided)

Environment:
  GOOGLE_MAPS_API_KEY  Required for any integration test mode (loaded from .env or shell env).
EOF
}

mode="full"
case "${1:-}" in
    --unit-only) mode="unit"; ;;
    --smoke)     mode="smoke"; ;;
    --full)      mode="full"; ;;
    -h|--help)   usage; exit 0; ;;
    "")          ;;
    *)           echo "Unknown argument: $1"; usage; exit 2; ;;
esac

echo "=== Satellite Provider Test Suite ==="
echo "Mode: $mode"
echo ""

if [[ "$mode" == "unit" ]]; then
    echo "Running unit tests only..."
    docker run --rm -v "$PROJECT_ROOT:/src" -w /src mcr.microsoft.com/dotnet/sdk:8.0 \
        dotnet test SatelliteProvider.Tests/SatelliteProvider.Tests.csproj \
        --no-restore --configuration Release \
        --logger "console;verbosity=normal"
    echo ""
    echo "=== Unit tests complete ==="
    exit 0
fi

if [[ -z "${GOOGLE_MAPS_API_KEY:-}" ]] && [[ -f "$PROJECT_ROOT/.env" ]]; then
    set -o allexport
    # shellcheck disable=SC1091
    source "$PROJECT_ROOT/.env"
    set +o allexport
fi

if [[ -z "${GOOGLE_MAPS_API_KEY:-}" ]]; then
    echo "ERROR: GOOGLE_MAPS_API_KEY is not set (export it or add to .env). Integration tests will fail."
    exit 3
fi

echo "Step 1: Unit tests"
docker run --rm -v "$PROJECT_ROOT:/src" -w /src mcr.microsoft.com/dotnet/sdk:8.0 \
    sh -c "dotnet restore SatelliteProvider.sln && dotnet test SatelliteProvider.Tests/SatelliteProvider.Tests.csproj --no-restore --configuration Release --logger 'console;verbosity=normal'"

echo ""
echo "Step 2: Integration tests (Docker Compose, mode=$mode)"
INTEGRATION_TESTS_MODE="$mode" docker compose \
    -f "$PROJECT_ROOT/docker-compose.yml" \
    -f "$PROJECT_ROOT/docker-compose.tests.yml" \
    up --build --abort-on-container-exit --exit-code-from integration-tests

echo ""
echo "=== All tests passed (mode=$mode) ==="