@0xsquid/react-hooks 6.4.3 → 6.4.4-sui-beta.0

package/dist/hooks/tokens/useBalance.js

@@ -1,39 +1,46 @@
 import { ChainType } from "@0xsquid/squid-types";
 import { useQuery } from "@tanstack/react-query";
-import { useBalance, useReadContract } from "wagmi";
+import { useReadContract, useBalance as useWagmiBalance } from "wagmi";
+import { useMemo } from "react";
 import { erc20Abi } from "viem";
 import { useCosmosContext } from "../../core";
 import { nativeEvmTokenAddress } from "../../core/constants";
 import { keys } from "../../core/queries/queries-keys";
 import { formatBNToReadable, isCosmosAddressValid } from "../../services";
-import { getCosmosTokenBalance, getSolanaTokenBalance, } from "../../services/external/rpcService";
+import { getCosmosTokenBalance, getSolanaTokenBalance, getSuiTokenBalance, } from "../../services/external/rpcService";
+import { isSuiAddressValid } from "../../services/internal/suiService";
 import { useWallet } from "../wallet/useWallet";
 import { useBitcoinNativeBalance } from "./useNativeBalance";
 const DEFAULT_REFRESH_INTERVAL_MS = 15000;
 export const useEvmBalance = ({ chain, token, userAddress, enabled = true, refreshIntervalMs = DEFAULT_REFRESH_INTERVAL_MS, }) => {
-    const { isConnected } = useWallet();
+    const { isChainTypeConnected } = useWallet();
     const isNativeToken = token?.address.toLowerCase() === nativeEvmTokenAddress.toLowerCase();
     const userAddressParsed = userAddress;
     // Only fetch using useBalance if it's a native token
-    const { data: nativeBalance } = useBalance({
+    const { data: nativeBalance, isLoading: isNativeTokenLoading } = useWagmiBalance({
         address: userAddressParsed,
         chainId: Number(chain?.chainId),
         query: {
-            enabled: isConnected(ChainType.EVM) && enabled && !!userAddress && isNativeToken,
+            enabled: enabled &&
+                !!userAddress &&
+                isNativeToken &&
+                !!chain &&
+                chain.chainType === ChainType.EVM &&
+                isChainTypeConnected(chain.chainType),
             refetchInterval: refreshIntervalMs,
             retry: 2,
         },
     });
     // Only fetch using useReadContract if it's not a native token
     // This is temporary, because of wagmi error: https://github.com/wevm/wagmi/issues/4353
-    const { data: erc20Balance } = useReadContract({
+    const { data: erc20Balance, isLoading: isErc20Loading } = useReadContract({
         address: token?.address,
         abi: erc20Abi,
         functionName: "balanceOf",
         args: [userAddressParsed],
         chainId: Number(chain?.chainId),
         query: {
-            enabled: isConnected(ChainType.EVM) &&
+            enabled: isChainTypeConnected(ChainType.EVM) &&
                 enabled &&
                 !!userAddress &&
                 !isNativeToken,
@@ -42,41 +49,44 @@ export const useEvmBalance = ({ chain, token, userAddress, enabled = true, refre
         },
     });
     const balance = isNativeToken ? nativeBalance?.value : erc20Balance;
-    const formattedBalance = formatBNToReadable(balance ?? BigInt(0), token?.decimals ?? 18);
-    return { balance: formattedBalance };
+    const isLoading = isNativeToken ? isNativeTokenLoading : isErc20Loading;
+    const formattedBalance = balance != null && token?.decimals != null
+        ? formatBNToReadable(balance, token.decimals)
+        : "0";
+    return { balance: formattedBalance, isLoading };
 };
-export const useCosmosBalance = ({ chainData, token, userAddress, enabled = true, refreshIntervalMs = DEFAULT_REFRESH_INTERVAL_MS, }) => {
+export const useCosmosBalance = ({ chain, token, userAddress, enabled = true, refreshIntervalMs = DEFAULT_REFRESH_INTERVAL_MS, }) => {
     const { isConnected } = useCosmosContext();
-    const query = useQuery({
-        queryKey: keys().balance(chainData?.chainId, token?.address, userAddress),
+    const { data: balance = "0", isLoading } = useQuery({
+        queryKey: keys().balance(chain?.chainId, token?.address, userAddress),
         queryFn: async () => {
-            if (!userAddress || !token || chainData?.chainType !== ChainType.COSMOS) {
+            if (!userAddress || !token || chain?.chainType !== ChainType.COSMOS) {
                 return "0";
             }
-            const isAddressValid = isCosmosAddressValid(chainData.bech32Config.bech32PrefixAccAddr, userAddress);
+            const isAddressValid = isCosmosAddressValid(chain.bech32Config.bech32PrefixAccAddr, userAddress);
             if (!isAddressValid)
                 return "0";
-            const balance = await getCosmosTokenBalance(chainData, userAddress, token.address);
-            return formatBNToReadable(balance, token.decimals);
+            const balanceBn = await getCosmosTokenBalance(chain, userAddress, token.address);
+            return formatBNToReadable(balanceBn, token.decimals);
         },
         enabled: isConnected &&
             enabled &&
             !!userAddress &&
             !!token &&
-            chainData?.chainType === ChainType.COSMOS,
+            chain?.chainType === ChainType.COSMOS,
         refetchInterval: refreshIntervalMs,
         retry: 2,
     });
-    return { ...query, balance: query.data ?? "0" };
+    return { balance, isLoading };
 };
 export const useSolanaBalance = ({ chain, token, userAddress, enabled = true, refreshIntervalMs = DEFAULT_REFRESH_INTERVAL_MS, }) => {
-    const query = useQuery({
+    const { data: balance = "0", isLoading } = useQuery({
         queryKey: keys().balance(chain?.chainId, token?.address, userAddress),
         queryFn: async () => {
             if (!userAddress || !token)
                 return "0";
-            const balance = await getSolanaTokenBalance(userAddress, token.address);
-            return formatBNToReadable(balance, token.decimals);
+            const balanceBn = await getSolanaTokenBalance(userAddress, token.address);
+            return formatBNToReadable(balanceBn, token.decimals);
         },
         enabled: enabled &&
             !!userAddress &&
@@ -85,11 +95,39 @@ export const useSolanaBalance = ({ chain, token, userAddress, enabled = true, re
         refetchInterval: refreshIntervalMs,
         retry: 2,
     });
-    return { ...query, balance: query.data ?? "0" };
+    return { balance, isLoading };
 };
 // TODO: implement fetching balances for all bitcoin tokens
-export const useBitcoinBalance = ({ address, chain, }) => {
-    const { data: balance } = useBitcoinNativeBalance(chain, address);
-    return { balance };
+export const useBitcoinBalance = ({ userAddress, chain, }) => {
+    const { balance: balanceBn, isLoading } = useBitcoinNativeBalance({
+        address: userAddress,
+        chain,
+    });
+    const balance = useMemo(() => {
+        if (!balanceBn)
+            return "0";
+        return formatBNToReadable(balanceBn?.value, balanceBn?.decimals);
+    }, [balanceBn]);
+    return { balance, isLoading };
+};
+export const useSuiBalance = ({ chain, token, userAddress, enabled = true, refreshIntervalMs = DEFAULT_REFRESH_INTERVAL_MS, }) => {
+    const { data: balance = "0", isLoading } = useQuery({
+        queryKey: keys().balance(chain?.chainId, token?.address, userAddress),
+        queryFn: async () => {
+            if (!userAddress || !token || !chain)
+                return "0";
+            const balanceBn = await getSuiTokenBalance(userAddress, token.address, chain.rpc);
+            return formatBNToReadable(balanceBn, token.decimals) ?? "0";
+        },
+        enabled: enabled &&
+            !!userAddress &&
+            !!token &&
+            !!chain?.rpc &&
+            chain?.chainType === ChainType.SUI &&
+            isSuiAddressValid(userAddress),
+        refetchInterval: refreshIntervalMs,
+        retry: 2,
+    });
+    return { balance, isLoading };
 };
 //# sourceMappingURL=useBalance.js.map

package/dist/hooks/tokens/useBalance.js.map

@@ -1 +1 @@
-{"version":3,"file":"useBalance.js","sourceRoot":"","sources":["../../../src/hooks/tokens/useBalance.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,OAAO,CAAC;AAEpD,OAAO,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAChC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAE,IAAI,EAAE,MAAM,iCAAiC,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAC1E,OAAO,EACL,qBAAqB,EACrB,qBAAqB,GACtB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE7D,MAAM,2BAA2B,GAAG,KAAK,CAAC;AAE1C,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAC5B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAOhD,EAAE,EAAE;IACH,MAAM,EAAE,WAAW,EAAE,GAAG,SAAS,EAAE,CAAC;IAEpC,MAAM,aAAa,GACjB,KAAK,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,qBAAqB,CAAC,WAAW,EAAE,CAAC;IACvE,MAAM,iBAAiB,GAAG,WAA4B,CAAC;IAEvD,qDAAqD;IACrD,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,UAAU,CAAC;QACzC,OAAO,EAAE,iBAAiB;QAC1B,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;QAC/B,KAAK,EAAE;YACL,OAAO,EACL,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,WAAW,IAAI,aAAa;YACzE,eAAe,EAAE,iBAAiB;YAClC,KAAK,EAAE,CAAC;SACT;KACF,CAAC,CAAC;IAEH,8DAA8D;IAC9D,uFAAuF;IACvF,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,eAAe,CAAC;QAC7C,OAAO,EAAE,KAAK,EAAE,OAAwB;QACxC,GAAG,EAAE,QAAQ;QACb,YAAY,EAAE,WAAW;QACzB,IAAI,EAAE,CAAC,iBAAiB,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;QAC/B,KAAK,EAAE;YACL,OAAO,EACL,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC;gBAC1B,OAAO;gBACP,CAAC,CAAC,WAAW;gBACb,CAAC,aAAa;YAChB,eAAe,EAAE,iBAAiB;YAClC,KAAK,EAAE,CAAC;SACT;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC;IAEpE,MAAM,gBAAgB,GAAG,kBAAkB,CACzC,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,EACpB,KAAK,EAAE,QAAQ,IAAI,EAAE,CACtB,CAAC;IAEF,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;AACvC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,EAC/B,SAAS,EACT,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAOhD,EAAE,EAAE;IACH,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAE3C,MAAM,KAAK,GAAG,QAAQ,CAAC;QACrB,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;QACzE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK,IAAI,SAAS,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,EAAE;gBACvE,OAAO,GAAG,CAAC;aACZ;YAED,MAAM,cAAc,GAAG,oBAAoB,CACzC,SAAS,CAAC,YAAY,CAAC,mBAAmB,EAC1C,WAAW,CACZ,CAAC;YAEF,IAAI,CAAC,cAAc;gBAAE,OAAO,GAAG,CAAC;YAEhC,MAAM,OAAO,GAAG,MAAM,qBAAqB,CACzC,SAAS,EACT,WAAW,EACX,KAAK,CAAC,OAAO,CACd,CAAC;YAEF,OAAO,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,EACL,WAAW;YACX,OAAO;YACP,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,KAAK;YACP,SAAS,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM;QAC3C,eAAe,EAAE,iBAAiB;QAClC,KAAK,EAAE,CAAC;KACT,CAAC,CAAC;IAEH,OAAO,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;AAClD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,EAC/B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAOhD,EAAE,EAAE;IACH,MAAM,KAAK,GAAG,QAAQ,CAAC;QACrB,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK;gBAAE,OAAO,GAAG,CAAC;YAEvC,MAAM,OAAO,GAAG,MAAM,qBAAqB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACxE,OAAO,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,EACL,OAAO;YACP,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,KAAK;YACP,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM;QACvC,eAAe,EAAE,iBAAiB;QAClC,KAAK,EAAE,CAAC;KACT,CAAC,CAAC;IAEH,OAAO,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;AAClD,CAAC,CAAC;AAEF,2DAA2D;AAC3D,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,EAChC,OAAO,EACP,KAAK,GAIN,EAAE,EAAE;IACH,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,uBAAuB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAElE,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC,CAAC"}
+{"version":3,"file":"useBalance.js","sourceRoot":"","sources":["../../../src/hooks/tokens/useBalance.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,UAAU,IAAI,eAAe,EAAE,MAAM,OAAO,CAAC;AAEvE,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAChC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAE,IAAI,EAAE,MAAM,iCAAiC,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAC1E,OAAO,EACL,qBAAqB,EACrB,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AACvE,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAc7D,MAAM,2BAA2B,GAAG,KAAK,CAAC;AAE1C,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAC5B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAC7B,EAAoB,EAAE;IACxC,MAAM,EAAE,oBAAoB,EAAE,GAAG,SAAS,EAAE,CAAC;IAE7C,MAAM,aAAa,GACjB,KAAK,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,qBAAqB,CAAC,WAAW,EAAE,CAAC;IACvE,MAAM,iBAAiB,GAAG,WAA4B,CAAC;IAEvD,qDAAqD;IACrD,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,oBAAoB,EAAE,GAC5D,eAAe,CAAC;QACd,OAAO,EAAE,iBAAiB;QAC1B,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;QAC/B,KAAK,EAAE;YACL,OAAO,EACL,OAAO;gBACP,CAAC,CAAC,WAAW;gBACb,aAAa;gBACb,CAAC,CAAC,KAAK;gBACP,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,GAAG;gBACjC,oBAAoB,CAAC,KAAK,CAAC,SAAS,CAAC;YACvC,eAAe,EAAE,iBAAiB;YAClC,KAAK,EAAE,CAAC;SACT;KACF,CAAC,CAAC;IAEL,8DAA8D;IAC9D,uFAAuF;IACvF,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,eAAe,CAAC;QACxE,OAAO,EAAE,KAAK,EAAE,OAAwB;QACxC,GAAG,EAAE,QAAQ;QACb,YAAY,EAAE,WAAW;QACzB,IAAI,EAAE,CAAC,iBAAiB,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;QAC/B,KAAK,EAAE;YACL,OAAO,EACL,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC;gBACnC,OAAO;gBACP,CAAC,CAAC,WAAW;gBACb,CAAC,aAAa;YAChB,eAAe,EAAE,iBAAiB;YAClC,KAAK,EAAE,CAAC;SACT;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC;IACpE,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,cAAc,CAAC;IAExE,MAAM,gBAAgB,GACpB,OAAO,IAAI,IAAI,IAAI,KAAK,EAAE,QAAQ,IAAI,IAAI;QACxC,CAAC,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC;QAC7C,CAAC,CAAC,GAAG,CAAC;IAEV,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,CAAC;AAClD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,EAC/B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAC7B,EAAoB,EAAE;IACxC,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAE3C,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,GAAG,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC;QAClD,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK,IAAI,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,EAAE;gBACnE,OAAO,GAAG,CAAC;aACZ;YAED,MAAM,cAAc,GAAG,oBAAoB,CACxC,KAAqB,CAAC,YAAY,CAAC,mBAAmB,EACvD,WAAW,CACZ,CAAC;YAEF,IAAI,CAAC,cAAc;gBAAE,OAAO,GAAG,CAAC;YAEhC,MAAM,SAAS,GAAG,MAAM,qBAAqB,CAC3C,KAAK,EACL,WAAW,EACX,KAAK,CAAC,OAAO,CACd,CAAC;YAEF,OAAO,kBAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,EACL,WAAW;YACX,OAAO;YACP,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,KAAK;YACP,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM;QACvC,eAAe,EAAE,iBAAiB;QAClC,KAAK,EAAE,CAAC;KACT,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,EAC/B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAC7B,EAAoB,EAAE;IACxC,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,GAAG,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC;QAClD,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK;gBAAE,OAAO,GAAG,CAAC;YAEvC,MAAM,SAAS,GAAG,MAAM,qBAAqB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1E,OAAO,kBAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,EACL,OAAO;YACP,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,KAAK;YACP,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM;QACvC,eAAe,EAAE,iBAAiB;QAClC,KAAK,EAAE,CAAC;KACT,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC,CAAC;AAEF,2DAA2D;AAC3D,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,EAChC,WAAW,EACX,KAAK,GACa,EAAoB,EAAE;IACxC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,uBAAuB,CAAC;QAChE,OAAO,EAAE,WAAW;QACpB,KAAK;KACN,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,EAAE;QAC3B,IAAI,CAAC,SAAS;YAAE,OAAO,GAAG,CAAC;QAE3B,OAAO,kBAAkB,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IACnE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAC5B,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,EACd,iBAAiB,GAAG,2BAA2B,GAC7B,EAAoB,EAAE;IACxC,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,GAAG,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC;QAClD,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK;gBAAE,OAAO,GAAG,CAAC;YAEjD,MAAM,SAAS,GAAG,MAAM,kBAAkB,CACxC,WAAW,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,GAAG,CACV,CAAC;YAEF,OAAO,kBAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC;QAC9D,CAAC;QACD,OAAO,EACL,OAAO;YACP,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,KAAK,EAAE,GAAG;YACZ,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,GAAG;YAClC,iBAAiB,CAAC,WAAW,CAAC;QAChC,eAAe,EAAE,iBAAiB;QAClC,KAAK,EAAE,CAAC;KACT,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC,CAAC"}

package/dist/hooks/tokens/useMultiChainBalance.d.ts

@@ -1,6 +1,6 @@
 import type { ChainData, Token } from "@0xsquid/squid-types";
 /**
- * Exposes a hook to get balance depending on the chain type (EVM or Cosmos)
+ * Hook to get balance depending on the chain type
  * @param {ChainData} chain
  * @param {Token} token
  * @param {string} userAddress
@@ -13,5 +13,5 @@ export declare const useMultiChainBalance: ({ chain, token, userAddress, enabled
     userAddress?: string | undefined;
     enabled?: boolean | undefined;
 }) => {
-    balance: string | undefined;
+    balance: string;
 };

package/dist/hooks/tokens/useMultiChainBalance.js

@@ -1,65 +1,68 @@
 import { ChainType } from "@0xsquid/squid-types";
-import { useBitcoinBalance, useCosmosBalance, useEvmBalance, useSolanaBalance, } from "../../hooks/tokens/useBalance";
+import { useMemo } from "react";
+import { useBitcoinBalance, useCosmosBalance, useEvmBalance, useSolanaBalance, useSuiBalance, } from "./useBalance";
 /**
- * Exposes a hook to get balance depending on the chain type (EVM or Cosmos)
+ * Hook to get balance depending on the chain type
  * @param {ChainData} chain
  * @param {Token} token
  * @param {string} userAddress
  * @param {boolean} enabled
  * @returns balance as string
  */
-export const useMultiChainBalance = ({ chain, token, userAddress = "", enabled = true, }) => {
-    /**
-     * Get balance for EVM
-     */
+export const useMultiChainBalance = ({ chain, token, userAddress, enabled = true, }) => {
     const { balance: evmBalance } = useEvmBalance({
-        chain: chain,
+        chain,
         token,
         userAddress,
         enabled: chain?.chainType === ChainType.EVM && enabled,
     });
-    /**
-     * Get balance for Cosmos
-     */
     const { balance: cosmosBalance } = useCosmosBalance({
-        chainData: chain,
+        chain,
         token,
         userAddress,
         enabled: chain?.chainType === ChainType.COSMOS && enabled,
     });
-    /**
-     * Get balance for Bitcoin
-     */
     const { balance: bitcoinBalance } = useBitcoinBalance({
         chain,
-        address: userAddress,
+        token,
+        userAddress,
+        enabled: chain?.chainType === ChainType.BTC && enabled,
     });
-    /**
-     * Get balance for Solana
-     */
     const { balance: solanaBalance } = useSolanaBalance({
         chain,
         token,
         userAddress,
         enabled: chain?.chainType === ChainType.SOLANA && enabled,
     });
-    let balance;
-    switch (chain?.chainType) {
-        case ChainType.EVM:
-            balance = evmBalance;
-            break;
-        case ChainType.COSMOS:
-            balance = cosmosBalance;
-            break;
-        case ChainType.BTC:
-            balance = bitcoinBalance;
-            break;
-        case ChainType.SOLANA:
-            balance = solanaBalance;
-            break;
-        default:
-            balance = undefined;
-    }
+    const { balance: suiBalance } = useSuiBalance({
+        chain,
+        token,
+        userAddress,
+        enabled: chain?.chainType === ChainType.SUI && enabled,
+    });
+    const balance = useMemo(() => {
+        if (!chain?.chainType)
+            return "0";
+        switch (chain.chainType) {
+            case ChainType.EVM:
+                return evmBalance;
+            case ChainType.COSMOS:
+                return cosmosBalance;
+            case ChainType.BTC:
+                return bitcoinBalance;
+            case ChainType.SOLANA:
+                return solanaBalance;
+            case ChainType.SUI:
+                return suiBalance;
+        }
+    }, [
+        chain?.chainType,
+        evmBalance,
+        cosmosBalance,
+        bitcoinBalance,
+        solanaBalance,
+        suiBalance,
+    ]);
     return { balance };
 };
 //# sourceMappingURL=useMultiChainBalance.js.map

package/dist/hooks/tokens/useMultiChainBalance.js.map

@@ -1 +1 @@
-{"version":3,"file":"useMultiChainBalance.js","sourceRoot":"","sources":["../../../src/hooks/tokens/useMultiChainBalance.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,aAAa,EACb,gBAAgB,GACjB,MAAM,+BAA+B,CAAC;AAEvC;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,EACnC,KAAK,EACL,KAAK,EACL,WAAW,GAAG,EAAE,EAChB,OAAO,GAAG,IAAI,GAMf,EAAE,EAAE;IACH;;OAEG;IACH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC;QAC5C,KAAK,EAAE,KAAiB;QACxB,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,GAAG,IAAI,OAAO;KACvD,CAAC,CAAC;IAEH;;OAEG;IACH,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC;QAClD,SAAS,EAAE,KAAoB;QAC/B,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,IAAI,OAAO;KAC1D,CAAC,CAAC;IAEH;;OAEG;IACH,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,GAAG,iBAAiB,CAAC;QACpD,KAAK;QACL,OAAO,EAAE,WAAW;KACrB,CAAC,CAAC;IAEH;;OAEG;IACH,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC;QAClD,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,IAAI,OAAO;KAC1D,CAAC,CAAC;IAEH,IAAI,OAA2B,CAAC;IAEhC,QAAQ,KAAK,EAAE,SAAS,EAAE;QACxB,KAAK,SAAS,CAAC,GAAG;YAChB,OAAO,GAAG,UAAU,CAAC;YACrB,MAAM;QACR,KAAK,SAAS,CAAC,MAAM;YACnB,OAAO,GAAG,aAAa,CAAC;YACxB,MAAM;QACR,KAAK,SAAS,CAAC,GAAG;YAChB,OAAO,GAAG,cAAc,CAAC;YACzB,MAAM;QACR,KAAK,SAAS,CAAC,MAAM;YACnB,OAAO,GAAG,aAAa,CAAC;YACxB,MAAM;QACR;YACE,OAAO,GAAG,SAAS,CAAC;KACvB;IAED,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC,CAAC"}
+{"version":3,"file":"useMultiChainBalance.js","sourceRoot":"","sources":["../../../src/hooks/tokens/useMultiChainBalance.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,aAAa,EACb,gBAAgB,EAChB,aAAa,GACd,MAAM,cAAc,CAAC;AAEtB;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,EACnC,KAAK,EACL,KAAK,EACL,WAAW,EACX,OAAO,GAAG,IAAI,GAMf,EAAE,EAAE;IACH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC;QAC5C,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,GAAG,IAAI,OAAO;KACvD,CAAC,CAAC;IAEH,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC;QAClD,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,IAAI,OAAO;KAC1D,CAAC,CAAC;IAEH,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,GAAG,iBAAiB,CAAC;QACpD,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,GAAG,IAAI,OAAO;KACvD,CAAC,CAAC;IAEH,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC;QAClD,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,MAAM,IAAI,OAAO;KAC1D,CAAC,CAAC;IAEH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC;QAC5C,KAAK;QACL,KAAK;QACL,WAAW;QACX,OAAO,EAAE,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,GAAG,IAAI,OAAO;KACvD,CAAC,CAAC;IAEH,MAAM,OAAO,GAAW,OAAO,CAAC,GAAG,EAAE;QACnC,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO,GAAG,CAAC;QAElC,QAAQ,KAAK,CAAC,SAAS,EAAE;YACvB,KAAK,SAAS,CAAC,GAAG;gBAChB,OAAO,UAAU,CAAC;YACpB,KAAK,SAAS,CAAC,MAAM;gBACnB,OAAO,aAAa,CAAC;YACvB,KAAK,SAAS,CAAC,GAAG;gBAChB,OAAO,cAAc,CAAC;YACxB,KAAK,SAAS,CAAC,MAAM;gBACnB,OAAO,aAAa,CAAC;YACvB,KAAK,SAAS,CAAC,GAAG;gBAChB,OAAO,UAAU,CAAC;SACrB;IACH,CAAC,EAAE;QACD,KAAK,EAAE,SAAS;QAChB,UAAU;QACV,aAAa;QACb,cAAc;QACd,aAAa;QACb,UAAU;KACX,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC,CAAC"}

package/dist/hooks/tokens/useNativeBalance.d.ts

@@ -1,24 +1,22 @@
 import { ChainData } from "@0xsquid/squid-types";
-export declare const useBitcoinNativeBalance: (chain?: ChainData, address?: string) => {
-    data: string | undefined;
+interface Balance {
+    decimals: number;
+    value: bigint;
+}
+interface UseNativeBalanceResult {
+    balance: Balance | undefined;
     isLoading: boolean;
-    nativeToken: import("@0xsquid/squid-types").Token | undefined;
-};
-export declare const useSolanaNativeBalance: (chain?: ChainData, address?: string) => {
-    data: string | undefined;
-    isLoading: boolean;
-    nativeToken: import("@0xsquid/squid-types").Token | undefined;
-};
+}
+interface UseNativeBalanceOptions {
+    chain: ChainData | undefined;
+    address: string | undefined;
+}
+export declare const useBitcoinNativeBalance: ({ chain, address, }: UseNativeBalanceOptions) => UseNativeBalanceResult;
+export declare const useSolanaNativeBalance: ({ chain, address, }: UseNativeBalanceOptions) => UseNativeBalanceResult;
+export declare const useSuiNativeBalance: ({ address, chain, }: UseNativeBalanceOptions) => UseNativeBalanceResult;
 export declare const useNativeBalance: (chain?: ChainData) => {
-    nativeBalance: {
-        decimals: number;
-        formatted: string;
-        symbol: string;
-        value: bigint;
-    } | {
-        decimals: number;
-        value: bigint;
-    } | undefined;
+    nativeBalance: Balance | undefined;
     nativeBalanceFormatted: string | undefined;
     isLoading: boolean;
 };
+export {};

package/dist/hooks/tokens/useNativeBalance.js

@@ -1,4 +1,4 @@
-import { ChainType, } from "@0xsquid/squid-types";
+import { ChainType } from "@0xsquid/squid-types";
 import { useQuery } from "@tanstack/react-query";
 import { useMemo } from "react";
 import { useBalance as useWagmiBalance } from "wagmi";
@@ -8,37 +8,58 @@ import { formatBNToReadable, isWalletAddressValid, parseToBigInt, } from "../../
 import { getBitcoinNativeBalance, getSolanaNativeBalance, } from "../../services/external/rpcService";
 import { useCosmosForChain } from "../cosmos/useCosmosForChain";
 import { useWallet } from "../wallet/useWallet";
-import { useCosmosBalance } from "./useBalance";
+import { useCosmosBalance, useSuiBalance } from "./useBalance";
 import { useNativeTokenForChain } from "./useNativeTokenForChain";
-const useEvmNativeBalance = (chain, evmAddress) => {
-    const { isConnected: isEvmConnected } = useWallet();
-    return useWagmiBalance({
-        address: evmAddress,
+const useEvmNativeBalance = ({ address, chain, }) => {
+    const { isChainTypeConnected } = useWallet();
+    const { data: nativeEvmBalance, isLoading } = useWagmiBalance({
+        address: address,
         chainId: Number(chain?.chainId),
         query: {
-            enabled: isEvmConnected(ChainType.EVM) && !!evmAddress,
+            enabled: isChainTypeConnected(ChainType.EVM) && !!address,
             refetchInterval: 10000,
             staleTime: 5000,
             refetchOnWindowFocus: true,
             refetchOnMount: true,
         },
     });
+    const balance = useMemo(() => {
+        if (nativeEvmBalance?.decimals && nativeEvmBalance?.value) {
+            return {
+                decimals: nativeEvmBalance.decimals,
+                value: nativeEvmBalance.value,
+            };
+        }
+        return;
+    }, [nativeEvmBalance?.decimals, nativeEvmBalance?.value]);
+    return {
+        balance,
+        isLoading,
+    };
 };
-const useCosmosNativeBalance = (chain, cosmosAddress) => {
+const useCosmosNativeBalance = ({ address, chain, }) => {
     const { isConnected: isCosmosConnected } = useCosmosContext();
     const { nativeToken: nativeCosmosToken } = useNativeTokenForChain(chain);
-    const { balance: nativeCosmosBalance, isLoading } = useCosmosBalance({
-        chainData: chain,
+    const { balance: rawBalance, isLoading } = useCosmosBalance({
+        chain,
         token: nativeCosmosToken,
-        userAddress: cosmosAddress,
+        userAddress: address,
         enabled: isCosmosConnected && chain?.chainType === ChainType.COSMOS,
         refreshIntervalMs: 10000,
     });
-    return { nativeCosmosBalance, nativeCosmosToken, isLoading };
+    const balance = useMemo(() => {
+        if (nativeCosmosToken?.decimals && rawBalance) {
+            return {
+                decimals: nativeCosmosToken.decimals,
+                value: parseToBigInt(rawBalance, nativeCosmosToken.decimals),
+            };
+        }
+    }, [nativeCosmosToken?.decimals, rawBalance]);
+    return { balance, isLoading };
 };
-export const useBitcoinNativeBalance = (chain, address) => {
+export const useBitcoinNativeBalance = ({ chain, address, }) => {
     const { nativeToken } = useNativeTokenForChain(chain);
-    const { data, isLoading } = useQuery(keys().balance(chain?.chainId, nativeToken?.address, address), async () => {
+    const { data: rawBalance = "0", isLoading } = useQuery(keys().balance(chain?.chainId, nativeToken?.address, address), async () => {
         if (!address || !nativeToken)
             return "0";
         const balance = await getBitcoinNativeBalance(address);
@@ -52,15 +73,22 @@ export const useBitcoinNativeBalance = (chain, address) => {
         refetchOnWindowFocus: true,
         refetchOnMount: true,
     });
+    const balance = useMemo(() => {
+        if (nativeToken?.decimals && rawBalance) {
+            return {
+                decimals: nativeToken.decimals,
+                value: parseToBigInt(rawBalance, nativeToken.decimals),
+            };
+        }
+    }, [nativeToken?.decimals, rawBalance]);
     return {
-        data,
+        balance,
         isLoading,
-        nativeToken,
     };
 };
-export const useSolanaNativeBalance = (chain, address) => {
+export const useSolanaNativeBalance = ({ chain, address, }) => {
     const { nativeToken } = useNativeTokenForChain(chain);
-    const { data, isLoading } = useQuery(keys().balance(chain?.chainId, nativeToken?.address, address), async () => {
+    const { data: rawBalance, isLoading } = useQuery(keys().balance(chain?.chainId, nativeToken?.address, address), async () => {
         const balance = await getSolanaNativeBalance(address);
         return formatBNToReadable(balance, nativeToken.decimals);
     }, {
@@ -72,61 +100,99 @@ export const useSolanaNativeBalance = (chain, address) => {
         refetchOnWindowFocus: true,
         refetchOnMount: true,
     });
+    const balance = useMemo(() => {
+        if (nativeToken?.decimals && rawBalance) {
+            return {
+                decimals: nativeToken.decimals,
+                value: parseToBigInt(rawBalance, nativeToken.decimals),
+            };
+        }
+    }, [nativeToken?.decimals, rawBalance]);
     return {
-        data,
+        balance,
+        isLoading,
+    };
+};
+export const useSuiNativeBalance = ({ address, chain, }) => {
+    const { nativeToken } = useNativeTokenForChain(chain);
+    const { balance: rawBalance, isLoading } = useSuiBalance({
+        chain,
+        token: nativeToken,
+        userAddress: address,
+    });
+    const balance = useMemo(() => {
+        if (nativeToken?.decimals && rawBalance) {
+            return {
+                decimals: nativeToken.decimals,
+                value: parseToBigInt(rawBalance, nativeToken.decimals),
+            };
+        }
+    }, [nativeToken?.decimals, rawBalance]);
+    return {
+        balance,
         isLoading,
-        nativeToken,
     };
 };
 export const useNativeBalance = (chain) => {
     const { connectedAddresses } = useWallet();
     const { data: cosmosAddressForChain } = useCosmosForChain(chain);
-    const { data: nativeEvmBalance, isLoading: isEvmLoading } = useEvmNativeBalance(chain, connectedAddresses[ChainType.EVM]);
-    const { nativeCosmosBalance, nativeCosmosToken, isLoading: isCosmosLoading, } = useCosmosNativeBalance(chain, cosmosAddressForChain);
-    const { data: nativeBitcoinBalance, isLoading: isBitcoinLoading, nativeToken: nativeBitcoinToken, } = useBitcoinNativeBalance(chain, connectedAddresses[ChainType.BTC]);
-    const { data: nativeSolanaBalance, isLoading: isSolanaLoading, nativeToken: nativeSolanaToken, } = useSolanaNativeBalance(chain, connectedAddresses[ChainType.SOLANA]);
-    const nativeBalance = useMemo(() => {
+    // Cosmos is a special case because the address changes on every chain
+    // so we can't use the default cosmos connected address
+    const { balance: nativeCosmosBalance, isLoading: isCosmosLoading } = useCosmosNativeBalance({
+        address: cosmosAddressForChain,
+        chain,
+    });
+    const { balance: nativeEvmBalance, isLoading: isEvmLoading } = useEvmNativeBalance({ address: connectedAddresses[ChainType.EVM], chain });
+    const { balance: nativeBitcoinBalance, isLoading: isBitcoinLoading } = useBitcoinNativeBalance({
+        address: connectedAddresses[ChainType.BTC],
+        chain,
+    });
+    const { balance: nativeSolanaBalance, isLoading: isSolanaLoading } = useSolanaNativeBalance({
+        address: connectedAddresses[ChainType.SOLANA],
+        chain,
+    });
+    const { balance: nativeSuiBalance, isLoading: isSuiLoading } = useSuiNativeBalance({
+        address: connectedAddresses[ChainType.SUI],
+        chain,
+    });
+    const { nativeBalance, nativeBalanceFormatted } = useMemo(() => {
+        let balance;
         switch (chain?.chainType) {
             case ChainType.EVM:
-                return nativeEvmBalance?.decimals && nativeEvmBalance?.value
-                    ? nativeEvmBalance
-                    : undefined;
+                balance = nativeEvmBalance;
+                break;
             case ChainType.COSMOS:
-                return nativeCosmosToken?.decimals && nativeCosmosBalance
-                    ? {
-                        decimals: nativeCosmosToken.decimals,
-                        value: parseToBigInt(nativeCosmosBalance, nativeCosmosToken.decimals),
-                    }
-                    : undefined;
+                balance = nativeCosmosBalance;
+                break;
             case ChainType.BTC:
-                return nativeBitcoinBalance && nativeBitcoinToken?.decimals
-                    ? {
-                        decimals: nativeBitcoinToken.decimals,
-                        value: parseToBigInt(nativeBitcoinBalance, nativeBitcoinToken.decimals),
-                    }
-                    : undefined;
+                balance = nativeBitcoinBalance;
+                break;
             case ChainType.SOLANA:
-                return nativeSolanaBalance && nativeSolanaToken?.decimals
-                    ? {
-                        decimals: nativeSolanaToken.decimals,
-                        value: parseToBigInt(nativeSolanaBalance, nativeSolanaToken.decimals),
-                    }
-                    : undefined;
-            default:
-                return undefined;
+                balance = nativeSolanaBalance;
+                break;
+            case ChainType.SUI:
+                balance = nativeSuiBalance;
+                break;
         }
+        const balanceFormatted = !!balance
+            ? formatBNToReadable(balance.value, balance.decimals)
+            : undefined;
+        return {
+            nativeBalance: balance,
+            nativeBalanceFormatted: balanceFormatted,
+        };
     }, [
         chain?.chainType,
         nativeEvmBalance,
-        nativeCosmosToken?.decimals,
         nativeCosmosBalance,
-        nativeBitcoinToken?.decimals,
         nativeBitcoinBalance,
         nativeSolanaBalance,
-        nativeSolanaToken?.decimals,
+        nativeSuiBalance,
     ]);
     const isLoading = useMemo(() => {
-        switch (chain?.chainType) {
+        if (!chain?.chainType)
+            return false;
+        switch (chain.chainType) {
             case ChainType.EVM:
                 return isEvmLoading;
             case ChainType.COSMOS:
@@ -135,8 +201,8 @@ export const useNativeBalance = (chain) => {
                 return isBitcoinLoading;
             case ChainType.SOLANA:
                 return isSolanaLoading;
-            default:
-                return false;
+            case ChainType.SUI:
+                return isSuiLoading;
         }
     }, [
         chain?.chainType,
@@ -144,9 +210,8 @@ export const useNativeBalance = (chain) => {
         isCosmosLoading,
         isBitcoinLoading,
         isSolanaLoading,
+        isSuiLoading,
     ]);
-    const nativeBalanceFormatted = nativeBalance &&
-        formatBNToReadable(nativeBalance?.value, nativeBalance?.decimals);
     return { nativeBalance, nativeBalanceFormatted, isLoading };
 };
 //# sourceMappingURL=useNativeBalance.js.map