param(
    [string]$Command = 'status'
)

$ErrorActionPreference = 'Stop'
$ProgressPreference = 'SilentlyContinue'

$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
$RootDir = (Resolve-Path $ScriptDir).Path
$EnvConfig = Join-Path $RootDir 'env_config.ps1'
if (Test-Path $EnvConfig) {
    . $EnvConfig
    Import-EnvFile -Path (Join-Path $RootDir '.env')
}
$PythonBin = Join-Path $RootDir '.venv-qwen35\Scripts\python.exe'
$GatewayRun = Join-Path $RootDir 'run_8080_toolhub_gateway.py'
$RuntimeDir = Join-Path $RootDir '.tmp\toolhub_gateway'
$PidFile = Join-Path $RuntimeDir 'gateway.pid'
$LogFile = Join-Path $RuntimeDir 'gateway.log'
$ErrLogFile = Join-Path $RuntimeDir 'gateway.err.log'
$ModelSwitch = Join-Path $RootDir 'switch_qwen35_webui.ps1'

$GatewayHost = if ($env:GATEWAY_HOST) { $env:GATEWAY_HOST } else { '127.0.0.1' }
$GatewayPort = if ($env:GATEWAY_PORT) { $env:GATEWAY_PORT } else { '8080' }
$BackendHost = if ($env:BACKEND_HOST) { $env:BACKEND_HOST } else { '127.0.0.1' }
$BackendPort = if ($env:BACKEND_PORT) { $env:BACKEND_PORT } else { '8081' }
$ThinkMode = if ($env:THINK_MODE) { $env:THINK_MODE } else { 'think-on' }
$BackendWaitHint = '.\start_8080_toolhub_stack.cmd logs'
$SpinnerFrameIntervalMs = 120
$SpinnerProbeIntervalMs = 1000

function Ensure-Dir {
    param([string]$Path)
    if (-not (Test-Path $Path)) {
        New-Item -Path $Path -ItemType Directory -Force | Out-Null
    }
}

function Test-GatewayRunning {
    if (-not (Test-Path $PidFile)) {
        return $false
    }
    $raw = Get-Content -Path $PidFile -ErrorAction SilentlyContinue | Select-Object -First 1
    $gatewayPid = 0
    if (-not [int]::TryParse([string]$raw, [ref]$gatewayPid)) {
        return $false
    }
    $proc = Get-Process -Id $gatewayPid -ErrorAction SilentlyContinue
    return $null -ne $proc
}

function Test-GatewayReady {
    try {
        $null = Invoke-RestMethod -Uri "http://$GatewayHost`:$GatewayPort/gateway/health" -Method Get -TimeoutSec 2
        return $true
    } catch {
        return $false
    }
}

function Show-GatewayFailureLogs {
    Write-Host '网关启动失败，最近日志如下:'
    if (Test-Path $LogFile) {
        Write-Host '=== 网关标准输出 ==='
        Get-Content -Path $LogFile -Tail 120 -ErrorAction SilentlyContinue
    }
    if (Test-Path $ErrLogFile) {
        Write-Host '=== 网关标准错误 ==='
        Get-Content -Path $ErrLogFile -Tail 120 -ErrorAction SilentlyContinue
    }
}

