@metamask-previews/assets-controller 14.0.2-preview-4f4c98e → 14.0.2-preview-eb263b0ee

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

@@ -0,0 +1,4 @@
+export { MulticallClient, TokensApiClient } from "./clients/index.mjs";
+export { BalanceFetcher, TokenDetector, StakedBalanceFetcher, getSupportedStakingChainIds, getStakingContractAddress, isStakingContractAssetId } from "./services/index.mjs";
+export { divideIntoBatches, reduceInBatchesSerially } from "./utils/index.mjs";
+//# sourceMappingURL=index.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../../src/data-sources/evm-rpc-services/index.ts"],"names":[],"mappings":"AAaA,OAAO,EACL,eAAe,EAEf,eAAe,EAGhB,4BAA2B;AAC5B,OAAO,EACL,cAAc,EACd,aAAa,EACb,oBAAoB,EACpB,2BAA2B,EAC3B,yBAAyB,EACzB,wBAAwB,EAQzB,6BAA4B;AAC7B,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,0BAAyB","sourcesContent":["export type {\n  Address,\n  AssetFetchEntry,\n  AssetsBalanceState,\n  ChainId,\n  GetProviderFunction,\n  Provider,\n  BalanceOfRequest,\n  BalanceOfResponse,\n  TokenListState,\n  BalanceFetchResult,\n  TokenDetectionResult,\n} from './types/index.js';\nexport {\n  MulticallClient,\n  type MulticallClientConfig,\n  TokensApiClient,\n  type TokensApiClientConfig,\n  type TokenListQueryClient,\n} from './clients/index.js';\nexport {\n  BalanceFetcher,\n  TokenDetector,\n  StakedBalanceFetcher,\n  getSupportedStakingChainIds,\n  getStakingContractAddress,\n  isStakingContractAssetId,\n  type BalancePollingInput,\n  type DetectionPollingInput,\n  type StakedBalancePollingInput,\n  type StakedBalanceFetchResult,\n  type OnBalanceUpdateCallback,\n  type OnDetectionUpdateCallback,\n  type OnStakedBalanceUpdateCallback,\n} from './services/index.js';\nexport { divideIntoBatches, reduceInBatchesSerially } from './utils/index.js';\n"]}

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

