################################################################################
##  File:  Install-DotnetSDK.ps1
##  Desc:  Install all released versions of the dotnet sdk and populate package
##         cache. Should run after VS and Node
##  Supply chain security: checksum validation
################################################################################

# Set environment variables
[System.Environment]::SetEnvironmentVariable("DOTNET_MULTILEVEL_LOOKUP", "0", "Machine")
[System.Environment]::SetEnvironmentVariable("DOTNET_NOLOGO", "1", "Machine")
[System.Environment]::SetEnvironmentVariable("DOTNET_SKIP_FIRST_TIME_EXPERIENCE", "1", "Machine")

[Net.ServicePointManager]::SecurityProtocol = [Net.ServicePointManager]::SecurityProtocol -bor "Tls12"

#region "Functions"
function Get-SDKVersionsToInstall (
    $DotnetVersion
) {
    $metadataJsonUri = "https://dotnetcli.blob.core.windows.net/dotnet/release-metadata/${DotnetVersion}/releases.json"
    $currentReleases = Invoke-DownloadWithRetry $metadataJsonUri | Get-Item | Get-Content | ConvertFrom-Json
    # filtering out the preview/rc releases
    $currentReleases = $currentReleases.'releases' | Where-Object { !$_.'release-version'.Contains('-') }

    $sdks = @()
    ForEach ($release in $currentReleases) {
        $sdks += $release.'sdk'
        $sdks += $release.'sdks'
    }

    return $sdks.version `
        | Sort-Object { [Version] $_ } -Unique `
        | Group-Object { $_.Substring(0, $_.LastIndexOf('.') + 2) } `
        | Foreach-Object { $_.Group[-1] }
}

function Invoke-Warmup (
    $SdkVersion
) {
    # warm up dotnet for first time experience
    $projectTypes = @('console', 'mstest', 'web', 'mvc', 'webapi')
    $projectTypes | ForEach-Object {
        $template = $_
        $projectPath = Join-Path -Path C:\temp -ChildPath $template
        New-Item -Path $projectPath -Force -ItemType Directory
        Push-Location -Path $projectPath
        & $env:ProgramFiles\dotnet\dotnet.exe new globaljson --sdk-version "$sdkVersion"
        & $env:ProgramFiles\dotnet\dotnet.exe new $template
        Pop-Location
        Remove-Item $projectPath -Force -Recurse
    }
}

function InstallSDKVersion (
    $SdkVersion,
    $dotnetVersion,
    $Warmup
) {
    if (!(Test-Path -Path "C:\Program Files\dotnet\sdk\$sdkVersion")) {
        Write-Host "Installing dotnet $sdkVersion"
        $ZipPath = [System.IO.Path]::combine([System.IO.Path]::GetTempPath(), [System.IO.Path]::GetRandomFileName())
        .\dotnet-install.ps1 -Version $sdkVersion -InstallDir $(Join-Path -Path $env:ProgramFiles -ChildPath 'dotnet') -ZipPath $ZipPath -KeepZip

        #region Supply chain security
        $distributorFileHash = (Invoke-RestMethod -Uri "https://dotnetcli.blob.core.windows.net/dotnet/release-metadata/$dotnetVersion/releases.json").releases.sdks.Where({ $_.version -eq $SdkVersion }).files.Where({ $_.name -eq 'dotnet-sdk-win-x64.zip' }).hash
        $localFileHash = (Get-FileHash -Path $ZipPath -Algorithm 'SHA512').Hash

        Use-ChecksumComparison -LocalFileHash $localFileHash -DistributorFileHash $distributorFileHash
        #endregion
    } else {
        Write-Host "Sdk version $sdkVersion already installed"
    }

    if ($Warmup) {
        Invoke-Warmup -SdkVersion $SdkVersion
    }
}

function InstallAllValidSdks() {
    # Consider all channels except preview/eol channels.
    # Sort the channels in ascending order
    $dotnetToolset = (Get-ToolsetContent).dotnet
    $dotnetVersions = $dotnetToolset.versions
    $warmup = $dotnetToolset.warmup

    # Download installation script.
    $installationName = "dotnet-install.ps1"
    $installationUrl = "https://dot.net/v1/${installationName}"
    Invoke-DownloadWithRetry -Url $installationUrl -Path ".\$installationName"

    ForEach ($dotnetVersion in $dotnetVersions)
    {
        $sdkVersionsToInstall = Get-SDKVersionsToInstall -DotnetVersion $dotnetVersion
        
        ForEach ($sdkVersion in $sdkVersionsToInstall)
        {
            InstallSDKVersion -SdkVersion $sdkVersion -DotnetVersion $dotnetVersion -Warmup $warmup
        }
    }
}

function InstallTools()
{
    $dotnetTools = (Get-ToolsetContent).dotnet.tools

    ForEach ($dotnetTool in $dotnetTools)
    {
        dotnet tool install $($dotnetTool.name) --tool-path "C:\Users\Default\.dotnet\tools" --add-source https://api.nuget.org/v3/index.json | Out-Null
    }
}

function RunPostInstallationSteps()
{
    # Add dotnet to PATH
    Add-MachinePathItem "C:\Program Files\dotnet"

    # Remove NuGet Folder
    $nugetPath = "$env:APPDATA\NuGet"
    if (Test-Path $nugetPath) {
        Remove-Item -Path $nugetPath -Force -Recurse
    }

    # Generate and copy new NuGet.Config config
    dotnet nuget list source | Out-Null
    Copy-Item -Path $nugetPath -Destination C:\Users\Default\AppData\Roaming -Force -Recurse

    # Add %USERPROFILE%\.dotnet\tools to USER PATH
    Add-DefaultPathItem "%USERPROFILE%\.dotnet\tools"
}
#endregion

InstallAllValidSdks
RunPostInstallationSteps
InstallTools

Invoke-PesterTests -TestFile "DotnetSDK"