Update labeled PR CI checks (#12344)

.github/workflows/trigger-ubuntu-win-build.yml

@@ -14,34 +14,106 @@ defaults:
 jobs:
   trigger-workflow:
     runs-on: ubuntu-latest
+    outputs:
+      ci_workflow_run_id: ${{ steps.resolve.outputs.ci_workflow_run_id }}
+      ci_workflow_run_url: ${{ steps.resolve.outputs.ci_workflow_run_url }}
+    env:
+      CI_PR_TOKEN: ${{ secrets.CI_PR_TOKEN }}
+      PR_TITLE: ${{ github.event.pull_request.title }}
+      CI_REPO: ${{ vars.CI_REPO }}
     steps:
-      - name: Trigger Build workflow
+      - name: Checkout Code
-        env:
+        uses: actions/checkout@v4
-          CI_PR_TOKEN: ${{ secrets.CI_PR_TOKEN }}
-          PR_TITLE: ${{ github.event.pull_request.title }}
-          CI_PR: ${{ secrets.CI_REPO }}
-        run: |
-          $headers = @{
-            Authorization="Bearer $env:CI_PR_TOKEN"
-          }
 
-          # Private repository for builds
+      - name: Trigger Build workflow
-          $apiRepoUrl = "https://api.github.com/repos/$env:CI_PR"
+        run: |
+          Import-Module ./helpers/GitHubApi.psm1
+          $gitHubApi = Get-GithubApi -Repository "${env:CI_REPO}" -AccessToken "${env:CI_PR_TOKEN}"
 
           $eventType = "trigger-${{ inputs.image_type }}-build"
-          $body = @{
+          [string] $prGuid = New-Guid
-            event_type = $eventType;
+          $clientPayload = @{
-            client_payload = @{
+              pr_title                = "${env:PR_TITLE} - " + $prGuid
-              pr_title = "$env:PR_TITLE"
+              custom_repo             = "${{ github.event.pull_request.head.repo.full_name }}"
-              custom_repo = "${{ github.event.pull_request.head.repo.full_name }}"
               custom_repo_commit_hash = "${{ github.event.pull_request.head.sha }}"
+          }
+
+          $gitHubApi.DispatchWorkflow($eventType, $clientPayload)
+          "PR_GUID=$prGuid" | Out-File -Append -FilePath $env:GITHUB_ENV
+
+      - name: Resolve Workflow Run ID
+        id: resolve
+        run: |
+          Import-Module ./helpers/GitHubApi.psm1
+          $gitHubApi = Get-GithubApi -Repository "${env:CI_REPO}" -AccessToken "${env:CI_PR_TOKEN}"
+
+          $workflowFileName = $("{0}.yml" -f "${{ inputs.image_type }}").ToLower()
+          $WorkflowSearchPattern = "${env:PR_GUID}"
+
+          # It might take a few minutes for the action to start
+          $attempt = 1
+          do {
+            $workflowRuns = $gitHubApi.GetWorkflowRuns($WorkflowFileName).workflow_runs
+            $workflowRunId = ($workflowRuns | Where-Object {$_.display_title -match $WorkflowSearchPattern}).id | Select-Object -First 1
+
+            if (-not ([string]::IsNullOrEmpty($workflowRunId))) {
+              $workflowRun = $gitHubApi.GetWorkflowRun($workflowRunId)
+              Write-Host "Found the workflow run with ID $workflowRunId on attempt $attempt. Workflow run link: $($workflowRun.html_url)"
+              "ci_workflow_run_id=$workflowRunId" | Out-File -Append -FilePath $env:GITHUB_OUTPUT
+              "ci_workflow_run_url=$($workflowRun.html_url)" | Out-File -Append -FilePath $env:GITHUB_OUTPUT
+              break
             }
+
+            Write-Host "Workflow run for $WorkflowSearchPattern pattern not found on attempt $attempt."
+            $attempt += 1
+            Start-Sleep 30
+          } until ($attempt -eq 10)
+
+          if ([string]::IsNullOrEmpty($workflowRunId)) {
+              throw "Failed to find a workflow run for '$WorkflowSearchPattern'."
           }
 
-          $bodyString = $body | ConvertTo-Json
+  wait-completion:
+    runs-on: ubuntu-latest
+    needs: trigger-workflow
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v4
 
-          try {
+      - name: Wait for workflow completion
-            Invoke-WebRequest -Uri "$apiRepoUrl/dispatches" -Method Post -Headers $headers -Body $bodyString | Out-Null
+        env:
-          } catch {
+          CI_PR_TOKEN: ${{ secrets.CI_PR_TOKEN }}
-            throw "$($_.exception[0].message)"
+          CI_REPO: ${{ vars.CI_REPO }}
-          }
+        run: |
+          ./helpers/WaitWorkflowCompletion.ps1 `
+            -WorkflowRunId "${{ needs.trigger-workflow.outputs.ci_workflow_run_id }}" `
+            -Repository "${env:CI_REPO}" `
+            -AccessToken "${env:CI_PR_TOKEN}"
+
+      - name: Add Summary
+        if: always()
+        run: |
+          "# Test Partner Image" >> $env:GITHUB_STEP_SUMMARY
+          "| Key | Value |" >> $env:GITHUB_STEP_SUMMARY
+          "| :-----------: | :--------: |" >> $env:GITHUB_STEP_SUMMARY
+          "| Workflow Run | [Link](${{ needs.trigger-workflow.outputs.ci_workflow_run_url }}) |" >> $env:GITHUB_STEP_SUMMARY
+          "| Workflow Result | $env:CI_WORKFLOW_RUN_RESULT |" >> $env:GITHUB_STEP_SUMMARY
+          "  " >> $env:GITHUB_STEP_SUMMARY
+
+  cancel-workflow:
+    runs-on: ubuntu-latest
+    needs: [trigger-workflow, wait-completion]
+    if: cancelled()
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v4
+
+      - name: Cancel workflow
+        env:
+          CI_PR_TOKEN: ${{ secrets.CI_PR_TOKEN }}
+          CI_REPO: ${{ vars.CI_REPO }}
+        run: |
+          Import-Module ./helpers/GitHubApi.psm1
+
+          $gitHubApi = Get-GithubApi -Repository "${env:CI_REPO}" -AccessToken "${env:CI_PR_TOKEN}"
+          $gitHubApi.CancelWorkflowRun("${{ needs.trigger-workflow.outputs.ci_workflow_run_id }}")

helpers/GitHubApi.psm1

@@ -0,0 +1,97 @@
+class GithubApi
+{
+    [string] $Repository
+    [object] hidden $AuthHeader
+
+    GithubApi(
+        [string] $Repository,
+        [string] $AccessToken
+    ) {
+        $this.Repository = $Repository
+        $this.AuthHeader = $this.BuildAuth($AccessToken)
+    }
+
+    [object] hidden BuildAuth([string]$AccessToken) {
+        if ([string]::IsNullOrEmpty($AccessToken)) {
+            return $null
+        }
+        $base64AuthInfo = [Convert]::ToBase64String([Text.Encoding]::ASCII.GetBytes("'':${AccessToken}"))
+        return @{
+            Authorization = "Basic ${base64AuthInfo}"
+        }
+    }
+
+    [string] hidden BuildBaseUrl([string]$Repository, [string]$ApiPrefix) {
+        return "https://$ApiPrefix.github.com/repos/$Repository"
+    }
+
+    [object] GetWorkflowRuns([string]$WorkflowId) {
+        $url = "actions/workflows/$WorkflowId/runs"
+        $response =  $this.InvokeRestMethod($url, 'GET', $null, $null)
+        return $response
+    }
+
+    [object] GetWorkflowRun([string]$WorkflowRunId) {
+        $url = "actions/runs/$WorkflowRunId"
+        $response =  $this.InvokeRestMethod($url, 'GET', $null, $null)
+        return $response
+    }
+
+    [object] DispatchWorkflow([string]$EventType, [object]$EventPayload) {
+        $url = "dispatches"
+        $body = @{
+            "event_type"     = $EventType
+            "client_payload" = $EventPayload
+        } | ConvertTo-Json
+        $response =  $this.InvokeRestMethod($url, 'POST', $null, $body)
+        return $response
+    }
+
+    [object] CancelWorkflowRun([string]$workflowRunId) {
+        $url = "actions/runs/$workflowRunId/cancel"
+        $response =  $this.InvokeRestMethod($url, 'POST', $null, $null)
+        return $response
+    }
+
+    [string] hidden BuildUrl([string]$url, [string]$RequestParams, [string]$ApiPrefix) {
+        $baseUrl = $this.BuildBaseUrl($this.Repository, $ApiPrefix)
+        if ([string]::IsNullOrEmpty($RequestParams)) {
+            return "$($baseUrl)/$($url)"
+        } else {
+            return "$($baseUrl)/$($url)?$($requestParams)"
+        }
+    }
+
+    [object] hidden InvokeRestMethod(
+        [string] $url,
+        [string] $Method,
+        [string] $RequestParams,
+        [string] $body
+    ) {
+        $requestUrl = $this.BuildUrl($url, $RequestParams, "api")
+        $params = @{
+            Method      = $Method
+            ContentType = "application/json"
+            Uri         = $requestUrl
+            Headers     = @{}
+        }
+        if ($this.AuthHeader) {
+            $params.Headers += $this.AuthHeader
+        }
+        if (![string]::IsNullOrEmpty($body)) {
+            $params.Body = $body
+        }
+
+        $response = Invoke-RestMethod @params
+        return $response
+    }
+}
+
+function Get-GithubApi {
+    param (
+        [string] $Repository,
+        [string] $AccessToken
+    )
+
+    return [GithubApi]::New($Repository, $AccessToken)
+}

helpers/WaitWorkflowCompletion.ps1

@@ -0,0 +1,47 @@
+Param (
+    [Parameter(Mandatory)]
+    [string] $WorkflowRunId,
+    [Parameter(Mandatory)]
+    [string] $Repository,
+    [Parameter(Mandatory)]
+    [string] $AccessToken,
+    [int] $RetryIntervalSeconds = 300,
+    [int] $MaxRetryCount = 0
+)
+
+Import-Module (Join-Path $PSScriptRoot "GitHubApi.psm1")
+
+function Wait-ForWorkflowCompletion($WorkflowRunId, $RetryIntervalSeconds) {
+    do {
+        Start-Sleep -Seconds $RetryIntervalSeconds
+        $workflowRun = $gitHubApi.GetWorkflowRun($WorkflowRunId)
+    } until ($workflowRun.status -eq "completed")
+
+    return $workflowRun
+}
+
+$gitHubApi = Get-GithubApi -Repository $Repository -AccessToken $AccessToken
+
+$attempt = 1
+do {
+    $finishedWorkflowRun = Wait-ForWorkflowCompletion -WorkflowRunId $WorkflowRunId -RetryIntervalSeconds $RetryIntervalSeconds
+    Write-Host "Workflow run finished with result: $($finishedWorkflowRun.conclusion)"
+    if ($finishedWorkflowRun.conclusion -in ("success", "cancelled", "timed_out")) {
+        break
+    } elseif ($finishedWorkflowRun.conclusion -eq "failure") {
+        if ($attempt -le $MaxRetryCount) {
+            Write-Host "Workflow run will be restarted. Attempt $attempt of $MaxRetryCount"
+            $gitHubApi.ReRunFailedJobs($WorkflowRunId)
+            $attempt += 1
+        } else {
+            break
+        }
+    }
+} while ($true)
+
+Write-Host "Last result: $($finishedWorkflowRun.conclusion)."
+"CI_WORKFLOW_RUN_RESULT=$($finishedWorkflowRun.conclusion)" | Out-File -Append -FilePath $env:GITHUB_ENV
+
+if ($finishedWorkflowRun.conclusion -in ("failure", "cancelled", "timed_out")) {
+    exit 1
+}