Redirect to Original URL After Session Expiry and Re-login

src/Components/Auth/Login.tsx

@@ -109,7 +109,15 @@ 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"
+          );
+          redirectParam
+            ? (window.location.href = new URL(
+                redirectParam,
+                window.location.origin
+              ).toString())
+            : (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.pathname}`;
               }
               Notification.Error({
                 msg: error.response.data.detail,

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();
 };