Merge branch 'release/v11.0.2' of https://github.com/utmstack/UTMStack into release/v11.0.2

Author: JocLRojas

plugins/azure/main.go

@@ -21,12 +21,49 @@ import (
 	"github.com/utmstack/UTMStack/plugins/azure/config"
 )
 
+type AzureCloud string
+
 const (
-	defaultTenant      string = "ce66672c-e36d-4761-a8c8-90058fee1a24"
-	urlCheckConnection        = "https://login.microsoftonline.com/"
-	wait                      = 1 * time.Second
+	defaultTenant string = "ce66672c-e36d-4761-a8c8-90058fee1a24"
+	wait                 = 1 * time.Second
+
+	AzurePublic     AzureCloud = "AzurePublic"
+	AzureGovernment AzureCloud = "AzureGovernment"
+	AzureChina      AzureCloud = "AzureChina"
 )
 
+type CloudEndpoints struct {
+	Name           AzureCloud
+	EventHubSuffix string
+	StorageSuffix  string
+	LoginAuthority string
+	Description    string
+}
+
+var SupportedClouds = []CloudEndpoints{
+	{
+		Name:           AzureGovernment,
+		EventHubSuffix: ".servicebus.usgovcloudapi.net",
+		StorageSuffix:  ".core.usgovcloudapi.net",
+		LoginAuthority: "https://login.microsoftonline.us/",
+		Description:    "Azure Government (US)",
+	},
+	{
+		Name:           AzureChina,
+		EventHubSuffix: ".servicebus.chinacloudapi.cn",
+		StorageSuffix:  ".core.chinacloudapi.cn",
+		LoginAuthority: "https://login.chinacloudapi.cn/",
+		Description:    "Azure China (21Vianet)",
+	},
+	{
+		Name:           AzurePublic,
+		EventHubSuffix: ".servicebus.windows.net",
+		StorageSuffix:  ".core.windows.net",
+		LoginAuthority: "https://login.microsoftonline.com/",
+		Description:    "Azure Public Cloud",
+	},
+}
+
 func main() {
 	mode := plugins.GetCfg().Env.Mode
 	if mode != "manager" {
@@ -46,12 +83,18 @@ func main() {
 	defer ticker.Stop()
 
 	for range ticker.C {
-		if err := connectionChecker(urlCheckConnection); err != nil {
-			_ = catcher.Error("External connection failure detected: %v", err, nil)
-		}
-
 		moduleConfig := config.GetConfig()
 		if moduleConfig != nil && moduleConfig.ModuleActive {
+			cloudsInUse := detectCloudsInUse(moduleConfig)
+			for cloudName, loginAuthority := range cloudsInUse {
+				if err := connectionChecker(loginAuthority); err != nil {
+					catcher.Info("Airgap or limited connectivity detected", map[string]any{
+						"cloud":          cloudName,
+						"loginAuthority": loginAuthority,
+					})
+				}
+			}
+
 			var wg sync.WaitGroup
 			wg.Add(len(moduleConfig.ModuleGroups))
 			for _, grp := range moduleConfig.ModuleGroups {
@@ -76,6 +119,47 @@ func main() {
 	}
 }
 
+func detectCloudsInUse(moduleConfig *config.ConfigurationSection) map[string]string {
+	cloudsMap := make(map[string]string)
+
+	for _, group := range moduleConfig.ModuleGroups {
+		for _, cnf := range group.ModuleGroupConfigurations {
+			if cnf.ConfKey == "eventHubConnection" || cnf.ConfKey == "storageConnection" {
+				if cloud, err := detectCloudFromConnectionString(cnf.ConfValue); err == nil {
+					cloudsMap[string(cloud.Name)] = cloud.LoginAuthority
+				}
+			}
+		}
+	}
+
+	return cloudsMap
+}
+
+func detectCloudFromConnectionString(connectionString string) (CloudEndpoints, error) {
+	if connectionString == "" {
+		return CloudEndpoints{}, fmt.Errorf("connection string is empty")
+	}
+
+	connStrLower := strings.ToLower(connectionString)
+
+	for _, cloud := range SupportedClouds {
+		if strings.Contains(connStrLower, strings.ToLower(cloud.EventHubSuffix)) {
+			return cloud, nil
+		}
+		if strings.Contains(connStrLower, strings.ToLower(cloud.StorageSuffix)) {
+			return cloud, nil
+		}
+	}
+
+	for _, cloud := range SupportedClouds {
+		if cloud.Name == AzurePublic {
+			return cloud, nil
+		}
+	}
+
+	return CloudEndpoints{}, fmt.Errorf("unable to detect Azure cloud from connection string")
+}
+
 func pull(group *config.ModuleGroup) {
 	agent := getAzureProcessor(group)