Redirect to Original URL After Session Expiry and Re-login

src/Components/Auth/Login.tsx

@@ -109,7 +109,20 @@ export const Login = (props: { forgot?: boolean }) => {
           window.location.pathname === "/" ||
           window.location.pathname === "/login"
         ) {
-          window.location.href = "/facility";
+          const redirectParam = new URLSearchParams(window.location.search).get(
+            "redirect"
+          );
+          try {
+            if (
+              redirectParam &&
+              new URL(redirectParam).origin === window.location.origin
+            )
+              window.location.href =
+                window.location.origin + new URL(redirectParam).pathname;
+            else window.location.href = "/facility";
+          } catch {
+            window.location.href = "/facility";
+          }
         } else {
           window.location.href = window.location.pathname.toString();
         }

src/Redux/fireRequest.tsx

@@ -152,7 +152,7 @@ export const fireRequest = (
           if (error.response.status > 400 && error.response.status < 500) {
             if (error.response.data && error.response.data.detail) {
               if (error.response.data.code === "token_not_valid") {
-                window.location.href = "/session-expired";
+                window.location.href = `/session-expired?redirect=${window.location.href}`;
               }
               Notification.Error({
                 msg: error.response.data.detail,

src/Utils/request/handleResponse.ts

@@ -29,7 +29,7 @@ export default function handleResponse(
   if (res.status >= 400) {
     // Invalid token
     if (!silent && error?.code === "token_not_valid") {
-      navigate("/session-expired");
+      navigate(`/session-expired?redirect=${window.location.href}`);
     }
 
     notify?.Error({ msg: error?.detail || "Something went wrong...!" });

src/Utils/utils.ts

@@ -107,7 +107,10 @@ export const handleSignOut = (forceReload: boolean) => {
   Object.values(LocalStorageKeys).forEach((key) =>
     localStorage.removeItem(key)
   );
-  navigate("/");
+  const redirectURL = new URLSearchParams(window.location.search).get(
+    "redirect"
+  );
+  redirectURL ? navigate(`/?redirect=${redirectURL}`) : navigate("/");
   if (forceReload) window.location.reload();
 };