function Write-SpinnerLine {
    param(
        [string]$Label,
        [double]$Current,
        [int]$Total,
        [int]$Tick
    )
    $frames = @('|', '/', '-', '\')
    $frame = $frames[$Tick % $frames.Count]
    $currentText = [string][int][Math]::Floor($Current)
    Write-Host -NoNewline "`r$Label $frame $currentText/$Total 秒"
}

function Complete-SpinnerLine {
    Write-Host ''
}

function Stop-OrphanGatewayProcesses {
    try {
        $rootPattern = [regex]::Escape($RootDir)
        $targets = Get-CimInstance Win32_Process -Filter "Name='python.exe'" -ErrorAction SilentlyContinue | Where-Object {
            $cmd = [string]$_.CommandLine
            $cmd -match 'run_8080_toolhub_gateway\.py' -and $cmd -match $rootPattern
        }
        foreach ($proc in $targets) {
            if ($proc.ProcessId) {
                Stop-Process -Id ([int]$proc.ProcessId) -Force -ErrorAction SilentlyContinue
            }
        }
    } catch {}
}

function Start-Backend {
    if ($env:MODEL_KEY -and $env:MODEL_KEY -ne '9b') {
        throw "当前交付包仅支持 MODEL_KEY=9b，收到: $($env:MODEL_KEY)"
    }
    $oldHost = $env:HOST
    $oldPort = $env:PORT
    try {
        $env:HOST = $BackendHost
        $env:PORT = $BackendPort
        & powershell.exe -NoProfile -ExecutionPolicy Bypass -File $ModelSwitch '9b' $ThinkMode
        if ($LASTEXITCODE -ne 0) {
            throw '后端启动失败，请先查看上面的直接原因'
        }
    } finally {
        $env:HOST = $oldHost
        $env:PORT = $oldPort
    }
}

function Start-Gateway {
    Ensure-Dir $RuntimeDir
    Stop-OrphanGatewayProcesses
    if (Test-GatewayRunning) {
        Write-Host '网关状态: 已运行'
        Write-Host "PID: $(Get-Content -Path $PidFile)"
        return
    }
    if (-not (Test-Path $PythonBin)) {
        throw "Python 环境不存在: $PythonBin"
    }

    $args = @(
        $GatewayRun,
        '--host', $GatewayHost,
        '--port', $GatewayPort,
        '--backend-base', "http://$BackendHost`:$BackendPort",
        '--model-server', "http://$BackendHost`:$BackendPort/v1"
    )
    if (Test-Path $ErrLogFile) {
        Remove-Item -Path $ErrLogFile -Force -ErrorAction SilentlyContinue
    }
    $oldWaitHint = $env:BACKEND_WAIT_HINT
    try {
        $env:BACKEND_WAIT_HINT = $BackendWaitHint
        $proc = Start-Process -FilePath $PythonBin -ArgumentList $args -WindowStyle Hidden -RedirectStandardOutput $LogFile -RedirectStandardError $ErrLogFile -PassThru
    } finally {
        $env:BACKEND_WAIT_HINT = $oldWaitHint
    }
    Set-Content -Path $PidFile -Value $proc.Id -Encoding ascii

    $timeoutSec = 60
    $stopwatch = [System.Diagnostics.Stopwatch]::StartNew()
    $nextProbeMs = 0
    $tick = 0
    while ($stopwatch.Elapsed.TotalSeconds -lt $timeoutSec) {
        Write-SpinnerLine -Label '网关启动中...' -Current $stopwatch.Elapsed.TotalSeconds -Total $timeoutSec -Tick $tick
        if ($stopwatch.ElapsedMilliseconds -ge $nextProbeMs) {
            if (-not (Test-GatewayRunning)) {
                break
            }
            if (Test-GatewayReady) {
                Complete-SpinnerLine
                return
            }
            $nextProbeMs += $SpinnerProbeIntervalMs
        }
        Start-Sleep -Milliseconds $SpinnerFrameIntervalMs
        $tick++
    }
    Complete-SpinnerLine

    Show-GatewayFailureLogs
    throw '网关启动失败。'
}

function Stop-Gateway {
    Stop-OrphanGatewayProcesses
    if (-not (Test-GatewayRunning)) {
        if (Test-Path $PidFile) {
            Remove-Item -Path $PidFile -Force -ErrorAction SilentlyContinue
        }
        Write-Host '网关状态: 未运行'
        return
    }

    $gatewayPid = [int](Get-Content -Path $PidFile | Select-Object -First 1)
    Stop-Process -Id $gatewayPid -Force -ErrorAction SilentlyContinue
    Start-Sleep -Seconds 1
    if (Test-Path $PidFile) {
        Remove-Item -Path $PidFile -Force -ErrorAction SilentlyContinue
    }
    Write-Host '网关状态: 已停止'
}

function Show-Status {
    Write-Host '=== 网关 ==='
    if (Test-GatewayRunning) {
        $state = if (Test-GatewayReady) { '可访问' } else { '初始化中' }
        Write-Host '状态: 运行中'
        Write-Host "PID: $(Get-Content -Path $PidFile)"
        Write-Host "地址: http://$GatewayHost`:$GatewayPort"
        Write-Host "健康: $state"
        Write-Host "日志: $LogFile"
        Write-Host "错误日志: $ErrLogFile"
    } else {
        Write-Host '状态: 未运行'
    }

    Write-Host ''
    Write-Host '=== 模型后端 ==='
    $oldHost = $env:HOST
    $oldPort = $env:PORT
    try {
        $env:HOST = $BackendHost
        $env:PORT = $BackendPort
        & powershell.exe -NoProfile -ExecutionPolicy Bypass -File $ModelSwitch 'status'
    } finally {
        $env:HOST = $oldHost
        $env:PORT = $oldPort
    }
}

function Show-Logs {
    Write-Host '=== 网关日志 ==='
    if (Test-Path $LogFile) {
        Get-Content -Path $LogFile -Tail 120
    }
    if (Test-Path $ErrLogFile) {
        Write-Host '=== 网关错误日志 ==='
        Get-Content -Path $ErrLogFile -Tail 120
        return
    }
    Write-Host '暂无日志'
}

function Stop-Backend {
    $oldHost = $env:HOST
    $oldPort = $env:PORT
    try {
        $env:HOST = $BackendHost
        $env:PORT = $BackendPort
        & powershell.exe -NoProfile -ExecutionPolicy Bypass -File $ModelSwitch 'stop'
    } finally {
        $env:HOST = $oldHost
        $env:PORT = $oldPort
    }
}

function Start-Stack {
    try {
        Write-Host '步骤 1/2: 启动模型后端'
        Start-Backend
        Write-Host '步骤 2/2: 启动网关服务'
        Start-Gateway
        Write-Host '栈已启动'
        Write-Host "网页入口: http://$GatewayHost`:$GatewayPort"
        Write-Host '可用状态检查命令: .\start_8080_toolhub_stack.cmd status'
        Write-Host '停止命令: .\start_8080_toolhub_stack.cmd stop'
    } catch {
        Write-Host $_.Exception.Message
        exit 1
    }
}

function Stop-Stack {
    Stop-Gateway
    Stop-Backend
}

switch ($Command) {
    'start' { Start-Stack; break }
    'stop' { Stop-Stack; break }
    'restart' { Stop-Stack; Start-Stack; break }
    'status' { Show-Status; break }
    'logs' { Show-Logs; break }
    default {
        Write-Host '用法:'
        Write-Host '  .\\start_8080_toolhub_stack.cmd {start|stop|restart|status|logs}'
        Write-Host ''
        Write-Host '可选环境变量:'
        Write-Host '  GATEWAY_HOST=127.0.0.1'
        Write-Host '  GATEWAY_PORT=8080'
        Write-Host '  BACKEND_HOST=127.0.0.1'
        Write-Host '  BACKEND_PORT=8081'
        Write-Host '  THINK_MODE=think-on'
        exit 1
    }
}