Allow the arm baseURI to be set dynamically (#920)

* Allow the arm baseURI to be set dynamically

* Defensive programming

src/Microsoft.SqlTools.ResourceProvider.Core/Authentication/IAzureUserAccount.cs

@@ -4,6 +4,7 @@
 
 using System;
 using System.Collections.Generic;
+using Microsoft.SqlTools.ResourceProvider.Core.Contracts;
 
 namespace Microsoft.SqlTools.ResourceProvider.Core.Authentication
 {
@@ -12,6 +13,11 @@ namespace Microsoft.SqlTools.ResourceProvider.Core.Authentication
     /// </summary>
     public interface IAzureUserAccount : IEquatable<IAzureUserAccount>, IUserAccount
     {
+        Account UnderlyingAccount
+        {
+            get;
+            set;
+        }
         /// <summary>
         /// User Account Display Info
         /// </summary>

src/Microsoft.SqlTools.ResourceProvider.Core/Contracts/Account.cs

@@ -58,9 +58,57 @@ namespace Microsoft.SqlTools.ResourceProvider.Core.Contracts
             get;
             set;
         }
+        /// <summary>
+        /// Information about the auth provider
+        /// </summary>
+        public ProviderSettings ProviderSettings;
 
     }
 
+    public class ProviderSettings
+    {
+        /// <summary>
+        /// Display name of the provider 
+        /// </summary>
+        public string DisplayName;
+
+        /// <summary>
+        /// ID of the provider
+        /// </summary>    
+        public string Id;
+        /// <summary>
+        /// Settings for the provider itself
+        /// </summary>
+        public ProviderSettingsObject Settings;
+    }
+
+    public class ProviderSettingsObject
+    {
+        public ResourceSetting ArmResource;
+        public ResourceSetting GraphResource;
+        public ResourceSetting OssRdbmsResource;
+        public ResourceSetting SqlResource;
+
+        /// <summary>
+        /// Actual sign in link
+        /// </summary>
+        public string Host;
+        /// <summary>
+        /// ClientID used
+        /// </summary>
+        public string ClientId;
+    }
+
+    public class ResourceSetting
+    {
+
+        /// <summary>
+        /// Endpoint of the resource
+        /// </summary>
+        public string Endpoint;
+        public string Id;
+    }
+
     /// <summary>
     /// Represents a key that identifies an account.
     /// </summary>
@@ -87,7 +135,6 @@ namespace Microsoft.SqlTools.ResourceProvider.Core.Contracts
         /// <summary>
         /// A display name that offers context for the account, such as "Contoso".
         /// </summary>
-            
         public string ContextualDisplayName { get; set; }
 
         // Note: ignoring ContextualLogo as it's not needed

src/Microsoft.SqlTools.ResourceProvider.DefaultImpl/AzureAuthenticationManager.cs

@@ -97,6 +97,7 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
             userAccount.DisplayInfo = ToDisplayInfo(account);
             userAccount.NeedsReauthentication = account.IsStale;
             userAccount.AllTenants = ProcessTenants(accountTokenWrapper, account);
+            userAccount.UnderlyingAccount = account;
             return userAccount;
         }
 

src/Microsoft.SqlTools.ResourceProvider.DefaultImpl/AzureResourceManager.cs

@@ -55,12 +55,26 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
             CommonUtil.CheckForNull(subscriptionContext, "subscriptionContext");
             try
             {
+                string armEndpoint = subscriptionContext.UserAccount.UnderlyingAccount.Properties.ProviderSettings?.Settings?.ArmResource?.Endpoint;
+                Uri armUri = null;
+                if (armEndpoint != null)
+                {
+                    try
+                    {
+                        armUri = new Uri(armEndpoint);
+                    }
+                    catch (Exception e)
+                    {
+                        Console.WriteLine($"Exception while parsing URI: {e.Message}");
+                    }
+                }
                 ServiceClientCredentials credentials = CreateCredentials(subscriptionContext);
-                SqlManagementClient sqlManagementClient = new SqlManagementClient(credentials)
+                SqlManagementClient sqlManagementClient = new SqlManagementClient(armUri ?? _resourceManagementUri, credentials)
                 {
                     SubscriptionId = subscriptionContext.Subscription.SubscriptionId
                 };
-                ResourceManagementClient resourceManagementClient = new ResourceManagementClient(credentials)
+
+                ResourceManagementClient resourceManagementClient = new ResourceManagementClient(armUri ?? _resourceManagementUri, credentials)
                 {
                     SubscriptionId = subscriptionContext.Subscription.SubscriptionId
                 };
@@ -128,7 +142,8 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
                         SR.FailedToGetAzureSqlServersWithError);
                     if (servers != null)
                     {
-                        sqlServers.AddRange(servers.Select(server => {
+                        sqlServers.AddRange(servers.Select(server =>
+                        {
                             var serverResource = new SqlAzureResource(server);
                             // TODO ResourceGroup name
                             return serverResource;
@@ -239,12 +254,25 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
             if (azureTenant != null)
             {
                 ServiceClientCredentials credentials = CreateCredentials(azureTenant);
-                using (SubscriptionClient client = new SubscriptionClient(_resourceManagementUri, credentials))
+                string armEndpoint = userAccount.UnderlyingAccount.Properties.ProviderSettings?.Settings?.ArmResource?.Endpoint;
+                Uri armUri = null;
+                if (armEndpoint != null)
+                {
+                    try
+                    {
+                        armUri = new Uri(armEndpoint);
+                    }
+                    catch (Exception e)
+                    {
+                        Console.WriteLine($"Exception while parsing URI: {e.Message}");
+                    }
+                }
+                using (SubscriptionClient client = new SubscriptionClient(armUri ?? _resourceManagementUri, credentials))
                 {
                     IEnumerable<Subscription> subs = await GetSubscriptionsAsync(client);
                     return new ServiceResponse<IAzureUserAccountSubscriptionContext>(subs.Select(sub =>
                     {
-                        AzureSubscriptionIdentifier subId = new AzureSubscriptionIdentifier(userAccount, azureTenant.TenantId, sub.SubscriptionId, _resourceManagementUri);
+                        AzureSubscriptionIdentifier subId = new AzureSubscriptionIdentifier(userAccount, azureTenant.TenantId, sub.SubscriptionId, armUri ?? _resourceManagementUri);
                         AzureUserAccountSubscriptionContext context = new AzureUserAccountSubscriptionContext(subId, credentials);
                         return context;
                     }));

src/Microsoft.SqlTools.ResourceProvider.DefaultImpl/AzureUserAccount.cs

@@ -6,6 +6,7 @@ using System;
 using System.Collections.Generic;
 using Microsoft.SqlTools.ResourceProvider.Core;
 using Microsoft.SqlTools.ResourceProvider.Core.Authentication;
+using Microsoft.SqlTools.ResourceProvider.Core.Contracts;
 
 namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
 {
@@ -39,6 +40,7 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
             this.TenantId = azureUserAccount.TenantId;
             this.AllTenants = azureUserAccount.AllTenants;
             this.UniqueId = azureUserAccount.UniqueId;
+            this.UnderlyingAccount = azureUserAccount.UnderlyingAccount;
             AzureUserAccount account = azureUserAccount as AzureUserAccount;
         }
         /// <summary>
@@ -99,5 +101,11 @@ namespace Microsoft.SqlTools.ResourceProvider.DefaultImpl
             get;
             set;
         }
+
+        public Account UnderlyingAccount
+        {
+            get;
+            set;
+        }
     }
 }