Ops: Update azure cloud token auth

Signed-off-by: mihaiandreiratoiu <mihai.ratoiu@uipath.com>
fluxcd · Aug 18, 2023 · d7eb4d5 · d7eb4d5
1 parent 0518e84
commit d7eb4d5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/oci/auth/azure/auth.go b/oci/auth/azure/auth.go
@@ -74,10 +74,16 @@ func (c *Client) getLoginAuth(ctx context.Context, registryURL string) (authn.Au
 		c.credential = cred
 	}
 
+	var configurationEnvironmentLookupMap = map[string]cloud.Configuration{
+		"io": cloud.AzurePublic,
+		"de": cloud.AzurePublic, // Azure Germany is not supported by Azure SDK for Go
+		"us": cloud.AzureGovernment,
+		"cn": cloud.AzureChina,
+	}
+	var azureEndpoint = registryURL[len(registryURL)-2:]
 	// Obtain access token using the token credential.
-	// TODO: Add support for other azure endpoints as well.
 	armToken, err := c.credential.GetToken(ctx, policy.TokenRequestOptions{
-		Scopes: []string{cloud.AzurePublic.Services[cloud.ResourceManager].Endpoint + "/" + ".default"},
+		Scopes: []string{configurationEnvironmentLookupMap[azureEndpoint].Services[cloud.ResourceManager].Endpoint + "/" + ".default"},
 	})
 	if err != nil {
 		return authConfig, err