test: add timeouts to winget arm64 probe

Author: Wangnov

.github/workflows/probe-winget-arm64.yml

@@ -45,14 +45,100 @@ jobs:
           "winget_available=true" | Add-Content -Encoding utf8 -Path $resultPath
           $wingetCommand | Format-List * | Out-File -Encoding utf8 -FilePath (Join-Path $logDir "winget-command.txt")
 
-          & winget --info *>&1 |
-            Tee-Object -FilePath (Join-Path $logDir "winget-info.log")
+          function Write-ProbeLog {
+            param(
+              [Parameter(Mandatory = $true)]
+              [string] $Path,
+              [Parameter(Mandatory = $true)]
+              [string] $Message
+            )
+
+            $Message | Add-Content -Encoding utf8 -Path $Path
+            Write-Host $Message
+          }
+
+          function Invoke-WinGetLogged {
+            param(
+              [Parameter(Mandatory = $true)]
+              [string[]] $Arguments,
+              [Parameter(Mandatory = $true)]
+              [string] $LogName,
+              [Parameter(Mandatory = $true)]
+              [int] $TimeoutSeconds
+            )
+
+            $logPath = Join-Path $logDir $LogName
+            $startedAt = Get-Date
+            Write-ProbeLog -Path $logPath -Message "command=winget $($Arguments -join ' ')"
+            Write-ProbeLog -Path $logPath -Message "startedAt=$($startedAt.ToUniversalTime().ToString("o"))"
+            Write-ProbeLog -Path $logPath -Message "timeoutSeconds=$TimeoutSeconds"
+
+            try {
+              $startInfo = [System.Diagnostics.ProcessStartInfo]::new()
+              $startInfo.FileName = $wingetCommand.Source
+              foreach ($argument in $Arguments) {
+                [void] $startInfo.ArgumentList.Add($argument)
+              }
+              $startInfo.RedirectStandardOutput = $true
+              $startInfo.RedirectStandardError = $true
+              $startInfo.UseShellExecute = $false
+              $startInfo.CreateNoWindow = $true
+
+              $process = [System.Diagnostics.Process]::new()
+              $process.StartInfo = $startInfo
+              [void] $process.Start()
+
+              $stdoutTask = $process.StandardOutput.ReadToEndAsync()
+              $stderrTask = $process.StandardError.ReadToEndAsync()
+              $timedOut = -not $process.WaitForExit($TimeoutSeconds * 1000)
+
+              if ($timedOut) {
+                Write-ProbeLog -Path $logPath -Message "timedOut=true"
+                try {
+                  $process.Kill($true)
+                } catch {
+                  Write-ProbeLog -Path $logPath -Message "killException=$($_.Exception.Message)"
+                }
+                $process.WaitForExit()
+              }
+
+              $stdout = $stdoutTask.GetAwaiter().GetResult()
+              $stderr = $stderrTask.GetAwaiter().GetResult()
+
+              if (-not [string]::IsNullOrEmpty($stdout)) {
+                Write-ProbeLog -Path $logPath -Message "stdout:"
+                Write-ProbeLog -Path $logPath -Message $stdout
+              }
+
+              if (-not [string]::IsNullOrEmpty($stderr)) {
+                Write-ProbeLog -Path $logPath -Message "stderr:"
+                Write-ProbeLog -Path $logPath -Message $stderr
+              }
+
+              $elapsedSeconds = [int] ((Get-Date) - $startedAt).TotalSeconds
+              Write-ProbeLog -Path $logPath -Message "elapsedSeconds=$elapsedSeconds"
+
+              if ($timedOut) {
+                Write-ProbeLog -Path $logPath -Message "exitCode=124"
+                return 124
+              }
+
+              Write-ProbeLog -Path $logPath -Message "exitCode=$($process.ExitCode)"
+              return $process.ExitCode
+            } catch {
+              Write-ProbeLog -Path $logPath -Message "exception=$($_.Exception.Message)"
+              return 125
+            }
+          }
+
+          $infoExitCode = Invoke-WinGetLogged -Arguments @("--info") -LogName "winget-info.log" -TimeoutSeconds 60
+          "infoExitCode=$infoExitCode" | Add-Content -Encoding utf8 -Path $resultPath
 
-          & winget source list *>&1 |
-            Tee-Object -FilePath (Join-Path $logDir "winget-source-list.log")
+          $sourceListExitCode = Invoke-WinGetLogged -Arguments @("source", "list") -LogName "winget-source-list.log" -TimeoutSeconds 120
+          "sourceListExitCode=$sourceListExitCode" | Add-Content -Encoding utf8 -Path $resultPath
 
-          & winget source update *>&1 |
-            Tee-Object -FilePath (Join-Path $logDir "winget-source-update.log")
+          $sourceUpdateExitCode = Invoke-WinGetLogged -Arguments @("source", "update") -LogName "winget-source-update.log" -TimeoutSeconds 180
+          "sourceUpdateExitCode=$sourceUpdateExitCode" | Add-Content -Encoding utf8 -Path $resultPath
 
           $downloadArgs = @(
             "download",
@@ -70,9 +156,7 @@ jobs:
           "winget $($downloadArgs -join ' ')" |
             Set-Content -Encoding utf8 -Path (Join-Path $logDir "winget-download-command.txt")
 
-          & winget @downloadArgs *>&1 |
-            Tee-Object -FilePath (Join-Path $logDir "winget-download-arm64.log")
-          $downloadExitCode = $LASTEXITCODE
+          $downloadExitCode = Invoke-WinGetLogged -Arguments $downloadArgs -LogName "winget-download-arm64.log" -TimeoutSeconds 600
           "exitCode=$downloadExitCode" | Add-Content -Encoding utf8 -Path $resultPath
 
           $files = @(Get-ChildItem -LiteralPath $outputDir -Recurse -File -ErrorAction SilentlyContinue)