refactor: use redirectUrl from sessionStorage

Author: raheeliftikhar5

client/packages/lowcoder/src/appView/AppViewInstance.tsx

@@ -11,6 +11,7 @@ import { AppView } from "./AppView";
 import { API_STATUS_CODES } from "constants/apiConstants";
 import { AUTH_LOGIN_URL } from "constants/routesURL";
 import { AuthSearchParams } from "constants/authConstants";
+import { saveAuthSearchParams } from "@lowcoder-ee/pages/userAuth/authUtils";
 
 export type OutputChangeHandler<O> = (output: O) => void;
 export type EventTriggerHandler = (eventName: string) => void;
@@ -71,9 +72,11 @@ export class AppViewInstance<I = any, O = any> {
         .then((i) => i.data)
         .catch((e) => {
           if (e.response?.status === API_STATUS_CODES.REQUEST_NOT_AUTHORISED) {
-            window.location.href = `${webUrl}${AUTH_LOGIN_URL}?${
-              AuthSearchParams.redirectUrl
-            }=${encodeURIComponent(window.location.href)}`;
+            saveAuthSearchParams({
+              [AuthSearchParams.redirectUrl]: encodeURIComponent(window.location.href),
+              [AuthSearchParams.loginType]: null,
+            })
+            window.location.href = `${webUrl}${AUTH_LOGIN_URL}`;
           }
         });
 

client/packages/lowcoder/src/constants/routesURL.ts

@@ -39,7 +39,6 @@ export const QR_CODE_OAUTH_URL = `${USER_AUTH_URL}/oauth/qrcode`;
 export const OAUTH_REDIRECT = `${USER_AUTH_URL}/oauth/redirect`;
 export const CAS_AUTH_REDIRECT = `${USER_AUTH_URL}/cas/redirect`;
 export const LDAP_AUTH_LOGIN_URL = `${USER_AUTH_URL}/ldap/login`;
-export const USER_INFO_COMPLETION = `${USER_AUTH_URL}/completion`;
 export const INVITE_LANDING_URL = "/invite/:invitationId";
 export const ORG_AUTH_LOGIN_URL = `/org/:orgId/auth/login`;
 export const ORG_AUTH_REGISTER_URL = `/org/:orgId/auth/register`;

client/packages/lowcoder/src/pages/userAuth/authUtils.ts

@@ -3,7 +3,6 @@ import {
   BASE_URL,
   CAS_AUTH_REDIRECT,
   OAUTH_REDIRECT,
-  USER_INFO_COMPLETION,
 } from "constants/routesURL";
 import { AxiosPromise, AxiosResponse } from "axios";
 import { ApiResponse } from "api/apiResponses";
@@ -80,12 +79,7 @@ export function authRespValidate(
 ) {
   let replaceUrl = redirectUrl || BASE_URL;
   const baseUrl = `${window.location.protocol}//${window.location.host}`;
-  if (infoCompleteCheck) {
-    // need complete info
-    replaceUrl = redirectUrl
-      ? `${USER_INFO_COMPLETION}?redirectUrl=${redirectUrl}`
-      : USER_INFO_COMPLETION;
-  }
+
   if (doValidResponse(resp)) {
     onAuthSuccess?.();
     history.replace(replaceUrl.replace(baseUrl, ''));

client/packages/lowcoder/src/pages/userAuth/index.tsx

@@ -3,7 +3,7 @@ import { Redirect, Route, Switch, useLocation, useParams } from "react-router-do
 import React, { useEffect, useMemo } from "react";
 import { useSelector, useDispatch } from "react-redux";
 import { selectSystemConfig } from "redux/selectors/configSelectors";
-import { AuthContext } from "pages/userAuth/authUtils";
+import { AuthContext, clearAuthSearchParams } from "pages/userAuth/authUtils";
 import { AuthRoutes } from "@lowcoder-ee/constants/authConstants";
 import { AuthLocationState } from "constants/authConstants";
 import { ProductLoading } from "components/ProductLoading";
@@ -37,6 +37,7 @@ export default function UserAuth() {
 
   const fetchUserAfterAuthSuccess = () => {
     dispatch(fetchUserAction());
+    clearAuthSearchParams();
   }
 
   return (

client/packages/lowcoder/src/pages/userAuth/thirdParty/authenticator/abstractAuthenticator.ts

@@ -39,7 +39,7 @@ export abstract class AbstractAuthenticator {
         authRespValidate(
           resp,
           this.needInfoCheck(this.authParams.sourceType),
-          this.authParams.afterLoginRedirect,
+          getSafeAuthRedirectURL(this.authParams.afterLoginRedirect),
           onAuthSuccess,
         );
       })

client/packages/lowcoder/src/pages/userAuth/thirdParty/thirdPartyAuth.tsx

@@ -1,11 +1,9 @@
 import {
-  AuthSearchParams,
   OAuthLocationState,
   ThirdPartyAuthGoal,
   ThirdPartyConfigType,
 } from "constants/authConstants";
-import { CommonGrayLabel, WhiteLoading } from "lowcoder-design";
-import { useLocation } from "react-router-dom";
+import { WhiteLoading } from "lowcoder-design";
 import history from "util/history";
 import { LoginLogoStyle, LoginLabelStyle, StyledLoginButton } from "pages/userAuth/authComponents";
 import { useSelector } from "react-redux";
@@ -16,6 +14,7 @@ import styled from "styled-components";
 import { trans } from "i18n";
 import { geneAuthStateAndSaveParam, getAuthUrl, getRedirectUrl } from "pages/userAuth/authUtils";
 import { Divider } from "antd";
+import { useRedirectUrl } from "util/hooks";
 
 const ThirdPartyLoginButtonWrapper = styled.div`
   button{
@@ -36,9 +35,7 @@ function ThirdPartyLoginButton(props: {
   label: string;
 }) {
   const { config, label } = props;
-  const location = useLocation();
-  const queryParams = new URLSearchParams(location.search);
-  const loginRedirectUrl = queryParams.get(AuthSearchParams.redirectUrl);
+  const loginRedirectUrl = useRedirectUrl();
   const redirectUrl = getRedirectUrl(config.authType);
   const onLoginClick = () => {
     const state = geneAuthStateAndSaveParam(

client/packages/lowcoder/src/redux/sagas/userSagas.ts

@@ -133,15 +133,10 @@ export function* logoutSaga(action: LogoutActionType) {
   try {
     let redirectURL = AUTH_LOGIN_URL;
     if (action.payload.notAuthorised) {
-      const currentUrl = window.location.href;
-      // redirectURL = `${AUTH_LOGIN_URL}`;
-      // redirectURL = `${AUTH_LOGIN_URL}?redirectUrl=${encodeURIComponent(currentUrl)}`;
+      const currentUrl = window.location.href
       const urlObj = new URL(currentUrl);
       // Add loginType param for auto login jump
       const loginType = urlObj.searchParams.get(AuthSearchParams.loginType);
-      // if (loginType) {
-      //   redirectURL = redirectURL + `&${AuthSearchParams.loginType}=${loginType}`;
-      // }
       saveAuthSearchParams({
         [AuthSearchParams.redirectUrl]: encodeURIComponent(currentUrl),
         [AuthSearchParams.loginType]: loginType

client/packages/lowcoder/src/util/hooks.ts

@@ -63,9 +63,6 @@ export function useRedirectUrl() {
   const authSearchParams = loadAuthSearchParams()
   const redirectUrl = authSearchParams && authSearchParams.redirectUrl
   return redirectUrl && decodeURIComponent(redirectUrl);
-  // const location = useLocation();
-  // const queryParams = new URLSearchParams(location.search);
-  // return queryParams.get(AuthSearchParams.redirectUrl);
 }
 
 export function useFixedDelay(callback: () => Promise<unknown>, delay: number | null) {