sql-server-samples/samples/manage/azure-sql-db-managed-instance/attach-jumpbox/attachJumpbox.ps1

$parameters = $args[0]
$scriptUrlBase = $args[1]

$subscriptionId = $parameters['subscriptionId']
$environmentName = $parameters['environmentName']
$resourceGroupName = $parameters['resourceGroupName']
$virtualMachineName = $parameters['virtualMachineName']
$virtualNetworkName = $parameters['virtualNetworkName']
$managementSubnetName = $parameters['subnetName']
$administratorLogin = $parameters['administratorLogin']
$administratorLoginPassword = $parameters['administratorLoginPassword']

if ($environmentName -eq '' -or ($null -eq $environmentName)) {
    $environmentName = 'AzureCloud'
    Write-Host "Environment: AzureCloud." -ForegroundColor Green
}

if ($virtualMachineName -eq '' -or ($null -eq $virtualMachineName)) {
    $virtualMachineName = 'Jumpbox'
    Write-Host "VM Name: 'Jumpbox'." -ForegroundColor Green
}

if ($managementSubnetName -eq '' -or ($null -eq $managementSubnetName)) {
    $managementSubnetName = 'Management'
    Write-Host "Using subnet 'Management' to deploy jumpbox VM." -ForegroundColor Green
}

function VerifyPSVersion {
    Write-Host "Verifying PowerShell version."
    if ($PSVersionTable.PSEdition -eq "Desktop") {
        if (($PSVersionTable.PSVersion.Major -ge 6) -or 
        (($PSVersionTable.PSVersion.Major -eq 5) -and ($PSVersionTable.PSVersion.Minor -ge 1))) {
            Write-Host "PowerShell version verified." -ForegroundColor Green
        }
        else {
            Write-Host "You need to install PowerShell version 5.1 or heigher." -ForegroundColor Red
            Break;
        }
    }
    else {
        if ($PSVersionTable.PSVersion.Major -ge 6) {
            Write-Host "PowerShell version verified." -ForegroundColor Green
        }
        else {
            Write-Host "You need to install PowerShell version 6.0 or heigher." -ForegroundColor Red
            Break;
        }        
    }
}

function EnsureAzModule {
    Write-Host "Checking if Az module is imported."
    $module = Get-Module Az
    If ($null -eq $module) {
        try {
            Import-Module Az -ErrorAction Stop
            Write-Host "Module Az imported." -ForegroundColor Green
        }
        catch {
            Install-Module Az -AllowClobber
            Write-Host "Module Az installed." -ForegroundColor Green
        }
    } else {
        Write-Host "Module Az imported." -ForegroundColor Green        
    }
}

function EnsureLogin {
    param (
        $environmentName
    )
    $context = Get-AzContext
    If ($null -eq $context.Subscription) {
        Write-Host "Sign-in..."
        If ($null -eq (Connect-AzAccount -Environment $environmentName -ErrorAction SilentlyContinue -ErrorVariable Errors)) {
            Write-Host ("Sign-in failed: {0}" -f $Errors[0].Exception.Message) -ForegroundColor Red
            Break
        }
    }
    Write-Host "Sign-in successful." -ForegroundColor Green
}

function SelectSubscriptionId {
    param (
        $subscriptionId
    )
    Write-Host "Selecting subscription '$subscriptionId'..."
    $context = Get-AzContext
    If ($context.Subscription.Id -ne $subscriptionId) {
        Try {
            $currentSubscriptionId = $context.Subscription.Id
            Write-Host "Switching subscription $currentSubscriptionId to '$subscriptionId'." -ForegroundColor Green
            Select-AzSubscription -SubscriptionId $subscriptionId -ErrorAction Stop | Out-null
        }
        Catch {
            Write-Host "Subscription selection failed: $_" -ForegroundColor Red
            Break
        }
    }
    Write-Host "Subscription selected." -ForegroundColor Green
}

function LoadVirtualNetwork {
    param (
        $resourceGroupName,
        $virtualNetworkName
    )
    Write-Host("Loading virtual network '{0}' in resource group '{1}'." -f $virtualNetworkName, $resourceGroupName)
    $virtualNetwork = Get-AzVirtualNetwork -ResourceGroupName $resourceGroupName -Name $virtualNetworkName -ErrorAction SilentlyContinue
    $id = $virtualNetwork.Id
    If ($null -ne $id) {
        Write-Host "Virtual network with id $id is loaded." -ForegroundColor Green
        If ($virtualNetwork.VirtualNetworkPeerings.Count -gt 0) {
            Write-Host "Virtual network is loaded, but it should not have peerings." -ForegroundColor Red
        }
        return $virtualNetwork
    }
    else {
        Write-Host "Virtual network $virtualNetworkName cannot be found." -ForegroundColor Red
        Break
    }
}

