Added redirect by return URL for AAD and AAD B2C flows. (#700)

Azure · Jun 11, 2020 · 8d9113f · 8d9113f
1 parent 7c905f6
commit 8d9113f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/services/aadService.ts b/src/services/aadService.ts
@@ -40,7 +40,8 @@ export class AadService {
 
         const response = await this.httpClient.send(request);
         const sasTokenHeader = response.headers.find(x => x.name.toLowerCase() === "ocp-apim-sas-token");
-
+        const returnUrl = this.routeHelper.getQueryParameter("returnUrl");
+
         if (!sasTokenHeader) { // User not registered with APIM.
             const jwtToken = Utils.parseJwt(idToken);
             const firstName = jwtToken.given_name;
@@ -49,7 +50,7 @@ export class AadService {
 
             if (firstName && lastName && email) {
                 await this.usersService.createUserWithOAuth(provider, idToken, firstName, lastName, email);
-                await this.router.navigateTo(Constants.pageUrlHome);
+                await this.router.navigateTo(returnUrl || Constants.pageUrlHome);
             }
             else {
                 const signupUrl = this.routeHelper.getIdTokenReferenceUrl(provider, idToken);
@@ -68,7 +69,7 @@ export class AadService {
 
         const sasToken = matches[1];
         await this.authenticator.setAccessToken(`SharedAccessSignature ${sasToken}`);
-        await this.router.navigateTo(Constants.pageUrlHome);
+        await this.router.navigateTo(returnUrl || Constants.pageUrlHome);
     }
 
     /**