@@ -0,0 +1,247 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    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_isNativeAsset, _BalanceFetcher_onBalanceUpdate, _BalanceFetcher_getAssetsToFetch, _BalanceFetcher_fetchBalances, _BalanceFetcher_processBalanceResponses, _BalanceFetcher_formatBalance;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BalanceFetcher = void 0;
+const polling_controller_1 = require("@metamask/polling-controller");
+const utils_1 = require("@metamask/utils");
+const constants_js_1 = require("../../../utils/constants.cjs");
+const index_js_1 = require("../utils/index.cjs");
+const DEFAULT_BALANCE_INTERVAL = 30000; // 30 seconds
+/**
+ * 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.
+ */
+class BalanceFetcher extends (0, polling_controller_1.StaticIntervalPollingControllerOnly)() {
+    constructor(multicallClient, messenger, config) {
+        super();
+        _BalanceFetcher_instances.add(this);
+        _BalanceFetcher_multicallClient.set(this, void 0);
+        _BalanceFetcher_messenger.set(this, void 0);
+        _BalanceFetcher_config.set(this, void 0);
+        _BalanceFetcher_isNativeAsset.set(this, void 0);
+        _BalanceFetcher_onBalanceUpdate.set(this, void 0);
+        __classPrivateFieldSet(this, _BalanceFetcher_multicallClient, multicallClient, "f");
+        __classPrivateFieldSet(this, _BalanceFetcher_messenger, messenger, "f");
+        __classPrivateFieldSet(this, _BalanceFetcher_config, {
+            defaultBatchSize: config.defaultBatchSize ?? 300,
+            defaultTimeoutMs: config.defaultTimeoutMs ?? 30000,
+        }, "f");
+        __classPrivateFieldSet(this, _BalanceFetcher_isNativeAsset, config.isNativeAsset, "f");
+        // Set the polling interval
+        this.setIntervalLength(config?.pollingInterval ?? DEFAULT_BALANCE_INTERVAL);
+    }
+    /**
+     * Set the callback to receive balance updates during polling.
+     *
+     * @param callback - Function to call with balance results.
+     */
+    setOnBalanceUpdate(callback) {
+        __classPrivateFieldSet(this, _BalanceFetcher_onBalanceUpdate, callback, "f");
+    }
+    /**
+     * Execute a poll cycle (required by base class).
+     * Fetches balances and calls the update callback.
+     *
+     * @param input - The polling input.
+     */
+    async _executePoll(input) {
+        const result = await __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_fetchBalances).call(this, input.chainId, input.accountId, input.accountAddress, input.customAssetsOnly === true);
+        if (__classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f") && result.balances.length > 0) {
+            await __classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f").call(this, result);
+        }
+    }
+    /**
+     * 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();
+        // Build a single map keyed by lowercase address that holds all info
+        // needed to match multicall responses back to their original entries.
+        const balanceRequests = [];
+        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 {
+                chainId,
+                accountId,
+                accountAddress,
+                balances: [],
+                failedAddresses: [],
+                timestamp,
+            };
+        }
+        const result = await (0, index_js_1.reduceInBatchesSerially)({
+            values: balanceRequests,
+            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, { fallbackToSingleCalls: true });
+                return __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_processBalanceResponses).call(this, responses, workingResult, chainId, accountId, timestamp, entryByAddress);
+            },
+        });
+        return {
+            chainId,
+            accountId,
+            accountAddress,
+            ...result,
+            timestamp,
+        };
+    }
+}
+exports.BalanceFetcher = BalanceFetcher;
+_BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new WeakMap(), _BalanceFetcher_config = new WeakMap(), _BalanceFetcher_isNativeAsset = new WeakMap(), _BalanceFetcher_onBalanceUpdate = new WeakMap(), _BalanceFetcher_instances = new WeakSet(), _BalanceFetcher_getAssetsToFetch = function _BalanceFetcher_getAssetsToFetch(chainId, accountId, customAssetsOnly) {
+    const state = __classPrivateFieldGet(this, _BalanceFetcher_messenger, "f").call('AssetsController:getState');
+    // 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();
+    const collect = (assetId) => {
+        const parsed = (0, utils_1.parseCaipAssetType)(assetId);
+        if (parsed.chain.reference !== chainIdDecimal) {
+            return;
+        }
+        const normalizedAssetId = assetId.toLowerCase();
+        if (assetsToFetch.has(normalizedAssetId)) {
+            return;
+        }
+        const isNative = __classPrivateFieldGet(this, _BalanceFetcher_isNativeAsset, "f").call(this, assetId);
+        assetsToFetch.set(normalizedAssetId, {
+            assetId,
+            address: isNative
+                ? constants_js_1.ZERO_ADDRESS
+                : parsed.assetReference.toLowerCase(),
+        });
+    };
+    // 1. Assets already tracked with a balance entry — only when not in
+    //    "customAssetsOnly" mode (otherwise another data source is already
+    //    refreshing them and we'd double-poll).
+    if (!customAssetsOnly) {
+        const accountBalances = state?.assetsBalance?.[accountId];
+        if (accountBalances) {
+            for (const assetId of Object.keys(accountBalances)) {
+                collect(assetId);
+            }
+        }
+    }
+    // 2. User-added custom assets — RPC is the sole balance fetcher for
+    //    these, so they must be polled even if they have no balance entry
+    //    yet (e.g. zero balance, first fetch failed, or state was hydrated
+    //    from disk before any successful fetch).
+    const customAssets = state?.customAssets?.[accountId];
+    if (customAssets) {
+        for (const assetId of customAssets) {
+            collect(assetId);
+        }
+    }
+    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.
+ * @param customAssetsOnly - When true, skip `assetsBalance` and only include `customAssets`.
+ * @returns Balance fetch result.
+ */
+async function _BalanceFetcher_fetchBalances(chainId, accountId, accountAddress, customAssetsOnly) {
+    const assets = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_getAssetsToFetch).call(this, chainId, accountId, customAssetsOnly);
+    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 isNative = lowerAddress === constants_js_1.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 (entry.decimals === undefined) {
+            formattedBalance = balance;
+        }
+        else {
+            decimals = entry.decimals;
+            formattedBalance = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_formatBalance).call(this, balance, decimals);
+        }
+        const balanceEntry = {
+            assetId: entry.assetId,
+            accountId,
+            chainId,
+            balance,
+            formattedBalance,
+            timestamp,
+        };
+        if (typeof decimals === 'number') {
+            balanceEntry.decimals = decimals;
+        }
+        balances.push(balanceEntry);
+    }
+    return { balances, failedAddresses };
+}, _BalanceFetcher_formatBalance = function _BalanceFetcher_formatBalance(rawBalance, decimals) {
+    if (rawBalance === '0') {
+        return '0';
+    }
+    try {
+        const balanceBigInt = BigInt(rawBalance);
+        const divisor = BigInt(10 ** decimals);
+        const integerPart = balanceBigInt / divisor;
+        const remainder = balanceBigInt % divisor;
+        const fractionalStr = remainder.toString().padStart(decimals, '0');
+        const trimmedFractional = fractionalStr.replace(/0+$/u, '');
+        if (trimmedFractional === '') {
+            return integerPart.toString();
+        }
+        return `${integerPart}.${trimmedFractional}`;
+    }
+    catch {
+        return rawBalance;
+    }
+};
+//# sourceMappingURL=BalanceFetcher.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BalanceFetcher.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,qEAAmF;AACnF,2CAAqD;AAErD,+DAA2D;AAc3D,iDAA4D;AAE5D,MAAM,wBAAwB,GAAG,KAAM,CAAC,CAAC,aAAa;AA4CtD;;;;;;;;GAQG;AACH,MAAa,cAAe,SAAQ,IAAA,wDAAmC,GAAuB;IAa5F,YACE,eAAgC,EAChC,SAAkC,EAClC,MAA4B;QAE5B,KAAK,EAAE,CAAC;;QAjBD,kDAAkC;QAElC,4CAAoC;QAEpC,yCAEP;QAEO,gDAAoD;QAE7D,kDAAsD;QAQpD,uBAAA,IAAI,mCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6BAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,0BAAW;YACb,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,GAAG;YAChD,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,KAAK;SACnD,MAAA,CAAC;QACF,uBAAA,IAAI,iCAAkB,MAAM,CAAC,aAAa,MAAA,CAAC;QAE3C,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,EACpB,KAAK,CAAC,gBAAgB,KAAK,IAAI,CAChC,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;IAmGD;;;;;;;;;;;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,IAAA,kCAAuB,EAG1C;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,EACL,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAChC,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;AAvUD,wCAuUC;4VAlQG,OAAgB,EAChB,SAAoB,EACpB,gBAAyB;IAEzB,MAAM,KAAK,GAAG,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAEhE,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,MAAM,OAAO,GAAG,CAAC,OAAsB,EAAQ,EAAE;QAC/C,MAAM,MAAM,GAAG,IAAA,0BAAkB,EAAC,OAAO,CAAC,CAAC;QAE3C,IAAI,MAAM,CAAC,KAAK,CAAC,SAAS,KAAK,cAAc,EAAE,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAEhD,IAAI,aAAa,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzC,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,uBAAA,IAAI,qCAAe,MAAnB,IAAI,EAAgB,OAAO,CAAC,CAAC;QAE9C,aAAa,CAAC,GAAG,CAAC,iBAAiB,EAAE;YACnC,OAAO;YACP,OAAO,EAAE,QAAQ;gBACf,CAAC,CAAC,2BAAY;gBACd,CAAC,CAAE,MAAM,CAAC,cAAc,CAAC,WAAW,EAAc;SACrD,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,oEAAoE;IACpE,uEAAuE;IACvE,4CAA4C;IAC5C,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,eAAe,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,eAAe,EAAE,CAAC;YACpB,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAoB,EAAE,CAAC;gBACtE,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,oEAAoE;IACpE,sEAAsE;IACtE,uEAAuE;IACvE,6CAA6C;IAC7C,MAAM,YAAY,GAAG,KAAK,EAAE,YAAY,EAAE,CAAC,SAAS,CAAC,CAAC;IACtD,IAAI,YAAY,EAAE,CAAC;QACjB,KAAK,MAAM,OAAO,IAAI,YAA+B,EAAE,CAAC;YACtD,OAAO,CAAC,OAAO,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC;AAED;;;;;;;;;GASG;AACH,KAAK,wCACH,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,gBAAyB;IAEzB,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAE5E,OAAO,IAAI,CAAC,sBAAsB,CAChC,OAAO,EACP,SAAS,EACT,cAAc,EACd,MAAM,CACP,CAAC;AACJ,CAAC,6FA2FC,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,2BAAY,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.js';\nimport type { MulticallClient } from '../clients/index.js';\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/index.js';\nimport { reduceInBatchesSerially } from '../utils/index.js';\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  /** Determines whether a CAIP-19 asset ID represents a native asset. */\n  isNativeAsset: (assetId: CaipAssetType) => boolean;\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   * When true, only fetch balances for entries in `state.customAssets`,\n   * skipping `state.assetsBalance`. Used by the supplemental RPC\n   * subscription on chains that another data source is already covering\n   * for regular balance refreshes.\n   */\n  customAssetsOnly?: boolean;\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<\n    Omit<BalanceFetcherConfig, 'pollingInterval' | 'isNativeAsset'>\n  >;\n\n  readonly #isNativeAsset: (assetId: CaipAssetType) => boolean;\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    this.#isNativeAsset = config.isNativeAsset;\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      input.customAssetsOnly === true,\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   * @param customAssetsOnly - When true, skip `assetsBalance` and only include `customAssets`.\n   * @returns Array of asset fetch entries from state for the requested chain.\n   */\n  #getAssetsToFetch(\n    chainId: ChainId,\n    accountId: AccountId,\n    customAssetsOnly: boolean,\n  ): AssetFetchEntry[] {\n    const state = this.#messenger.call('AssetsController:getState');\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    const collect = (assetId: CaipAssetType): void => {\n      const parsed = parseCaipAssetType(assetId);\n\n      if (parsed.chain.reference !== chainIdDecimal) {\n        return;\n      }\n\n      const normalizedAssetId = assetId.toLowerCase();\n\n      if (assetsToFetch.has(normalizedAssetId)) {\n        return;\n      }\n\n      const isNative = this.#isNativeAsset(assetId);\n\n      assetsToFetch.set(normalizedAssetId, {\n        assetId,\n        address: isNative\n          ? ZERO_ADDRESS\n          : (parsed.assetReference.toLowerCase() as Address),\n      });\n    };\n\n    // 1. Assets already tracked with a balance entry — only when not in\n    //    \"customAssetsOnly\" mode (otherwise another data source is already\n    //    refreshing them and we'd double-poll).\n    if (!customAssetsOnly) {\n      const accountBalances = state?.assetsBalance?.[accountId];\n      if (accountBalances) {\n        for (const assetId of Object.keys(accountBalances) as CaipAssetType[]) {\n          collect(assetId);\n        }\n      }\n    }\n\n    // 2. User-added custom assets — RPC is the sole balance fetcher for\n    //    these, so they must be polled even if they have no balance entry\n    //    yet (e.g. zero balance, first fetch failed, or state was hydrated\n    //    from disk before any successful fetch).\n    const customAssets = state?.customAssets?.[accountId];\n    if (customAssets) {\n      for (const assetId of customAssets as CaipAssetType[]) {\n        collect(assetId);\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   * @param customAssetsOnly - When true, skip `assetsBalance` and only include `customAssets`.\n   * @returns Balance fetch result.\n   */\n  async #fetchBalances(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    customAssetsOnly: boolean,\n  ): Promise<BalanceFetchResult> {\n    const assets = this.#getAssetsToFetch(chainId, accountId, customAssetsOnly);\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          { fallbackToSingleCalls: true },\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/services/{BalanceFetcher.d.ts → BalanceFetcher.d.cts}

