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

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"}

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

@@ -10,10 +10,11 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
 var _StakedBalanceFetcher_providerGetter, _StakedBalanceFetcher_onStakedBalanceUpdate;
-import { Interface } from '@ethersproject/abi';
-import { StaticIntervalPollingControllerOnly } from '@metamask/polling-controller';
-import { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId, weiToHumanReadable, } from '../utils/index.js';
-export { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId, };
+import { Interface } from "@ethersproject/abi";
+import { Web3Provider } from "@ethersproject/providers";
+import { StaticIntervalPollingControllerOnly } from "@metamask/polling-controller";
+import { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId, weiToHumanReadable } from "../utils/index.mjs";
+export { getStakingContractAddress, getSupportedStakingChainIds, isStakingContractAssetId };
 /** Staking contract ABI: getShares(account) and convertToAssets(shares). */
 const STAKING_CONTRACT_ABI = [
     {
@@ -122,4 +123,4 @@ export class StakedBalanceFetcher extends StaticIntervalPollingControllerOnly()
     }
 }
 _StakedBalanceFetcher_providerGetter = new WeakMap(), _StakedBalanceFetcher_onStakedBalanceUpdate = new WeakMap();
-//# sourceMappingURL=StakedBalanceFetcher.js.map
+//# sourceMappingURL=StakedBalanceFetcher.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"StakedBalanceFetcher.mjs","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;AACxD,OAAO,EAAE,mCAAmC,EAAE,qCAAqC;AAGnF,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,EACxB,kBAAkB,EACnB,2BAA0B;AAE3B,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC3B,wBAAwB,EACzB,CAAC;AAiCF,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;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,SAAS,CAAC,oBAAoB,CAAC,CAAC;AAErE,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAS5B,MAAM,+BAA+B,GAAG,MAAO,CAAC,CAAC,YAAY;AAE7D,MAAM,OAAO,oBAAqB,SAAQ,mCAAmC,EAA6B;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,yBAAyB,CAAC,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,iBAAiB,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,iBAAiB,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,iBAAiB,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,iBAAiB,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,kBAAkB,CAAC,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","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/TokenDetector.cjs

