allow tenant override by env var

ekristen · May 17, 2022 · feaac36 · feaac36
1 parent cfa1a63
commit feaac36
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/pkg/azure/client_certificate.go b/pkg/azure/client_certificate.go
@@ -21,6 +21,9 @@ func AcquireTokenClientCertificate(ctx context.Context, tenantID, resource strin
 		if v := os.Getenv("AZURE_AUTHORITY_HOST"); v != "" {
 			authorityHost = v
 		}
+		if v := os.Getenv("AZURE_TENANT_ID"); v != "" {
+			tenantID = v
+		}
 
 		// trim the suffix / if exists
 		resource = strings.TrimSuffix(resource, "/")

diff --git a/pkg/azure/client_secret.go b/pkg/azure/client_secret.go
@@ -17,6 +17,9 @@ func AcquireTokenClientSecret(ctx context.Context, tenantID, resource string) fu
 		if v := os.Getenv("AZURE_AUTHORITY_HOST"); v != "" {
 			authorityHost = v
 		}
+		if v := os.Getenv("AZURE_TENANT_ID"); v != "" {
+			tenantID = v
+		}
 
 		// trim the suffix / if exists
 		resource = strings.TrimSuffix(resource, "/")

diff --git a/pkg/azure/client_token.go b/pkg/azure/client_token.go
@@ -26,6 +26,9 @@ func AcquireTokenFederatedToken(ctx context.Context, tenantID string) func(tenan
 		if v := os.Getenv("AZURE_AUTHORITY_HOST"); v != "" {
 			authorityHost = v
 		}
+		if v := os.Getenv("AZURE_TENANT_ID"); v != "" {
+			tenantID = v
+		}
 
 		// trim the suffix / if exists
 		resource = strings.TrimSuffix(resource, "/")