From 5114206866a8a9a74de1b71ffda22cee45c11ca4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sr=C4=91an=20Bo=C5=BEovi=C4=87?= <srdjan.bozovic@live.com>
Date: Wed, 1 Apr 2020 10:45:09 +0200
Subject: [PATCH] add environment option in script

---
 .../attach-jumpbox/attachJumpbox.ps1              | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/samples/manage/azure-sql-db-managed-instance/attach-jumpbox/attachJumpbox.ps1 b/samples/manage/azure-sql-db-managed-instance/attach-jumpbox/attachJumpbox.ps1
index 252e9c9b..ae02ea63 100644
--- a/samples/manage/azure-sql-db-managed-instance/attach-jumpbox/attachJumpbox.ps1
+++ b/samples/manage/azure-sql-db-managed-instance/attach-jumpbox/attachJumpbox.ps1
@@ -2,6 +2,7 @@ $parameters = $args[0]
 $scriptUrlBase = $args[1]
 
 $subscriptionId = $parameters['subscriptionId']
+$environmentName = $parameters['environmentName']
 $resourceGroupName = $parameters['resourceGroupName']
 $virtualMachineName = $parameters['virtualMachineName']
 $virtualNetworkName = $parameters['virtualNetworkName']
@@ -9,6 +10,11 @@ $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
@@ -59,11 +65,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
         }
@@ -180,7 +189,7 @@ function CalculateVpnClientAddressPoolPrefix {
 
 VerifyPSVersion
 EnsureAzModule
-EnsureLogin
+EnsureLogin -environmentName $environmentName
 SelectSubscriptionId -subscriptionId $subscriptionId
 
 $virtualNetwork = LoadVirtualNetwork -resourceGroupName $resourceGroupName -virtualNetworkName $virtualNetworkName