@@ -0,0 +1,230 @@
+"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 _TokenDetector_instances, _TokenDetector_multicallClient, _TokenDetector_tokensApiClient, _TokenDetector_config, _TokenDetector_tokenListCache, _TokenDetector_onDetectionUpdate, _TokenDetector_fetchAndCacheTokenList, _TokenDetector_processBalanceResponses, _TokenDetector_formatBalance, _TokenDetector_getTokenMetadata, _TokenDetector_createAsset;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TokenDetector = void 0;
+const polling_controller_1 = require("@metamask/polling-controller");
+const logger_js_1 = require("../../../logger.cjs");
+const index_js_1 = require("../utils/index.cjs");
+const log = (0, logger_js_1.createModuleLogger)(logger_js_1.projectLogger, 'TokenDetector');
+const DEFAULT_DETECTION_INTERVAL = 180000; // 3 minutes
+/**
+ * TokenDetector - Detects tokens with non-zero balances via multicall.
+ * Fetches the token list from the Tokens API and uses multicall to check balances.
+ * Extends StaticIntervalPollingControllerOnly for built-in polling support.
+ */
+class TokenDetector extends (0, polling_controller_1.StaticIntervalPollingControllerOnly)() {
+    constructor(multicallClient, tokensApiClient, config) {
+        super();
+        _TokenDetector_instances.add(this);
+        _TokenDetector_multicallClient.set(this, void 0);
+        _TokenDetector_tokensApiClient.set(this, void 0);
+        _TokenDetector_config.set(this, void 0);
+        _TokenDetector_tokenListCache.set(this, new Map());
+        _TokenDetector_onDetectionUpdate.set(this, void 0);
+        __classPrivateFieldSet(this, _TokenDetector_multicallClient, multicallClient, "f");
+        __classPrivateFieldSet(this, _TokenDetector_tokensApiClient, tokensApiClient, "f");
+        __classPrivateFieldSet(this, _TokenDetector_config, {
+            tokenDetectionEnabled: config?.tokenDetectionEnabled ?? (() => true),
+            useExternalService: config?.useExternalService ?? (() => true),
+            defaultBatchSize: config?.defaultBatchSize ?? 300,
+            defaultTimeoutMs: config?.defaultTimeoutMs ?? 30000,
+        }, "f");
+        this.setIntervalLength(config?.pollingInterval ?? DEFAULT_DETECTION_INTERVAL);
+    }
+    /**
+     * Set the callback to receive detection updates during polling.
+     *
+     * @param callback - Function to call with detection results.
+     */
+    setOnDetectionUpdate(callback) {
+        __classPrivateFieldSet(this, _TokenDetector_onDetectionUpdate, callback, "f");
+    }
+    /**
+     * Execute a poll cycle (required by base class).
+     * Detects tokens and calls the update callback.
+     *
+     * @param input - The polling input.
+     */
+    async _executePoll(input) {
+        try {
+            const result = await this.detectTokens(input.chainId, input.accountId, input.accountAddress);
+            if (__classPrivateFieldGet(this, _TokenDetector_onDetectionUpdate, "f") && result.detectedAssets.length > 0) {
+                __classPrivateFieldGet(this, _TokenDetector_onDetectionUpdate, "f").call(this, result);
+            }
+        }
+        catch (error) {
+            log('Token detection poll failed', { chainId: input.chainId, error });
+        }
+    }
+    /**
+     * Fetch the list of token addresses to check for the given chain.
+     * Calls the Tokens API and caches the result for metadata lookups.
+     *
+     * @param chainId - Chain ID in hex format.
+     * @returns Array of token contract addresses.
+     */
+    async getTokensToCheck(chainId) {
+        const tokenList = await __classPrivateFieldGet(this, _TokenDetector_instances, "m", _TokenDetector_fetchAndCacheTokenList).call(this, chainId);
+        return tokenList.map((entry) => entry.address);
+    }
+    async detectTokens(chainId, accountId, accountAddress, options) {
+        const tokenDetectionEnabled = options?.tokenDetectionEnabled ?? __classPrivateFieldGet(this, _TokenDetector_config, "f").tokenDetectionEnabled();
+        const useExternalService = options?.useExternalService ?? __classPrivateFieldGet(this, _TokenDetector_config, "f").useExternalService();
+        if (!tokenDetectionEnabled || !useExternalService) {
+            return {
+                chainId,
+                accountId,
+                accountAddress,
+                detectedAssets: [],
+                detectedBalances: [],
+                zeroBalanceAddresses: [],
+                failedAddresses: [],
+                timestamp: Date.now(),
+            };
+        }
+        const batchSize = options?.batchSize ?? __classPrivateFieldGet(this, _TokenDetector_config, "f").defaultBatchSize;
+        const timestamp = Date.now();
+        const tokensToCheck = await this.getTokensToCheck(chainId);
+        if (tokensToCheck.length === 0) {
+            return {
+                chainId,
+                accountId,
+                accountAddress,
+                detectedAssets: [],
+                detectedBalances: [],
+                zeroBalanceAddresses: [],
+                failedAddresses: [],
+                timestamp,
+            };
+        }
+        const balanceRequests = tokensToCheck.map((tokenAddress) => ({
+            tokenAddress,
+            accountAddress,
+        }));
+        const result = await (0, index_js_1.reduceInBatchesSerially)({
+            values: balanceRequests,
+            batchSize,
+            initialResult: {
+                detectedAssets: [],
+                detectedBalances: [],
+                zeroBalanceAddresses: [],
+                failedAddresses: [],
+            },
+            eachBatch: async (workingResult, batch) => {
+                const responses = await __classPrivateFieldGet(this, _TokenDetector_multicallClient, "f").batchBalanceOf(chainId, batch);
+                return __classPrivateFieldGet(this, _TokenDetector_instances, "m", _TokenDetector_processBalanceResponses).call(this, responses, workingResult, chainId, accountId, timestamp);
+            },
+        });
+        return {
+            chainId,
+            accountId,
+            accountAddress,
+            ...result,
+            timestamp,
+        };
+    }
+}
+exports.TokenDetector = TokenDetector;
+_TokenDetector_multicallClient = new WeakMap(), _TokenDetector_tokensApiClient = new WeakMap(), _TokenDetector_config = new WeakMap(), _TokenDetector_tokenListCache = new WeakMap(), _TokenDetector_onDetectionUpdate = new WeakMap(), _TokenDetector_instances = new WeakSet(), _TokenDetector_fetchAndCacheTokenList = async function _TokenDetector_fetchAndCacheTokenList(chainId) {
+    try {
+        const list = await __classPrivateFieldGet(this, _TokenDetector_tokensApiClient, "f").fetchTokenList(chainId);
+        __classPrivateFieldGet(this, _TokenDetector_tokenListCache, "f").set(chainId, list);
+        return list;
+    }
+    catch (error) {
+        const cached = __classPrivateFieldGet(this, _TokenDetector_tokenListCache, "f").get(chainId);
+        log('Failed to fetch token list; using stale cache', {
+            chainId,
+            cachedCount: cached?.length ?? 0,
+            error,
+        });
+        return cached ?? [];
+    }
+}, _TokenDetector_processBalanceResponses = function _TokenDetector_processBalanceResponses(responses, accumulator, chainId, accountId, timestamp) {
+    const { detectedAssets, detectedBalances, zeroBalanceAddresses, failedAddresses, } = accumulator;
+    for (const response of responses) {
+        if (!response.success) {
+            failedAddresses.push(response.tokenAddress);
+            continue;
+        }
+        const balance = response.balance ?? '0';
+        if (balance === '0' || balance === '') {
+            zeroBalanceAddresses.push(response.tokenAddress);
+            continue;
+        }
+        const tokenMetadata = __classPrivateFieldGet(this, _TokenDetector_instances, "m", _TokenDetector_getTokenMetadata).call(this, chainId, response.tokenAddress);
+        const asset = __classPrivateFieldGet(this, _TokenDetector_instances, "m", _TokenDetector_createAsset).call(this, chainId, response.tokenAddress, tokenMetadata);
+        detectedAssets.push(asset);
+        if (tokenMetadata?.decimals === undefined) {
+            continue;
+        }
+        const { decimals } = tokenMetadata;
+        const formattedBalance = __classPrivateFieldGet(this, _TokenDetector_instances, "m", _TokenDetector_formatBalance).call(this, balance, decimals);
+        detectedBalances.push({
+            assetId: asset.assetId,
+            accountId,
+            chainId,
+            balance,
+            formattedBalance,
+            decimals,
+            timestamp,
+        });
+    }
+    return {
+        detectedAssets,
+        detectedBalances,
+        zeroBalanceAddresses,
+        failedAddresses,
+    };
+}, _TokenDetector_formatBalance = function _TokenDetector_formatBalance(rawBalance, decimals) {
+    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;
+    }
+}, _TokenDetector_getTokenMetadata = function _TokenDetector_getTokenMetadata(chainId, tokenAddress) {
+    const list = __classPrivateFieldGet(this, _TokenDetector_tokenListCache, "f").get(chainId) ?? [];
+    const lowerAddress = tokenAddress.toLowerCase();
+    const exact = list.find((entry) => entry.address === tokenAddress);
+    if (exact) {
+        return exact;
+    }
+    return list.find((entry) => entry.address.toLowerCase() === lowerAddress);
+}, _TokenDetector_createAsset = function _TokenDetector_createAsset(chainId, tokenAddress, metadata) {
+    const chainIdDecimal = parseInt(chainId, 16);
+    const assetId = `eip155:${chainIdDecimal}/erc20:${tokenAddress.toLowerCase()}`;
+    return {
+        assetId,
+        chainId,
+        address: tokenAddress,
+        type: 'erc20',
+        symbol: metadata?.symbol,
+        name: metadata?.name,
+        decimals: metadata?.decimals,
+        image: metadata?.iconUrl,
+        isNative: false,
+        aggregators: metadata?.aggregators,
+    };
+};
+//# sourceMappingURL=TokenDetector.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"TokenDetector.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/TokenDetector.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,qEAAmF;AAGnF,mDAAuE;AAevE,iDAA4D;AAE5D,MAAM,GAAG,GAAG,IAAA,8BAAkB,EAAC,yBAAa,EAAE,eAAe,CAAC,CAAC;AAE/D,MAAM,0BAA0B,GAAG,MAAO,CAAC,CAAC,YAAY;AA8BxD;;;;GAIG;AACH,MAAa,aAAc,SAAQ,IAAA,wDAAmC,GAAyB;IAW7F,YACE,eAAgC,EAChC,eAAgC,EAChC,MAA4B;QAE5B,KAAK,EAAE,CAAC;;QAfD,iDAAkC;QAElC,iDAAkC;QAElC,wCAAgE;QAEhE,wCAAkD,IAAI,GAAG,EAAE,EAAC;QAErE,mDAA0D;QAQxD,uBAAA,IAAI,kCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,kCAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,yBAAW;YACb,qBAAqB,EACnB,MAAM,EAAE,qBAAqB,IAAI,CAAC,GAAY,EAAE,CAAC,IAAI,CAAC;YACxD,kBAAkB,EAAE,MAAM,EAAE,kBAAkB,IAAI,CAAC,GAAY,EAAE,CAAC,IAAI,CAAC;YACvE,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,GAAG;YACjD,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,KAAK;SACpD,MAAA,CAAC;QAEF,IAAI,CAAC,iBAAiB,CACpB,MAAM,EAAE,eAAe,IAAI,0BAA0B,CACtD,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,oBAAoB,CAAC,QAAmC;QACtD,uBAAA,IAAI,oCAAsB,QAAQ,MAAA,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,KAA4B;QAC7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CACpC,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,CACrB,CAAC;YAEF,IAAI,uBAAA,IAAI,wCAAmB,IAAI,MAAM,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChE,uBAAA,IAAI,wCAAmB,MAAvB,IAAI,EAAoB,MAAM,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,6BAA6B,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,gBAAgB,CAAC,OAAgB;QACrC,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,uEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,CAAC;QAC9D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAkB,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,OAA+B;QAE/B,MAAM,qBAAqB,GACzB,OAAO,EAAE,qBAAqB,IAAI,uBAAA,IAAI,6BAAQ,CAAC,qBAAqB,EAAE,CAAC;QACzE,MAAM,kBAAkB,GACtB,OAAO,EAAE,kBAAkB,IAAI,uBAAA,IAAI,6BAAQ,CAAC,kBAAkB,EAAE,CAAC;QACnE,IAAI,CAAC,qBAAqB,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAClD,OAAO;gBACL,OAAO;gBACP,SAAS;gBACT,cAAc;gBACd,cAAc,EAAE,EAAE;gBAClB,gBAAgB,EAAE,EAAE;gBACpB,oBAAoB,EAAE,EAAE;gBACxB,eAAe,EAAE,EAAE;gBACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,uBAAA,IAAI,6BAAQ,CAAC,gBAAgB,CAAC;QACtE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAE3D,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,OAAO;gBACP,SAAS;gBACT,cAAc;gBACd,cAAc,EAAE,EAAE;gBAClB,gBAAgB,EAAE,EAAE;gBACpB,oBAAoB,EAAE,EAAE;gBACxB,eAAe,EAAE,EAAE;gBACnB,SAAS;aACV,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAuB,aAAa,CAAC,GAAG,CAC3D,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACjB,YAAY;YACZ,cAAc;SACf,CAAC,CACH,CAAC;QASF,MAAM,MAAM,GAAG,MAAM,IAAA,kCAAuB,EAG1C;YACA,MAAM,EAAE,eAAe;YACvB,SAAS;YACT,aAAa,EAAE;gBACb,cAAc,EAAE,EAAE;gBAClB,gBAAgB,EAAE,EAAE;gBACpB,oBAAoB,EAAE,EAAE;gBACxB,eAAe,EAAE,EAAE;aACpB;YACD,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,sCAAiB,CAAC,cAAc,CAC1D,OAAO,EACP,KAAK,CACN,CAAC;gBAEF,OAAO,uBAAA,IAAI,wEAAyB,MAA7B,IAAI,EACT,SAAS,EACT,aAAqC,EACrC,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,OAAO;YACL,OAAO;YACP,SAAS;YACT,cAAc;YACd,GAAG,MAAM;YACT,SAAS;SACV,CAAC;IACJ,CAAC;CAsJF;AA1TD,sCA0TC;0TApJC,KAAK,gDAAyB,OAAgB;IAC5C,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,uBAAA,IAAI,sCAAiB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjD,GAAG,CAAC,+CAA+C,EAAE;YACnD,OAAO;YACP,WAAW,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;YAChC,KAAK;SACN,CAAC,CAAC;QACH,OAAO,MAAM,IAAI,EAAE,CAAC;IACtB,CAAC;AACH,CAAC,2FAGC,SAA8B,EAC9B,WAKC,EACD,OAAgB,EAChB,SAAoB,EACpB,SAAiB;IAOjB,MAAM,EACJ,cAAc,EACd,gBAAgB,EAChB,oBAAoB,EACpB,eAAe,GAChB,GAAG,WAAW,CAAC;IAEhB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,GAAG,CAAC;QAExC,IAAI,OAAO,KAAK,GAAG,IAAI,OAAO,KAAK,EAAE,EAAE,CAAC;YACtC,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YACjD,SAAS;QACX,CAAC;QAED,MAAM,aAAa,GAAG,uBAAA,IAAI,iEAAkB,MAAtB,IAAI,EACxB,OAAO,EACP,QAAQ,CAAC,YAAY,CACtB,CAAC;QAEF,MAAM,KAAK,GAAG,uBAAA,IAAI,4DAAa,MAAjB,IAAI,EAChB,OAAO,EACP,QAAQ,CAAC,YAAY,EACrB,aAAa,CACd,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE3B,IAAI,aAAa,EAAE,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC1C,SAAS;QACX,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,GAAG,aAAa,CAAC;QACnC,MAAM,gBAAgB,GAAG,uBAAA,IAAI,8DAAe,MAAnB,IAAI,EAAgB,OAAO,EAAE,QAAQ,CAAC,CAAC;QAChE,gBAAgB,CAAC,IAAI,CAAC;YACpB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS;YACT,OAAO;YACP,OAAO;YACP,gBAAgB;YAChB,QAAQ;YACR,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,cAAc;QACd,gBAAgB;QAChB,oBAAoB;QACpB,eAAe;KAChB,CAAC;AACJ,CAAC,uEAEc,UAAkB,EAAE,QAAgB;IACjD,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,6EAGC,OAAgB,EAChB,YAAqB;IAErB,MAAM,IAAI,GAAG,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IACrD,MAAM,YAAY,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IAEhD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,YAAY,CAAC,CAAC;IACnE,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,YAAY,CAAC,CAAC;AAC5E,CAAC,mEAGC,OAAgB,EAChB,YAAqB,EACrB,QAAoC;IAEpC,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAE7C,MAAM,OAAO,GACX,UAAU,cAAc,UAAU,YAAY,CAAC,WAAW,EAAE,EAAmB,CAAC;IAElF,OAAO;QACL,OAAO;QACP,OAAO;QACP,OAAO,EAAE,YAAY;QACrB,IAAI,EAAE,OAAO;QACb,MAAM,EAAE,QAAQ,EAAE,MAAM;QACxB,IAAI,EAAE,QAAQ,EAAE,IAAI;QACpB,QAAQ,EAAE,QAAQ,EAAE,QAAQ;QAC5B,KAAK,EAAE,QAAQ,EAAE,OAAO;QACxB,QAAQ,EAAE,KAAK;QACf,WAAW,EAAE,QAAQ,EAAE,WAAW;KACnC,CAAC;AACJ,CAAC","sourcesContent":["import { StaticIntervalPollingControllerOnly } from '@metamask/polling-controller';\nimport type { CaipAssetType } from '@metamask/utils';\n\nimport { projectLogger, createModuleLogger } from '../../../logger.js';\nimport type { MulticallClient } from '../clients/index.js';\nimport type { TokensApiClient } from '../clients/TokensApiClient.js';\nimport type {\n  AccountId,\n  Address,\n  Asset,\n  AssetBalance,\n  BalanceOfRequest,\n  BalanceOfResponse,\n  ChainId,\n  TokenDetectionOptions,\n  TokenDetectionResult,\n  TokenListEntry,\n} from '../types/index.js';\nimport { reduceInBatchesSerially } from '../utils/index.js';\n\nconst log = createModuleLogger(projectLogger, 'TokenDetector');\n\nconst DEFAULT_DETECTION_INTERVAL = 180_000; // 3 minutes\n\nexport type TokenDetectorConfig = {\n  /** Function returning whether token detection is enabled (avoids stale value) */\n  tokenDetectionEnabled?: () => boolean;\n  /** Function returning whether external services are allowed (avoids stale value; default: () => true) */\n  useExternalService?: () => boolean;\n  defaultBatchSize?: number;\n  defaultTimeoutMs?: number;\n  /** Polling interval in ms (default: 3 minutes) */\n  pollingInterval?: number;\n};\n\n/**\n * Polling input for TokenDetector - identifies what to poll for.\n */\nexport type DetectionPollingInput = {\n  /** Chain ID (hex format) */\n  chainId: ChainId;\n  /** Account ID */\n  accountId: AccountId;\n  /** Account address */\n  accountAddress: Address;\n};\n\n/**\n * Callback type for token detection updates.\n */\nexport type OnDetectionUpdateCallback = (result: TokenDetectionResult) => void;\n\n/**\n * TokenDetector - Detects tokens with non-zero balances via multicall.\n * Fetches the token list from the Tokens API and uses multicall to check balances.\n * Extends StaticIntervalPollingControllerOnly for built-in polling support.\n */\nexport class TokenDetector extends StaticIntervalPollingControllerOnly<DetectionPollingInput>() {\n  readonly #multicallClient: MulticallClient;\n\n  readonly #tokensApiClient: TokensApiClient;\n\n  readonly #config: Required<Omit<TokenDetectorConfig, 'pollingInterval'>>;\n\n  readonly #tokenListCache: Map<ChainId, TokenListEntry[]> = new Map();\n\n  #onDetectionUpdate: OnDetectionUpdateCallback | undefined;\n\n  constructor(\n    multicallClient: MulticallClient,\n    tokensApiClient: TokensApiClient,\n    config?: TokenDetectorConfig,\n  ) {\n    super();\n    this.#multicallClient = multicallClient;\n    this.#tokensApiClient = tokensApiClient;\n    this.#config = {\n      tokenDetectionEnabled:\n        config?.tokenDetectionEnabled ?? ((): boolean => true),\n      useExternalService: config?.useExternalService ?? ((): boolean => true),\n      defaultBatchSize: config?.defaultBatchSize ?? 300,\n      defaultTimeoutMs: config?.defaultTimeoutMs ?? 30000,\n    };\n\n    this.setIntervalLength(\n      config?.pollingInterval ?? DEFAULT_DETECTION_INTERVAL,\n    );\n  }\n\n  /**\n   * Set the callback to receive detection updates during polling.\n   *\n   * @param callback - Function to call with detection results.\n   */\n  setOnDetectionUpdate(callback: OnDetectionUpdateCallback): void {\n    this.#onDetectionUpdate = callback;\n  }\n\n  /**\n   * Execute a poll cycle (required by base class).\n   * Detects tokens and calls the update callback.\n   *\n   * @param input - The polling input.\n   */\n  async _executePoll(input: DetectionPollingInput): Promise<void> {\n    try {\n      const result = await this.detectTokens(\n        input.chainId,\n        input.accountId,\n        input.accountAddress,\n      );\n\n      if (this.#onDetectionUpdate && result.detectedAssets.length > 0) {\n        this.#onDetectionUpdate(result);\n      }\n    } catch (error) {\n      log('Token detection poll failed', { chainId: input.chainId, error });\n    }\n  }\n\n  /**\n   * Fetch the list of token addresses to check for the given chain.\n   * Calls the Tokens API and caches the result for metadata lookups.\n   *\n   * @param chainId - Chain ID in hex format.\n   * @returns Array of token contract addresses.\n   */\n  async getTokensToCheck(chainId: ChainId): Promise<Address[]> {\n    const tokenList = await this.#fetchAndCacheTokenList(chainId);\n    return tokenList.map((entry) => entry.address as Address);\n  }\n\n  async detectTokens(\n    chainId: ChainId,\n    accountId: AccountId,\n    accountAddress: Address,\n    options?: TokenDetectionOptions,\n  ): Promise<TokenDetectionResult> {\n    const tokenDetectionEnabled =\n      options?.tokenDetectionEnabled ?? this.#config.tokenDetectionEnabled();\n    const useExternalService =\n      options?.useExternalService ?? this.#config.useExternalService();\n    if (!tokenDetectionEnabled || !useExternalService) {\n      return {\n        chainId,\n        accountId,\n        accountAddress,\n        detectedAssets: [],\n        detectedBalances: [],\n        zeroBalanceAddresses: [],\n        failedAddresses: [],\n        timestamp: Date.now(),\n      };\n    }\n    const batchSize = options?.batchSize ?? this.#config.defaultBatchSize;\n    const timestamp = Date.now();\n\n    const tokensToCheck = await this.getTokensToCheck(chainId);\n\n    if (tokensToCheck.length === 0) {\n      return {\n        chainId,\n        accountId,\n        accountAddress,\n        detectedAssets: [],\n        detectedBalances: [],\n        zeroBalanceAddresses: [],\n        failedAddresses: [],\n        timestamp,\n      };\n    }\n\n    const balanceRequests: BalanceOfRequest[] = tokensToCheck.map(\n      (tokenAddress) => ({\n        tokenAddress,\n        accountAddress,\n      }),\n    );\n\n    type DetectionAccumulator = {\n      detectedAssets: Asset[];\n      detectedBalances: AssetBalance[];\n      zeroBalanceAddresses: Address[];\n      failedAddresses: Address[];\n    };\n\n    const result = await reduceInBatchesSerially<\n      BalanceOfRequest,\n      DetectionAccumulator\n    >({\n      values: balanceRequests,\n      batchSize,\n      initialResult: {\n        detectedAssets: [],\n        detectedBalances: [],\n        zeroBalanceAddresses: [],\n        failedAddresses: [],\n      },\n      eachBatch: async (workingResult, batch) => {\n        const responses = await this.#multicallClient.batchBalanceOf(\n          chainId,\n          batch,\n        );\n\n        return this.#processBalanceResponses(\n          responses,\n          workingResult as DetectionAccumulator,\n          chainId,\n          accountId,\n          timestamp,\n        );\n      },\n    });\n\n    return {\n      chainId,\n      accountId,\n      accountAddress,\n      ...result,\n      timestamp,\n    };\n  }\n\n  async #fetchAndCacheTokenList(chainId: ChainId): Promise<TokenListEntry[]> {\n    try {\n      const list = await this.#tokensApiClient.fetchTokenList(chainId);\n      this.#tokenListCache.set(chainId, list);\n      return list;\n    } catch (error) {\n      const cached = this.#tokenListCache.get(chainId);\n      log('Failed to fetch token list; using stale cache', {\n        chainId,\n        cachedCount: cached?.length ?? 0,\n        error,\n      });\n      return cached ?? [];\n    }\n  }\n\n  #processBalanceResponses(\n    responses: BalanceOfResponse[],\n    accumulator: {\n      detectedAssets: Asset[];\n      detectedBalances: AssetBalance[];\n      zeroBalanceAddresses: Address[];\n      failedAddresses: Address[];\n    },\n    chainId: ChainId,\n    accountId: AccountId,\n    timestamp: number,\n  ): {\n    detectedAssets: Asset[];\n    detectedBalances: AssetBalance[];\n    zeroBalanceAddresses: Address[];\n    failedAddresses: Address[];\n  } {\n    const {\n      detectedAssets,\n      detectedBalances,\n      zeroBalanceAddresses,\n      failedAddresses,\n    } = accumulator;\n\n    for (const response of responses) {\n      if (!response.success) {\n        failedAddresses.push(response.tokenAddress);\n        continue;\n      }\n\n      const balance = response.balance ?? '0';\n\n      if (balance === '0' || balance === '') {\n        zeroBalanceAddresses.push(response.tokenAddress);\n        continue;\n      }\n\n      const tokenMetadata = this.#getTokenMetadata(\n        chainId,\n        response.tokenAddress,\n      );\n\n      const asset = this.#createAsset(\n        chainId,\n        response.tokenAddress,\n        tokenMetadata,\n      );\n      detectedAssets.push(asset);\n\n      if (tokenMetadata?.decimals === undefined) {\n        continue;\n      }\n\n      const { decimals } = tokenMetadata;\n      const formattedBalance = this.#formatBalance(balance, decimals);\n      detectedBalances.push({\n        assetId: asset.assetId,\n        accountId,\n        chainId,\n        balance,\n        formattedBalance,\n        decimals,\n        timestamp,\n      });\n    }\n\n    return {\n      detectedAssets,\n      detectedBalances,\n      zeroBalanceAddresses,\n      failedAddresses,\n    };\n  }\n\n  #formatBalance(rawBalance: string, decimals: number): string {\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  #getTokenMetadata(\n    chainId: ChainId,\n    tokenAddress: Address,\n  ): TokenListEntry | undefined {\n    const list = this.#tokenListCache.get(chainId) ?? [];\n    const lowerAddress = tokenAddress.toLowerCase();\n\n    const exact = list.find((entry) => entry.address === tokenAddress);\n    if (exact) {\n      return exact;\n    }\n\n    return list.find((entry) => entry.address.toLowerCase() === lowerAddress);\n  }\n\n  #createAsset(\n    chainId: ChainId,\n    tokenAddress: Address,\n    metadata: TokenListEntry | undefined,\n  ): Asset {\n    const chainIdDecimal = parseInt(chainId, 16);\n\n    const assetId =\n      `eip155:${chainIdDecimal}/erc20:${tokenAddress.toLowerCase()}` as CaipAssetType;\n\n    return {\n      assetId,\n      chainId,\n      address: tokenAddress,\n      type: 'erc20',\n      symbol: metadata?.symbol,\n      name: metadata?.name,\n      decimals: metadata?.decimals,\n      image: metadata?.iconUrl,\n      isNative: false,\n      aggregators: metadata?.aggregators,\n    };\n  }\n}\n"]}