#!/bin/bash
# Run frontend test with optional port exposure
# Usage: ./run-test-frontend.sh [--expose-port {on|off}]
# Default: off
# 
# --expose-port on:  Keep services running after test (manual cleanup required)
# --expose-port off: Auto cleanup after test (default)

set -e

EXPOSE_PORT="off"
COMPOSE_FILE="docker-compose-frontend.yml"

# Load environment variables from .env.TEST if exists
if [ -f ".env.TEST" ]; then
    echo "📋 Loading .env.TEST..."
    set -a
    source .env.TEST
    set +a
fi

# Parse arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        --expose-port)
            EXPOSE_PORT="$2"
            shift 2
            ;;
        --expose-port=*)
            EXPOSE_PORT="${1#*=}"
            shift
            ;;
        *)
            echo "Unknown option: $1"
            echo "Usage: $0 [--expose-port {on|off}]"
            exit 1
            ;;
    esac
done

# Validate expose-port value
if [[ "$EXPOSE_PORT" != "on" && "$EXPOSE_PORT" != "off" ]]; then
    echo "Error: --expose-port must be 'on' or 'off'"
    exit 1
fi

# Select compose file based on expose-port
if [[ "$EXPOSE_PORT" == "on" ]]; then
    COMPOSE_FILE="docker-compose-frontend-expose.yml"
    echo "🔌 Port exposure: ON (services will keep running)"
else
    echo "🔌 Port exposure: OFF (auto cleanup after test)"
fi

echo "📦 Using compose file: $COMPOSE_FILE"

# Clean any previous containers first
echo "🧹 Cleaning up previous containers..."
docker compose -f "$COMPOSE_FILE" down -v 2>/dev/null || true

# Build frontend with correct API base URL (force no cache, remove image first)
echo "🔨 Building frontend..."
docker rmi harborforge-test-frontend:dev 2>/dev/null || true
docker compose -f "$COMPOSE_FILE" build --no-cache --build-arg VITE_API_BASE=http://backend:8000 frontend

# Build backend (force no cache, remove image first)
echo "🔨 Building backend..."
docker rmi harborforge-test-backend:dev 2>/dev/null || true
docker compose -f "$COMPOSE_FILE" build --no-cache backend

# Build test runner (force no cache, remove image first)
echo "🔨 Building test runner..."
docker rmi harborforge-test-runner:dev 2>/dev/null || true
docker compose -f "$COMPOSE_FILE" build --no-cache test

# Start services
echo "📦 Starting services..."
docker compose -f "$COMPOSE_FILE" up -d

# Wait for frontend to be ready
echo "⏳ Waiting for services..."
MAX_RETRIES=30
RETRY_COUNT=0
until docker run --rm --network harborforgetest_test-network curlimages/curl -s -o /dev/null -w "%{http_code}" http://frontend:3000/ 2>/dev/null | grep -q "200" || [ $RETRY_COUNT -eq $MAX_RETRIES ]; do
    echo "  Waiting for frontend... ($RETRY_COUNT/$MAX_RETRIES)"
    sleep 2
    RETRY_COUNT=$((RETRY_COUNT+1))
done

if [ $RETRY_COUNT -eq $MAX_RETRIES ]; then
    echo "❌ Frontend failed to start"
    docker compose -f "$COMPOSE_FILE" logs
    docker compose -f "$COMPOSE_FILE" down -v
    exit 1
fi

echo "✅ Services ready!"

# Run test
echo "🧪 Running test..."
docker compose -f "$COMPOSE_FILE" run --rm -e WORKERS=1 test
TEST_EXIT_CODE=$?

# Cleanup decision based on expose-port
if [[ "$EXPOSE_PORT" == "on" ]]; then
    echo ""
    echo "🔌 Port exposure is ON - keeping services running!"
    echo "   Use './run-test-frontend.sh --expose-port on' to cleanup"
    echo "   Or manually: docker compose -f $COMPOSE_FILE down -v"
    echo ""
    
    if [ $TEST_EXIT_CODE -eq 0 ]; then
        echo "✅ Test passed!"
    else
        echo "❌ Test failed with exit code: $TEST_EXIT_CODE"
    fi
else
    echo ""
    echo "🧹 Cleaning up containers and volumes..."
    docker compose -f "$COMPOSE_FILE" down -v
    
    if [ $TEST_EXIT_CODE -eq 0 ]; then
        echo "✅ Test passed!"
    else
        echo "❌ Test failed with exit code: $TEST_EXIT_CODE"
    fi
fi

exit $TEST_EXIT_CODE