$ErrorActionPreference = "SilentlyContinue"

Write-Host "== AIAgent stop ==" -ForegroundColor Cyan

function Get-PidsListeningOnPort([int]$Port) {
  $pids = New-Object System.Collections.Generic.HashSet[int]
  try {
    netstat -ano | ForEach-Object {
      $ln = $_.Trim()
      if ($ln -notmatch "LISTENING") { return }
      # 匹配 :8037 或 :3001 等端口后的 LISTENING 行末 PID
      if ($ln -match ":$Port\s+.*LISTENING\s+(\d+)\s*$") {
        [void]$pids.Add([int]$Matches[1])
      }
    }
  } catch { }
  return @($pids)
}

function Stop-OnPorts([int[]]$ports, [string]$name) {
  $all = New-Object System.Collections.Generic.HashSet[int]
  foreach ($p in $ports) {
    foreach ($pid in (Get-PidsListeningOnPort $p)) {
      [void]$all.Add($pid)
    }
  }
  if ($all.Count -eq 0) {
    Write-Host "[SKIP] ${name}: no listener on ports $($ports -join ',')" -ForegroundColor DarkGray
    return
  }
  foreach ($pid in $all) {
    if ($pid -le 4) { continue }
    try {
      Stop-Process -Id $pid -Force -ErrorAction SilentlyContinue
      Write-Host "[OK] stopped ${name} PID=$pid" -ForegroundColor Green
    } catch {
      Write-Host "[WARN] failed to stop ${name} PID=$pid" -ForegroundColor Yellow
    }
  }
}

# 后端 API（8037 / 8041 备用）
Stop-OnPorts @(8037, 8041) "backend-api"

# 前端 Vite（3001）
Stop-OnPorts @(3001) "frontend-dev"

# Redis（6379，仅当监听在本地开发端口时结束；若与其它项目共用请谨慎）
Stop-OnPorts @(6379) "redis"

# Celery Worker / Beat（不监听端口，需要按命令行匹配）
$celeryPatterns = @(
  @{Pattern='celery\s+-A\s+app\.core\.celery_app\s+worker'; Name='celery-worker'},
  @{Pattern='celery\s+-A\s+app\.core\.celery_app\s+beat'; Name='celery-beat'}
)
foreach ($cp in $celeryPatterns) {
  $targets = Get-CimInstance Win32_Process | Where-Object {
    $_.CommandLine -and $_.CommandLine -match $cp.Pattern
  }
  if (-not $targets) {
    Write-Host "[SKIP] $($cp.Name): no matching process" -ForegroundColor DarkGray
  }
  foreach ($p in $targets) {
    try {
      Stop-Process -Id $p.ProcessId -Force -ErrorAction SilentlyContinue
      Write-Host "[OK] stopped $($cp.Name) PID=$($p.ProcessId)" -ForegroundColor Green
    } catch {
      Write-Host "[WARN] failed to stop $($cp.Name) PID=$($p.ProcessId)" -ForegroundColor Yellow
    }
  }
}

Start-Sleep -Milliseconds 600

Write-Host ""
Write-Host "Port check:" -ForegroundColor Cyan
foreach ($port in 3001, 8037, 8041, 6379) {
  $line = netstat -ano | Select-String ":$port\s+.*LISTENING" | Select-Object -First 1
  if ($line) {
    Write-Host " - ${port}: LISTEN" -ForegroundColor Yellow
  } else {
    Write-Host " - ${port}: free" -ForegroundColor Green
  }
}

Write-Host ""
Write-Host "DONE: stop script finished" -ForegroundColor Green