param(
    [string]$ApiBaseUrl = "http://192.168.200.61:18121/api/v1",
    [string]$ClusterID = "cfc0743d-d960-49fb-9de8-96e063d5e4aa",
    [string]$ActorUserID = "f67d943f-5397-4b3a-a229-695fe67ad700",
    [string]$EntryNodeName = "test-1",
    [string]$ExitNodeName = "test-2",
    [string]$EntryBaseUrl = "http://192.168.200.61:19131",
    [string]$ResultPath = "artifacts\c19x-remote-workspace-mailbox-consumer-resume-smoke-result.json",
    [switch]$SkipClose
)

Set-StrictMode -Version Latest
$ErrorActionPreference = "Stop"

$repoRoot = (Resolve-Path (Join-Path $PSScriptRoot "..\..")).ProviderPath
$runId = "c19x-" + (Get-Date -Format "yyyyMMdd-HHmmss")
$consumerID = "rdp-worker-probe"

function Invoke-Api {
    param([string]$Method, [string]$Path, [object]$Body = $null)
    $uri = "$ApiBaseUrl$Path"
    if ($null -eq $Body) { return Invoke-RestMethod -Method $Method -Uri $uri -TimeoutSec 30 }
    return Invoke-RestMethod -Method $Method -Uri $uri -ContentType "application/json" -Body ($Body | ConvertTo-Json -Depth 80) -TimeoutSec 30
}

function Get-PropertyValue {
    param([object]$Item, [string]$Name, [object]$Default = $null)
    if ($null -eq $Item) { return $Default }
    $property = $Item.PSObject.Properties[$Name]
    if ($null -eq $property) { return $Default }
    return $property.Value
}

function ConvertTo-Base64UrlJson {
    param([object]$Value)
    if ($Value -is [string]) { $json = $Value } else { $json = $Value | ConvertTo-Json -Depth 80 -Compress }
    $bytes = [System.Text.Encoding]::UTF8.GetBytes($json)
    return [Convert]::ToBase64String($bytes).TrimEnd("=").Replace("+", "-").Replace("/", "_")
}

function Get-Events {
    param([object]$Item)
    $events = Get-PropertyValue -Item $Item -Name "events" -Default $null
    if ($null -eq $events) { return @() }
    return @($events)
}

function Get-NodeByName {
    param([string]$Name)
    $nodes = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/nodes?actor_user_id=$ActorUserID").nodes
    $node = @($nodes | Where-Object { $_.name -eq $Name }) | Select-Object -First 1
    if ($null -eq $node) { throw "Node '$Name' was not found in cluster $ClusterID" }
    return $node
}

function Disable-ExistingRemoteWorkspaceRoutes {
    param([string]$SourceNodeID, [string]$DestinationNodeID)
    $items = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/mesh/route-intents?actor_user_id=$ActorUserID").route_intents
    foreach ($item in @($items)) {
        if ([string](Get-PropertyValue -Item $item -Name "status" -Default "") -ne "active") { continue }
        if ([string](Get-PropertyValue -Item $item -Name "service_class" -Default "") -ne "remote_workspace") { continue }
        $sourceSelector = Get-PropertyValue -Item $item -Name "source_selector" -Default $null
        $destinationSelector = Get-PropertyValue -Item $item -Name "destination_selector" -Default $null
        if ([string](Get-PropertyValue -Item $sourceSelector -Name "node_id" -Default "") -ne $SourceNodeID) { continue }
        if ([string](Get-PropertyValue -Item $destinationSelector -Name "node_id" -Default "") -ne $DestinationNodeID) { continue }
        [void](Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents/$($item.id)/disable" -Body @{
            actor_user_id = $ActorUserID
            reason = "c19x isolate remote workspace mailbox consumer resume smoke"
        })
    }
}

