runner-images/images/win/scripts/Installers/Validate-Toolset.ps1

################################################################################
##  File:  Validate-Toolset.ps1
##  Team:  CI-Build
##  Desc:  Validate Toolset
################################################################################

function Run-ExecutableTests {
    param (
        [Parameter(Mandatory)] [string[]] $Executables,
        [Parameter(Mandatory)] [string] $ToolPath
    )
    $versionCommand = $Executables["command"]
    foreach ($executable in $Executables["tools"]) {
        $executablePath = Join-Path $ToolPath $executable

        Write-Host "Check $executable..."
        if (Test-Path $executablePath) {
            Write-Host "$executable is successfully installed: $(& $executablePath $versionCommand)"
        } else {
            Write-Host "$executablePath is not installed!"
            exit 1
        }
    }
}

function Validate-SystemDefaultTool {
    param (
        [Parameter(Mandatory)] [string] $ToolName,
        [Parameter(Mandatory)] [string] $ExpectedVersion
    )

    $versionCommand = $toolsExecutables[$ToolName]["command"]
    $binName = $ToolName.ToLower()

    # Check if tool on path
    if (Get-Command -Name $binName) {
        $versionOnPath = $(& $binName $versionCommand 2>&1) | Select-String -Pattern ".*(\d+\.\d+[\.\d+]+)"

        # Check if version is correct
        if ($versionOnPath.matches.Groups[1].Value -notlike $ExpectedVersion) {
            Write-Error "$ToolName $ExpectedVersion is not in the PATH"
            exit 1
        }

        Write-Host "$ToolName $versionOnPath on path"
    } else {
        Write-Host "$ToolName is not on path"
        exit 1
    }
}

$ErrorActionPreference = "Stop"

# Define executables for cached tools
$toolsExecutables = @{
    Python = @{
        tools = @("python.exe", "Scripts\pip.exe")
        command = "--version"
    }
    node = @{
        tools = @("node.exe", "npm")
        command = "--version"
    }
    PyPy = @{
        tools = @("python.exe", "Scripts\pip.exe")
        command = "--version"
    }
    go = @{
        tools = @("bin\go.exe")
        command = "version"
    }
}

# Get toolcache content from toolset
$tools = Get-ToolsetContent | Select-Object -ExpandProperty toolcache

foreach($tool in $tools) {
    # Get executables for current tool
    $toolExecs = $toolsExecutables[$tool.name]

    foreach ($version in $tool.versions) {
        $foundVersionArchPath = Get-ToolsetToolFullPath -Name $tool.name -Version $version -Arch $tool.arch

        if ($toolExecs) {
            Write-Host "Run validation test for $($tool.name)($($tool.arch)) $($foundVersion.name) executables..."
            Run-ExecutableTests -Executables $toolExecs -ToolPath $foundVersionArchPath
        }
    }

    if (-not ([string]::IsNullOrEmpty($tool.default))) {
        Write-Host "Validate system default $($tool.name)($($tool.arch)) $($tool.default)..."
        Validate-SystemDefaultTool -ToolName $tool.name -ExpectedVersion $tool.default
    }
}