Skip to content

Commit d21e254

Browse files
committed
Refactored environment audience map to be clearer
1 parent 9c39f5d commit d21e254

File tree

1 file changed

+21
-13
lines changed

1 file changed

+21
-13
lines changed

src/Common/Commands.Common.Authentication.Abstractions/Extensions/AzureEnvironmentExtensions.cs

Lines changed: 21 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -283,20 +283,28 @@ public static void SetEndpoint(this IAzureEnvironment environment, string endpoi
283283
/// <returns>The correct token audience for tokens bound for the given endpoint.</returns>
284284
public static string GetTokenAudience(this IAzureEnvironment environment, string targetEndpoint)
285285
{
286-
string resource = AzureEnvironment.Endpoint.ActiveDirectoryServiceEndpointResourceId;
287-
if (targetEndpoint == AzureEnvironment.Endpoint.Graph)
286+
string resource;
287+
switch (targetEndpoint)
288288
{
289-
resource = AzureEnvironment.Endpoint.GraphEndpointResourceId;
290-
}
291-
else if (targetEndpoint == AzureEnvironment.Endpoint.AzureDataLakeAnalyticsCatalogAndJobEndpointSuffix ||
292-
targetEndpoint == AzureEnvironment.Endpoint.AzureDataLakeStoreFileSystemEndpointSuffix ||
293-
targetEndpoint == AzureEnvironment.Endpoint.DataLakeEndpointResourceId)
294-
{
295-
resource = AzureEnvironment.Endpoint.DataLakeEndpointResourceId;
296-
}
297-
else if (targetEndpoint == AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId)
298-
{
299-
resource = AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId;
289+
case AzureEnvironment.Endpoint.Graph:
290+
resource = AzureEnvironment.Endpoint.GraphEndpointResourceId;
291+
break;
292+
case AzureEnvironment.Endpoint.AzureDataLakeAnalyticsCatalogAndJobEndpointSuffix:
293+
case AzureEnvironment.Endpoint.AzureDataLakeStoreFileSystemEndpointSuffix:
294+
case AzureEnvironment.Endpoint.DataLakeEndpointResourceId:
295+
resource = AzureEnvironment.Endpoint.DataLakeEndpointResourceId;
296+
break;
297+
case AzureEnvironment.Endpoint.AzureKeyVaultDnsSuffix:
298+
case AzureEnvironment.Endpoint.AzureKeyVaultServiceEndpointResourceId:
299+
resource = AzureEnvironment.Endpoint.AzureKeyVaultServiceEndpointResourceId;
300+
break;
301+
case AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpoint:
302+
case AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId:
303+
resource = AzureEnvironment.ExtendedEndpoint.OperationalInsightsEndpointResourceId;
304+
break;
305+
default:
306+
resource = AzureEnvironment.Endpoint.ActiveDirectoryServiceEndpointResourceId;
307+
break;
300308
}
301309

302310
return resource;

0 commit comments

Comments
 (0)