@metamask/assets-controller 3.1.0 → 3.2.0

package/dist/data-sources/evm-rpc-services/services/BalanceFetcher.mjs

@@ -9,14 +9,20 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _BalanceFetcher_instances, _BalanceFetcher_multicallClient, _BalanceFetcher_messenger, _BalanceFetcher_config, _BalanceFetcher_onBalanceUpdate, _BalanceFetcher_processBalanceResponses, _BalanceFetcher_formatBalance;
+var _BalanceFetcher_instances, _BalanceFetcher_multicallClient, _BalanceFetcher_messenger, _BalanceFetcher_config, _BalanceFetcher_onBalanceUpdate, _BalanceFetcher_getAssetsToFetch, _BalanceFetcher_fetchBalances, _BalanceFetcher_processBalanceResponses, _BalanceFetcher_formatBalance;
 import { StaticIntervalPollingControllerOnly } from "@metamask/polling-controller";
+import { parseCaipAssetType } from "@metamask/utils";
+import { ZERO_ADDRESS } from "../../../utils/constants.mjs";
 import { reduceInBatchesSerially } from "../utils/index.mjs";
 const DEFAULT_BALANCE_INTERVAL = 30000; // 30 seconds
-const ZERO_ADDRESS = '0x0000000000000000000000000000000000000000';
 /**
  * BalanceFetcher - Fetches token balances via multicall.
  * Extends StaticIntervalPollingControllerOnly for built-in polling support.
+ *
+ * Callers provide CAIP-19 asset IDs; the fetcher extracts on-chain addresses
+ * (or uses the zero address for native assets) and maps multicall responses
+ * back to the original asset IDs. This ensures the returned balance entries
+ * always carry the correct identifier regardless of chain.
  */
 export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
     constructor(multicallClient, messenger, config) {
@@ -31,7 +37,6 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
         __classPrivateFieldSet(this, _BalanceFetcher_config, {
             defaultBatchSize: config?.defaultBatchSize ?? 300,
             defaultTimeoutMs: config?.defaultTimeoutMs ?? 30000,
-            includeNativeByDefault: config?.includeNativeByDefault ?? true,
         }, "f");
         // Set the polling interval
         this.setIntervalLength(config?.pollingInterval ?? DEFAULT_BALANCE_INTERVAL);
@@ -51,69 +56,36 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
      * @param input - The polling input.
      */
     async _executePoll(input) {
-        const result = await this.fetchBalances(input.chainId, input.accountId, input.accountAddress);
+        const result = await __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_fetchBalances).call(this, input.chainId, input.accountId, input.accountAddress);
         if (__classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f") && result.balances.length > 0) {
             await __classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f").call(this, result);
         }
     }
-    getTokensToFetch(chainId, accountId) {
-        const state = __classPrivateFieldGet(this, _BalanceFetcher_messenger, "f").call('AssetsController:getState');
-        if (!state?.assetsBalance) {
-            return [];
-        }
-        const accountBalances = state.assetsBalance[accountId];
-        if (!accountBalances) {
-            return [];
-        }
-        // Convert hex chainId to decimal for CAIP-2 matching
-        const chainIdDecimal = parseInt(chainId, 16);
-        const caipChainPrefix = `eip155:${chainIdDecimal}/`;
-        const tokenMap = new Map();
-        for (const assetId of Object.keys(accountBalances)) {
-            // Only process ERC20 tokens on the current chain
-            if (assetId.startsWith(caipChainPrefix) && assetId.includes('/erc20:')) {
-                // Parse token address from CAIP-19: eip155:1/erc20:0x...
-                const tokenAddress = assetId.split('/erc20:')[1];
-                if (tokenAddress) {
-                    const lowerAddress = tokenAddress.toLowerCase();
-                    if (!tokenMap.has(lowerAddress)) {
-                        tokenMap.set(lowerAddress, {
-                            address: tokenAddress,
-                            symbol: '',
-                        });
-                    }
-                }
-            }
-        }
-        return Array.from(tokenMap.values());
-    }
-    async fetchBalances(chainId, accountId, accountAddress, options) {
-        const tokens = this.getTokensToFetch(chainId, accountId);
-        const tokenAddresses = tokens.map((token) => token.address);
-        return this.fetchBalancesForTokens(chainId, accountId, accountAddress, tokenAddresses, options, tokens);
-    }
-    async fetchBalancesForTokens(chainId, accountId, accountAddress, tokenAddresses, options, tokenInfos) {
-        const batchSize = options?.batchSize ?? __classPrivateFieldGet(this, _BalanceFetcher_config, "f").defaultBatchSize;
-        const includeNative = options?.includeNative ?? __classPrivateFieldGet(this, _BalanceFetcher_config, "f").includeNativeByDefault;
+    /**
+     * Fetch balances for the given assets via multicall.
+     *
+     * Each entry bundles a CAIP-19 asset ID with its on-chain address and
+     * optional decimals.
+     *
+     * @param chainId - Hex chain ID.
+     * @param accountId - Account UUID.
+     * @param accountAddress - On-chain address of the account.
+     * @param assets - Asset fetch entries to fetch balances for.
+     * @returns Balance fetch result.
+     */
+    async fetchBalancesForAssets(chainId, accountId, accountAddress, assets) {
         const timestamp = Date.now();
-        const tokenInfoMap = new Map();
-        if (tokenInfos) {
-            for (const info of tokenInfos) {
-                tokenInfoMap.set(info.address.toLowerCase(), info);
-            }
-        }
+        // Build a single map keyed by lowercase address that holds all info
+        // needed to match multicall responses back to their original entries.
         const balanceRequests = [];
-        if (includeNative) {
-            balanceRequests.push({
-                tokenAddress: ZERO_ADDRESS,
-                accountAddress,
-            });
-        }
-        for (const tokenAddress of tokenAddresses) {
-            balanceRequests.push({
-                tokenAddress,
-                accountAddress,
-            });
+        const entryByAddress = new Map();
+        for (const entry of assets) {
+            const lowerAddress = entry.address.toLowerCase();
+            if (entryByAddress.has(lowerAddress)) {
+                continue; // deduplicate
+            }
+            entryByAddress.set(lowerAddress, entry);
+            balanceRequests.push({ tokenAddress: entry.address, accountAddress });
         }
         if (balanceRequests.length === 0) {
             return {
@@ -127,14 +99,14 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
         }
         const result = await reduceInBatchesSerially({
             values: balanceRequests,
-            batchSize,
+            batchSize: __classPrivateFieldGet(this, _BalanceFetcher_config, "f").defaultBatchSize,
             initialResult: {
                 balances: [],
                 failedAddresses: [],
             },
             eachBatch: async (workingResult, batch) => {
                 const responses = await __classPrivateFieldGet(this, _BalanceFetcher_multicallClient, "f").batchBalanceOf(chainId, batch);
-                return __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_processBalanceResponses).call(this, responses, workingResult, chainId, accountId, timestamp, tokenInfoMap);
+                return __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_processBalanceResponses).call(this, responses, workingResult, chainId, accountId, timestamp, entryByAddress);
             },
         });
         return {
@@ -146,37 +118,76 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
         };
     }
 }
