add environment parameter

Author: srdan-bozovic-msft

samples/manage/azure-sql-db-managed-instance/attach-vpn-gateway/attachVPNGateway.ps1

@@ -2,11 +2,17 @@ $parameters = $args[0]
 $scriptUrlBase = $args[1]
 
 $subscriptionId = $parameters['subscriptionId']
+$environmentName = $parameters['environmentName']
 $resourceGroupName = $parameters['resourceGroupName']
 $virtualNetworkName = $parameters['virtualNetworkName']
 $certificateNamePrefix = $parameters['certificateNamePrefix']
 $clientCertificatePassword = $parameters['clientCertificatePassword'] #used only when certificates are created using openssl
 
+if ($environmentName -eq '' -or ($null -eq $environmentName)) {
+    $environmentName = 'AzureCloud'
+    Write-Host "Environment: AzureCloud." -ForegroundColor Green
+}
+
 if ($clientCertificatePassword -eq '' -or ($null -eq $clientCertificatePassword)) {
     $clientCertificatePassword = 'S0m3Str0nGP@ssw0rd'
 }
@@ -52,11 +58,14 @@ function EnsureAzModule {
     }
 }
 
-function EnsureLogin () {
+function EnsureLogin {
+    param (
+        $environmentName
+    )
     $context = Get-AzContext
     If ($null -eq $context.Subscription) {
         Write-Host "Sign-in..."
-        If ($null -eq (Connect-AzAccount -ErrorAction SilentlyContinue -ErrorVariable Errors)) {
+        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
         }
@@ -220,9 +229,7 @@ function CreateCertificate() {
 
 VerifyPSVersion
 EnsureAzModule
-EnsureLogin
-SelectSubscriptionId -subscriptionId $subscriptionId
-
+EnsureLogin -environmentName $environmentNameSelectSubscriptionId -subscriptionId $subscriptionId
 $virtualNetwork = LoadVirtualNetwork -resourceGroupName $resourceGroupName -virtualNetworkName $virtualNetworkName
 
 $subnets = $virtualNetwork.Subnets.Name
@@ -265,4 +272,4 @@ $templateParameters = @{
 
 New-AzResourceGroupDeployment -ResourceGroupName $resourceGroupName -TemplateUri ($scriptUrlBase + '/azuredeploy.json?t=' + [DateTime]::Now.Ticks) -TemplateParameterObject $templateParameters
 
-Write-Host "Deployment completed."
+Write-Host "Deployment completed."