Adding ability to define a custom AzureEnvironment.

src/Promitor.Core/Extensions/AzureCloudExtensions.cs

@@ -25,6 +25,8 @@
                     return AzureEnvironment.AzureGermanCloud;
                 case AzureCloud.UsGov:
                     return AzureEnvironment.AzureUSGovernment;
+                case AzureCloud.Custom:
+                    return AzureEnvironmentExtensions.AzureCustomCloud;
                 default:
                     throw new ArgumentOutOfRangeException(nameof(azureCloud), "No Azure environment is known for in legacy SDK");
             }
@@ -81,6 +83,8 @@
                     return AzureAuthorityHosts.AzureGermany;
                 case AzureCloud.UsGov:
                     return AzureAuthorityHosts.AzureGovernment;
+                case AzureCloud.Custom:
+                    return new Uri(Environment.GetEnvironmentVariable("PROMITOR_AZURE_AUTH_ENDPOINT"));
                 default:
                     throw new ArgumentOutOfRangeException(nameof(azureCloud), "No Azure environment is known for");
             }

src/Promitor.Core/Extensions/AzureEnvironmentExtensions.cs

@@ -1,5 +1,6 @@
 using Humanizer;
 using Microsoft.Azure.Management.ResourceManager.Fluent;
+using System;
 
 namespace Promitor.Core.Extensions
 {
@@ -14,5 +15,16 @@ public static string GetDisplayName(this AzureEnvironment azureCloud)
         {
             return azureCloud.Name.Replace("Azure", "").Replace("Cloud", "").Humanize(LetterCasing.Title);
         }
+
+        public static AzureEnvironment AzureCustomCloud = new AzureEnvironment()
+        {
+            Name = nameof(AzureCustomCloud),
+            AuthenticationEndpoint = Environment.GetEnvironmentVariable("PROMITOR_AUTH_ENDPOINT"),
+            ResourceManagerEndpoint = Environment.GetEnvironmentVariable("PROMITOR_RESOURCE_MANAGER_ENDPOINT"),
+            ManagementEndpoint = Environment.GetEnvironmentVariable("PROMITOR_MANAGEMENT_ENDPOINT"),
+            GraphEndpoint = Environment.GetEnvironmentVariable("PROMITOR_GRAPH_ENDPOINT"),
+            StorageEndpointSuffix = Environment.GetEnvironmentVariable("PROMITOR_STORAGE_ENDPOINT_SUFFIX"),
+            KeyVaultSuffix = Environment.GetEnvironmentVariable("PROMITOR_KEY_VAULT_SUFFIX")
+        };
     }
 }

src/Promitor.Core/Serialization/Enum/AzureCloud.cs

@@ -6,6 +6,7 @@ public enum AzureCloud
         Global,
         China,
         UsGov,
-        Germany
+        Germany,
+        Custom
     }
 }

src/Promitor.Integrations.LogAnalytics/LogAnalyticsClient.cs

@@ -15,6 +15,7 @@
         public readonly string ColumnNameResult = "result";
         private readonly Uri _defaultEndpoint = new("https://api.loganalytics.io");
         private readonly Uri _govEndpoint = new("https://api.loganalytics.us");
+        private readonly Uri _customEndpoint = new (Environment.GetEnvironmentVariable("PROMITOR_LOG_ANALYTICS_ENDPOINT"));
 
         private readonly LogsQueryClient _logsQueryClient;
 
@@ -31,6 +32,7 @@
             {
                 nameof(AzureEnvironment.AzureGlobalCloud) => _defaultEndpoint,
                 nameof(AzureEnvironment.AzureUSGovernment) => _govEndpoint,
+                "AzureCustomCloud" => _customEndpoint,
                 _ => throw new NotSupportedException($"Environment {azureEnvironment.Name} is not supported for scraping Azure Log Analytics resource(s)")
             };