-_BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new WeakMap(), _BalanceFetcher_config = new WeakMap(), _BalanceFetcher_onBalanceUpdate = new WeakMap(), _BalanceFetcher_instances = new WeakSet(), _BalanceFetcher_processBalanceResponses = function _BalanceFetcher_processBalanceResponses(responses, accumulator, chainId, accountId, timestamp, tokenInfoMap) {
+_BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new WeakMap(), _BalanceFetcher_config = new WeakMap(), _BalanceFetcher_onBalanceUpdate = new WeakMap(), _BalanceFetcher_instances = new WeakSet(), _BalanceFetcher_getAssetsToFetch = function _BalanceFetcher_getAssetsToFetch(chainId, accountId) {
+    const state = __classPrivateFieldGet(this, _BalanceFetcher_messenger, "f").call('AssetsController:getState');
+    const accountBalances = state?.assetsBalance?.[accountId];
+    if (!accountBalances) {
+        return [];
+    }
+    // Convert hex chainId to decimal for CAIP-2 matching
+    // This is safe because we are filtring with an accountId that is for evm balances only
+    const chainIdDecimal = parseInt(chainId, 16).toString();
+    const assetsToFetch = new Map();
+    for (const assetId of Object.keys(accountBalances)) {
+        const { chain: { reference: chainReference }, assetNamespace, assetReference, } = parseCaipAssetType(assetId);
+        if (chainReference === chainIdDecimal) {
+            const assetIdLowerCase = assetId.toLowerCase();
+            if (assetsToFetch.has(assetIdLowerCase)) {
+                continue;
+            }
+            const isNative = assetNamespace === 'slip44';
+            const tokenAddress = isNative
+                ? ZERO_ADDRESS
+                : assetReference.toLowerCase();
+            assetsToFetch.set(assetIdLowerCase, {
+                assetId,
+                address: tokenAddress,
+            });
+        }
+    }
+    return Array.from(assetsToFetch.values());
+}, _BalanceFetcher_fetchBalances = 
+/**
+ * Fetch balances for assets already tracked in state for the given
+ * account and chain.
+ *
+ * @param chainId - Hex chain ID.
+ * @param accountId - Account UUID.
+ * @param accountAddress - On-chain address of the account.
+ * @returns Balance fetch result.
+ */
+async function _BalanceFetcher_fetchBalances(chainId, accountId, accountAddress) {
+    const assets = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_getAssetsToFetch).call(this, chainId, accountId);
+    return this.fetchBalancesForAssets(chainId, accountId, accountAddress, assets);
+}, _BalanceFetcher_processBalanceResponses = function _BalanceFetcher_processBalanceResponses(responses, accumulator, chainId, accountId, timestamp, entryByAddress) {
     const { balances, failedAddresses } = accumulator;
     for (const response of responses) {
         if (!response.success) {
             failedAddresses.push(response.tokenAddress);
             continue;
         }
+        const lowerAddress = response.tokenAddress.toLowerCase();
+        const entry = entryByAddress.get(lowerAddress);
+        if (!entry) {
+            continue;
+        }
         const balance = response.balance ?? '0';
-        const chainIdDecimal = parseInt(chainId, 16);
-        const isNative = response.tokenAddress.toLowerCase() === ZERO_ADDRESS.toLowerCase();
-        const tokenInfo = tokenInfoMap.get(response.tokenAddress.toLowerCase());
-        // Native uses 18. ERC-20: format when decimals are known; otherwise pass raw
-        // `balance` through so RpcDataSource can resolve decimals (state → list → RPC).
+        const isNative = lowerAddress === ZERO_ADDRESS.toLowerCase();
         let decimals;
         let formattedBalance;
         if (isNative) {
             decimals = 18;
             formattedBalance = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_formatBalance).call(this, balance, decimals);
         }
-        else if (tokenInfo?.decimals === undefined) {
+        else if (entry.decimals === undefined) {
             formattedBalance = balance;
         }
         else {
-            decimals = tokenInfo.decimals;
+            decimals = entry.decimals;
             formattedBalance = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_formatBalance).call(this, balance, decimals);
         }
-        const assetId = isNative
-            ? `eip155:${chainIdDecimal}/slip44:60`
-            : `eip155:${chainIdDecimal}/erc20:${response.tokenAddress.toLowerCase()}`;
         const balanceEntry = {
-            assetId,
+            assetId: entry.assetId,
             accountId,
             chainId,
             balance,

package/dist/data-sources/evm-rpc-services/services/BalanceFetcher.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"BalanceFetcher.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,mCAAmC,EAAE,qCAAqC;AAgBnF,OAAO,EAAE,uBAAuB,EAAE,2BAAiB;AAEnD,MAAM,wBAAwB,GAAG,KAAM,CAAC,CAAC,aAAa;AAEtD,MAAM,YAAY,GAChB,4CAAuD,CAAC;AAoC1D;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,mCAAmC,EAAuB;IAS5F,YACE,eAAgC,EAChC,SAAkC,EAClC,MAA6B;QAE7B,KAAK,EAAE,CAAC;;QAbD,kDAAkC;QAElC,4CAAoC;QAEpC,yCAAiE;QAE1E,kDAAsD;QAQpD,uBAAA,IAAI,mCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6BAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,0BAAW;YACb,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,GAAG;YACjD,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,KAAK;YACnD,sBAAsB,EAAE,MAAM,EAAE,sBAAsB,IAAI,IAAI;SAC/D,MAAA,CAAC;QAEF,2BAA2B;QAC3B,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,eAAe,IAAI,wBAAwB,CAAC,CAAC;IAC9E,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,QAAiC;QAClD,uBAAA,IAAI,mCAAoB,QAAQ,MAAA,CAAC;IACnC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,KAA0B;QAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CACrC,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,CACrB,CAAC;QAEF,IAAI,uBAAA,IAAI,uCAAiB,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,uBAAA,IAAI,uCAAiB,MAArB,IAAI,EAAkB,MAAM,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,OAAgB,EAAE,SAAoB;QACrD,MAAM,KAAK,GAAG,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAEhE,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,CAAC;YAC1B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,eAAe,GAAG,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,qDAAqD;QACrD,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,eAAe,GAAG,UAAU,cAAc,GAAG,CAAC;QAEpD,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;QAEnD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACnD,iDAAiD;YACjD,IAAI,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACvE,yDAAyD;gBACzD,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAY,CAAC;gBAC5D,IAAI,YAAY,EAAE,CAAC;oBACjB,MAAM,YAAY,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;oBAChD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;wBAChC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE;4BACzB,OAAO,EAAE,YAAY;4BACrB,MAAM,EAAE,EAAE;yBACX,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,OAA6B;QAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACzD,MAAM,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAE5D,OAAO,IAAI,CAAC,sBAAsB,CAChC,OAAO,EACP,SAAS,EACT,cAAc,EACd,cAAc,EACd,OAAO,EACP,MAAM,CACP,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,cAAyB,EACzB,OAA6B,EAC7B,UAA6B;QAE7B,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,uBAAA,IAAI,8BAAQ,CAAC,gBAAgB,CAAC;QACtE,MAAM,aAAa,GACjB,OAAO,EAAE,aAAa,IAAI,uBAAA,IAAI,8BAAQ,CAAC,sBAAsB,CAAC;QAChE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0B,CAAC;QACvD,IAAI,UAAU,EAAE,CAAC;YACf,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,MAAM,eAAe,GAAuB,EAAE,CAAC;QAE/C,IAAI,aAAa,EAAE,CAAC;YAClB,eAAe,CAAC,IAAI,CAAC;gBACnB,YAAY,EAAE,YAAY;gBAC1B,cAAc;aACf,CAAC,CAAC;QACL,CAAC;QAED,KAAK,MAAM,YAAY,IAAI,cAAc,EAAE,CAAC;YAC1C,eAAe,CAAC,IAAI,CAAC;gBACnB,YAAY;gBACZ,cAAc;aACf,CAAC,CAAC;QACL,CAAC;QAED,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO;gBACP,SAAS;gBACT,cAAc;gBACd,QAAQ,EAAE,EAAE;gBACZ,eAAe,EAAE,EAAE;gBACnB,SAAS;aACV,CAAC;QACJ,CAAC;QAOD,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAG1C;YACA,MAAM,EAAE,eAAe;YACvB,SAAS;YACT,aAAa,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,eAAe,EAAE,EAAE;aACpB;YACD,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,uCAAiB,CAAC,cAAc,CAC1D,OAAO,EACP,KAAK,CACN,CAAC;gBAEF,OAAO,uBAAA,IAAI,0EAAyB,MAA7B,IAAI,EACT,SAAS,EACT,aAAiC,EACjC,OAAO,EACP,SAAS,EACT,SAAS,EACT,YAAY,CACb,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,OAAO;YACL,OAAO;YACP,SAAS;YACT,cAAc;YACd,GAAG,MAAM;YACT,SAAS;SACV,CAAC;IACJ,CAAC;CAuFF;2TApFG,SAA8B,EAC9B,WAGC,EACD,OAAgB,EAChB,SAAoB,EACpB,SAAiB,EACjB,YAAyC;IAKzC,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,WAAW,CAAC;IAElD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,GAAG,CAAC;QACxC,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,QAAQ,GACZ,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE,KAAK,YAAY,CAAC,WAAW,EAAE,CAAC;QACrE,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC;QAExE,6EAA6E;QAC7E,gFAAgF;QAChF,IAAI,QAA4B,CAAC;QACjC,IAAI,gBAAwB,CAAC;QAC7B,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,GAAG,EAAE,CAAC;YACd,gBAAgB,GAAG,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC;aAAM,IAAI,SAAS,EAAE,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC7C,gBAAgB,GAAG,OAAO,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;YAC9B,gBAAgB,GAAG,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC;QACD,MAAM,OAAO,GAAkB,QAAQ;YACrC,CAAC,CAAE,UAAU,cAAc,YAA8B;YACzD,CAAC,CAAE,UAAU,cAAc,UAAU,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE,EAAoB,CAAC;QAE/F,MAAM,YAAY,GAAiB;YACjC,OAAO;YACP,SAAS;YACT,OAAO;YACP,OAAO;YACP,gBAAgB;YAChB,SAAS;SACV,CAAC;QACF,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACjC,YAAY,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACnC,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9B,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;AACvC,CAAC,yEAEc,UAAkB,EAAE,QAAgB;IACjD,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;QACvB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,MAAM,CAAC,EAAE,IAAI,QAAQ,CAAC,CAAC;QAEvC,MAAM,WAAW,GAAG,aAAa,GAAG,OAAO,CAAC;QAC5C,MAAM,SAAS,GAAG,aAAa,GAAG,OAAO,CAAC;QAC1C,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnE,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAE5D,IAAI,iBAAiB,KAAK,EAAE,EAAE,CAAC;YAC7B,OAAO,WAAW,CAAC,QAAQ,EAAE,CAAC;QAChC,CAAC;QAED,OAAO,GAAG,WAAW,IAAI,iBAAiB,EAAE,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,UAAU,CAAC;IACpB,CAAC;AACH,CAAC","sourcesContent":["import { StaticIntervalPollingControllerOnly } from '@metamask/polling-controller';\nimport type { CaipAssetType } from '@metamask/utils';\n\nimport type { MulticallClient } from '../clients';\nimport type {\n  AccountId,\n  Address,\n  AssetBalance,\n  AssetsBalanceState,\n  BalanceFetchOptions,\n  BalanceFetchResult,\n  BalanceOfRequest,\n  BalanceOfResponse,\n  ChainId,\n  TokenFetchInfo,\n} from '../types';\nimport { reduceInBatchesSerially } from '../utils';\n\nconst DEFAULT_BALANCE_INTERVAL = 30_000; // 30 seconds\n\nconst ZERO_ADDRESS: Address =\n  '0x0000000000000000000000000000000000000000' as Address;\n\n/**\n * Minimal messenger interface for BalanceFetcher.\n */\nexport type BalanceFetcherMessenger = {\n  call: (action: 'AssetsController:getState') => AssetsBalanceState;\n};\n\nexport type BalanceFetcherConfig = {\n  defaultBatchSize?: number;\n  defaultTimeoutMs?: number;\n  includeNativeByDefault?: boolean;\n  /** Polling interval in ms (default: 30s) */\n  pollingInterval?: number;\n};\n\n/**\n * Polling input for BalanceFetcher - identifies what to poll for.\n */\nexport type BalancePollingInput = {\n  /** Chain ID (hex format) */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n};\n\n/**\n * Callback type for balance updates.\n */\nexport type OnBalanceUpdateCallback = (\n  result: BalanceFetchResult,\n) => void | Promise<void>;\n\n/**\n * BalanceFetcher - Fetches token balances via multicall.\n * Extends StaticIntervalPollingControllerOnly for built-in polling support.\n */\nexport class BalanceFetcher extends StaticIntervalPollingControllerOnly<BalancePollingInput>() {\n  readonly #multicallClient: MulticallClient;\n\n  readonly #messenger: BalanceFetcherMessenger;\n\n  readonly #config: Required<Omit<BalanceFetcherConfig, 'pollingInterval'>>;\n\n  #onBalanceUpdate: OnBalanceUpdateCallback | undefined;\n\n  constructor(\n    multicallClient: MulticallClient,\n    messenger: BalanceFetcherMessenger,\n    config?: BalanceFetcherConfig,\n  ) {\n    super();\n    this.#multicallClient = multicallClient;\n    this.#messenger = messenger;\n    this.#config = {\n      defaultBatchSize: config?.defaultBatchSize ?? 300,\n      defaultTimeoutMs: config?.defaultTimeoutMs ?? 30000,\n      includeNativeByDefault: config?.includeNativeByDefault ?? true,\n    };\n\n    // Set the polling interval\n    this.setIntervalLength(config?.pollingInterval ?? DEFAULT_BALANCE_INTERVAL);\n  }\n\n  /**\n   * Set the callback to receive balance updates during polling.\n   *\n   * @param callback - Function to call with balance results.\n   */\n  setOnBalanceUpdate(callback: OnBalanceUpdateCallback): void {\n    this.#onBalanceUpdate = callback;\n  }\n\n  /**\n   * Execute a poll cycle (required by base class).\n   * Fetches balances and calls the update callback.\n   *\n   * @param input - The polling input.\n   */\n  async _executePoll(input: BalancePollingInput): Promise<void> {\n    const result = await this.fetchBalances(\n      input.chainId,\n      input.accountId,\n      input.accountAddress,\n    );\n\n    if (this.#onBalanceUpdate && result.balances.length > 0) {\n      await this.#onBalanceUpdate(result);\n    }\n  }\n\n  getTokensToFetch(chainId: ChainId, accountId: AccountId): TokenFetchInfo[] {\n    const state = this.#messenger.call('AssetsController:getState');\n\n    if (!state?.assetsBalance) {\n      return [];\n    }\n\n    const accountBalances = state.assetsBalance[accountId];\n    if (!accountBalances) {\n      return [];\n    }\n\n    // Convert hex chainId to decimal for CAIP-2 matching\n    const chainIdDecimal = parseInt(chainId, 16);\n    const caipChainPrefix = `eip155:${chainIdDecimal}/`;\n\n    const tokenMap = new Map<string, TokenFetchInfo>();\n\n    for (const assetId of Object.keys(accountBalances)) {\n      // Only process ERC20 tokens on the current chain\n      if (assetId.startsWith(caipChainPrefix) && assetId.includes('/erc20:')) {\n        // Parse token address from CAIP-19: eip155:1/erc20:0x...\n        const tokenAddress = assetId.split('/erc20:')[1] as Address;\n        if (tokenAddress) {\n          const lowerAddress = tokenAddress.toLowerCase();\n          if (!tokenMap.has(lowerAddress)) {\n            tokenMap.set(lowerAddress, {\n              address: tokenAddress,\n              symbol: '',\n            });\n          }\n        }\n      }\n    }\n\n    return Array.from(tokenMap.values());\n  }\n\n  async fetchBalances(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    options?: BalanceFetchOptions,\n  ): Promise<BalanceFetchResult> {\n    const tokens = this.getTokensToFetch(chainId, accountId);\n    const tokenAddresses = tokens.map((token) => token.address);\n\n    return this.fetchBalancesForTokens(\n      chainId,\n      accountId,\n      accountAddress,\n      tokenAddresses,\n      options,\n      tokens,\n    );\n  }\n\n  async fetchBalancesForTokens(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    tokenAddresses: Address[],\n    options?: BalanceFetchOptions,\n    tokenInfos?: TokenFetchInfo[],\n  ): Promise<BalanceFetchResult> {\n    const batchSize = options?.batchSize ?? this.#config.defaultBatchSize;\n    const includeNative =\n      options?.includeNative ?? this.#config.includeNativeByDefault;\n    const timestamp = Date.now();\n\n    const tokenInfoMap = new Map<string, TokenFetchInfo>();\n    if (tokenInfos) {\n      for (const info of tokenInfos) {\n        tokenInfoMap.set(info.address.toLowerCase(), info);\n      }\n    }\n\n    const balanceRequests: BalanceOfRequest[] = [];\n\n    if (includeNative) {\n      balanceRequests.push({\n        tokenAddress: ZERO_ADDRESS,\n        accountAddress,\n      });\n    }\n\n    for (const tokenAddress of tokenAddresses) {\n      balanceRequests.push({\n        tokenAddress,\n        accountAddress,\n      });\n    }\n\n    if (balanceRequests.length === 0) {\n      return {\n        chainId,\n        accountId,\n        accountAddress,\n        balances: [],\n        failedAddresses: [],\n        timestamp,\n      };\n    }\n\n    type FetchAccumulator = {\n      balances: AssetBalance[];\n      failedAddresses: Address[];\n    };\n\n    const result = await reduceInBatchesSerially<\n      BalanceOfRequest,\n      FetchAccumulator\n    >({\n      values: balanceRequests,\n      batchSize,\n      initialResult: {\n        balances: [],\n        failedAddresses: [],\n      },\n      eachBatch: async (workingResult, batch) => {\n        const responses = await this.#multicallClient.batchBalanceOf(\n          chainId,\n          batch,\n        );\n\n        return this.#processBalanceResponses(\n          responses,\n          workingResult as FetchAccumulator,\n          chainId,\n          accountId,\n          timestamp,\n          tokenInfoMap,\n        );\n      },\n    });\n\n    return {\n      chainId,\n      accountId,\n      accountAddress,\n      ...result,\n      timestamp,\n    };\n  }\n\n  #processBalanceResponses(\n    responses: BalanceOfResponse[],\n    accumulator: {\n      balances: AssetBalance[];\n      failedAddresses: Address[];\n    },\n    chainId: ChainId,\n    accountId: AccountId,\n    timestamp: number,\n    tokenInfoMap: Map<string, TokenFetchInfo>,\n  ): {\n    balances: AssetBalance[];\n    failedAddresses: Address[];\n  } {\n    const { balances, failedAddresses } = accumulator;\n\n    for (const response of responses) {\n      if (!response.success) {\n        failedAddresses.push(response.tokenAddress);\n        continue;\n      }\n\n      const balance = response.balance ?? '0';\n      const chainIdDecimal = parseInt(chainId, 16);\n      const isNative =\n        response.tokenAddress.toLowerCase() === ZERO_ADDRESS.toLowerCase();\n      const tokenInfo = tokenInfoMap.get(response.tokenAddress.toLowerCase());\n\n      // Native uses 18. ERC-20: format when decimals are known; otherwise pass raw\n      // `balance` through so RpcDataSource can resolve decimals (state → list → RPC).\n      let decimals: number | undefined;\n      let formattedBalance: string;\n      if (isNative) {\n        decimals = 18;\n        formattedBalance = this.#formatBalance(balance, decimals);\n      } else if (tokenInfo?.decimals === undefined) {\n        formattedBalance = balance;\n      } else {\n        decimals = tokenInfo.decimals;\n        formattedBalance = this.#formatBalance(balance, decimals);\n      }\n      const assetId: CaipAssetType = isNative\n        ? (`eip155:${chainIdDecimal}/slip44:60` as CaipAssetType)\n        : (`eip155:${chainIdDecimal}/erc20:${response.tokenAddress.toLowerCase()}` as CaipAssetType);\n\n      const balanceEntry: AssetBalance = {\n        assetId,\n        accountId,\n        chainId,\n        balance,\n        formattedBalance,\n        timestamp,\n      };\n      if (typeof decimals === 'number') {\n        balanceEntry.decimals = decimals;\n      }\n      balances.push(balanceEntry);\n    }\n\n    return { balances, failedAddresses };\n  }\n\n  #formatBalance(rawBalance: string, decimals: number): string {\n    if (rawBalance === '0') {\n      return '0';\n    }\n\n    try {\n      const balanceBigInt = BigInt(rawBalance);\n      const divisor = BigInt(10 ** decimals);\n\n      const integerPart = balanceBigInt / divisor;\n      const remainder = balanceBigInt % divisor;\n      const fractionalStr = remainder.toString().padStart(decimals, '0');\n      const trimmedFractional = fractionalStr.replace(/0+$/u, '');\n\n      if (trimmedFractional === '') {\n        return integerPart.toString();\n      }\n\n      return `${integerPart}.${trimmedFractional}`;\n    } catch {\n      return rawBalance;\n    }\n  }\n}\n"]}
+{"version":3,"file":"BalanceFetcher.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,mCAAmC,EAAE,qCAAqC;AACnF,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAErD,OAAO,EAAE,YAAY,EAAE,qCAAiC;AAcxD,OAAO,EAAE,uBAAuB,EAAE,2BAAiB;AAEnD,MAAM,wBAAwB,GAAG,KAAM,CAAC,CAAC,aAAa;AAmCtD;;;;;;;;GAQG;AACH,MAAM,OAAO,cAAe,SAAQ,mCAAmC,EAAuB;IAS5F,YACE,eAAgC,EAChC,SAAkC,EAClC,MAA6B;QAE7B,KAAK,EAAE,CAAC;;QAbD,kDAAkC;QAElC,4CAAoC;QAEpC,yCAAiE;QAE1E,kDAAsD;QAQpD,uBAAA,IAAI,mCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6BAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,0BAAW;YACb,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,GAAG;YACjD,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,KAAK;SACpD,MAAA,CAAC;QAEF,2BAA2B;QAC3B,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,eAAe,IAAI,wBAAwB,CAAC,CAAC;IAC9E,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,QAAiC;QAClD,uBAAA,IAAI,mCAAoB,QAAQ,MAAA,CAAC;IACnC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,KAA0B;QAC3C,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EACvB,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,CACrB,CAAC;QAEF,IAAI,uBAAA,IAAI,uCAAiB,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,uBAAA,IAAI,uCAAiB,MAArB,IAAI,EAAkB,MAAM,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IA4ED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,sBAAsB,CAC1B,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,MAAyB;QAEzB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,oEAAoE;QACpE,sEAAsE;QACtE,MAAM,eAAe,GAAuB,EAAE,CAAC;QAC/C,MAAM,cAAc,GAAG,IAAI,GAAG,EAA2B,CAAC;QAE1D,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACjD,IAAI,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;gBACrC,SAAS,CAAC,cAAc;YAC1B,CAAC;YAED,cAAc,CAAC,GAAG,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;YACxC,eAAe,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO;gBACP,SAAS;gBACT,cAAc;gBACd,QAAQ,EAAE,EAAE;gBACZ,eAAe,EAAE,EAAE;gBACnB,SAAS;aACV,CAAC;QACJ,CAAC;QAOD,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAG1C;YACA,MAAM,EAAE,eAAe;YACvB,SAAS,EAAE,uBAAA,IAAI,8BAAQ,CAAC,gBAAgB;YACxC,aAAa,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,eAAe,EAAE,EAAE;aACpB;YACD,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,uCAAiB,CAAC,cAAc,CAC1D,OAAO,EACP,KAAK,CACN,CAAC;gBAEF,OAAO,uBAAA,IAAI,0EAAyB,MAA7B,IAAI,EACT,SAAS,EACT,aAAiC,EACjC,OAAO,EACP,SAAS,EACT,SAAS,EACT,cAAc,CACf,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,OAAO;YACL,OAAO;YACP,SAAS;YACT,cAAc;YACd,GAAG,MAAM;YACT,SAAS;SACV,CAAC;IACJ,CAAC;CAqFF;6SA5OmB,OAAgB,EAAE,SAAoB;IACtD,MAAM,KAAK,GAAG,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAEhE,MAAM,eAAe,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;IAC1D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,qDAAqD;IACrD,uFAAuF;IACvF,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;IAExD,MAAM,aAAa,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEzD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAoB,EAAE,CAAC;QACtE,MAAM,EACJ,KAAK,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE,EACpC,cAAc,EACd,cAAc,GACf,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAEhC,IAAI,cAAc,KAAK,cAAc,EAAE,CAAC;YACtC,MAAM,gBAAgB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/C,IAAI,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACxC,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,cAAc,KAAK,QAAQ,CAAC;YAC7C,MAAM,YAAY,GAAG,QAAQ;gBAC3B,CAAC,CAAC,YAAY;gBACd,CAAC,CAAE,cAAc,CAAC,WAAW,EAAc,CAAC;YAE9C,aAAa,CAAC,GAAG,CAAC,gBAAgB,EAAE;gBAClC,OAAO;gBACP,OAAO,EAAE,YAAY;aACtB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC;AAED;;;;;;;;GAQG;AACH,KAAK,wCACH,OAAgB,EAChB,SAAoB,EACpB,cAAuB;IAEvB,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,OAAO,EAAE,SAAS,CAAC,CAAC;IAE1D,OAAO,IAAI,CAAC,sBAAsB,CAChC,OAAO,EACP,SAAS,EACT,cAAc,EACd,MAAM,CACP,CAAC;AACJ,CAAC,6FA0FC,SAA8B,EAC9B,WAGC,EACD,OAAgB,EAChB,SAAoB,EACpB,SAAiB,EACjB,cAA4C;IAK5C,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,WAAW,CAAC;IAElD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QAED,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,GAAG,CAAC;QACxC,MAAM,QAAQ,GAAG,YAAY,KAAK,YAAY,CAAC,WAAW,EAAE,CAAC;QAE7D,IAAI,QAA4B,CAAC;QACjC,IAAI,gBAAwB,CAAC;QAC7B,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,GAAG,EAAE,CAAC;YACd,gBAAgB,GAAG,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACxC,gBAAgB,GAAG,OAAO,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;YAC1B,gBAAgB,GAAG,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS;YACT,OAAO;YACP,OAAO;YACP,gBAAgB;YAChB,SAAS;SACV,CAAC;QACF,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACjC,YAAY,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACnC,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9B,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;AACvC,CAAC,yEAEc,UAAkB,EAAE,QAAgB;IACjD,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;QACvB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,MAAM,CAAC,EAAE,IAAI,QAAQ,CAAC,CAAC;QAEvC,MAAM,WAAW,GAAG,aAAa,GAAG,OAAO,CAAC;QAC5C,MAAM,SAAS,GAAG,aAAa,GAAG,OAAO,CAAC;QAC1C,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnE,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAE5D,IAAI,iBAAiB,KAAK,EAAE,EAAE,CAAC;YAC7B,OAAO,WAAW,CAAC,QAAQ,EAAE,CAAC;QAChC,CAAC;QAED,OAAO,GAAG,WAAW,IAAI,iBAAiB,EAAE,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,UAAU,CAAC;IACpB,CAAC;AACH,CAAC","sourcesContent":["import { StaticIntervalPollingControllerOnly } from '@metamask/polling-controller';\nimport { parseCaipAssetType } from '@metamask/utils';\n\nimport { ZERO_ADDRESS } from '../../../utils/constants';\nimport type { MulticallClient } from '../clients';\nimport type {\n  AccountId,\n  Address,\n  AssetBalance,\n  AssetFetchEntry,\n  AssetsBalanceState,\n  BalanceFetchResult,\n  BalanceOfRequest,\n  BalanceOfResponse,\n  CaipAssetType,\n  ChainId,\n} from '../types';\nimport { reduceInBatchesSerially } from '../utils';\n\nconst DEFAULT_BALANCE_INTERVAL = 30_000; // 30 seconds\n\n/**\n * Minimal messenger interface for BalanceFetcher.\n */\nexport type BalanceFetcherMessenger = {\n  call: (action: 'AssetsController:getState') => AssetsBalanceState;\n};\n\nexport type BalanceFetcherConfig = {\n  defaultBatchSize?: number;\n  defaultTimeoutMs?: number;\n  /** Polling interval in ms (default: 30s) */\n  pollingInterval?: number;\n};\n\n/**\n * Polling input for BalanceFetcher - identifies what to poll for.\n */\nexport type BalancePollingInput = {\n  /** Chain ID (hex format) */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n};\n\n/**\n * Callback type for balance updates.\n */\nexport type OnBalanceUpdateCallback = (\n  result: BalanceFetchResult,\n) => void | Promise<void>;\n\n/**\n * BalanceFetcher - Fetches token balances via multicall.\n * Extends StaticIntervalPollingControllerOnly for built-in polling support.\n *\n * Callers provide CAIP-19 asset IDs; the fetcher extracts on-chain addresses\n * (or uses the zero address for native assets) and maps multicall responses\n * back to the original asset IDs. This ensures the returned balance entries\n * always carry the correct identifier regardless of chain.\n */\nexport class BalanceFetcher extends StaticIntervalPollingControllerOnly<BalancePollingInput>() {\n  readonly #multicallClient: MulticallClient;\n\n  readonly #messenger: BalanceFetcherMessenger;\n\n  readonly #config: Required<Omit<BalanceFetcherConfig, 'pollingInterval'>>;\n\n  #onBalanceUpdate: OnBalanceUpdateCallback | undefined;\n\n  constructor(\n    multicallClient: MulticallClient,\n    messenger: BalanceFetcherMessenger,\n    config?: BalanceFetcherConfig,\n  ) {\n    super();\n    this.#multicallClient = multicallClient;\n    this.#messenger = messenger;\n    this.#config = {\n      defaultBatchSize: config?.defaultBatchSize ?? 300,\n      defaultTimeoutMs: config?.defaultTimeoutMs ?? 30000,\n    };\n\n    // Set the polling interval\n    this.setIntervalLength(config?.pollingInterval ?? DEFAULT_BALANCE_INTERVAL);\n  }\n\n  /**\n   * Set the callback to receive balance updates during polling.\n   *\n   * @param callback - Function to call with balance results.\n   */\n  setOnBalanceUpdate(callback: OnBalanceUpdateCallback): void {\n    this.#onBalanceUpdate = callback;\n  }\n\n  /**\n   * Execute a poll cycle (required by base class).\n   * Fetches balances and calls the update callback.\n   *\n   * @param input - The polling input.\n   */\n  async _executePoll(input: BalancePollingInput): Promise<void> {\n    const result = await this.#fetchBalances(\n      input.chainId,\n      input.accountId,\n      input.accountAddress,\n    );\n\n    if (this.#onBalanceUpdate && result.balances.length > 0) {\n      await this.#onBalanceUpdate(result);\n    }\n  }\n\n  /**\n   * Return asset fetch entries tracked in state for the given account and\n   * chain. Both native (`slip44:`) and ERC-20 (`erc20:`) entries are included.\n   *\n   * @param chainId - Hex chain ID (e.g. \"0x1\").\n   * @param accountId - Account UUID.\n   * @returns Array of asset fetch entries from state for the requested chain.\n   */\n  #getAssetsToFetch(chainId: ChainId, accountId: AccountId): AssetFetchEntry[] {\n    const state = this.#messenger.call('AssetsController:getState');\n\n    const accountBalances = state?.assetsBalance?.[accountId];\n    if (!accountBalances) {\n      return [];\n    }\n\n    // Convert hex chainId to decimal for CAIP-2 matching\n    // This is safe because we are filtring with an accountId that is for evm balances only\n    const chainIdDecimal = parseInt(chainId, 16).toString();\n\n    const assetsToFetch = new Map<string, AssetFetchEntry>();\n\n    for (const assetId of Object.keys(accountBalances) as CaipAssetType[]) {\n      const {\n        chain: { reference: chainReference },\n        assetNamespace,\n        assetReference,\n      } = parseCaipAssetType(assetId);\n\n      if (chainReference === chainIdDecimal) {\n        const assetIdLowerCase = assetId.toLowerCase();\n        if (assetsToFetch.has(assetIdLowerCase)) {\n          continue;\n        }\n\n        const isNative = assetNamespace === 'slip44';\n        const tokenAddress = isNative\n          ? ZERO_ADDRESS\n          : (assetReference.toLowerCase() as Address);\n\n        assetsToFetch.set(assetIdLowerCase, {\n          assetId,\n          address: tokenAddress,\n        });\n      }\n    }\n\n    return Array.from(assetsToFetch.values());\n  }\n\n  /**\n   * Fetch balances for assets already tracked in state for the given\n   * account and chain.\n   *\n   * @param chainId - Hex chain ID.\n   * @param accountId - Account UUID.\n   * @param accountAddress - On-chain address of the account.\n   * @returns Balance fetch result.\n   */\n  async #fetchBalances(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n  ): Promise<BalanceFetchResult> {\n    const assets = this.#getAssetsToFetch(chainId, accountId);\n\n    return this.fetchBalancesForAssets(\n      chainId,\n      accountId,\n      accountAddress,\n      assets,\n    );\n  }\n\n  /**\n   * Fetch balances for the given assets via multicall.\n   *\n   * Each entry bundles a CAIP-19 asset ID with its on-chain address and\n   * optional decimals.\n   *\n   * @param chainId - Hex chain ID.\n   * @param accountId - Account UUID.\n   * @param accountAddress - On-chain address of the account.\n   * @param assets - Asset fetch entries to fetch balances for.\n   * @returns Balance fetch result.\n   */\n  async fetchBalancesForAssets(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    assets: AssetFetchEntry[],\n  ): Promise<BalanceFetchResult> {\n    const timestamp = Date.now();\n\n    // Build a single map keyed by lowercase address that holds all info\n    // needed to match multicall responses back to their original entries.\n    const balanceRequests: BalanceOfRequest[] = [];\n    const entryByAddress = new Map<string, AssetFetchEntry>();\n\n    for (const entry of assets) {\n      const lowerAddress = entry.address.toLowerCase();\n      if (entryByAddress.has(lowerAddress)) {\n        continue; // deduplicate\n      }\n\n      entryByAddress.set(lowerAddress, entry);\n      balanceRequests.push({ tokenAddress: entry.address, accountAddress });\n    }\n\n    if (balanceRequests.length === 0) {\n      return {\n        chainId,\n        accountId,\n        accountAddress,\n        balances: [],\n        failedAddresses: [],\n        timestamp,\n      };\n    }\n\n    type FetchAccumulator = {\n      balances: AssetBalance[];\n      failedAddresses: Address[];\n    };\n\n    const result = await reduceInBatchesSerially<\n      BalanceOfRequest,\n      FetchAccumulator\n    >({\n      values: balanceRequests,\n      batchSize: this.#config.defaultBatchSize,\n      initialResult: {\n        balances: [],\n        failedAddresses: [],\n      },\n      eachBatch: async (workingResult, batch) => {\n        const responses = await this.#multicallClient.batchBalanceOf(\n          chainId,\n          batch,\n        );\n\n        return this.#processBalanceResponses(\n          responses,\n          workingResult as FetchAccumulator,\n          chainId,\n          accountId,\n          timestamp,\n          entryByAddress,\n        );\n      },\n    });\n\n    return {\n      chainId,\n      accountId,\n      accountAddress,\n      ...result,\n      timestamp,\n    };\n  }\n\n  #processBalanceResponses(\n    responses: BalanceOfResponse[],\n    accumulator: {\n      balances: AssetBalance[];\n      failedAddresses: Address[];\n    },\n    chainId: ChainId,\n    accountId: AccountId,\n    timestamp: number,\n    entryByAddress: Map<string, AssetFetchEntry>,\n  ): {\n    balances: AssetBalance[];\n    failedAddresses: Address[];\n  } {\n    const { balances, failedAddresses } = accumulator;\n\n    for (const response of responses) {\n      if (!response.success) {\n        failedAddresses.push(response.tokenAddress);\n        continue;\n      }\n\n      const lowerAddress = response.tokenAddress.toLowerCase();\n      const entry = entryByAddress.get(lowerAddress);\n      if (!entry) {\n        continue;\n      }\n\n      const balance = response.balance ?? '0';\n      const isNative = lowerAddress === ZERO_ADDRESS.toLowerCase();\n\n      let decimals: number | undefined;\n      let formattedBalance: string;\n      if (isNative) {\n        decimals = 18;\n        formattedBalance = this.#formatBalance(balance, decimals);\n      } else if (entry.decimals === undefined) {\n        formattedBalance = balance;\n      } else {\n        decimals = entry.decimals;\n        formattedBalance = this.#formatBalance(balance, decimals);\n      }\n\n      const balanceEntry: AssetBalance = {\n        assetId: entry.assetId,\n        accountId,\n        chainId,\n        balance,\n        formattedBalance,\n        timestamp,\n      };\n      if (typeof decimals === 'number') {\n        balanceEntry.decimals = decimals;\n      }\n      balances.push(balanceEntry);\n    }\n\n    return { balances, failedAddresses };\n  }\n\n  #formatBalance(rawBalance: string, decimals: number): string {\n    if (rawBalance === '0') {\n      return '0';\n    }\n\n    try {\n      const balanceBigInt = BigInt(rawBalance);\n      const divisor = BigInt(10 ** decimals);\n\n      const integerPart = balanceBigInt / divisor;\n      const remainder = balanceBigInt % divisor;\n      const fractionalStr = remainder.toString().padStart(decimals, '0');\n      const trimmedFractional = fractionalStr.replace(/0+$/u, '');\n\n      if (trimmedFractional === '') {\n        return integerPart.toString();\n      }\n\n      return `${integerPart}.${trimmedFractional}`;\n    } catch {\n      return rawBalance;\n    }\n  }\n}\n"]}

package/dist/data-sources/evm-rpc-services/types/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"","sourcesContent":["// Core types\nexport type {\n  AccountId,\n  Address,\n  AssetType,\n  CaipAssetType,\n  ChainId,\n} from './core';\n\n// Asset types\nexport type { Asset, AssetBalance } from './assets';\n\n// State types\nexport type {\n  AssetsBalanceState,\n  TokenChainsCacheEntry,\n  TokenListEntry,\n  TokenListState,\n} from './state';\n\n// Config types\nexport type { GetProviderFunction, Provider } from './config';\n\n// Multicall types\nexport type { BalanceOfRequest, BalanceOfResponse } from './multicall';\n\n// Service types\nexport type {\n  BalanceFetchOptions,\n  BalanceFetchResult,\n  TokenDetectionOptions,\n  TokenDetectionResult,\n  TokenFetchInfo,\n} from './services';\n"]}
+{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"","sourcesContent":["// Core types\nexport type {\n  AccountId,\n  Address,\n  AssetType,\n  CaipAssetType,\n  ChainId,\n} from './core';\n\n// Asset types\nexport type { Asset, AssetBalance } from './assets';\n\n// State types\nexport type {\n  AssetsBalanceState,\n  TokenChainsCacheEntry,\n  TokenListEntry,\n  TokenListState,\n} from './state';\n\n// Config types\nexport type { GetProviderFunction, Provider } from './config';\n\n// Multicall types\nexport type { BalanceOfRequest, BalanceOfResponse } from './multicall';\n\n// Service types\nexport type {\n  AssetFetchEntry,\n  BalanceFetchResult,\n  TokenDetectionOptions,\n  TokenDetectionResult,\n} from './services';\n"]}

