@metamask-previews/assets-controller 14.0.2-preview-3866c0ff1 → 14.0.2-preview-e3275932a

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

@@ -0,0 +1,243 @@
+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;
+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.
+ * 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) {
+        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 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,
+        };
+    }
+}
+_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 = 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
+                ? 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 === 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.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"]}

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

@@ -0,0 +1,132 @@
+"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 _StakedBalanceFetcher_providerGetter, _StakedBalanceFetcher_onStakedBalanceUpdate;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StakedBalanceFetcher = exports.STAKING_INTERFACE = exports.isStakingContractAssetId = exports.getSupportedStakingChainIds = exports.getStakingContractAddress = void 0;
+const abi_1 = require("@ethersproject/abi");
+const providers_1 = require("@ethersproject/providers");
+const polling_controller_1 = require("@metamask/polling-controller");
+const index_js_1 = require("../utils/index.cjs");
+Object.defineProperty(exports, "getStakingContractAddress", { enumerable: true, get: function () { return index_js_1.getStakingContractAddress; } });
+Object.defineProperty(exports, "getSupportedStakingChainIds", { enumerable: true, get: function () { return index_js_1.getSupportedStakingChainIds; } });
+Object.defineProperty(exports, "isStakingContractAssetId", { enumerable: true, get: function () { return index_js_1.isStakingContractAssetId; } });
+/** Staking contract ABI: getShares(account) and convertToAssets(shares). */
+const STAKING_CONTRACT_ABI = [
+    {
+        inputs: [{ internalType: 'address', name: 'account', type: 'address' }],
+        name: 'getShares',
+        outputs: [{ internalType: 'uint256', name: '', type: 'uint256' }],
+        stateMutability: 'view',
+        type: 'function',
+    },
+    {
+        inputs: [{ internalType: 'uint256', name: 'shares', type: 'uint256' }],
+        name: 'convertToAssets',
+        outputs: [{ internalType: 'uint256', name: 'assets', type: 'uint256' }],
+        stateMutability: 'view',
+        type: 'function',
+    },
+];
+exports.STAKING_INTERFACE = new abi_1.Interface(STAKING_CONTRACT_ABI);
+const STAKING_DECIMALS = 18;
+const DEFAULT_STAKED_BALANCE_INTERVAL = 180000; // 3 minutes
+class StakedBalanceFetcher extends (0, polling_controller_1.StaticIntervalPollingControllerOnly)() {
+    constructor(config) {
+        super();
+        _StakedBalanceFetcher_providerGetter.set(this, void 0);
+        _StakedBalanceFetcher_onStakedBalanceUpdate.set(this, void 0);
+        __classPrivateFieldSet(this, _StakedBalanceFetcher_providerGetter, config?.getNetworkProvider, "f");
+        this.setIntervalLength(config?.pollingInterval ?? DEFAULT_STAKED_BALANCE_INTERVAL);
+    }
+    /**
+     * Register a callback that is invoked after every successful poll with
+     * the staked balance (including zero). Zero is reported so that merged
+     * updates can clear prior non-zero state.
+     *
+     * @param callback - The callback to invoke.
+     */
+    setOnStakedBalanceUpdate(callback) {
+        __classPrivateFieldSet(this, _StakedBalanceFetcher_onStakedBalanceUpdate, callback, "f");
+    }
+    async _executePoll(input) {
+        let result;
+        try {
+            result = await this.fetchStakedBalance(input);
+        }
+        catch {
+            // Do not push an update on provider/RPC failure; otherwise we would
+            // overwrite existing non-zero staked balances with zero in state.
+            return;
+        }
+        if (__classPrivateFieldGet(this, _StakedBalanceFetcher_onStakedBalanceUpdate, "f")) {
+            __classPrivateFieldGet(this, _StakedBalanceFetcher_onStakedBalanceUpdate, "f").call(this, {
+                accountId: input.accountId,
+                chainId: input.chainId,
+                balance: result,
+            });
+        }
+    }
+    /**
+     * Fetches the staked balance for an account on a chain using the same
+     * staking contract as AccountTrackerController (getShares then convertToAssets).
+     * Returns a human-readable amount string (e.g. "1.5" for 1.5 ETH).
+     * Throws when no provider is available or when the RPC/contract call fails, so
+     * callers do not persist a false zero and overwrite existing balances.
+     *
+     * @param input - Chain, account ID, and address to query.
+     * @returns Human-readable staked balance (amount string).
+     * @throws When provider is missing or when getShares/convertToAssets fails.
+     */
+    async fetchStakedBalance(input) {
+        const { chainId, accountAddress } = input;
+        const provider = __classPrivateFieldGet(this, _StakedBalanceFetcher_providerGetter, "f")?.call(this, chainId);
+        if (!provider) {
+            throw new Error('StakedBalanceFetcher: no provider available for chain');
+        }
+        const contractAddress = (0, index_js_1.getStakingContractAddress)(chainId);
+        if (!contractAddress) {
+            return { amount: '0' };
+        }
+        try {
+            const sharesCalldata = exports.STAKING_INTERFACE.encodeFunctionData('getShares', [
+                accountAddress,
+            ]);
+            const sharesResult = await provider.call({
+                to: contractAddress,
+                data: sharesCalldata,
+            });
+            const sharesRaw = exports.STAKING_INTERFACE.decodeFunctionResult('getShares', sharesResult)[0];
+            const sharesBigNum = BigInt(sharesRaw.toString());
+            if (sharesBigNum === 0n) {
+                return { amount: '0' };
+            }
+            const assetsCalldata = exports.STAKING_INTERFACE.encodeFunctionData('convertToAssets', [sharesBigNum]);
+            const assetsResult = await provider.call({
+                to: contractAddress,
+                data: assetsCalldata,
+            });
+            const assetsRaw = exports.STAKING_INTERFACE.decodeFunctionResult('convertToAssets', assetsResult)[0];
+            const assetsWei = BigInt(assetsRaw.toString());
+            const amount = (0, index_js_1.weiToHumanReadable)(assetsWei, STAKING_DECIMALS);
+            return { amount };
+        }
+        catch (error) {
+            throw error instanceof Error
+                ? error
+                : new Error('StakedBalanceFetcher: failed to fetch staked balance');
+        }
+    }
+}
+exports.StakedBalanceFetcher = StakedBalanceFetcher;
+_StakedBalanceFetcher_providerGetter = new WeakMap(), _StakedBalanceFetcher_onStakedBalanceUpdate = new WeakMap();
+//# sourceMappingURL=StakedBalanceFetcher.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"StakedBalanceFetcher.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/StakedBalanceFetcher.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,4CAA+C;AAC/C,wDAAwD;AACxD,qEAAmF;AAGnF,iDAK2B;AAGzB,0GAPA,oCAAyB,OAOA;AACzB,4GAPA,sCAA2B,OAOA;AAC3B,yGAPA,mCAAwB,OAOA;AAkC1B,4EAA4E;AAC5E,MAAM,oBAAoB,GAAG;IAC3B;QACE,MAAM,EAAE,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACvE,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACjE,eAAe,EAAE,MAAM;QACvB,IAAI,EAAE,UAAU;KACjB;IACD;QACE,MAAM,EAAE,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACtE,IAAI,EAAE,iBAAiB;QACvB,OAAO,EAAE,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACvE,eAAe,EAAE,MAAM;QACvB,IAAI,EAAE,UAAU;KACjB;CACF,CAAC;AAEW,QAAA,iBAAiB,GAAG,IAAI,eAAS,CAAC,oBAAoB,CAAC,CAAC;AAErE,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAS5B,MAAM,+BAA+B,GAAG,MAAO,CAAC,CAAC,YAAY;AAE7D,MAAa,oBAAqB,SAAQ,IAAA,wDAAmC,GAA6B;IAKxG,YAAY,MAAmC;QAC7C,KAAK,EAAE,CAAC;QALD,uDAAiE;QAE1E,8DAAkE;QAIhE,uBAAA,IAAI,wCAAmB,MAAM,EAAE,kBAAkB,MAAA,CAAC;QAElD,IAAI,CAAC,iBAAiB,CACpB,MAAM,EAAE,eAAe,IAAI,+BAA+B,CAC3D,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,wBAAwB,CAAC,QAAuC;QAC9D,uBAAA,IAAI,+CAA0B,QAAQ,MAAA,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAgC;QACjD,IAAI,MAAqB,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,oEAAoE;YACpE,kEAAkE;YAClE,OAAO;QACT,CAAC;QAED,IAAI,uBAAA,IAAI,mDAAuB,EAAE,CAAC;YAChC,uBAAA,IAAI,mDAAuB,MAA3B,IAAI,EAAwB;gBAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,kBAAkB,CACtB,KAAgC;QAEhC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC;QAC1C,MAAM,QAAQ,GAAG,uBAAA,IAAI,4CAAgB,EAAE,KAAtB,IAAI,EAAmB,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,eAAe,GAAG,IAAA,oCAAyB,EAAC,OAAO,CAAC,CAAC;QAE3D,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;QACzB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,yBAAiB,CAAC,kBAAkB,CAAC,WAAW,EAAE;gBACvE,cAAc;aACf,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;gBACvC,EAAE,EAAE,eAAe;gBACnB,IAAI,EAAE,cAAc;aACrB,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,yBAAiB,CAAC,oBAAoB,CACtD,WAAW,EACX,YAAY,CACb,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;YAElD,IAAI,YAAY,KAAK,EAAE,EAAE,CAAC;gBACxB,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;YACzB,CAAC;YAED,MAAM,cAAc,GAAG,yBAAiB,CAAC,kBAAkB,CACzD,iBAAiB,EACjB,CAAC,YAAY,CAAC,CACf,CAAC;YACF,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;gBACvC,EAAE,EAAE,eAAe;gBACnB,IAAI,EAAE,cAAc;aACrB,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,yBAAiB,CAAC,oBAAoB,CACtD,iBAAiB,EACjB,YAAY,CACb,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE/C,MAAM,MAAM,GAAG,IAAA,6BAAkB,EAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;YAC/D,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,YAAY,KAAK;gBAC1B,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;CACF;AA7GD,oDA6GC","sourcesContent":["import { Interface } from '@ethersproject/abi';\nimport { Web3Provider } from '@ethersproject/providers';\nimport { StaticIntervalPollingControllerOnly } from '@metamask/polling-controller';\n\nimport type { Address, AccountId, ChainId } from '../types/index.js';\nimport {\n  getStakingContractAddress,\n  getSupportedStakingChainIds,\n  isStakingContractAssetId,\n  weiToHumanReadable,\n} from '../utils/index.js';\n\nexport {\n  getStakingContractAddress,\n  getSupportedStakingChainIds,\n  isStakingContractAssetId,\n};\n\nexport type StakedBalancePollingInput = {\n  /** Chain ID (hex format, e.g. 0x1) */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n};\n\n/** Human-readable staked balance (e.g. \"1.5\" for 1.5 ETH). */\nexport type StakedBalance = {\n  amount: string;\n};\n\n/** Result reported via the update callback. */\nexport type StakedBalanceFetchResult = {\n  /** Account ID (UUID). */\n  accountId: AccountId;\n  /** Hex chain ID. */\n  chainId: ChainId;\n  /** Human-readable staked balance. */\n  balance: StakedBalance;\n};\n\n/**\n * Callback type for staked balance updates.\n */\nexport type OnStakedBalanceUpdateCallback = (\n  result: StakedBalanceFetchResult,\n) => void;\n\n/** Staking contract ABI: getShares(account) and convertToAssets(shares). */\nconst STAKING_CONTRACT_ABI = [\n  {\n    inputs: [{ internalType: 'address', name: 'account', type: 'address' }],\n    name: 'getShares',\n    outputs: [{ internalType: 'uint256', name: '', type: 'uint256' }],\n    stateMutability: 'view',\n    type: 'function',\n  },\n  {\n    inputs: [{ internalType: 'uint256', name: 'shares', type: 'uint256' }],\n    name: 'convertToAssets',\n    outputs: [{ internalType: 'uint256', name: 'assets', type: 'uint256' }],\n    stateMutability: 'view',\n    type: 'function',\n  },\n];\n\nexport const STAKING_INTERFACE = new Interface(STAKING_CONTRACT_ABI);\n\nconst STAKING_DECIMALS = 18;\n\nexport type StakedBalanceFetcherConfig = {\n  /** Polling interval in ms (default: 180s) */\n  pollingInterval?: number;\n  /** Returns the network provider for the given chain. Required for fetchStakedBalance. */\n  getNetworkProvider?: (chainId: ChainId) => Web3Provider | undefined;\n};\n\nconst DEFAULT_STAKED_BALANCE_INTERVAL = 180_000; // 3 minutes\n\nexport class StakedBalanceFetcher extends StaticIntervalPollingControllerOnly<StakedBalancePollingInput>() {\n  readonly #providerGetter?: (chainId: ChainId) => Web3Provider | undefined;\n\n  #onStakedBalanceUpdate: OnStakedBalanceUpdateCallback | undefined;\n\n  constructor(config?: StakedBalanceFetcherConfig) {\n    super();\n    this.#providerGetter = config?.getNetworkProvider;\n\n    this.setIntervalLength(\n      config?.pollingInterval ?? DEFAULT_STAKED_BALANCE_INTERVAL,\n    );\n  }\n\n  /**\n   * Register a callback that is invoked after every successful poll with\n   * the staked balance (including zero). Zero is reported so that merged\n   * updates can clear prior non-zero state.\n   *\n   * @param callback - The callback to invoke.\n   */\n  setOnStakedBalanceUpdate(callback: OnStakedBalanceUpdateCallback): void {\n    this.#onStakedBalanceUpdate = callback;\n  }\n\n  async _executePoll(input: StakedBalancePollingInput): Promise<void> {\n    let result: StakedBalance;\n    try {\n      result = await this.fetchStakedBalance(input);\n    } catch {\n      // Do not push an update on provider/RPC failure; otherwise we would\n      // overwrite existing non-zero staked balances with zero in state.\n      return;\n    }\n\n    if (this.#onStakedBalanceUpdate) {\n      this.#onStakedBalanceUpdate({\n        accountId: input.accountId,\n        chainId: input.chainId,\n        balance: result,\n      });\n    }\n  }\n\n  /**\n   * Fetches the staked balance for an account on a chain using the same\n   * staking contract as AccountTrackerController (getShares then convertToAssets).\n   * Returns a human-readable amount string (e.g. \"1.5\" for 1.5 ETH).\n   * Throws when no provider is available or when the RPC/contract call fails, so\n   * callers do not persist a false zero and overwrite existing balances.\n   *\n   * @param input - Chain, account ID, and address to query.\n   * @returns Human-readable staked balance (amount string).\n   * @throws When provider is missing or when getShares/convertToAssets fails.\n   */\n  async fetchStakedBalance(\n    input: StakedBalancePollingInput,\n  ): Promise<StakedBalance> {\n    const { chainId, accountAddress } = input;\n    const provider = this.#providerGetter?.(chainId);\n    if (!provider) {\n      throw new Error('StakedBalanceFetcher: no provider available for chain');\n    }\n    const contractAddress = getStakingContractAddress(chainId);\n\n    if (!contractAddress) {\n      return { amount: '0' };\n    }\n\n    try {\n      const sharesCalldata = STAKING_INTERFACE.encodeFunctionData('getShares', [\n        accountAddress,\n      ]);\n      const sharesResult = await provider.call({\n        to: contractAddress,\n        data: sharesCalldata,\n      });\n      const sharesRaw = STAKING_INTERFACE.decodeFunctionResult(\n        'getShares',\n        sharesResult,\n      )[0];\n      const sharesBigNum = BigInt(sharesRaw.toString());\n\n      if (sharesBigNum === 0n) {\n        return { amount: '0' };\n      }\n\n      const assetsCalldata = STAKING_INTERFACE.encodeFunctionData(\n        'convertToAssets',\n        [sharesBigNum],\n      );\n      const assetsResult = await provider.call({\n        to: contractAddress,\n        data: assetsCalldata,\n      });\n      const assetsRaw = STAKING_INTERFACE.decodeFunctionResult(\n        'convertToAssets',\n        assetsResult,\n      )[0];\n      const assetsWei = BigInt(assetsRaw.toString());\n\n      const amount = weiToHumanReadable(assetsWei, STAKING_DECIMALS);\n      return { amount };\n    } catch (error) {\n      throw error instanceof Error\n        ? error\n        : new Error('StakedBalanceFetcher: failed to fetch staked balance');\n    }\n  }\n}\n"]}

