dev.ps1

param(
    [Parameter(Mandatory=$true)]
    [ValidateSet("start", "stop", "restart", "reset", "logs", "rebuild", "status")]
    [string]$cmd,

    [int]$workers = 2
)

function Assert-Docker {
    docker compose version | Out-Null
    if ($LASTEXITCODE -ne 0) {
        throw "Docker Compose is required. Start Docker Desktop and try again."
    }
}

function Wait-Postgres {
    param([int]$TimeoutSeconds = 90)

    Write-Host "Waiting for Postgres health..."
    $deadline = (Get-Date).AddSeconds($TimeoutSeconds)
    while ((Get-Date) -lt $deadline) {
        $status = docker inspect -f "{{.State.Health.Status}}" mlis-postgres 2>$null
        if ($LASTEXITCODE -eq 0 -and $status -eq "healthy") {
            Write-Host "Postgres is healthy."
            return
        }
        Start-Sleep -Seconds 2
    }

    throw "Timed out waiting for Postgres to become healthy."
}

function Wait-ApiReady {
    param([int]$TimeoutSeconds = 90)

    Write-Host "Waiting for API readiness..."
    $deadline = (Get-Date).AddSeconds($TimeoutSeconds)
    while ((Get-Date) -lt $deadline) {
        try {
            $response = Invoke-WebRequest -UseBasicParsing -Uri "http://localhost:8001/health/ready/full" -TimeoutSec 3
            if ($response.StatusCode -ge 200 -and $response.StatusCode -lt 300) {
                Write-Host "API is ready."
                return
            }
        } catch {
            Start-Sleep -Seconds 2
            continue
        }
        Start-Sleep -Seconds 2
    }

    throw "Timed out waiting for API readiness. Run .\dev.ps1 logs to inspect app/worker output."
}

function Write-StartupSummary {
    Write-Host ""
    Write-Host "Push scheduler demo stack is running."
    Write-Host "Console:        http://localhost:8001/console"
    Write-Host "API readiness:  http://localhost:8001/health/ready/full"
    Write-Host "Scheduler mode: push"
    Write-Host "Job source:     db"
    Write-Host "Database:       Postgres"
    Write-Host "Workers:        $workers"
    Write-Host ""
    Write-Host "Use .\dev.ps1 logs to view logs, .\dev.ps1 stop to stop the stack."
}

function Start-App {
    Assert-Docker

    Write-Host "Building image..."
    docker compose build
    if ($LASTEXITCODE -ne 0) { throw "docker compose build failed." }

    Write-Host "Starting Postgres..."
    docker compose up -d postgres
    if ($LASTEXITCODE -ne 0) { throw "Failed to start Postgres." }
    Wait-Postgres

    Write-Host "Starting API and worker containers in push mode..."
    docker compose up -d --scale worker=$workers app worker
    if ($LASTEXITCODE -ne 0) { throw "Failed to start app/worker containers." }
    Wait-ApiReady
    Write-StartupSummary
}

function Stop-App {
    Assert-Docker

    Write-Host "Stopping containers..."
    docker compose down
}

function Restart-App {
    Assert-Docker

    Write-Host "Restarting app and worker containers..."
    docker compose restart app worker
    Wait-ApiReady
    Write-StartupSummary
}

function Reset-App {
    Assert-Docker

    Write-Host "WARNING: This will delete database volume."
    docker compose down -v

    Write-Host "Rebuilding (no cache)..."
    docker compose build --no-cache
    if ($LASTEXITCODE -ne 0) { throw "docker compose build --no-cache failed." }

    Write-Host "Starting fresh push-mode stack..."
    docker compose up -d postgres
    if ($LASTEXITCODE -ne 0) { throw "Failed to start Postgres." }
    Wait-Postgres
    docker compose up -d --scale worker=$workers app worker
    if ($LASTEXITCODE -ne 0) { throw "Failed to start app/worker containers." }
    Wait-ApiReady
    Write-StartupSummary
}

function Rebuild-App {
    Assert-Docker

    Write-Host "Rebuilding image (no cache)..."
    docker compose build --no-cache
    if ($LASTEXITCODE -ne 0) { throw "docker compose build --no-cache failed." }
    docker compose up -d postgres
    if ($LASTEXITCODE -ne 0) { throw "Failed to start Postgres." }
    Wait-Postgres
    docker compose up -d --scale worker=$workers app worker
    if ($LASTEXITCODE -ne 0) { throw "Failed to start app/worker containers." }
    Wait-ApiReady
    Write-StartupSummary
}

function Show-Logs {
    Assert-Docker
    docker compose logs -f postgres app worker
}

function Show-Status {
    Assert-Docker
    docker compose ps
}

switch ($cmd) {
    "start"   { Start-App }
    "stop"    { Stop-App }
    "restart" { Restart-App }
    "reset"   { Reset-App }
    "rebuild" { Rebuild-App }
    "logs"    { Show-Logs }
    "status"  { Show-Status }
}