package/dist/data-sources/evm-rpc-services/types/index.d.cts

@@ -3,5 +3,5 @@ export type { Asset, AssetBalance } from "./assets.cjs";
 export type { AssetsBalanceState, TokenChainsCacheEntry, TokenListEntry, TokenListState, } from "./state.cjs";
 export type { GetProviderFunction, Provider } from "./config.cjs";
 export type { BalanceOfRequest, BalanceOfResponse } from "./multicall.cjs";
-export type { BalanceFetchOptions, BalanceFetchResult, TokenDetectionOptions, TokenDetectionResult, TokenFetchInfo, } from "./services.cjs";
+export type { AssetFetchEntry, BalanceFetchResult, TokenDetectionOptions, TokenDetectionResult, } from "./services.cjs";
 //# sourceMappingURL=index.d.cts.map

package/dist/data-sources/evm-rpc-services/types/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"AACA,YAAY,EACV,SAAS,EACT,OAAO,EACP,SAAS,EACT,aAAa,EACb,OAAO,GACR,mBAAe;AAGhB,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AAGpD,YAAY,EACV,kBAAkB,EAClB,qBAAqB,EACrB,cAAc,EACd,cAAc,GACf,oBAAgB;AAGjB,YAAY,EAAE,mBAAmB,EAAE,QAAQ,EAAE,qBAAiB;AAG9D,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,wBAAoB;AAGvE,YAAY,EACV,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,cAAc,GACf,uBAAmB"}
+{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"AACA,YAAY,EACV,SAAS,EACT,OAAO,EACP,SAAS,EACT,aAAa,EACb,OAAO,GACR,mBAAe;AAGhB,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AAGpD,YAAY,EACV,kBAAkB,EAClB,qBAAqB,EACrB,cAAc,EACd,cAAc,GACf,oBAAgB;AAGjB,YAAY,EAAE,mBAAmB,EAAE,QAAQ,EAAE,qBAAiB;AAG9D,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,wBAAoB;AAGvE,YAAY,EACV,eAAe,EACf,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,GACrB,uBAAmB"}

