fix(ui-react): prevent re-calculate friendly address on wallet unchan…

…ged in useTonAddress
ton-connect · Apr 21, 2024 · dfbe5ba · dfbe5ba
1 parent a881c7f
commit dfbe5ba
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/packages/ui-react/src/hooks/useTonAddress.ts b/packages/ui-react/src/hooks/useTonAddress.ts
@@ -1,18 +1,23 @@
 import { CHAIN, toUserFriendlyAddress } from '@tonconnect/ui';
 import { useTonWallet } from './useTonWallet';
+import { useMemo } from 'react';
 
 /**
  * Use it to get user's current ton wallet address. If wallet is not connected hook will return empty string.
  * @param [userFriendly=true] allows to choose format of the address.
  */
 export function useTonAddress(userFriendly = true): string {
     const wallet = useTonWallet();
-
-    if (wallet) {
-        return userFriendly
-            ? toUserFriendlyAddress(wallet.account.address, wallet.account.chain === CHAIN.TESTNET)
-            : wallet.account.address;
-    } else {
-        return '';
-    }
+    return useMemo(() => {
+        if (wallet) {
+            return userFriendly
+                ? toUserFriendlyAddress(
+                      wallet.account.address,
+                      wallet.account.chain === CHAIN.TESTNET
+                  )
+                : wallet.account.address;
+        } else {
+            return '';
+        }
+    }, [wallet, userFriendly, wallet?.account.address, wallet?.account.chain]);
 }