update environment variables after installing Swift on Windows

Author: matthewbastien

.github/workflows/scripts/windows/swift/install-swift.ps1

@@ -38,5 +38,50 @@ function Install-Swift {
         Write-Host ('FAILED ({0})' -f $Process.ExitCode)
         exit 1
     }
+    Write-Host -NoNewLine 'Adding Swift to PATH ... '
+    # Older toolchains modify the Machine PATH to include \Library\Developer\Toolchains
+    foreach ($Item in [Environment]::GetEnvironmentVariables('Machine')['Path'].Split(';')) {
+        if ($Item.Contains('Swift') -or $Item.Contains('Library\Developer\Toolchains')) {
+            GitHubActions-AddPath -Item $Item
+        }
+    }
+    # Newer toolchains modify the User PATH to include a path to a Swift directory
+    foreach ($Item in [Environment]::GetEnvironmentVariables('User')['Path'].Split(';')) {
+        if ($Item.Contains('Swift')) {
+            GitHubActions-AddPath -Item $Item
+        }
+    }
+    # Find and export the SDKROOT environment variable
+    $MachineSDKROOT = [Environment]::GetEnvironmentVariables('Machine')['SDKROOT']
+    $UserSDKROOT = [Environment]::GetEnvironmentVariables('User')['SDKROOT']
+    if ($MachineSDKROOT) {
+        GitHubActions-ExportVariable -Name "SDKROOT" -Value $MachineSDKROOT
+    } elseif ($UserSDKROOT) {
+        GitHubActions-ExportVariable -Name "SDKROOT" -Value $UserSDKROOT
+    }
+    Write-Host 'SUCCESS'
     Remove-Item -Force installer.exe
-}
+}
+
+function GitHubActions-AddPath {
+    param (
+        [string]$Item
+    )
+    $FilePath = $env:GITHUB_PATH
+    if ($FilePath) {
+        Add-Content -Path "$FilePath" -Value "$Item`r`n"
+    }
+    $env:Path = $Item + ";" + $env:Path
+}
+
+function GitHubActions-ExportVariable {
+    param (
+        [string]$Name,
+        [string]$Value
+    )
+    $FilePath = $env:GITHUB_ENV
+    if ($FilePath) {
+        Add-Content -Path "$FilePath" -Value "$Name=$Value`r`n"
+    }
+    Set-Content env:\$Name $Value
+}

.github/workflows/swift_package_test.yml

@@ -731,7 +731,4 @@ jobs:
       - name: Build / Test
         timeout-minutes: 60
         if: ${{ !inputs.enable_windows_docker }}
-        run: |
-          Import-Module $env:ChocolateyInstall\helpers\chocolateyProfile.psm1
-          RefreshEnv
-          powershell.exe -NoLogo -File $env:TEMP\test-script\run.ps1; exit $LastExitCode
+        run: powershell.exe -NoLogo -File $env:TEMP\test-script\run.ps1; exit $LastExitCode