package/dist/data-sources/evm-rpc-services/types/index.d.mts

@@ -3,5 +3,5 @@ export type { Asset, AssetBalance } from "./assets.mjs";
 export type { AssetsBalanceState, TokenChainsCacheEntry, TokenListEntry, TokenListState, } from "./state.mjs";
 export type { GetProviderFunction, Provider } from "./config.mjs";
 export type { BalanceOfRequest, BalanceOfResponse } from "./multicall.mjs";
-export type { BalanceFetchOptions, BalanceFetchResult, TokenDetectionOptions, TokenDetectionResult, TokenFetchInfo, } from "./services.mjs";
+export type { AssetFetchEntry, BalanceFetchResult, TokenDetectionOptions, TokenDetectionResult, } from "./services.mjs";
 //# sourceMappingURL=index.d.mts.map

package/dist/data-sources/evm-rpc-services/types/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"AACA,YAAY,EACV,SAAS,EACT,OAAO,EACP,SAAS,EACT,aAAa,EACb,OAAO,GACR,mBAAe;AAGhB,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AAGpD,YAAY,EACV,kBAAkB,EAClB,qBAAqB,EACrB,cAAc,EACd,cAAc,GACf,oBAAgB;AAGjB,YAAY,EAAE,mBAAmB,EAAE,QAAQ,EAAE,qBAAiB;AAG9D,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,wBAAoB;AAGvE,YAAY,EACV,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,cAAc,GACf,uBAAmB"}
+{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"AACA,YAAY,EACV,SAAS,EACT,OAAO,EACP,SAAS,EACT,aAAa,EACb,OAAO,GACR,mBAAe;AAGhB,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AAGpD,YAAY,EACV,kBAAkB,EAClB,qBAAqB,EACrB,cAAc,EACd,cAAc,GACf,oBAAgB;AAGjB,YAAY,EAAE,mBAAmB,EAAE,QAAQ,EAAE,qBAAiB;AAG9D,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,wBAAoB;AAGvE,YAAY,EACV,eAAe,EACf,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,GACrB,uBAAmB"}