@@ -1,5 +1,5 @@
-import type { MulticallClient } from '../clients/index.js';
-import type { AccountId, Address, AssetFetchEntry, AssetsBalanceState, BalanceFetchResult, CaipAssetType, ChainId } from '../types/index.js';
+import type { MulticallClient } from "../clients/index.cjs";
+import type { AccountId, Address, AssetFetchEntry, AssetsBalanceState, BalanceFetchResult, CaipAssetType, ChainId } from "../types/index.cjs";
 /**
  * Minimal messenger interface for BalanceFetcher.
  */
@@ -37,18 +37,18 @@ export type BalancePollingInput = {
  */
 export type OnBalanceUpdateCallback = (result: BalanceFetchResult) => void | Promise<void>;
 declare const BalanceFetcher_base: (abstract new (...args: any[]) => {
-    readonly _callbacks: Map<string, Set<(input: BalancePollingInput) => void>>;
+    readonly "__#6@#intervalIds": Record<string, NodeJS.Timeout>;
+    "__#6@#intervalLength": number | undefined;
+    setIntervalLength(intervalLength: number): void;
+    getIntervalLength(): number | undefined;
+    _startPolling(input: BalancePollingInput): void;
+    _stopPollingByPollingTokenSetId(key: string): void;
+    readonly "__#3@#pollingTokenSets": Map<string, Set<string>>;
+    readonly "__#3@#callbacks": Map<string, Set<(input: BalancePollingInput) => void>>;
     _executePoll(input: BalancePollingInput): Promise<void>;
     startPolling(input: BalancePollingInput): string;
     stopAllPolling(): void;
     stopPollingByPollingToken(pollingToken: string): void;
-    readonly _intervalIds: Record<import("@metamask/polling-controller").PollingTokenSetId, NodeJS.Timeout>;
-    _intervalLength: number | undefined;
-    setIntervalLength(intervalLength: number): void;
-    getIntervalLength(): number | undefined;
-    _startPolling(input: BalancePollingInput): void;
-    _stopPollingByPollingTokenSetId(key: import("@metamask/polling-controller").PollingTokenSetId): void;
-    readonly _pollingTokenSets: Map<import("@metamask/polling-controller").PollingTokenSetId, Set<string>>;
     onPollingComplete(input: BalancePollingInput, callback: (input: BalancePollingInput) => void): void;
 }) & {
     new (): {};
@@ -93,4 +93,4 @@ export declare class BalanceFetcher extends BalanceFetcher_base {
     fetchBalancesForAssets(chainId: ChainId, accountId: AccountId, accountAddress: Address, assets: AssetFetchEntry[]): Promise<BalanceFetchResult>;
 }
 export {};
-//# sourceMappingURL=BalanceFetcher.d.ts.map
+//# sourceMappingURL=BalanceFetcher.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BalanceFetcher.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,eAAe,EAAE,6BAA4B;AAC3D,OAAO,KAAK,EACV,SAAS,EACT,OAAO,EAEP,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAGlB,aAAa,EACb,OAAO,EACR,2BAA0B;AAK3B;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,EAAE,CAAC,MAAM,EAAE,2BAA2B,KAAK,kBAAkB,CAAC;CACnE,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,4CAA4C;IAC5C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uEAAuE;IACvE,aAAa,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC;CACpD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,4BAA4B;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB;;;;;OAKG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,CACpC,MAAM,EAAE,kBAAkB,KACvB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;;;;;;;;;;;;;;;;;;AAE1B;;;;;;;;GAQG;AACH,qBAAa,cAAe,SAAQ,mBAA0D;;gBAc1F,eAAe,EAAE,eAAe,EAChC,SAAS,EAAE,uBAAuB,EAClC,MAAM,EAAE,oBAAoB;IAe9B;;;;OAIG;IACH,kBAAkB,CAAC,QAAQ,EAAE,uBAAuB,GAAG,IAAI;IAI3D;;;;;OAKG;IACG,YAAY,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8G7D;;;;;;;;;;;OAWG;IACG,sBAAsB,CAC1B,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,SAAS,EACpB,cAAc,EAAE,OAAO,EACvB,MAAM,EAAE,eAAe,EAAE,GACxB,OAAO,CAAC,kBAAkB,CAAC;CA0J/B"}

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

@@ -0,0 +1,96 @@
+import type { MulticallClient } from "../clients/index.mjs";
+import type { AccountId, Address, AssetFetchEntry, AssetsBalanceState, BalanceFetchResult, CaipAssetType, ChainId } from "../types/index.mjs";
+/**
+ * Minimal messenger interface for BalanceFetcher.
+ */
+export type BalanceFetcherMessenger = {
+    call: (action: 'AssetsController:getState') => AssetsBalanceState;
+};
+export type BalanceFetcherConfig = {
+    defaultBatchSize?: number;
+    defaultTimeoutMs?: number;
+    /** Polling interval in ms (default: 30s) */
+    pollingInterval?: number;
+    /** Determines whether a CAIP-19 asset ID represents a native asset. */
+    isNativeAsset: (assetId: CaipAssetType) => boolean;
+};
+/**
+ * Polling input for BalanceFetcher - identifies what to poll for.
+ */
+export type BalancePollingInput = {
+    /** Chain ID (hex format) */
+    chainId: ChainId;
+    /** Account ID */
+    accountId: AccountId;
+    /** Account address */
+    accountAddress: Address;
+    /**
+     * When true, only fetch balances for entries in `state.customAssets`,
+     * skipping `state.assetsBalance`. Used by the supplemental RPC
+     * subscription on chains that another data source is already covering
+     * for regular balance refreshes.
+     */
+    customAssetsOnly?: boolean;
+};
+/**
+ * Callback type for balance updates.
+ */
+export type OnBalanceUpdateCallback = (result: BalanceFetchResult) => void | Promise<void>;
+declare const BalanceFetcher_base: (abstract new (...args: any[]) => {
+    readonly "__#6@#intervalIds": Record<string, NodeJS.Timeout>;
+    "__#6@#intervalLength": number | undefined;
+    setIntervalLength(intervalLength: number): void;
+    getIntervalLength(): number | undefined;
+    _startPolling(input: BalancePollingInput): void;
+    _stopPollingByPollingTokenSetId(key: string): void;
+    readonly "__#3@#pollingTokenSets": Map<string, Set<string>>;
+    readonly "__#3@#callbacks": Map<string, Set<(input: BalancePollingInput) => void>>;
+    _executePoll(input: BalancePollingInput): Promise<void>;
+    startPolling(input: BalancePollingInput): string;
+    stopAllPolling(): void;
+    stopPollingByPollingToken(pollingToken: string): void;
+    onPollingComplete(input: BalancePollingInput, callback: (input: BalancePollingInput) => void): void;
+}) & {
+    new (): {};
+};
+/**
+ * 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 declare class BalanceFetcher extends BalanceFetcher_base {
+    #private;
+    constructor(multicallClient: MulticallClient, messenger: BalanceFetcherMessenger, config: BalanceFetcherConfig);
+    /**
+     * Set the callback to receive balance updates during polling.
+     *
+     * @param callback - Function to call with balance results.
+     */
+    setOnBalanceUpdate(callback: OnBalanceUpdateCallback): void;
+    /**
+     * Execute a poll cycle (required by base class).
+     * Fetches balances and calls the update callback.
+     *
+     * @param input - The polling input.
+     */
+    _executePoll(input: BalancePollingInput): Promise<void>;
+    /**
+     * 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.
+     */
+    fetchBalancesForAssets(chainId: ChainId, accountId: AccountId, accountAddress: Address, assets: AssetFetchEntry[]): Promise<BalanceFetchResult>;
+}
+export {};
+//# sourceMappingURL=BalanceFetcher.d.mts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BalanceFetcher.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,eAAe,EAAE,6BAA4B;AAC3D,OAAO,KAAK,EACV,SAAS,EACT,OAAO,EAEP,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAGlB,aAAa,EACb,OAAO,EACR,2BAA0B;AAK3B;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,EAAE,CAAC,MAAM,EAAE,2BAA2B,KAAK,kBAAkB,CAAC;CACnE,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,4CAA4C;IAC5C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uEAAuE;IACvE,aAAa,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC;CACpD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,4BAA4B;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB;;;;;OAKG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,CACpC,MAAM,EAAE,kBAAkB,KACvB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;;;;;;;;;;;;;;;;;;AAE1B;;;;;;;;GAQG;AACH,qBAAa,cAAe,SAAQ,mBAA0D;;gBAc1F,eAAe,EAAE,eAAe,EAChC,SAAS,EAAE,uBAAuB,EAClC,MAAM,EAAE,oBAAoB;IAe9B;;;;OAIG;IACH,kBAAkB,CAAC,QAAQ,EAAE,uBAAuB,GAAG,IAAI;IAI3D;;;;;OAKG;IACG,YAAY,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8G7D;;;;;;;;;;;OAWG;IACG,sBAAsB,CAC1B,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,SAAS,EACpB,cAAc,EAAE,OAAO,EACvB,MAAM,EAAE,eAAe,EAAE,GACxB,OAAO,CAAC,kBAAkB,CAAC;CA0J/B"}

package/dist/data-sources/evm-rpc-services/services/{BalanceFetcher.js → BalanceFetcher.mjs}

@@ -10,10 +10,10 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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_isNativeAsset, _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.js';
-import { reduceInBatchesSerially } from '../utils/index.js';
+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
 /**
  * BalanceFetcher - Fetches token balances via multicall.
@@ -240,4 +240,4 @@ async function _BalanceFetcher_fetchBalances(chainId, accountId, accountAddress,
         return rawBalance;
     }
 };
-//# sourceMappingURL=BalanceFetcher.js.map
+//# sourceMappingURL=BalanceFetcher.mjs.map

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

@@ -0,0 +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;AACnF,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAErD,OAAO,EAAE,YAAY,EAAE,qCAAoC;AAc3D,OAAO,EAAE,uBAAuB,EAAE,2BAA0B;AAE5D,MAAM,wBAAwB,GAAG,KAAM,CAAC,CAAC,aAAa;AA4CtD;;;;;;;;GAQG;AACH,MAAM,OAAO,cAAe,SAAQ,mCAAmC,EAAuB;IAa5F,YACE,eAAgC,EAChC,SAAkC,EAClC,MAA4B;QAE5B,KAAK,EAAE,CAAC;;QAjBD,kDAAkC;QAElC,4CAAoC;QAEpC,yCAEP;QAEO,gDAAoD;QAE7D,kDAAsD;QAQpD,uBAAA,IAAI,mCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6BAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,0BAAW;YACb,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,GAAG;YAChD,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,KAAK;SACnD,MAAA,CAAC;QACF,uBAAA,IAAI,iCAAkB,MAAM,CAAC,aAAa,MAAA,CAAC;QAE3C,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,EACpB,KAAK,CAAC,gBAAgB,KAAK,IAAI,CAChC,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;IAmGD;;;;;;;;;;;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,EACL,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAChC,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;4VAlQG,OAAgB,EAChB,SAAoB,EACpB,gBAAyB;IAEzB,MAAM,KAAK,GAAG,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAEhE,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,MAAM,OAAO,GAAG,CAAC,OAAsB,EAAQ,EAAE;QAC/C,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAE3C,IAAI,MAAM,CAAC,KAAK,CAAC,SAAS,KAAK,cAAc,EAAE,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAEhD,IAAI,aAAa,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzC,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,uBAAA,IAAI,qCAAe,MAAnB,IAAI,EAAgB,OAAO,CAAC,CAAC;QAE9C,aAAa,CAAC,GAAG,CAAC,iBAAiB,EAAE;YACnC,OAAO;YACP,OAAO,EAAE,QAAQ;gBACf,CAAC,CAAC,YAAY;gBACd,CAAC,CAAE,MAAM,CAAC,cAAc,CAAC,WAAW,EAAc;SACrD,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,oEAAoE;IACpE,uEAAuE;IACvE,4CAA4C;IAC5C,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,eAAe,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,eAAe,EAAE,CAAC;YACpB,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAoB,EAAE,CAAC;gBACtE,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,oEAAoE;IACpE,sEAAsE;IACtE,uEAAuE;IACvE,6CAA6C;IAC7C,MAAM,YAAY,GAAG,KAAK,EAAE,YAAY,EAAE,CAAC,SAAS,CAAC,CAAC;IACtD,IAAI,YAAY,EAAE,CAAC;QACjB,KAAK,MAAM,OAAO,IAAI,YAA+B,EAAE,CAAC;YACtD,OAAO,CAAC,OAAO,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC;AAED;;;;;;;;;GASG;AACH,KAAK,wCACH,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,gBAAyB;IAEzB,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAE5E,OAAO,IAAI,CAAC,sBAAsB,CAChC,OAAO,EACP,SAAS,EACT,cAAc,EACd,MAAM,CACP,CAAC;AACJ,CAAC,6FA2FC,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.js';\nimport type { MulticallClient } from '../clients/index.js';\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/index.js';\nimport { reduceInBatchesSerially } from '../utils/index.js';\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  /** Determines whether a CAIP-19 asset ID represents a native asset. */\n  isNativeAsset: (assetId: CaipAssetType) => boolean;\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   * When true, only fetch balances for entries in `state.customAssets`,\n   * skipping `state.assetsBalance`. Used by the supplemental RPC\n   * subscription on chains that another data source is already covering\n   * for regular balance refreshes.\n   */\n  customAssetsOnly?: boolean;\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<\n    Omit<BalanceFetcherConfig, 'pollingInterval' | 'isNativeAsset'>\n  >;\n\n  readonly #isNativeAsset: (assetId: CaipAssetType) => boolean;\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    this.#isNativeAsset = config.isNativeAsset;\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      input.customAssetsOnly === true,\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   * @param customAssetsOnly - When true, skip `assetsBalance` and only include `customAssets`.\n   * @returns Array of asset fetch entries from state for the requested chain.\n   */\n  #getAssetsToFetch(\n    chainId: ChainId,\n    accountId: AccountId,\n    customAssetsOnly: boolean,\n  ): AssetFetchEntry[] {\n    const state = this.#messenger.call('AssetsController:getState');\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    const collect = (assetId: CaipAssetType): void => {\n      const parsed = parseCaipAssetType(assetId);\n\n      if (parsed.chain.reference !== chainIdDecimal) {\n        return;\n      }\n\n      const normalizedAssetId = assetId.toLowerCase();\n\n      if (assetsToFetch.has(normalizedAssetId)) {\n        return;\n      }\n\n      const isNative = this.#isNativeAsset(assetId);\n\n      assetsToFetch.set(normalizedAssetId, {\n        assetId,\n        address: isNative\n          ? ZERO_ADDRESS\n          : (parsed.assetReference.toLowerCase() as Address),\n      });\n    };\n\n    // 1. Assets already tracked with a balance entry — only when not in\n    //    \"customAssetsOnly\" mode (otherwise another data source is already\n    //    refreshing them and we'd double-poll).\n    if (!customAssetsOnly) {\n      const accountBalances = state?.assetsBalance?.[accountId];\n      if (accountBalances) {\n        for (const assetId of Object.keys(accountBalances) as CaipAssetType[]) {\n          collect(assetId);\n        }\n      }\n    }\n\n    // 2. User-added custom assets — RPC is the sole balance fetcher for\n    //    these, so they must be polled even if they have no balance entry\n    //    yet (e.g. zero balance, first fetch failed, or state was hydrated\n    //    from disk before any successful fetch).\n    const customAssets = state?.customAssets?.[accountId];\n    if (customAssets) {\n      for (const assetId of customAssets as CaipAssetType[]) {\n        collect(assetId);\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   * @param customAssetsOnly - When true, skip `assetsBalance` and only include `customAssets`.\n   * @returns Balance fetch result.\n   */\n  async #fetchBalances(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    customAssetsOnly: boolean,\n  ): Promise<BalanceFetchResult> {\n    const assets = this.#getAssetsToFetch(chainId, accountId, customAssetsOnly);\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          { fallbackToSingleCalls: true },\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"]}