fix: tomo ssr issue (#264)

babylonlabs-io · Oct 28, 2024 · e274321 · e274321
1 parent 91fb335
commit e274321
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/components/common/AuthGuard/index.tsx b/src/components/common/AuthGuard/index.tsx
@@ -1,4 +1,9 @@
-import type { PropsWithChildren, ReactNode } from "react";
+import {
+  useEffect,
+  useState,
+  type PropsWithChildren,
+  type ReactNode,
+} from "react";
 
 import { useWalletConnection } from "@/app/context/wallet/WalletConnectionProvider";
 
@@ -11,6 +16,11 @@ export function AuthGuard({
   fallback,
 }: PropsWithChildren<AuthGuardProps>) {
   const { isConnected } = useWalletConnection();
+  const [displayComponent, setDisplayComponent] = useState(false);
 
-  return isConnected ? children : fallback;
+  useEffect(() => {
+    setDisplayComponent(isConnected);
+  }, [isConnected]);
+
+  return displayComponent ? children : fallback;
 }