package/dist/data-sources/evm-rpc-services/types/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"","sourcesContent":["// Core types\nexport type {\n  AccountId,\n  Address,\n  AssetType,\n  CaipAssetType,\n  ChainId,\n} from './core';\n\n// Asset types\nexport type { Asset, AssetBalance } from './assets';\n\n// State types\nexport type {\n  AssetsBalanceState,\n  TokenChainsCacheEntry,\n  TokenListEntry,\n  TokenListState,\n} from './state';\n\n// Config types\nexport type { GetProviderFunction, Provider } from './config';\n\n// Multicall types\nexport type { BalanceOfRequest, BalanceOfResponse } from './multicall';\n\n// Service types\nexport type {\n  BalanceFetchOptions,\n  BalanceFetchResult,\n  TokenDetectionOptions,\n  TokenDetectionResult,\n  TokenFetchInfo,\n} from './services';\n"]}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/index.ts"],"names":[],"mappings":"","sourcesContent":["// Core types\nexport type {\n  AccountId,\n  Address,\n  AssetType,\n  CaipAssetType,\n  ChainId,\n} from './core';\n\n// Asset types\nexport type { Asset, AssetBalance } from './assets';\n\n// State types\nexport type {\n  AssetsBalanceState,\n  TokenChainsCacheEntry,\n  TokenListEntry,\n  TokenListState,\n} from './state';\n\n// Config types\nexport type { GetProviderFunction, Provider } from './config';\n\n// Multicall types\nexport type { BalanceOfRequest, BalanceOfResponse } from './multicall';\n\n// Service types\nexport type {\n  AssetFetchEntry,\n  BalanceFetchResult,\n  TokenDetectionOptions,\n  TokenDetectionResult,\n} from './services';\n"]}