function New-RemoteWorkspaceRouteIntent {
    param([string]$SourceNodeID, [string]$DestinationNodeID)
    $expiresAt = (Get-Date).ToUniversalTime().AddMinutes(5).ToString("o")
    return Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents" -Body @{
        actor_user_id = $ActorUserID
        source_selector = @{ node_id = $SourceNodeID }
        destination_selector = @{ node_id = $DestinationNodeID }
        service_class = "remote_workspace"
        priority = 2100000000
        policy = @{
            synthetic_enabled = $true
            route_version = "$runId-remote-workspace"
            policy_version = "$runId-remote-workspace"
            peer_directory_version = "$runId-remote-workspace"
            hops = @($SourceNodeID, $DestinationNodeID)
            allowed_channels = @("control", "interactive", "reliable", "bulk", "droppable")
            max_ttl = 8
            max_hops = 8
            expires_at = $expiresAt
            metadata = @{ smoke = "c19x_remote_workspace_mailbox_consumer_resume"; run_id = $runId }
        }
    }
}

function New-FrameBatch {
    param([int]$Index)
    return [ordered]@{
        schema_version = "rap.remote_workspace_frame_batch.v1"
        probe_only = $true
        service_class = "remote_workspace"
        channel_class = "interactive"
        adapter_contract_id = "rap.rdp_worker.remote_workspace_adapter_contract_probe.v1"
        frames = @(@{ channel = "display"; direction = "adapter_to_client"; payload_encoding = "none"; payload_length = $Index; droppable = $true })
    }
}

function Invoke-FrameBatch {
    param([object]$FrameBatch, [hashtable]$Headers, [string]$Url)
    $response = Invoke-WebRequest -Method POST -Uri $Url -Headers $Headers -Body ($FrameBatch | ConvertTo-Json -Depth 80 -Compress) -ContentType "application/vnd.rap.remote-workspace-frame-batch.v1+json" -TimeoutSec 30
    return [ordered]@{ status_code = [int]$response.StatusCode; body = ($response.Content | ConvertFrom-Json) }
}

function Invoke-Mailbox {
    param([string]$SessionID, [string]$Query = "")
    $url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/mailbox$Query"
    $response = Invoke-WebRequest -Method GET -Uri $url -TimeoutSec 35
    $json = $null
    if ($response.Content) { $json = $response.Content | ConvertFrom-Json }
    return [ordered]@{ status_code = [int]$response.StatusCode; body = $response.Content; json = $json }
}

function Invoke-MailboxRaw {
    param([string]$SessionID, [string]$Query = "")
    $url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/mailbox$Query"
    try {
        $response = Invoke-WebRequest -Method GET -Uri $url -TimeoutSec 35
        return [ordered]@{ status_code = [int]$response.StatusCode; body = $response.Content }
    } catch {
        $response = $_.Exception.Response
        if ($null -eq $response) { throw }
        $reader = [System.IO.StreamReader]::new($response.GetResponseStream())
        try { $body = $reader.ReadToEnd() } finally { $reader.Dispose() }
        return [ordered]@{ status_code = [int]$response.StatusCode; body = $body }
    }
}

function Invoke-Control {
    param([string]$SessionID)
    $url = "$EntryBaseUrl/mesh/v1/remote-workspace/adapter-sessions/$SessionID/control"
    $body = @{ action = "close"; reason = "c19x mailbox consumer resume close" } | ConvertTo-Json -Compress
    return Invoke-RestMethod -Method POST -Uri $url -ContentType "application/json" -Body $body -TimeoutSec 30
}

