Redirect to previous page on connecting wallet

src/App.tsx

@@ -20,7 +20,7 @@ function App() {
       <WalletConnectProvider>
         <Routes>
           <Route path="/" element={<TermsAndConditions />} />
-          <Route path="/connect-wallet" element={<ConnectWallet />} />
+          <Route path="/connect-wallet/:redirectTo?" element={<ConnectWallet />} />
           <Route element={<SignPageLayout />}>
             <Route path="/sign-with-nitro-key" element={<SignWithNitroKey />} />
             <Route

src/layout/SignPageLayout.tsx

@@ -1,5 +1,5 @@
 import React from "react";
-import { Outlet, useNavigate } from "react-router-dom";
+import { Outlet, useLocation, useNavigate } from "react-router-dom";
 
 import {
   Toolbar,
@@ -14,10 +14,14 @@ import { useWalletConnectContext } from "../context/WalletConnectContext";
 const SignPageLayout = () => {
   const { disconnect, session } = useWalletConnectContext();
   const navigate = useNavigate();
+  const location = useLocation();
 
   const disconnectHandler = async () => {
+    const { pathname } = location;
+    const redirectTo = pathname ? pathname.substring(1) : "";
+
     await disconnect();
-    navigate("/");
+    navigate(`/connect-wallet/${redirectTo}`);
   };
 
   return (

src/pages/ConnectWallet.tsx

@@ -1,5 +1,5 @@
 import React, { useEffect } from "react";
-import {useNavigate } from "react-router-dom";
+import {useLocation, useNavigate, useParams } from "react-router-dom";
 
 import { Button, Box, Container, Typography, colors } from "@mui/material";
 
@@ -11,13 +11,19 @@ const ConnectWallet = () => {
 
   const navigate = useNavigate();
 
+  const {redirectTo} = useParams();
 
   useEffect(() => {
 
     if (session) {
-      navigate("/sign-with-nitro-key");
+      if (redirectTo){
+        navigate(`/${redirectTo}`);
+      }
+      else {
+        navigate("/sign-with-nitro-key");
+      }
     }
-  }, [session, navigate,]);
+  }, [session, navigate, redirectTo]);
 
   const handler = async () => {
     await connect();