redirect after login to original route (#739) (#759)

Co-authored-by: Aashish Upadhyay <[email protected]>
Co-authored-by: mnida <[email protected]>

frontend/src/App.tsx

@@ -61,7 +61,11 @@ function App() {
           error?.response?.status === 401 &&
           !publicRoutes.includes(pathname)
         ) {
-          navigate("/");
+          navigate("/", {
+            state: {
+              redirectTo: pathname,
+            },
+          });
         }
         return error;
       });

frontend/src/pages/Login.tsx

@@ -1,5 +1,5 @@
 import React, { FC, useState } from "react";
-import { useNavigate, Link } from "react-router-dom";
+import { useNavigate, Link, useLocation } from "react-router-dom";
 import { Authentication, instance } from "../api/api";
 import { Card, Input, Button, Form } from "antd";
 import "./Login.css";
@@ -24,7 +24,9 @@ interface LoginForm extends HTMLFormControlsCollection {
 interface FormElements extends HTMLFormElement {
   readonly elements: LoginForm;
 }
-
+interface LocationState {
+  redirectTo?: string;
+}
 interface LoginProps {
   username?: string;
   password?: string;
@@ -36,6 +38,7 @@ const Login: FC<LoginProps> = (props) => {
   const [error, setError] = useState("");
   const setUsernameToStore = useGlobalStore((state) => state.setUsername);
   const queryClient = useQueryClient();
+  const location = useLocation();
 
   const [isAuthenticated, setIsAuthenticated] = useState(false);
   const navigate = useNavigate();
@@ -48,8 +51,8 @@ const Login: FC<LoginProps> = (props) => {
     setUsername(event.target.value);
   };
 
-  const redirectDashboard = () => {
-    navigate("/dashboard");
+  const redirectAfterLogin = () => {
+    navigate((location.state as LocationState).redirectTo!);
   };
 
   const isDemo = (import.meta as any).env.VITE_IS_DEMO === "true";
@@ -79,7 +82,7 @@ const Login: FC<LoginProps> = (props) => {
         cookies.set("Token", token);
         instance.defaults.headers.common.Authorization = `Token ${token}`;
         queryClient.refetchQueries(["session"]);
-        redirectDashboard();
+        redirectAfterLogin();
       },
       onError: (error: QueryErrors) => {
         // setError(error.message);