package/dist/data-sources/evm-rpc-services/types/services.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"services.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"","sourcesContent":["import type { Asset, AssetBalance } from './assets';\nimport type { AccountId, Address, ChainId } from './core';\n\n/**\n * Token detection result.\n */\nexport type TokenDetectionResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address used for detection */\n  accountAddress: Address;\n  /** Newly detected tokens with non-zero balances */\n  detectedAssets: Asset[];\n  /** Balances for newly detected tokens */\n  detectedBalances: AssetBalance[];\n  /** Tokens that were checked but had zero balance */\n  zeroBalanceAddresses: Address[];\n  /** Tokens that failed to be checked */\n  failedAddresses: Address[];\n  /** Timestamp of detection */\n  timestamp: number;\n};\n\n/**\n * Token detection options.\n */\nexport type TokenDetectionOptions = {\n  /** Whether token detection is enabled */\n  tokenDetectionEnabled?: boolean;\n  /** Whether external services (e.g. token list API) are allowed; detection stops when false */\n  useExternalService?: boolean;\n  /** Maximum number of tokens to check per batch */\n  batchSize?: number;\n  /** Timeout for detection in milliseconds */\n  timeout?: number;\n};\n\n/**\n * Balance fetch result.\n */\nexport type BalanceFetchResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n  /** Fetched balances */\n  balances: AssetBalance[];\n  /** Token addresses that failed to fetch */\n  failedAddresses: Address[];\n  /** Block number when balances were fetched */\n  blockNumber?: number;\n  /** Timestamp of fetch */\n  timestamp: number;\n};\n\n/**\n * Balance fetch options.\n */\nexport type BalanceFetchOptions = {\n  /** Maximum number of tokens to fetch per batch */\n  batchSize?: number;\n  /** Timeout for fetch in milliseconds */\n  timeout?: number;\n  /** Include native token balance */\n  includeNative?: boolean;\n};\n\n/**\n * Token info for balance fetching.\n */\nexport type TokenFetchInfo = {\n  /** Token contract address */\n  address: Address;\n  /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */\n  decimals?: number;\n  /** Token symbol (optional) */\n  symbol?: string;\n};\n"]}
+{"version":3,"file":"services.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"","sourcesContent":["import type { Asset, AssetBalance } from './assets';\nimport type { AccountId, Address, CaipAssetType, ChainId } from './core';\n\n/**\n * Token detection result.\n */\nexport type TokenDetectionResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address used for detection */\n  accountAddress: Address;\n  /** Newly detected tokens with non-zero balances */\n  detectedAssets: Asset[];\n  /** Balances for newly detected tokens */\n  detectedBalances: AssetBalance[];\n  /** Tokens that were checked but had zero balance */\n  zeroBalanceAddresses: Address[];\n  /** Tokens that failed to be checked */\n  failedAddresses: Address[];\n  /** Timestamp of detection */\n  timestamp: number;\n};\n\n/**\n * Token detection options.\n */\nexport type TokenDetectionOptions = {\n  /** Whether token detection is enabled */\n  tokenDetectionEnabled?: boolean;\n  /** Whether external services (e.g. token list API) are allowed; detection stops when false */\n  useExternalService?: boolean;\n  /** Maximum number of tokens to check per batch */\n  batchSize?: number;\n  /** Timeout for detection in milliseconds */\n  timeout?: number;\n};\n\n/**\n * Balance fetch result.\n */\nexport type BalanceFetchResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n  /** Fetched balances */\n  balances: AssetBalance[];\n  /** Token addresses that failed to fetch */\n  failedAddresses: Address[];\n  /** Block number when balances were fetched */\n  blockNumber?: number;\n  /** Timestamp of fetch */\n  timestamp: number;\n};\n\n/**\n * Entry describing a single asset to fetch a balance for.\n * Bundles the CAIP-19 asset ID with the on-chain address (zero address for\n * native assets even when the chain has a native asset with a non-zero address)\n * and optional metadata\n */\nexport type AssetFetchEntry = {\n  /** CAIP-19 asset type identifier */\n  assetId: CaipAssetType;\n  /** On-chain contract address (zero address for native assets regardless of the chain's native asset address) */\n  address: Address;\n  /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */\n  decimals?: number;\n};\n"]}