$routeID = ""
try {
    $entryNode = Get-NodeByName -Name $EntryNodeName
    $exitNode = Get-NodeByName -Name $ExitNodeName
    Disable-ExistingRemoteWorkspaceRoutes -SourceNodeID $entryNode.id -DestinationNodeID $exitNode.id
    $route = (New-RemoteWorkspaceRouteIntent -SourceNodeID $entryNode.id -DestinationNodeID $exitNode.id).route_intent
    $routeID = [string]$route.id

    $leaseResponse = Invoke-Api -Method POST -Path "/clusters/$ClusterID/fabric/service-channels/leases" -Body @{
        actor_user_id = $ActorUserID
        organization_id = "org-home"
        user_id = "user-x"
        resource_id = "$runId-remote-workspace"
        service_class = "remote_workspace"
        entry_node_ids = @([string]$entryNode.id)
        exit_node_ids = @([string]$exitNode.id)
        preferred_entry_node_id = [string]$entryNode.id
        preferred_exit_node_id = [string]$exitNode.id
        ttl_seconds = 120
        metadata = @{ smoke = "c19x_remote_workspace_mailbox_consumer_resume"; run_id = $runId }
    }
    $lease = $leaseResponse.fabric_service_channel_lease
    $authorityPayload = Get-PropertyValue -Item $lease -Name "authority_payload" -Default $null
    if ($authorityPayload -is [string] -and $authorityPayload.Length -gt 0) { $decodedAuthority = $authorityPayload | ConvertFrom-Json } else { $decodedAuthority = $authorityPayload }

    $ingressUrl = "$EntryBaseUrl/api/v1/clusters/$ClusterID/fabric/service-channels/$($lease.channel_id)/remote-workspaces/$($lease.resource_id)/streams/interactive"
    $headers = @{
        "X-RAP-Service-Channel-Token" = [string]$lease.token.token
        "X-RAP-Fabric-Channel-ID" = [string]$lease.channel_id
        "X-RAP-Service-Channel-Authority-Payload" = ConvertTo-Base64UrlJson -Value $decodedAuthority
        "X-RAP-Service-Channel-Authority-Signature" = ConvertTo-Base64UrlJson -Value (Get-PropertyValue -Item $lease -Name "authority_signature" -Default $null)
        "X-RAP-Service-Class" = "remote_workspace"
        "X-RAP-Channel-Class" = "interactive"
    }

    $deliveries = @()
    for ($i = 1; $i -le 3; $i++) {
        $deliveries += ,(Invoke-FrameBatch -FrameBatch (New-FrameBatch -Index $i) -Headers $headers -Url $ingressUrl)
    }
    $adapterSessionID = [string](Get-PropertyValue -Item $deliveries[0].body -Name "adapter_session_id" -Default "")
    $baseline = Invoke-Mailbox -SessionID $adapterSessionID -Query "?limit=10"
    $baselineEvents = @(Get-Events -Item $baseline.json)
    $firstSequence = if ($baselineEvents.Count -ge 1) { [int64]$baselineEvents[0].sequence } else { 0 }
    $secondSequence = if ($baselineEvents.Count -ge 2) { [int64]$baselineEvents[1].sequence } else { 0 }
    $thirdSequence = if ($baselineEvents.Count -ge 3) { [int64]$baselineEvents[2].sequence } else { 0 }

    $checkpointSeed = Invoke-Mailbox -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&limit=2"
    $ackSeed = Invoke-Mailbox -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&ack_sequence=$firstSequence&limit=1"
    $resumeCheckpoint = Invoke-Mailbox -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&resume_from=checkpoint&limit=10"
    $resumeAck = Invoke-Mailbox -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&resume_from=ack&limit=10"
    $invalidResume = Invoke-MailboxRaw -SessionID $adapterSessionID -Query "?consumer_id=$consumerID&resume_from=ack&after_sequence=$firstSequence"
    $control = $null
    if (-not $SkipClose) {
        $control = Invoke-Control -SessionID $adapterSessionID
    }

    $checkpointEvents = @(Get-Events -Item $resumeCheckpoint.json)
    $ackEvents = @(Get-Events -Item $resumeAck.json)

    $checks = [ordered]@{
        lease_ready = ([string]$lease.status -eq "ready")
        deliveries_accepted = (@($deliveries | Where-Object { [int]$_.status_code -eq 202 }).Count -eq 3 -and $adapterSessionID -match "^rap-rw-adapter-session-[0-9a-f]{24}$")
        baseline_sequences_visible = ($baselineEvents.Count -ge 3 -and $firstSequence -gt 0 -and $secondSequence -gt $firstSequence -and $thirdSequence -gt $secondSequence)
        checkpoint_seed_visible = ([int]$checkpointSeed.status_code -eq 200 -and [int64]$checkpointSeed.json.consumer_checkpoint_sequence -eq $secondSequence)
        ack_seed_visible = ([int]$ackSeed.status_code -eq 200 -and [int64]$ackSeed.json.consumer_ack_sequence -eq $firstSequence)
        resume_checkpoint_filters = ([int]$resumeCheckpoint.status_code -eq 200 -and [string]$resumeCheckpoint.json.resume_from -eq "checkpoint" -and [int64]$resumeCheckpoint.json.resume_sequence -eq $secondSequence -and [int64]$resumeCheckpoint.json.after_sequence -eq $secondSequence -and [int]$resumeCheckpoint.json.returned_count -eq 1 -and $checkpointEvents.Count -eq 1 -and [int64]$checkpointEvents[0].sequence -eq $thirdSequence)
        resume_ack_filters = ([int]$resumeAck.status_code -eq 200 -and [string]$resumeAck.json.resume_from -eq "ack" -and [int64]$resumeAck.json.resume_sequence -eq $firstSequence -and [int64]$resumeAck.json.after_sequence -eq $firstSequence -and [int]$resumeAck.json.returned_count -eq 2 -and $ackEvents.Count -eq 2 -and [int64]$ackEvents[0].sequence -eq $secondSequence -and [int64]$ackEvents[1].sequence -eq $thirdSequence)
        invalid_resume_rejected = ([int]$invalidResume.status_code -eq 400)
        close_accepted = ($SkipClose -or ([bool]$control.accepted -and [string]$control.session_state -eq "closed"))
    }
    $failed = @($checks.GetEnumerator() | Where-Object { -not $_.Value } | ForEach-Object { $_.Key })

    $result = [ordered]@{
        schema_version = "c19x.remote_workspace_mailbox_consumer_resume_smoke.v1"
        run_id = $runId
        cluster_id = $ClusterID
        entry_node = @{ id = $entryNode.id; name = $EntryNodeName }
        exit_node = @{ id = $exitNode.id; name = $ExitNodeName }
        channel_id = $lease.channel_id
        route_id = $routeID
        adapter_session_id = $adapterSessionID
        baseline = $baseline
        checkpoint_seed = $checkpointSeed
        ack_seed = $ackSeed
        resume_checkpoint = $resumeCheckpoint
        resume_ack = $resumeAck
        invalid_resume = $invalidResume
        control = $control
        checks = $checks
        failed_checks = $failed
        passed = ($failed.Count -eq 0)
    }
} finally {
    if ($routeID) {
        try {
            [void](Invoke-Api -Method POST -Path "/clusters/$ClusterID/mesh/route-intents/$routeID/disable" -Body @{
                actor_user_id = $ActorUserID
                reason = "c19x mailbox consumer resume cleanup"
            })
        } catch {
            Write-Warning "cleanup failed after c19x smoke: $($_.Exception.Message)"
        }
    }
}

$fullResultPath = Join-Path $repoRoot $ResultPath
$resultDir = Split-Path -Parent $fullResultPath
if ($resultDir) { New-Item -ItemType Directory -Force -Path $resultDir | Out-Null }
$result | ConvertTo-Json -Depth 100 | Set-Content -Encoding UTF8 -Path $fullResultPath

if (-not $result.passed) {
    throw "C19X remote workspace mailbox consumer resume smoke failed. Result: $fullResultPath Failed: $($failed -join ', ')"
}

Write-Host "C19X remote workspace mailbox consumer resume smoke passed. Result: $fullResultPath"
$result