[CmdletBinding()]

param(

    [Parameter(Mandatory=$true)]
    [string]
    $SQLServerVersion,

    [Parameter()]
    [string]
    $SQL2016Media,

    [Parameter()]
    [string]
    $SQL2017Media,

    [Parameter()]
    [string]
    $SQL2019Media

)

try {
    Start-Transcript -Path C:\AWSQuickstart\log\DownloadSQLEE.ps1.txt -Append

    $ErrorActionPreference = "Stop"

    $DestPath = "C:\SQLMedia"
    New-Item "$DestPath" -Type directory -Force

    $ssmssource = "https://download.microsoft.com/download/3/C/7/3C77BAD3-4E0F-4C6B-84DD-42796815AFF6/SSMS-Setup-ENU.exe"

    if ($SQLServerVersion -eq "2016") {
        $source = $SQL2016Media
    }
    elseif ($SQLServerVersion -eq "2017") {
        $source = $SQL2017Media
    }
    else {
        $source = $SQL2019Media
    }

    $tries = 5
    while ($tries -ge 1) {
        try {
            Start-BitsTransfer -Source $source -Destination "$DestPath" -ErrorAction Stop
            Start-BitsTransfer -Source $ssmssource -Destination "$DestPath" -ErrorAction Stop
            break
        } 
        catch {
            $tries--
            Write-Verbose "Exception:"
            Write-Verbose "$_"
            if ($tries -lt 1) {
                throw $_
            }
            else {
                Write-Verbose "Failed download. Retrying again in 5 seconds"
                Start-Sleep 5
            }
        }
    }
}
catch {
    Write-Verbose "$($_.exception.message)@ $(Get-Date)"
}