function SetVirtualNetwork {
    param($virtualNetwork)

    Write-Host "Applying changes to the virtual network."
    Try {
        Set-AzVirtualNetwork -VirtualNetwork $virtualNetwork -ErrorAction Stop | Out-Null
    }
    Catch {
        Write-Host "Failed to configure Virtual Network: $_" -ForegroundColor Red
    }
}

function ConvertCidrToUint32Array {
    param($cidrRange)
    $cidrRangeParts = $cidrRange.Split("/")
    $ipParts = $cidrRangeParts[0].Split(".")
    $ipnum = ([Convert]::ToUInt32($ipParts[0]) -shl 24) -bor `
    ([Convert]::ToUInt32($ipParts[1]) -shl 16) -bor `
    ([Convert]::ToUInt32($ipParts[2]) -shl 8) -bor `
        [Convert]::ToUInt32($ipParts[3])

    $maskbits = [System.Convert]::ToInt32($cidrRangeParts[1])
    $mask = 0xffffffff
    $mask = $mask -shl (32 - $maskbits)
    $ipstart = $ipnum -band $mask
    $ipend = $ipnum -bor ($mask -bxor 0xffffffff)
    return @($ipstart, $ipend)
}

function ConvertUInt32ToIPAddress {
    param($uint32IP)
    $v1 = $uint32IP -band 0xff
    $v2 = ($uint32IP -shr 8) -band 0xff
    $v3 = ($uint32IP -shr 16) -band 0xff
    $v4 = ($uint32IP -shr 24)
    return "$v4.$v3.$v2.$v1"
}

function CalculateNextAddressPrefix {
    param($virtualNetwork, $prefixLength)
    Write-Host "Calculating address prefix with length $prefixLength..."
    $startIPAddress = 0
    ForEach ($addressPrefix in $virtualNetwork.AddressSpace.AddressPrefixes) {
        $endIPAddress = (ConvertCidrToUint32Array $addressPrefix)[1]
        If ($endIPAddress -gt $startIPAddress) {
            $startIPAddress = $endIPAddress
        }
    }
    $startIPAddress += 1
    $addressPrefixResult = (ConvertUInt32ToIPAddress $startIPAddress) + "/" + $prefixLength
    Write-Host "Using address prefix $addressPrefixResult." -ForegroundColor Green
    return $addressPrefixResult
}

function CalculateVpnClientAddressPoolPrefix {
    param($gatewaySubnetPrefix)
    Write-Host "Calculating VPN client address pool prefix."
    If ($gatewaySubnetPrefix.StartsWith("10.")) {
        return "192.168.0.0/24"
    }
    else {
        return "172.16.0.0/24"
    }

}

VerifyPSVersion
EnsureAzModule
EnsureLogin -environmentName $environmentName
SelectSubscriptionId -subscriptionId $subscriptionId

$virtualNetwork = LoadVirtualNetwork -resourceGroupName $resourceGroupName -virtualNetworkName $virtualNetworkName

$subnets = $virtualNetwork.Subnets.Name

If ($false -eq $subnets.Contains($managementSubnetName)) {
    Write-Host "$managementSubnetName is not one of the subnets in $subnets" -ForegroundColor Yellow
    $managementSubnetPrefix = CalculateNextAddressPrefix $virtualNetwork 28
    Write-Host "Creating subnet $managementSubnetName ($managementSubnetPrefix) in the virtual network ..." -ForegroundColor Green

    $virtualNetwork.AddressSpace.AddressPrefixes.Add($managementSubnetPrefix)
    Add-AzVirtualNetworkSubnetConfig -Name $managementSubnetName -VirtualNetwork $virtualNetwork -AddressPrefix $managementSubnetPrefix | Out-Null

    SetVirtualNetwork $virtualNetwork
    Write-Host "Added subnet $managementSubnetName into virtual network." -ForegroundColor Green
}
else {
    Write-Host "The subnet $managementSubnetName exists in the virtual network." -ForegroundColor Green
}

Write-Host

# Start the deployment
Write-Host "Starting deployment..."
Write-Host "Deployment will take about 20m." -ForegroundColor Yellow

$templateParameters = @{
    location                   = $virtualNetwork.Location
    virtualNetworkName         = $virtualNetworkName
    managementSubnetName       = $managementSubnetName
    virtualMachineName         = $virtualMachineName
    administratorLogin         = $administratorLogin
    administratorLoginPassword = $administratorLoginPassword
}

New-AzResourceGroupDeployment -ResourceGroupName $resourceGroupName -TemplateUri ($scriptUrlBase + '/azuredeploy.json?t=' + [DateTime]::Now.Ticks) -TemplateParameterObject $templateParameters

Write-Host "Deployment completed."