[Test only] #2

# Requires PowerShell 7+ and an authenticated GitHub CLI (gh).
# Preview: pwsh -File .\refresh-arm-linter-issue.ps1
# Update:  pwsh -File .\refresh-arm-linter-issue.ps1 -Apply
[CmdletBinding()]
param([switch]$Apply)

$ErrorActionPreference = 'Stop'
$repo = 'Azure/typespec-azure'
$issueRepo = 'msyyc/typespec-azure'
$issueNumber = 2
$expectedRows = 181
$statusOrder = @{ 'Done' = 0; 'Hold on' = 1; 'Doing' = 2; 'To do' = 3; 'Need investigation' = 4 }
$pinnedPrs = @{
    'ConsistentResponseSchemaForPut' = 5423
    'PutRequestResponseSchemeArm' = 5423
    'XmsResourceInPutResponse' = 5423
    'PatchBodyParametersSchema' = 5294
    'ConsistentPatchProperties' = 5294
    'UnSupportedPatchProperties' = 5294
    'DescriptionMustNotBeNodeName' = 5663
    'DescriptiveDescriptionRequired' = 5663
    'NoErrorCodeResponses' = 5556
    'GetResponseCodes' = 5556
    'LroExtension' = 5652
    'XMSLongRunningOperationProperty' = 5652
    'PutInOperationName' = 5673
    'PatchInOperationName' = 5673
    'DeleteInOperationName' = 5673
}

