eng/common/scripts/stress-testing/find-all-stress-packages.ps1

param(
    [string]$searchDirectory = '.',
    [hashtable]$filters = @{}
)

class StressTestPackageInfo {
    [string]$Namespace
    [string]$Directory
    [string]$ReleaseName
    [string]$Dockerfile
    [string]$DockerBuildDir
    [string]$Deployer
}

. $PSScriptRoot/../job-matrix/job-matrix-functions.ps1
. $PSScriptRoot/generate-scenario-matrix.ps1

function FindStressPackages(
    [string]$directory,
    [hashtable]$filters = @{},
    [switch]$CI,
    [string]$namespaceOverride,
    [string]$MatrixSelection,
    [Parameter(Mandatory=$False)][string]$MatrixFileName,
    [Parameter(Mandatory=$False)][string]$MatrixDisplayNameFilter,
    [Parameter(Mandatory=$False)][array]$MatrixFilters,
    [Parameter(Mandatory=$False)][array]$MatrixReplace,
    [Parameter(Mandatory=$False)][array]$MatrixNonSparseParameters
) {
    # Bare minimum filter for stress tests
    $filters['stressTest'] = 'true'
    $packages = @()
    $chartFiles = Get-ChildItem -Recurse -Filter 'Chart.yaml' $directory
    Write-Host "Found chart files:"
    Write-Host ($chartFiles -join "`n")

    if (!$MatrixFileName) {
        $MatrixFileName = 'scenarios-matrix.yaml'
    }

    foreach ($chartFile in $chartFiles) {
        $chart = ParseChart $chartFile

        if (!(matchesAnnotations $chart $filters)) {
            Write-Host "Skipping chart file '$chartFile'"
            continue
        }

        VerifyAddonsVersion $chart $chartFile

        # Default to "sparse" matrix selection type, unless a default is specified
        # in Chart.yaml annotations, or an override is passed in from the command line
        $selection = if ($MatrixSelection) {
            $MatrixSelection
        } elseif ($chart['annotations'] -and $chart['annotations']['matrixSelection']) {
            $chart['annotations']['matrixSelection']
        } else {
            "sparse"
        }

        $matrixFilePath = (Join-Path $chartFile.Directory.FullName $MatrixFileName)
        if (Test-Path $matrixFilePath) {
            GenerateScenarioMatrix `
                -matrixFilePath $matrixFilePath `
                -Selection $selection `
                -DisplayNameFilter $MatrixDisplayNameFilter `
                -Filters $MatrixFilters `
                -Replace $MatrixReplace `
                -NonSparseParameters $MatrixNonSparseParameters
        }

        $packages += NewStressTestPackageInfo `
                        -chart $chart `
                        -chartFile $chartFile `
                        -CI:$CI `
                        -namespaceOverride $namespaceOverride
    }

    return $packages
}

function ParseChart([string]$chartFile) {
    return ConvertFrom-Yaml (Get-Content -Raw $chartFile)
}

function MatchesAnnotations([hashtable]$chart, [hashtable]$filters) {
    foreach ($filter in $filters.GetEnumerator()) {
        if (!$chart["annotations"] -or $chart["annotations"][$filter.Key] -notmatch $filter.Value) {
            return $false
        }
    }

    return $true
}

function VerifyAddonsVersion([hashtable]$chart, [string]$chartFile) {
    foreach ($dependency in $chart.dependencies) {
        if ($dependency.name -eq "stress-test-addons" -and
            $dependency.version -like '0.1.*' -or
            $dependency.version -like '^0.1.*' -or
            $dependency.version -like '0.2.*' -or
            $dependency.version -like '^0.2.*') {
            throw "The stress-test-addons version in use for '$chartFile' is $($dependency.version), use the version ~0.3.0 to avoid breaking changes"
        }
    }
}

function GetUsername() {
    # Check GITHUB_USER for users in codespaces environments, since the default user is `codespaces` and
    # we would like to avoid namespace overlaps for different codespaces users.
    $stressUser = $env:GITHUB_USER ?? $env:USER ?? $env:USERNAME
    # Remove spaces, underscores, etc. that may be in $namespace.
    # Value must be a valid RFC 1123 DNS label: https://kubernetes.io/docs/concepts/overview/working-with-objects/names/#dns-label-names
    $stressUser = $stressUser -replace '_|\W', '-'

    return $stressUser.ToLower()
}

function NewStressTestPackageInfo(
    [hashtable]$chart,
    [System.IO.FileInfo]$chartFile,
    [switch]$CI,
    [object]$namespaceOverride
) {
    $namespace = if ($namespaceOverride) {
        $namespaceOverride
    } elseif ($CI) {
        $chart.annotations.namespace
    } else {
        GetUsername
    }

    return [StressTestPackageInfo]@{
        Namespace = $namespace.ToLower()
        Directory = $chartFile.DirectoryName
        ReleaseName = $chart.name
        Dockerfile = "dockerfile" -in $chart.annotations.keys ? $chart.annotations.dockerfile : $null
        DockerBuildDir = "dockerbuilddir" -in $chart.annotations.keys ? $chart.annotations.dockerbuilddir : $null
    }
}

# Don't call functions when the script is being dot sourced
if ($MyInvocation.InvocationName -ne ".") {
    FindStressPackages $searchDirectory $filters
}