Merge branch 'main' into dependabot/cargo/src/agent/quick-xml-0.30.0

microsoft · Jul 25, 2023 · eb6c230 · eb6c230
2 parents 9a6b74e + 226d0f2
commit eb6c230
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/deployment/deploy.py b/src/deployment/deploy.py
@@ -536,6 +536,13 @@ def create_new_app_registration(
     ) -> Dict[str, Any]:
         logger.info("creating Application registration")
 
+        appRegRedirectUris = [f"{self.get_instance_url()}/.auth/login/aad/callback"]
+
+        if self.custom_domain:
+            appRegRedirectUris.append(
+                f"https://{self.custom_domain}/.auth/login/aad/callback"
+            )
+
         params = {
             "displayName": self.application_name,
             "identifierUris": [self.get_identifier_url()],
@@ -560,7 +567,7 @@ def create_new_app_registration(
                     "enableAccessTokenIssuance": False,
                     "enableIdTokenIssuance": True,
                 },
-                "redirectUris": [f"{self.get_instance_url()}/.auth/login/aad/callback"],
+                "redirectUris": appRegRedirectUris,
             },
             "requiredResourceAccess": [
                 {