################################################################################
##  File:  Install-VisualStudio.ps1
##  Desc:  Install Visual Studio
################################################################################
$vsToolset = (Get-ToolsetContent).visualStudio

if (Test-IsWin19) {
    # Install Visual Studio for Windows 19
    Install-VisualStudio `
        -Version $vsToolset.subversion `
        -Edition $vsToolset.edition `
        -Channel $vsToolset.channel `
        -RequiredComponents $vsToolset.workloads `
        -ExtraArgs "--allWorkloads --includeRecommended --remove Component.CPython3.x64"
}

if ( (Test-IsWin22) -or (Test-IsWin25) ) {
    # Install Visual Studio for Windows 22 and 25 with InstallChannel
    Install-VisualStudio `
        -Version $vsToolset.subversion `
        -Edition $vsToolset.edition `
        -Channel $vsToolset.channel `
        -InstallChannel $vsToolset.installChannel `
        -RequiredComponents $vsToolset.workloads `
        -ExtraArgs "--allWorkloads --includeRecommended --remove Component.CPython3.x64"
}

# Find the version of VS installed for this instance
# Only supports a single instance
$vsProgramData = Get-Item -Path "C:\ProgramData\Microsoft\VisualStudio\Packages\_Instances"
$instanceFolders = Get-ChildItem -Path $vsProgramData.FullName

if ($instanceFolders -is [array]) {
    throw "More than one instance installed"
}

# Updating content of MachineState.json file to disable autoupdate of VSIX extensions
$vsInstallRoot = (Get-VisualStudioInstance).InstallationPath
$newContent = '{"Extensions":[{"Key":"1e906ff5-9da8-4091-a299-5c253c55fdc9","Value":{"ShouldAutoUpdate":false}},{"Key":"Microsoft.VisualStudio.Web.AzureFunctions","Value":{"ShouldAutoUpdate":false}}],"ShouldAutoUpdate":false,"ShouldCheckForUpdates":false}'
Set-Content -Path "$vsInstallRoot\Common7\IDE\Extensions\MachineState.json" -Value $newContent

if (Test-IsWin19) {
    # Install Windows 10 SDK version 10.0.14393.795
    Install-Binary -Type EXE `
        -Url 'https://go.microsoft.com/fwlink/p/?LinkId=838916' `
        -InstallArgs @("/q", "/norestart", "/ceip off", "/features OptionId.WindowsSoftwareDevelopmentKit") `
        -ExpectedSubject 'CN=Microsoft Corporation, OU=MOPR, O=Microsoft Corporation, L=Redmond, S=Washington, C=US'

    # Install Windows 11 SDK version 10.0.22621.0
    Install-Binary -Type EXE `
        -Url 'https://go.microsoft.com/fwlink/p/?linkid=2196241' `
        -InstallArgs @("/q", "/norestart", "/ceip off", "/features OptionId.UWPManaged OptionId.UWPCPP OptionId.UWPLocalized OptionId.DesktopCPPx86 OptionId.DesktopCPPx64 OptionId.DesktopCPParm64") `
        -ExpectedSubject $(Get-MicrosoftPublisher)
}

if (Test-IsWin22) {
    # Install Windows 10 SDK version 10.0.17763
    Install-Binary -Type EXE `
    -Url 'https://go.microsoft.com/fwlink/p/?LinkID=2033908' `
    -InstallArgs @("/q", "/norestart", "/ceip off", "/features OptionId.UWPManaged OptionId.UWPCPP OptionId.UWPLocalized OptionId.DesktopCPPx86 OptionId.DesktopCPPx64 OptionId.DesktopCPParm64") `
    -ExpectedSubject $(Get-MicrosoftPublisher)
}

if (-not (Test-IsWin19)) {
     # Install Windows 11 SDK version 10.0.26100
     Install-Binary -Type EXE `
        -Url 'https://go.microsoft.com/fwlink/?linkid=2286561' `
        -InstallArgs @("/q", "/norestart", "/ceip off", "/features OptionId.UWPManaged OptionId.UWPCPP OptionId.UWPLocalized OptionId.DesktopCPPx86 OptionId.DesktopCPPx64 OptionId.DesktopCPParm64") `
        -ExpectedSubject $(Get-MicrosoftPublisher)
}

Invoke-PesterTests -TestFile "VisualStudio"