# Test throttle functionality
# 测试限流功能
#
# 主要测试内容：
# 1. 限流记录清除功能
# 2. 正常注册请求（在限流范围内）
# 3. 批量请求测试限流阈值
# 4. 验证限流配置是否正确生效
#
# 当前限流配置：
# - 注册接口：10次/5分钟（开发环境已放宽）
# - 登录接口：5次/分钟
# - 发送验证码：1次/分钟
# - 密码重置：3次/小时
$baseUrl = "http://localhost:3000"

Write-Host "🚦 Testing Throttle Functionality" -ForegroundColor Green
Write-Host "==================================" -ForegroundColor Green

# Clear throttle first
Write-Host "`n🔄 Clearing throttle records..." -ForegroundColor Blue
try {
    $clearResponse = Invoke-RestMethod -Uri "$baseUrl/auth/debug-clear-throttle" -Method POST
    Write-Host "✅ $($clearResponse.message)" -ForegroundColor Green
} catch {
    Write-Host "⚠️ Could not clear throttle records" -ForegroundColor Yellow
}

# Test normal registration (should work with increased limit)
Write-Host "`n📋 Test 1: Normal registration with increased throttle limit" -ForegroundColor Yellow
$registerData = @{
    username = "testuser_throttle_$(Get-Random)"
    password = "password123"
    nickname = "Test User Throttle"
} | ConvertTo-Json

try {
    $response = Invoke-RestMethod -Uri "$baseUrl/auth/register" -Method POST -Body $registerData -ContentType "application/json" -ErrorAction Stop
    Write-Host "✅ SUCCESS: Registration completed" -ForegroundColor Green
    Write-Host "Message: $($response.message)" -ForegroundColor Cyan
} catch {
    $statusCode = $_.Exception.Response.StatusCode.value__
    Write-Host "❌ FAILED ($statusCode)" -ForegroundColor $(if ($statusCode -eq 429) { "Yellow" } else { "Red" })
    
    if ($_.Exception.Response) {
        $reader = New-Object System.IO.StreamReader($_.Exception.Response.GetResponseStream())
        $responseBody = $reader.ReadToEnd()
        $reader.Close()
        
        try {
            $errorResponse = $responseBody | ConvertFrom-Json
            Write-Host "Message: $($errorResponse.message)" -ForegroundColor Cyan
            if ($errorResponse.throttle_info) {
                Write-Host "Throttle Info:" -ForegroundColor Gray
                Write-Host "  Limit: $($errorResponse.throttle_info.limit)" -ForegroundColor Gray
                Write-Host "  Window: $($errorResponse.throttle_info.window_seconds)s" -ForegroundColor Gray
                Write-Host "  Current: $($errorResponse.throttle_info.current_requests)" -ForegroundColor Gray
                Write-Host "  Reset: $($errorResponse.throttle_info.reset_time)" -ForegroundColor Gray
            }
        } catch {
            Write-Host "Raw Response: $responseBody" -ForegroundColor Gray
        }
    }
}

# Test throttle limits by making multiple requests
Write-Host "`n📋 Test 2: Testing throttle limits (register endpoint: 10 requests/5min)" -ForegroundColor Yellow
$successCount = 0
$throttleCount = 0

for ($i = 1; $i -le 12; $i++) {
    $testData = @{
        username = "throttletest_$i"
        password = "password123"
        nickname = "Throttle Test $i"
    } | ConvertTo-Json
    
    try {
        $response = Invoke-RestMethod -Uri "$baseUrl/auth/register" -Method POST -Body $testData -ContentType "application/json" -ErrorAction Stop
        $successCount++
        Write-Host "  Request $i`: ✅ Success" -ForegroundColor Green
    } catch {
        $statusCode = $_.Exception.Response.StatusCode.value__
        if ($statusCode -eq 429) {
            $throttleCount++
            Write-Host "  Request $i`: 🚦 Throttled (429)" -ForegroundColor Yellow
        } else {
            Write-Host "  Request $i`: ❌ Failed ($statusCode)" -ForegroundColor Red
        }
    }
    
    # Small delay between requests
    Start-Sleep -Milliseconds 100
}

Write-Host "`n📊 Results:" -ForegroundColor Cyan
Write-Host "  Successful requests: $successCount" -ForegroundColor Green
Write-Host "  Throttled requests: $throttleCount" -ForegroundColor Yellow
Write-Host "  Expected behavior: ~10 success, ~2 throttled" -ForegroundColor Gray

if ($successCount -ge 8 -and $throttleCount -ge 1) {
    Write-Host "✅ PASS: Throttle is working correctly" -ForegroundColor Green
} else {
    Write-Host "⚠️ WARNING: Throttle behavior may need adjustment" -ForegroundColor Yellow
}

Write-Host "`n🎯 Throttle Configuration:" -ForegroundColor Green
Write-Host "  Register: 10 requests / 5 minutes" -ForegroundColor White
Write-Host "  Login: 5 requests / 1 minute" -ForegroundColor White
Write-Host "  Send Code: 1 request / 1 minute" -ForegroundColor White
Write-Host "  Password Reset: 3 requests / 1 hour" -ForegroundColor White