[CmdletBinding()]
param(
    [string]$ContainerName = "ndc-mission-core-perception-worker",
    [string]$HealthUrl = "http://127.0.0.1:18020/health"
)

$ErrorActionPreference = "Stop"
$stageIds = @(
    "source-ingress",
    "camera-decode",
    "preprocessing",
    "detector",
    "semantic-model",
    "sensor-fusion",
    "tracking",
    "temporal-state",
    "result-publication"
)
$health = $null
$collectorState = "unavailable"
try {
    $healthJson = docker exec $ContainerName python3 -c `
        "import urllib.request;print(urllib.request.urlopen('$HealthUrl',timeout=2).read().decode())" `
        2>$null
    if ($LASTEXITCODE -eq 0 -and $healthJson) {
        $health = $healthJson | ConvertFrom-Json
        $collectorState = "live"
    }
}
catch {
    $health = $null
}

$serviceState = if ($health -and $health.state) {
    [string]$health.state
}
else {
    "unavailable"
}
$currentStage = if ($health -and $health.current_stage) {
    [string]$health.current_stage
}
else {
    ""
}
$activeStages = if ($health -and $health.active_stages) {
    @($health.active_stages | ForEach-Object { [string]$_ })
}
else {
    @()
}
$activeRequestId = if ($health -and $health.active_request_id) {
    [string]$health.active_request_id
}
else {
    ""
}
$activeFrameIndex = if (
    $health -and
    $null -ne $health.active_frame_index
) {
    [int64]$health.active_frame_index
}
else {
    [int64]-1
}
$completedRuns = if ($health -and $null -ne $health.completed_runs) {
    [int64]$health.completed_runs
}
else {
    [int64]0
}
$failedRuns = if ($health -and $null -ne $health.failed_runs) {
    [int64]$health.failed_runs
}
else {
    [int64]0
}
$modelLoadSeconds = if ($health -and $null -ne $health.model_load_seconds) {
    [double]$health.model_load_seconds
}
else {
    [double]0
}

$samples = foreach ($stageId in $stageIds) {
    $stageState = if ($collectorState -ne "live") {
        "unavailable"
    }
    elseif ($currentStage -eq $stageId -or $activeStages -contains $stageId) {
        "active"
    }
    elseif ($serviceState -eq "busy") {
        "waiting"
    }
    else {
        "ready"
    }
    $sample = [ordered]@{
        stage_id = $stageId
        stage_state = $stageState
        service_state = $serviceState
        current_stage = $currentStage
        active_request_id = $activeRequestId
        active_frame_index = $activeFrameIndex
        completed_runs = $completedRuns
        failed_runs = $failedRuns
        model_load_seconds = $modelLoadSeconds
        collector_state = $collectorState
        health_ok = [bool]($health -and $health.ok)
    }
    if ($health -and $health.stage_metrics) {
        $property = $health.stage_metrics.PSObject.Properties[$stageId]
        if ($property) {
            $metric = $property.Value
            if ($null -ne $metric.elapsed_seconds) {
                $sample.elapsed_seconds = [double]$metric.elapsed_seconds
            }
            if ($null -ne $metric.activations) {
                $sample.activations = [int64]$metric.activations
            }
            if ($null -ne $metric.share_percent) {
                $sample.share_percent = [double]$metric.share_percent
            }
        }
    }
    [pscustomobject]$sample
}

@($samples) | ConvertTo-Json -Compress -Depth 8