update protectedRoute

TogetherCrew · Sep 1, 2024 · cc347c2 · cc347c2
1 parent 08b9687
commit cc347c2
Showing 1 changed file with 8 additions and 12 deletions.
diff --git a/src/ProtectedRoute.tsx b/src/ProtectedRoute.tsx
@@ -15,23 +15,19 @@ const ProtectedRoute = ({ children }: ProtectedRouteProps) => {
   const { isConnected } = useAccount();
 
   useEffect(() => {
-    if (!isConnected) {
-      setIsAuthenticated(false);
-      localStorage.removeItem('OCI_TOKEN');
-    }
-  }, [isConnected]);
-
-  useEffect(() => {
-    const checkAuthStatus = async () => {
-      const token = localStorage.getItem('OCI_TOKEN');
-      if (token) {
-        setIsAuthenticated(true);
+    const checkAuthStatus = () => {
+      if (isConnected) {
+        const token = localStorage.getItem('OCI_TOKEN');
+        setIsAuthenticated(!!token);
+      } else {
+        setIsAuthenticated(false);
+        localStorage.removeItem('OCI_TOKEN');
       }
       setLoading(false);
     };
 
     checkAuthStatus();
-  }, []);
+  }, [isConnected]);
 
   if (loading) {
     return (