package/dist/data-sources/evm-rpc-services/types/services.d.cts

@@ -1,5 +1,5 @@
 import type { Asset, AssetBalance } from "./assets.cjs";
-import type { AccountId, Address, ChainId } from "./core.cjs";
+import type { AccountId, Address, CaipAssetType, ChainId } from "./core.cjs";
 /**
  * Token detection result.
  */
@@ -54,25 +54,17 @@ export type BalanceFetchResult = {
     timestamp: number;
 };
 /**
- * Balance fetch options.
+ * Entry describing a single asset to fetch a balance for.
+ * Bundles the CAIP-19 asset ID with the on-chain address (zero address for
+ * native assets even when the chain has a native asset with a non-zero address)
+ * and optional metadata
  */
-export type BalanceFetchOptions = {
-    /** Maximum number of tokens to fetch per batch */
-    batchSize?: number;
-    /** Timeout for fetch in milliseconds */
-    timeout?: number;
-    /** Include native token balance */
-    includeNative?: boolean;
-};
-/**
- * Token info for balance fetching.
- */
-export type TokenFetchInfo = {
-    /** Token contract address */
+export type AssetFetchEntry = {
+    /** CAIP-19 asset type identifier */
+    assetId: CaipAssetType;
+    /** On-chain contract address (zero address for native assets regardless of the chain's native asset address) */
     address: Address;
     /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */
     decimals?: number;
-    /** Token symbol (optional) */
-    symbol?: string;
 };
 //# sourceMappingURL=services.d.cts.map

package/dist/data-sources/evm-rpc-services/types/services.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"services.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AACpD,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,mBAAe;AAE1D;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,yCAAyC;IACzC,cAAc,EAAE,OAAO,CAAC;IACxB,mDAAmD;IACnD,cAAc,EAAE,KAAK,EAAE,CAAC;IACxB,yCAAyC;IACzC,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,oDAAoD;IACpD,oBAAoB,EAAE,OAAO,EAAE,CAAC;IAChC,uCAAuC;IACvC,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,6BAA6B;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,8FAA8F;IAC9F,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,uBAAuB;IACvB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,2CAA2C;IAC3C,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,8CAA8C;IAC9C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,wCAAwC;IACxC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mCAAmC;IACnC,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,6BAA6B;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,6GAA6G;IAC7G,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,8BAA8B;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC"}
+{"version":3,"file":"services.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AACpD,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,mBAAe;AAEzE;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,yCAAyC;IACzC,cAAc,EAAE,OAAO,CAAC;IACxB,mDAAmD;IACnD,cAAc,EAAE,KAAK,EAAE,CAAC;IACxB,yCAAyC;IACzC,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,oDAAoD;IACpD,oBAAoB,EAAE,OAAO,EAAE,CAAC;IAChC,uCAAuC;IACvC,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,6BAA6B;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,8FAA8F;IAC9F,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,uBAAuB;IACvB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,2CAA2C;IAC3C,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,8CAA8C;IAC9C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,oCAAoC;IACpC,OAAO,EAAE,aAAa,CAAC;IACvB,gHAAgH;IAChH,OAAO,EAAE,OAAO,CAAC;IACjB,6GAA6G;IAC7G,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC"}

package/dist/data-sources/evm-rpc-services/types/services.d.mts

@@ -1,5 +1,5 @@
 import type { Asset, AssetBalance } from "./assets.mjs";
-import type { AccountId, Address, ChainId } from "./core.mjs";
+import type { AccountId, Address, CaipAssetType, ChainId } from "./core.mjs";
 /**
  * Token detection result.
  */
@@ -54,25 +54,17 @@ export type BalanceFetchResult = {
     timestamp: number;
 };
 /**
- * Balance fetch options.
+ * Entry describing a single asset to fetch a balance for.
+ * Bundles the CAIP-19 asset ID with the on-chain address (zero address for
+ * native assets even when the chain has a native asset with a non-zero address)
+ * and optional metadata
  */
