feat: store wallet pk as custom prop

mrgnlabs · Sep 10, 2024 · 7bdd91f · 7bdd91f
1 parent 1083822
commit 7bdd91f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/apps/marginfi-v2-trading/src/context/TradeProvider.tsx b/apps/marginfi-v2-trading/src/context/TradeProvider.tsx
@@ -46,8 +46,14 @@ export const TradePovider: React.FC<{
     trackReferral(referralCode, wallet.publicKey.toBase58());
 
     if (!isLoggedIn && wallet.publicKey) {
+      const walletAddress = wallet.publicKey.toBase58();
       setIsLoggedIn(true);
-      identify(wallet.publicKey.toBase58());
+
+      if (!walletAddress) return;
+
+      identify(walletAddress, {
+        wallet: walletAddress,
+      });
     }
   }, [router.asPath, wallet, connected, isLoggedIn]);
 

diff --git a/apps/marginfi-v2-ui/src/context/MrgnlendProvider.tsx b/apps/marginfi-v2-ui/src/context/MrgnlendProvider.tsx
@@ -24,7 +24,11 @@ export const MrgnlendProvider: React.FC<{
 
   // identify user if logged in
   React.useEffect(() => {
-    identify(wallet.publicKey.toBase58());
+    const walletAddress = wallet.publicKey?.toBase58();
+    if (!walletAddress) return;
+    identify(walletAddress, {
+      wallet: walletAddress,
+    });
   }, [wallet.publicKey]);
 
   // if account set in query param then store inn local storage and remove from url