package/dist/data-sources/evm-rpc-services/services/{StakedBalanceFetcher.d.ts → StakedBalanceFetcher.d.cts}

@@ -1,7 +1,7 @@
-import { Interface } from '@ethersproject/abi';
-import { Web3Provider } from '@ethersproject/providers';
-import type { Address, AccountId, ChainId } from '../types/index.js';
-import { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId } from '../utils/index.js';
+import { Interface } from "@ethersproject/abi";
+import { Web3Provider } from "@ethersproject/providers";
+import type { Address, AccountId, ChainId } from "../types/index.cjs";
+import { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId } from "../utils/index.cjs";
 export { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId, };
 export type StakedBalancePollingInput = {
     /** Chain ID (hex format, e.g. 0x1) */
@@ -36,18 +36,19 @@ export type StakedBalanceFetcherConfig = {
     getNetworkProvider?: (chainId: ChainId) => Web3Provider | undefined;
 };
 declare const StakedBalanceFetcher_base: (abstract new (...args: any[]) => {
-    readonly _callbacks: Map<string, Set<(input: StakedBalancePollingInput) => void>>;
+    readonly "__#6@#intervalIds": Record<string, NodeJS.Timeout>;
+    "__#6@#intervalLength": number | undefined;
+    setIntervalLength(intervalLength: number): void;
+    getIntervalLength(): number | undefined;
+    _startPolling(input: StakedBalancePollingInput): void;
+    _stopPollingByPollingTokenSetId(key: string): void;
+    readonly "__#3@#pollingTokenSets": Map<string, Set<string>>;
+    /** Human-readable staked balance (e.g. "1.5" for 1.5 ETH). */
+    readonly "__#3@#callbacks": Map<string, Set<(input: StakedBalancePollingInput) => void>>;
     _executePoll(input: StakedBalancePollingInput): Promise<void>;
     startPolling(input: StakedBalancePollingInput): 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: StakedBalancePollingInput): void;
-    _stopPollingByPollingTokenSetId(key: import("@metamask/polling-controller").PollingTokenSetId): void;
-    readonly _pollingTokenSets: Map<import("@metamask/polling-controller").PollingTokenSetId, Set<string>>;
     onPollingComplete(input: StakedBalancePollingInput, callback: (input: StakedBalancePollingInput) => void): void;
 }) & {
     new (): {};
@@ -77,4 +78,4 @@ export declare class StakedBalanceFetcher extends StakedBalanceFetcher_base {
      */
     fetchStakedBalance(input: StakedBalancePollingInput): Promise<StakedBalance>;
 }
-//# sourceMappingURL=StakedBalanceFetcher.d.ts.map
+//# sourceMappingURL=StakedBalanceFetcher.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"StakedBalanceFetcher.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/StakedBalanceFetcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,2BAA2B;AAC/C,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAGxD,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,2BAA0B;AACrE,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,EAEzB,2BAA0B;AAE3B,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,GACzB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,sCAAsC;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;CACzB,CAAC;AAEF,8DAA8D;AAC9D,MAAM,MAAM,aAAa,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,+CAA+C;AAC/C,MAAM,MAAM,wBAAwB,GAAG;IACrC,yBAAyB;IACzB,SAAS,EAAE,SAAS,CAAC;IACrB,oBAAoB;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,qCAAqC;IACrC,OAAO,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG,CAC1C,MAAM,EAAE,wBAAwB,KAC7B,IAAI,CAAC;AAoBV,eAAO,MAAM,iBAAiB,WAAsC,CAAC;AAIrE,MAAM,MAAM,0BAA0B,GAAG;IACvC,6CAA6C;IAC7C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,yFAAyF;IACzF,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,YAAY,GAAG,SAAS,CAAC;CACrE,CAAC;;;;;;;;;IAjDF,8DAA8D;;;;;;;;;;AAqD9D,qBAAa,oBAAqB,SAAQ,yBAAgE;;gBAK5F,MAAM,CAAC,EAAE,0BAA0B;IAS/C;;;;;;OAMG;IACH,wBAAwB,CAAC,QAAQ,EAAE,6BAA6B,GAAG,IAAI;IAIjE,YAAY,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBnE;;;;;;;;;;OAUG;IACG,kBAAkB,CACtB,KAAK,EAAE,yBAAyB,GAC/B,OAAO,CAAC,aAAa,CAAC;CAoD1B"}

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

@@ -0,0 +1,81 @@
+import { Interface } from "@ethersproject/abi";
+import { Web3Provider } from "@ethersproject/providers";
+import type { Address, AccountId, ChainId } from "../types/index.mjs";
+import { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId } from "../utils/index.mjs";
+export { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId, };
+export type StakedBalancePollingInput = {
+    /** Chain ID (hex format, e.g. 0x1) */
+    chainId: ChainId;
+    /** Account ID */
+    accountId: AccountId;
+    /** Account address */
+    accountAddress: Address;
+};
+/** Human-readable staked balance (e.g. "1.5" for 1.5 ETH). */
+export type StakedBalance = {
+    amount: string;
+};
+/** Result reported via the update callback. */
+export type StakedBalanceFetchResult = {
+    /** Account ID (UUID). */
+    accountId: AccountId;
+    /** Hex chain ID. */
+    chainId: ChainId;
+    /** Human-readable staked balance. */
+    balance: StakedBalance;
+};
+/**
+ * Callback type for staked balance updates.
+ */
+export type OnStakedBalanceUpdateCallback = (result: StakedBalanceFetchResult) => void;
+export declare const STAKING_INTERFACE: Interface;
+export type StakedBalanceFetcherConfig = {
+    /** Polling interval in ms (default: 180s) */
+    pollingInterval?: number;
+    /** Returns the network provider for the given chain. Required for fetchStakedBalance. */
+    getNetworkProvider?: (chainId: ChainId) => Web3Provider | undefined;
+};
+declare const StakedBalanceFetcher_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: StakedBalancePollingInput): void;
+    _stopPollingByPollingTokenSetId(key: string): void;
+    readonly "__#3@#pollingTokenSets": Map<string, Set<string>>;
+    /** Human-readable staked balance (e.g. "1.5" for 1.5 ETH). */
+    readonly "__#3@#callbacks": Map<string, Set<(input: StakedBalancePollingInput) => void>>;
+    _executePoll(input: StakedBalancePollingInput): Promise<void>;
+    startPolling(input: StakedBalancePollingInput): string;
+    stopAllPolling(): void;
+    stopPollingByPollingToken(pollingToken: string): void;
+    onPollingComplete(input: StakedBalancePollingInput, callback: (input: StakedBalancePollingInput) => void): void;
+}) & {
+    new (): {};
+};
+export declare class StakedBalanceFetcher extends StakedBalanceFetcher_base {
+    #private;
+    constructor(config?: StakedBalanceFetcherConfig);
+    /**
+     * Register a callback that is invoked after every successful poll with
+     * the staked balance (including zero). Zero is reported so that merged
+     * updates can clear prior non-zero state.
+     *
+     * @param callback - The callback to invoke.
+     */
+    setOnStakedBalanceUpdate(callback: OnStakedBalanceUpdateCallback): void;
+    _executePoll(input: StakedBalancePollingInput): Promise<void>;
+    /**
+     * Fetches the staked balance for an account on a chain using the same
+     * staking contract as AccountTrackerController (getShares then convertToAssets).
+     * Returns a human-readable amount string (e.g. "1.5" for 1.5 ETH).
+     * Throws when no provider is available or when the RPC/contract call fails, so
+     * callers do not persist a false zero and overwrite existing balances.
+     *
+     * @param input - Chain, account ID, and address to query.
+     * @returns Human-readable staked balance (amount string).
+     * @throws When provider is missing or when getShares/convertToAssets fails.
+     */
+    fetchStakedBalance(input: StakedBalancePollingInput): Promise<StakedBalance>;
+}
+//# sourceMappingURL=StakedBalanceFetcher.d.mts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"StakedBalanceFetcher.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/StakedBalanceFetcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,2BAA2B;AAC/C,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAGxD,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,2BAA0B;AACrE,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,EAEzB,2BAA0B;AAE3B,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,GACzB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,sCAAsC;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,iBAAiB;IACjB,SAAS,EAAE,SAAS,CAAC;IACrB,sBAAsB;IACtB,cAAc,EAAE,OAAO,CAAC;CACzB,CAAC;AAEF,8DAA8D;AAC9D,MAAM,MAAM,aAAa,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,+CAA+C;AAC/C,MAAM,MAAM,wBAAwB,GAAG;IACrC,yBAAyB;IACzB,SAAS,EAAE,SAAS,CAAC;IACrB,oBAAoB;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,qCAAqC;IACrC,OAAO,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG,CAC1C,MAAM,EAAE,wBAAwB,KAC7B,IAAI,CAAC;AAoBV,eAAO,MAAM,iBAAiB,WAAsC,CAAC;AAIrE,MAAM,MAAM,0BAA0B,GAAG;IACvC,6CAA6C;IAC7C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,yFAAyF;IACzF,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,YAAY,GAAG,SAAS,CAAC;CACrE,CAAC;;;;;;;;;IAjDF,8DAA8D;;;;;;;;;;AAqD9D,qBAAa,oBAAqB,SAAQ,yBAAgE;;gBAK5F,MAAM,CAAC,EAAE,0BAA0B;IAS/C;;;;;;OAMG;IACH,wBAAwB,CAAC,QAAQ,EAAE,6BAA6B,GAAG,IAAI;IAIjE,YAAY,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBnE;;;;;;;;;;OAUG;IACG,kBAAkB,CACtB,KAAK,EAAE,yBAAyB,GAC/B,OAAO,CAAC,aAAa,CAAC;CAoD1B"}