Add purview endpoints for data plane auth (#303) (#317)

Co-authored-by: Parv Saxena <[email protected]> Co-authored-by: Parv Saxena <[email protected]> Co-authored-by: Parv Saxena <[email protected]> Co-authored-by: Yeming Liu <[email protected]>
Azure · May 9, 2022 · a346bb0 · a346bb0
1 parent f4ec4f7
commit a346bb0
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/src/Authentication.Abstractions/AzureEnvironment.cs b/src/Authentication.Abstractions/AzureEnvironment.cs
@@ -252,6 +252,9 @@ private static void SetExtendedProperties(IDictionary<string, AzureEnvironment>
                 azureEnvironments[EnvironmentName.AzureCloud].SetProperty(ExtendedEndpoint.ManagedHsmServiceEndpointSuffix, AzureEnvironmentConstants.AzureManagedHsmDnsSuffix);
                 azureEnvironments[EnvironmentName.AzureCloud].SetProperty(ExtendedEndpoint.MicrosoftGraphEndpointResourceId, AzureEnvironmentConstants.AzureMicrosoftGraphEndpointResourceId);
                 azureEnvironments[EnvironmentName.AzureCloud].SetProperty(ExtendedEndpoint.MicrosoftGraphUrl, AzureEnvironmentConstants.AzureMicrosoftGraphUrl);
+                azureEnvironments[EnvironmentName.AzureCloud].SetProperty(ExtendedEndpoint.AzurePurviewEndpointSuffix, AzureEnvironmentConstants.AzurePurviewEndpointSuffix);
+                azureEnvironments[EnvironmentName.AzureCloud].SetProperty(ExtendedEndpoint.AzurePurviewEndpointResourceId, AzureEnvironmentConstants.AzurePurviewEndpointResourceId);
+
             }
 
             if (azureEnvironments.ContainsKey(EnvironmentName.AzureChinaCloud))
@@ -672,7 +675,9 @@ public static class ExtendedEndpoint
                 AzureSynapseAnalyticsEndpointSuffix = "AzureSynapseAnalyticsEndpointSuffix",
                 AzureSynapseAnalyticsEndpointResourceId = "AzureSynapseAnalyticsEndpointResourceId",
                 MicrosoftGraphUrl = "MicrosoftGraphUrl",
-                MicrosoftGraphEndpointResourceId = "MicrosoftGraphEndpointResourceId";
+                MicrosoftGraphEndpointResourceId = "MicrosoftGraphEndpointResourceId",
+                AzurePurviewEndpointSuffix = "AzurePurviewEndpointSuffix",
+                AzurePurviewEndpointResourceId = "AzurePurviewEndpointResourceId";
         }
     }
 }
diff --git a/src/Authentication.Abstractions/AzureEnvironmentConstants.cs b/src/Authentication.Abstractions/AzureEnvironmentConstants.cs
@@ -236,5 +236,15 @@ public static class AzureEnvironmentConstants
         public const string ChinaMicrosoftGraphEndpointResourceId = "https://microsoftgraph.chinacloudapi.cn/";
         public const string USGovernmentMicrosoftGraphEndpointResourceId = "https://graph.microsoft.us/";
         public const string GermanMicrosoftGraphEndpointResourceId = "https://graph.microsoft.de/";
+
+        /// <summary>
+        /// The domain name suffix for Azure Purview Services
+        /// </summary>
+        public const string AzurePurviewEndpointSuffix = "purview.azure.net";
+
+        /// <summary>
+        /// The token audience for authorizing Purview Service requests
+        /// </summary>
+        public const string AzurePurviewEndpointResourceId = "https://purview.azure.net";
     }
 }
diff --git a/src/Authentication.Abstractions/Extensions/AzureEnvironmentExtensions.cs b/src/Authentication.Abstractions/Extensions/AzureEnvironmentExtensions.cs
@@ -307,6 +307,12 @@ public static void SetEndpoint(this IAzureEnvironment environment, string endpoi
                     case AzureEnvironment.ExtendedEndpoint.MicrosoftGraphEndpointResourceId:
                         environment.SetProperty(AzureEnvironment.ExtendedEndpoint.MicrosoftGraphEndpointResourceId, propertyValue);
                         break;
+                    case AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointSuffix:
+                        environment.SetProperty(AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointSuffix, propertyValue);
+                        break;
+                    case AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointResourceId:
+                        environment.SetProperty(AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointResourceId, propertyValue);
+                        break;
                 }
             }
         }
@@ -360,6 +366,10 @@ public static string GetTokenAudience(this IAzureEnvironment environment, string
                 case AzureEnvironment.ExtendedEndpoint.MicrosoftGraphUrl:
                     resource = AzureEnvironment.ExtendedEndpoint.MicrosoftGraphEndpointResourceId;
                     break;
+                case AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointSuffix:
+                case AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointResourceId:
+                    resource = AzureEnvironment.ExtendedEndpoint.AzurePurviewEndpointResourceId;
+                    break;
                 default:
                     resource = AzureEnvironment.Endpoint.ActiveDirectoryServiceEndpointResourceId;
                     break;