$migrationGroups = @(
    @{ Id = 'M1'; Label = 'Redundant client name'; Members = @('XmsClientNameParameter', 'XmsClientNameProperty', 'XmsClientName') }
    @{ Id = 'M2'; Label = 'Legacy SDK settings'; Members = @('DeprecatedXmsCodeGenerationSetting', 'LicenseHeaderMustNotBeSpecified') }
    @{ Id = 'M3'; Label = 'Request-body prohibition'; Members = @('GetMustNotHaveRequestBody', 'DeleteMustNotHaveRequestBody') }
    @{ Id = 'M4'; Label = 'PATCH response codes'; Members = @('PatchResponseCodes', 'LroPatch202') }
    @{ Id = 'M5'; Label = 'Tracked-resource lifecycle'; Members = @('TrackedResourcesMustHavePut', 'TrackedResourcePatchOperation') }
    @{ Id = 'C1'; Label = 'Documentation links'; Members = @('AvoidMsdnReferences', 'docLinkLocale') }
    @{ Id = 'C4'; Label = 'PATCH envelope'; Members = @('PatchIdentityProperty', 'PatchSkuProperty') }
    @{ Id = 'C5'; Label = 'Default-error response'; Members = @('RequiredDefaultResponse', 'MissingDefaultResponse') }
    @{ Id = 'C6'; Label = 'Required collections'; Members = @('NestedResourcesMustHaveListOperation', 'TopLevelResourcesListByResourceGroup', 'TopLevelResourcesListBySubscription') }
    @{ Id = 'C7'; Label = 'Pagination'; Members = @('PageableOperation', 'PageableRequires200Response', 'XmsPageableMustHaveCorrespondingResponse') }
    @{ Id = 'C8'; Label = 'Canonical operations response'; Members = @('OperationsApiResponseSchema', 'OperationsApiSchemaUsesCommonTypes') }
)
$reviewedRules = @(
    'AvoidMsdnReferences'
    'DeleteResponseBodyEmpty'
    'GetOperationMustNotBeLongRunning'
    'InvalidSkuModel'
    'ListInOperationName'
    'LocationMustHaveXmsMutability'
    'LongRunningOperationsOptionsValidator'
    'LroPatch202'
    'MissingSegmentsInNestedResourceListOperation'
    'MissingXmsErrorResponse'
    'MutabilityWithReadOnly'
    'NestedResourcesMustHaveListOperation'
    'OperationIdNounVerb'
    'OperationsApiResponseSchema'
    'OperationsApiSchemaUsesCommonTypes'
    'PageableRequires200Response'
    'ParameterNotUsingCommonTypes'
    'PatchIdentityProperty'
    'PatchResponseCodes'
    'PatchSkuProperty'
    'PathContainsResourceGroup'
    'PathContainsResourceType'
    'PathForNestedResource'
    'PathForTrackedResourceTypes'
    'PathResourceProviderNamePascalCase'
    'PostOperationIdContainsUrlVerb'
    'PreviewVersionOverOneYear'
    'PutRequestResponseScheme'
    'RequestBodyMustExistForPutPatch'
    'ReservedResourceNamesModelAsEnum'
    'ResourceHasXMsResourceEnabled'
    'TopLevelResourcesListByResourceGroup'
    'TrackedResourcePatchOperation'
    'TrackedResourcesMustHavePut'
    'ValidFormats'
    'XmsPageableMustHaveCorrespondingResponse'
    'XmsPathsMustOverloadPaths'
    'AzureResourceTagsSchema'
    'DefaultErrorResponseSchema'
    'DefaultInEnum'
    'DeleteMustNotHaveRequestBody'
    'DeleteOperationResponses'
    'DeprecatedXmsCodeGenerationSetting'
    'docLinkLocale'
    'GetCollectionResponseSchema'
    'GetMustNotHaveRequestBody'
    'LicenseHeaderMustNotBeSpecified'
    'LroWithOriginalUriAsFinalState'
    'MissingDefaultResponse'
    'PageableOperation'
    'PathContainsSubscriptionId'
    'PathResourceProviderMatchNamespace'
    'PrivateEndpointResourceSchemaValidation'
    'RequiredDefaultResponse'
    'RequiredPropertiesMissingInResourceModel'
    'RequiredReadOnlySystemData'
    'Rpaas_ResourceProvisioningState'
    'SecurityDefinitionsStructure'
    'SystemDataInPropertiesBag'
    'TopLevelResourcesListBySubscription'
    'UniqueXmsEnumName'
    'UniqueXmsExample'
    'XmsClientNameParameter'
    'XmsClientNameProperty'
    'XmsEnumValidation'
    'XmsPageableListByRGAndSubscriptions'
    'XmsParameterLocation'
    'XmsClientName'
)
$reviewedRank = @{}
for ($i = 0; $i -lt $reviewedRules.Count; $i++) {
    if ($reviewedRank.ContainsKey($reviewedRules[$i])) { throw 'Duplicate reviewed rule.' }
    $reviewedRank[$reviewedRules[$i]] = $i
}
$groupByRule = @{}
$groupRank = @{}
foreach ($group in $migrationGroups) {
    $anchorRank = ($group.Members | ForEach-Object {
        if (-not $reviewedRank.ContainsKey($_)) { throw "Unknown group member: $_" }
        $reviewedRank[$_]
    } | Measure-Object -Minimum).Minimum
    foreach ($rule in $group.Members) {
        if ($groupByRule.ContainsKey($rule)) { throw "Duplicate group membership: $rule" }
        $groupByRule[$rule] = $group
        $groupRank[$rule] = $anchorRank
    }
}

function Format-MigrationLibrary([string]$Rule, [string]$Library) {
    if (-not $groupByRule.ContainsKey($Rule)) { return $Library }
    $group = $groupByRule[$Rule]
    $annotation = "[$($group.Id) - $($group.Label)][migration-$($group.Id)]"
    if ($Library -ceq $annotation) { return $annotation }
    if ($Library.EndsWith("<br>$annotation", [StringComparison]::Ordinal)) { return $Library }
    if ($Library -eq '-' -or [string]::IsNullOrWhiteSpace($Library)) { return $annotation }
    "$Library<br>$annotation"
}