-export type BalanceFetchOptions = {
-    /** Maximum number of tokens to fetch per batch */
-    batchSize?: number;
-    /** Timeout for fetch in milliseconds */
-    timeout?: number;
-    /** Include native token balance */
-    includeNative?: boolean;
-};
-/**
- * Token info for balance fetching.
- */
-export type TokenFetchInfo = {
-    /** Token contract address */
+export type AssetFetchEntry = {
+    /** CAIP-19 asset type identifier */
+    assetId: CaipAssetType;
+    /** On-chain contract address (zero address for native assets regardless of the chain's native asset address) */
     address: Address;
     /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */
     decimals?: number;
-    /** Token symbol (optional) */
-    symbol?: string;
 };
 //# sourceMappingURL=services.d.mts.map

package/dist/data-sources/evm-rpc-services/types/services.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"services.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AACpD,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,mBAAe;AAE1D;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,yCAAyC;IACzC,cAAc,EAAE,OAAO,CAAC;IACxB,mDAAmD;IACnD,cAAc,EAAE,KAAK,EAAE,CAAC;IACxB,yCAAyC;IACzC,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,oDAAoD;IACpD,oBAAoB,EAAE,OAAO,EAAE,CAAC;IAChC,uCAAuC;IACvC,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,6BAA6B;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,8FAA8F;IAC9F,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,uBAAuB;IACvB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,2CAA2C;IAC3C,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,8CAA8C;IAC9C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,wCAAwC;IACxC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mCAAmC;IACnC,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,6BAA6B;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,6GAA6G;IAC7G,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,8BAA8B;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC"}
+{"version":3,"file":"services.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,qBAAiB;AACpD,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,mBAAe;AAEzE;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,yCAAyC;IACzC,cAAc,EAAE,OAAO,CAAC;IACxB,mDAAmD;IACnD,cAAc,EAAE,KAAK,EAAE,CAAC;IACxB,yCAAyC;IACzC,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,oDAAoD;IACpD,oBAAoB,EAAE,OAAO,EAAE,CAAC;IAChC,uCAAuC;IACvC,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,6BAA6B;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,8FAA8F;IAC9F,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kDAAkD;IAClD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,uBAAuB;IACvB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,2CAA2C;IAC3C,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,8CAA8C;IAC9C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,oCAAoC;IACpC,OAAO,EAAE,aAAa,CAAC;IACvB,gHAAgH;IAChH,OAAO,EAAE,OAAO,CAAC;IACjB,6GAA6G;IAC7G,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC"}

package/dist/data-sources/evm-rpc-services/types/services.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"services.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"","sourcesContent":["import type { Asset, AssetBalance } from './assets';\nimport type { AccountId, Address, ChainId } from './core';\n\n/**\n * Token detection result.\n */\nexport type TokenDetectionResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address used for detection */\n  accountAddress: Address;\n  /** Newly detected tokens with non-zero balances */\n  detectedAssets: Asset[];\n  /** Balances for newly detected tokens */\n  detectedBalances: AssetBalance[];\n  /** Tokens that were checked but had zero balance */\n  zeroBalanceAddresses: Address[];\n  /** Tokens that failed to be checked */\n  failedAddresses: Address[];\n  /** Timestamp of detection */\n  timestamp: number;\n};\n\n/**\n * Token detection options.\n */\nexport type TokenDetectionOptions = {\n  /** Whether token detection is enabled */\n  tokenDetectionEnabled?: boolean;\n  /** Whether external services (e.g. token list API) are allowed; detection stops when false */\n  useExternalService?: boolean;\n  /** Maximum number of tokens to check per batch */\n  batchSize?: number;\n  /** Timeout for detection in milliseconds */\n  timeout?: number;\n};\n\n/**\n * Balance fetch result.\n */\nexport type BalanceFetchResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n  /** Fetched balances */\n  balances: AssetBalance[];\n  /** Token addresses that failed to fetch */\n  failedAddresses: Address[];\n  /** Block number when balances were fetched */\n  blockNumber?: number;\n  /** Timestamp of fetch */\n  timestamp: number;\n};\n\n/**\n * Balance fetch options.\n */\nexport type BalanceFetchOptions = {\n  /** Maximum number of tokens to fetch per batch */\n  batchSize?: number;\n  /** Timeout for fetch in milliseconds */\n  timeout?: number;\n  /** Include native token balance */\n  includeNative?: boolean;\n};\n\n/**\n * Token info for balance fetching.\n */\nexport type TokenFetchInfo = {\n  /** Token contract address */\n  address: Address;\n  /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */\n  decimals?: number;\n  /** Token symbol (optional) */\n  symbol?: string;\n};\n"]}
+{"version":3,"file":"services.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/services.ts"],"names":[],"mappings":"","sourcesContent":["import type { Asset, AssetBalance } from './assets';\nimport type { AccountId, Address, CaipAssetType, ChainId } from './core';\n\n/**\n * Token detection result.\n */\nexport type TokenDetectionResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address used for detection */\n  accountAddress: Address;\n  /** Newly detected tokens with non-zero balances */\n  detectedAssets: Asset[];\n  /** Balances for newly detected tokens */\n  detectedBalances: AssetBalance[];\n  /** Tokens that were checked but had zero balance */\n  zeroBalanceAddresses: Address[];\n  /** Tokens that failed to be checked */\n  failedAddresses: Address[];\n  /** Timestamp of detection */\n  timestamp: number;\n};\n\n/**\n * Token detection options.\n */\nexport type TokenDetectionOptions = {\n  /** Whether token detection is enabled */\n  tokenDetectionEnabled?: boolean;\n  /** Whether external services (e.g. token list API) are allowed; detection stops when false */\n  useExternalService?: boolean;\n  /** Maximum number of tokens to check per batch */\n  batchSize?: number;\n  /** Timeout for detection in milliseconds */\n  timeout?: number;\n};\n\n/**\n * Balance fetch result.\n */\nexport type BalanceFetchResult = {\n  /** Chain ID */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n  /** Fetched balances */\n  balances: AssetBalance[];\n  /** Token addresses that failed to fetch */\n  failedAddresses: Address[];\n  /** Block number when balances were fetched */\n  blockNumber?: number;\n  /** Timestamp of fetch */\n  timestamp: number;\n};\n\n/**\n * Entry describing a single asset to fetch a balance for.\n * Bundles the CAIP-19 asset ID with the on-chain address (zero address for\n * native assets even when the chain has a native asset with a non-zero address)\n * and optional metadata\n */\nexport type AssetFetchEntry = {\n  /** CAIP-19 asset type identifier */\n  assetId: CaipAssetType;\n  /** On-chain contract address (zero address for native assets regardless of the chain's native asset address) */\n  address: Address;\n  /** Token decimals (omit when unknown — balance fetcher returns raw balance for RpcDataSource to resolve). */\n  decimals?: number;\n};\n"]}

package/dist/selectors/balance.cjs

@@ -7,6 +7,10 @@ exports.getAggregatedBalanceForAccount = exports.getInternalAccountsForGroup = e
 const controller_utils_1 = require("@metamask/controller-utils");
 const utils_1 = require("@metamask/utils");
 const bignumber_js_1 = __importDefault(require("bignumber.js"));
+// ============================================================================
+// TRACE NAMES — used in Sentry spans (search these strings in Discover)
+// ============================================================================
+const TRACE_AGGREGATED_BALANCE_SELECTOR = 'AggregatedBalanceSelector';
 const ZERO_PRICE = { price: 0, pricePercentChange1d: 0 };
 const getAmountFromBalance = (balance) => {
     if (typeof balance === 'object' && balance !== null && 'amount' in balance) {
@@ -234,7 +238,8 @@ function resolveAccountsToAggregate(args) {
     }
     return [selectedInternalAccount];
 }
-function getAggregatedBalanceForAccount(state, selectedInternalAccount, enabledNetworkMap, accountTreeState, internalAccountsOrAccountIds, accountsById) {
+function getAggregatedBalanceForAccount(state, selectedInternalAccount, enabledNetworkMap, accountTreeState, internalAccountsOrAccountIds, accountsById, trace) {
+    const startTime = trace ? performance.now() : 0;
     const { assetsBalance, assetsInfo, assetPreferences, assetsPrice } = state;
     const metadata = (assetsInfo ?? {});
     const hasPrices = Boolean(assetsPrice) &&
@@ -292,6 +297,26 @@ function getAggregatedBalanceForAccount(state, selectedInternalAccount, enabledN
             }
         }
     }
+    if (trace) {
+        const durationMs = performance.now() - startTime;
+        const uniqueNetworks = new Set();
+        for (const assetId of merged.keys()) {
+            const info = getAssetInfo(assetInfoCache, assetId);
+            uniqueNetworks.add(info.chainId);
+        }
+        trace({
+            name: TRACE_AGGREGATED_BALANCE_SELECTOR,
+            data: {
+                duration_ms: durationMs,
+                asset_count: merged.size,
+                network_count: uniqueNetworks.size,
+                account_count: accountsToAggregate.length,
+            },
+            tags: { controller: 'AssetsController' },
+        }, () => undefined).catch(() => {
+            // Telemetry failure must not break.
+        });
+    }
     if (hasPrices) {
         const pricePercentChange1d = totalBalanceInFiat > 0 ? weightedNumerator / totalBalanceInFiat : 0;
         return {