Add a check for the azure_environment option

Elliott Barrere · Elliott Barrere · commit 341db0bb779d · 2024-04-23T20:26:32.000Z
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -94,10 +94,18 @@ func Load(ctx context.Context, p, domain string, opts ...Option) (config AAD, er
 	if config.AppID == "" {
 		return AAD{}, fmt.Errorf("missing required 'app_id' entry in configuration file")
 	}
+
 	if config.AzureEnvironment == "" {
 		config.AzureEnvironment = "Commercial"
 	}
-
+	switch config.AzureEnvironment {
+	case "Commercial":
+                logger.Debug(ctx, "Using Azure Commercial environment")
+	case "GCC-H":
+                logger.Debug(ctx, "Using Azure GCC-H environment")
+	default:
+		return AAD{}, fmt.Errorf("unknown value '%s' for 'azure_environment'", config.AzureEnvironment)
+	}
 	return config, nil
 }
 

Original file line number	Diff line number	Diff line change
`@@ -94,10 +94,18 @@ func Load(ctx context.Context, p, domain string, opts ...Option) (config AAD, er`
`94`	`94`	`if config.AppID == "" {`
`95`	`95`	`return AAD{}, fmt.Errorf("missing required 'app_id' entry in configuration file")`
`96`	`96`	`}`
	`97`	`+`
`97`	`98`	`if config.AzureEnvironment == "" {`
`98`	`99`	`config.AzureEnvironment = "Commercial"`
`99`	`100`	`}`
`100`		`-`
	`101`	`+ switch config.AzureEnvironment {`
	`102`	`+ case "Commercial":`
	`103`	`+ logger.Debug(ctx, "Using Azure Commercial environment")`
	`104`	`+ case "GCC-H":`
	`105`	`+ logger.Debug(ctx, "Using Azure GCC-H environment")`
	`106`	`+ default:`
	`107`	`+ return AAD{}, fmt.Errorf("unknown value '%s' for 'azure_environment'", config.AzureEnvironment)`
	`108`	`+ }`
`101`	`109`	`return config, nil`
`102`	`110`	`}`
`103`	`111`