function Get-OrderedRows([object[]]$Rows) {
    $Rows | Sort-Object { $statusOrder[$_.Cells[8]] }, {
        if ($reviewedRank.ContainsKey($_.Rule)) { 1 } else { 0 }
    }, {
        if ($groupRank.ContainsKey($_.Rule)) { $groupRank[$_.Rule] }
        elseif ($reviewedRank.ContainsKey($_.Rule)) { $reviewedRank[$_.Rule] }
        else { $_.OriginalId }
    }, {
        if ($reviewedRank.ContainsKey($_.Rule)) { $reviewedRank[$_.Rule] }
        else { $_.OriginalId }
    }
}

function Assert-MigrationLayout([object[]]$Rows) {
    foreach ($rule in $reviewedRules) {
        if ($rule -notin $Rows.Rule) { throw "Missing reviewed rule: $rule" }
    }
    foreach ($row in $Rows) {
        if ($row.Cells[2] -cne (Format-MigrationLibrary $row.Rule $row.Cells[2])) {
            throw "Missing migration-group annotation: $($row.Rule)"
        }
    }
    for ($i = 1; $i -lt $Rows.Count; $i++) {
        if ($statusOrder[$Rows[$i].Cells[8]] -lt $statusOrder[$Rows[$i - 1].Cells[8]]) {
            throw 'Statuses must form contiguous blocks: Done, Hold on, Doing, To do, Need investigation.'
        }
    }
    $ordered = @(Get-OrderedRows $Rows)
    if (($Rows.Rule -join '|') -cne ($ordered.Rule -join '|')) {
        throw 'Incorrect status-first/migration-group ordering.'
    }
}

function Get-GitHubJson([string[]]$Arguments) {
    $raw = & gh @Arguments
    if ($LASTEXITCODE -ne 0) { throw "GitHub command failed: gh $($Arguments -join ' ')" }
    ($raw -join "`n") | ConvertFrom-Json
}

function Get-Api([string]$Path, [switch]$Paginate) {
    if ($Paginate) {
        $pages = Get-GitHubJson @('api', $Path, '--paginate', '--slurp')
        foreach ($page in $pages) { foreach ($item in $page) { $item } }
    } else {
        Get-GitHubJson @('api', $Path)
    }
}

function Test-EligibleUser($User, [string]$Author) {
    $User.type -eq 'User' -and $User.login -ne $Author -and $User.login -notmatch '\[bot\]$'
}

function Get-Rows([string]$Body) {
    foreach ($line in ($Body -split "`n")) {
        if ($line -notmatch '^\| \d+ \|') { continue }
        if ($line -notmatch '^\|(?:[^|\r\n]*\|){10}\r?$') { throw 'Malformed table row.' }
        $cells = @($line.Split('|')[1..10] | ForEach-Object { $_.Trim() })
        if (-not $statusOrder.ContainsKey($cells[8])) { throw "Unknown status: $($cells[8])" }
        $rule = if ($cells[1] -match '^\[([^\]]+)\]') { $Matches[1] } else { $cells[1] }
        [pscustomobject]@{ OriginalId = [int]$cells[0]; Rule = $rule; Cells = $cells }
    }
}

function Assert-Table([object[]]$Rows) {
    if ($Rows.Count -ne $expectedRows -or @($Rows.Rule | Sort-Object -Unique).Count -ne $expectedRows) {
        throw "Expected exactly $expectedRows unique rules."
    }
    for ($i = 0; $i -lt $Rows.Count; $i++) {
        if ($Rows[$i].OriginalId -ne $i + 1) { throw 'IDs must be consecutive from 1.' }
    }
}

$issue = Get-Api "repos/$issueRepo/issues/$issueNumber"
$originalBody = $issue.body
$rows = @(Get-Rows $originalBody)
Assert-Table $rows
$allPrs = @(Get-GitHubJson @(
    'pr', 'list', '--repo', $repo, '--state', 'all',
    '--search', '"[Swagger Linter Migration]" in:title',
    '--limit', '1000', '--json', 'number,title'
))
if ($allPrs.Count -ge 1000) { throw 'PR search may be truncated; narrow or paginate the search.' }
$candidates = @($allPrs | Where-Object { $_.title -notmatch '\(origin\)' })
$changes = [Collections.Generic.List[object]]::new()

foreach ($row in $rows) {
    if ($row.Cells[8] -in @('Done', 'Hold on')) { continue }
    $row.Cells[2] = Format-MigrationLibrary $row.Rule $row.Cells[2]
    # Match the original Swagger name in legacy titles and OldRule -> new-rule mappings.
    $pattern = '^\[Swagger Linter Migration\]\s+' + [regex]::Escape($row.Rule) + '(?=\s|->|$)'
    $isPinned = $pinnedPrs.ContainsKey($row.Rule)
    if ($isPinned) {
        $number = $pinnedPrs[$row.Rule]
    } else {
        $matching = @($candidates | Where-Object { $_.title -match $pattern })
        if ($matching.Count -gt 1) { throw "Multiple matching PRs for $($row.Rule); resolve manually." }
        if ($matching.Count -eq 0) {
            if ($row.Cells[3] -match '/pull/\d+') {
                throw "Existing PR for $($row.Rule) was not found in search; investigate before updating."
            }
            continue
        }
        $number = $matching[0].number
    }

    $pr = Get-Api "repos/$repo/pulls/$number"
    if (-not $isPinned -and ($pr.title -notmatch $pattern -or $pr.title -match '\(origin\)')) {
        throw "PR #$number title changed."
    }
    if (-not $pr.merged -and $pr.state -ne 'open') {
        throw "PR #$number is closed without merging; resolve its table status manually."
    }
    $files = @(Get-Api "repos/$repo/pulls/$number/files?per_page=100" -Paginate)
    if ($files.Count -ne $pr.changed_files) { throw "Incomplete changed-file list for PR #$number." }
    $libs = @(
        foreach ($package in @('typespec-azure-core', 'typespec-azure-resource-manager', 'typespec-client-generator-core')) {
            if ($files.filename -like "packages/$package/*") { "@azure-tools/$package" }
        }
    )
    if (-not $libs.Count) { throw "Unrecognized library in PR #$number; extend the package mapping." }

    $reviews = @(Get-Api "repos/$repo/pulls/$number/reviews?per_page=100" -Paginate)
    # A later change request or dismissal supersedes an approval; a comment does not.
    $approvals = @($reviews | Where-Object {
        (Test-EligibleUser $_.user $pr.user.login) -and
        $_.state -in @('APPROVED', 'CHANGES_REQUESTED', 'DISMISSED')
    } | Group-Object { $_.user.login } | ForEach-Object {
        $last = $_.Group | Sort-Object { [DateTimeOffset]$_.submitted_at }, id | Select-Object -Last 1
        if ($last.state -eq 'APPROVED') { $last.user.login }
    } | Sort-Object -Unique)

    $feedbackFlag = '-'
    if (-not $pr.merged) {
        $comments = @(Get-Api "repos/$repo/issues/$number/comments?per_page=100" -Paginate)
        $inline = @(Get-Api "repos/$repo/pulls/$number/comments?per_page=100" -Paginate)
        $head = Get-Api "repos/$repo/commits/$($pr.head.sha)"
        $latest = [DateTimeOffset]$head.commit.committer.date
        $feedback = @(
            @($comments) + @($inline) | Where-Object {
                (Test-EligibleUser $_.user $pr.user.login) -and
                [DateTimeOffset]$_.created_at -gt $latest
            }
            $reviews | Where-Object {
                (Test-EligibleUser $_.user $pr.user.login) -and
                -not [string]::IsNullOrWhiteSpace($_.body) -and
                $_.submitted_at -and [DateTimeOffset]$_.submitted_at -gt $latest
            }
        )
        $feedbackFlag = if ($feedback.Count) { 'Yes' } else { 'No' }
    }
    $fresh = Get-Api "repos/$repo/pulls/$number"
    if ($fresh.head.sha -ne $pr.head.sha -or $fresh.state -ne $pr.state -or
        $fresh.merged -ne $pr.merged -or $fresh.title -cne $pr.title -or $fresh.body -cne $pr.body) {
        throw "PR #$number changed while reading; rerun."
    }

    $before = $row.Cells[2..9] -join ' | '
    $row.Cells[2] = Format-MigrationLibrary $row.Rule ($libs -join ', ')
    $row.Cells[3] = "[$number](https://github.com/$repo/pull/$number)"
    $row.Cells[4] = ([DateTimeOffset]$pr.created_at).ToString('yyyy-MM-dd')
    $row.Cells[5] = if ($pr.merged_at) { ([DateTimeOffset]$pr.merged_at).ToString('yyyy-MM-dd') } else { '-' }
    $row.Cells[6] = if ($approvals.Count) { $approvals -join ', ' } else { '-' }
    $row.Cells[7] = $feedbackFlag
    $row.Cells[8] = if ($pr.merged) { 'Done' } else { 'Doing' }
    $after = $row.Cells[2..9] -join ' | '
    if ($before -cne $after) {
        $changes.Add([pscustomobject]@{ Rule = $row.Rule; Before = $before; After = $after })
    }
}

$sorted = @(Get-OrderedRows $rows)
$index = 0
$newLines = foreach ($line in ($originalBody -split "`n")) {
    if ($line -match '^\| \d+ \|') {
        $sorted[$index].Cells[0] = ($index + 1).ToString()
        '| ' + ($sorted[$index].Cells -join ' | ') + ' |'
        $index++
    } else { $line }
}
$newBody = $newLines -join "`n"
$newRows = @(Get-Rows $newBody)
Assert-Table $newRows
foreach ($old in @(Get-Rows $originalBody)) {
    $new = $newRows | Where-Object Rule -eq $old.Rule
    if (-not $new) { throw "Lost rule: $($old.Rule)" }
    if ($old.Cells[8] -in @('Done', 'Hold on') -and ($old.Cells[1..8] -join '|') -cne ($new.Cells[1..8] -join '|')) {
        throw "Modified a protected Done/Hold on row: $($old.Rule)"
    }
    if ($old.Cells[9] -cne $new.Cells[9]) {
        throw "Modified preserved notes: $($old.Rule)"
    }
}
Assert-MigrationLayout $newRows
$changes | Format-List
$newRows | Group-Object { $_.Cells[8] } | Select-Object Name, Count | Format-Table
if ($newBody -ceq $originalBody) { Write-Output 'Already up to date; no edit needed.'; return }
if (-not $Apply) { Write-Output 'Preview only. Rerun with -Apply to update the issue.'; return }

$current = Get-Api "repos/$issueRepo/issues/$issueNumber"
if ($current.body -cne $originalBody) { throw 'Issue changed during refresh; rerun to avoid overwriting edits.' }
$tempFile = [IO.Path]::GetTempFileName()
try {
    [IO.File]::WriteAllText($tempFile, $newBody, [Text.UTF8Encoding]::new($false))
    & gh issue edit $issueNumber --repo $issueRepo --body-file $tempFile
    if ($LASTEXITCODE -ne 0) { throw 'Issue update failed.' }
    $saved = Get-Api "repos/$issueRepo/issues/$issueNumber"
    if ($saved.body -cne $newBody) { throw 'Saved issue differs from proposed body; inspect it before retrying.' }
    Assert-Table @(Get-Rows $saved.body)
    Assert-MigrationLayout @(Get-Rows $saved.body)
    Write-Output "Updated https://github.com/$issueRepo/issues/$issueNumber"
} finally {
    Remove-Item -LiteralPath $tempFile
}