@metamask-previews/assets-controller 14.0.3-preview-e12b5ece6 → 14.0.3-preview-fe4b3be11

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

@@ -24,11 +24,16 @@ export type BalancePollingInput = {
     accountId: AccountId;
     /** Account address */
     accountAddress: Address;
+    /**
+     * When present, fetch exactly these assets instead of the tracked
+     * `assetsBalance` entries (Accounts API v6 asset-scoped RPC poll). Must be
+     * deterministically ordered — the polling input is the dedupe key.
+     */
+    assetIds?: CaipAssetType[];
     /**
      * When true, only fetch balances for entries in `state.customAssets`,
      * skipping `state.assetsBalance`. Used by the supplemental RPC
-     * subscription on chains that another data source is already covering
-     * for regular balance refreshes.
+     * subscription on the Accounts API v5 path.
      */
     customAssetsOnly?: boolean;
 };

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

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

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

@@ -24,11 +24,16 @@ export type BalancePollingInput = {
     accountId: AccountId;
     /** Account address */
     accountAddress: Address;
+    /**
+     * When present, fetch exactly these assets instead of the tracked
+     * `assetsBalance` entries (Accounts API v6 asset-scoped RPC poll). Must be
+     * deterministically ordered — the polling input is the dedupe key.
+     */
+    assetIds?: CaipAssetType[];
     /**
      * When true, only fetch balances for entries in `state.customAssets`,
      * skipping `state.assetsBalance`. Used by the supplemental RPC
-     * subscription on chains that another data source is already covering
-     * for regular balance refreshes.
+     * subscription on the Accounts API v5 path.
      */
     customAssetsOnly?: boolean;
 };

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

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

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

@@ -58,7 +58,7 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
      * @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);
+        const result = await __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_fetchBalances).call(this, input.chainId, input.accountId, input.accountAddress, input.assetIds, input.customAssetsOnly === true);
         if (__classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f") && result.balances.length > 0) {
             await __classPrivateFieldGet(this, _BalanceFetcher_onBalanceUpdate, "f").call(this, result);
         }
@@ -120,8 +120,7 @@ export class BalanceFetcher extends StaticIntervalPollingControllerOnly() {
         };
     }
 }
-_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');
+_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, assetIds, customAssetsOnly = false) {
     // 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();
@@ -143,9 +142,18 @@ _BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new
                 : 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).
+    // Asset-scoped poll: fetch only the input's assets; tracked balances on
+    // this chain are another source's responsibility.
+    if (assetIds) {
+        for (const assetId of assetIds) {
+            collect(assetId);
+        }
+        return Array.from(assetsToFetch.values());
+    }
+    const state = __classPrivateFieldGet(this, _BalanceFetcher_messenger, "f").call('AssetsController:getState');
+    // Assets 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) {
@@ -154,10 +162,9 @@ _BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new
             }
         }
     }
-    // 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).
+    // User-added custom assets — on the v5 path RPC is the sole balance
+    // fetcher for these, so they must be polled even if they have no balance
+    // entry yet. On the v6 path they are usually already in `assetsBalance`.
     const customAssets = state?.customAssets?.[accountId];
     if (customAssets) {
         for (const assetId of customAssets) {
@@ -173,11 +180,13 @@ _BalanceFetcher_multicallClient = new WeakMap(), _BalanceFetcher_messenger = new
  * @param chainId - Hex chain ID.
  * @param accountId - Account UUID.
  * @param accountAddress - On-chain address of the account.
+ * @param assetIds - When present, fetch exactly these assets instead of the
+ * tracked `assetsBalance` entries.
  * @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);
+async function _BalanceFetcher_fetchBalances(chainId, accountId, accountAddress, assetIds, customAssetsOnly = false) {
+    const assets = __classPrivateFieldGet(this, _BalanceFetcher_instances, "m", _BalanceFetcher_getAssetsToFetch).call(this, chainId, accountId, assetIds, customAssetsOnly);
     return this.fetchBalancesForAssets(chainId, accountId, accountAddress, assets);
 }, _BalanceFetcher_processBalanceResponses = function _BalanceFetcher_processBalanceResponses(responses, accumulator, chainId, accountId, timestamp, entryByAddress) {
     const { balances, failedAddresses } = accumulator;

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

@@ -1 +1 @@
-{"version":3,"file":"BalanceFetcher.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/services/BalanceFetcher.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,mCAAmC,EAAE,qCAAqC;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"]}
+{"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;AAiDtD;;;;;;;;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,QAAQ,EACd,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;IAsHD;;;;;;;;;;;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;4VAnRG,OAAgB,EAChB,SAAoB,EACpB,QAA0B,EAC1B,gBAAgB,GAAG,KAAK;IAExB,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,wEAAwE;IACxE,kDAAkD;IAClD,IAAI,QAAQ,EAAE,CAAC;QACb,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,OAAO,CAAC,CAAC;QACnB,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,KAAK,GAAG,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAEhE,yDAAyD;IACzD,oEAAoE;IACpE,yCAAyC;IACzC,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,yEAAyE;IACzE,yEAAyE;IACzE,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;;;;;;;;;;;GAWG;AACH,KAAK,wCACH,OAAgB,EAChB,SAAoB,EACpB,cAAuB,EACvB,QAA0B,EAC1B,gBAAgB,GAAG,KAAK;IAExB,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EACjB,OAAO,EACP,SAAS,EACT,QAAQ,EACR,gBAAgB,CACjB,CAAC;IAEF,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 present, fetch exactly these assets instead of the tracked\n   * `assetsBalance` entries (Accounts API v6 asset-scoped RPC poll). Must be\n   * deterministically ordered — the polling input is the dedupe key.\n   */\n  assetIds?: CaipAssetType[];\n  /**\n   * When true, only fetch balances for entries in `state.customAssets`,\n   * skipping `state.assetsBalance`. Used by the supplemental RPC\n   * subscription on the Accounts API v5 path.\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.assetIds,\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 assetIds - When present, fetch exactly these assets instead of the\n   * tracked `assetsBalance` entries.\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    assetIds?: CaipAssetType[],\n    customAssetsOnly = false,\n  ): AssetFetchEntry[] {\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    // Asset-scoped poll: fetch only the input's assets; tracked balances on\n    // this chain are another source's responsibility.\n    if (assetIds) {\n      for (const assetId of assetIds) {\n        collect(assetId);\n      }\n      return Array.from(assetsToFetch.values());\n    }\n\n    const state = this.#messenger.call('AssetsController:getState');\n\n    // Assets 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    // User-added custom assets — on the v5 path RPC is the sole balance\n    // fetcher for these, so they must be polled even if they have no balance\n    // entry yet. On the v6 path they are usually already in `assetsBalance`.\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 assetIds - When present, fetch exactly these assets instead of the\n   * tracked `assetsBalance` entries.\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    assetIds?: CaipAssetType[],\n    customAssetsOnly = false,\n  ): Promise<BalanceFetchResult> {\n    const assets = this.#getAssetsToFetch(\n      chainId,\n      accountId,\n      assetIds,\n      customAssetsOnly,\n    );\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/types/state.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"state.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"","sourcesContent":["import type { ChainId } from './core.js';\n\n/**\n * Single token entry from token list.\n */\nexport type TokenListEntry = {\n  /** Contract address */\n  address: string;\n  /** Token symbol */\n  symbol: string;\n  /** Token name */\n  name: string;\n  /** Token decimals */\n  decimals: number;\n  /** Logo URL */\n  iconUrl?: string;\n  /** Aggregator sources */\n  aggregators?: string[];\n  /** Occurrence count in lists */\n  occurrences?: number;\n};\n\n/**\n * Cache entry for a chain's token list.\n */\nexport type TokenChainsCacheEntry = {\n  /** Timestamp when the cache was last updated */\n  timestamp: number;\n  /** Token list data: address -> TokenListEntry */\n  data: Record<string, TokenListEntry>;\n};\n\n/**\n * Token list state shape (from TokenListController).\n */\nexport type TokenListState = {\n  /** Map of chain ID to token list cache entry */\n  tokensChainsCache: Record<ChainId, TokenChainsCacheEntry>;\n};\n\n/**\n * Single asset balance entry.\n */\nexport type AssetBalanceEntry = {\n  /** Human-readable balance amount */\n  amount: string;\n};\n\n/**\n * Assets balance state shape (from AssetsController).\n * Maps accountId -> assetId (CAIP-19) -> balance entry.\n */\nexport type AssetsBalanceState = {\n  /** Balance data per account: accountId -> assetId -> balance */\n  assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;\n  /**\n   * User-added custom assets per account: accountId -> CAIP-19 asset IDs.\n   * Used to ensure RPC polling refreshes custom assets even when they have\n   * no entry in `assetsBalance` yet (e.g. zero balance, first fetch failed).\n   */\n  customAssets?: Record<string, string[]>;\n};\n"]}
+{"version":3,"file":"state.cjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"","sourcesContent":["import type { ChainId } from './core.js';\n\n/**\n * Single token entry from token list.\n */\nexport type TokenListEntry = {\n  /** Contract address */\n  address: string;\n  /** Token symbol */\n  symbol: string;\n  /** Token name */\n  name: string;\n  /** Token decimals */\n  decimals: number;\n  /** Logo URL */\n  iconUrl?: string;\n  /** Aggregator sources */\n  aggregators?: string[];\n  /** Occurrence count in lists */\n  occurrences?: number;\n};\n\n/**\n * Cache entry for a chain's token list.\n */\nexport type TokenChainsCacheEntry = {\n  /** Timestamp when the cache was last updated */\n  timestamp: number;\n  /** Token list data: address -> TokenListEntry */\n  data: Record<string, TokenListEntry>;\n};\n\n/**\n * Token list state shape (from TokenListController).\n */\nexport type TokenListState = {\n  /** Map of chain ID to token list cache entry */\n  tokensChainsCache: Record<ChainId, TokenChainsCacheEntry>;\n};\n\n/**\n * Single asset balance entry.\n */\nexport type AssetBalanceEntry = {\n  /** Human-readable balance amount */\n  amount: string;\n};\n\n/**\n * Assets balance state shape (from AssetsController).\n * Maps accountId -> assetId (CAIP-19) -> balance entry.\n */\nexport type AssetsBalanceState = {\n  /** Balance data per account: accountId -> assetId -> balance */\n  assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;\n  /**\n   * User-added custom assets per account: accountId -> CAIP-19 asset IDs.\n   * Not read by asset-scoped polls (`BalancePollingInput.assetIds`). Regular\n   * polls still include these so v5 supplemental/`customAssetsOnly` refreshes\n   * pins that have no `assetsBalance` row yet.\n   */\n  customAssets?: Record<string, string[]>;\n};\n"]}

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

@@ -50,8 +50,9 @@ export type AssetsBalanceState = {
     assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;
     /**
      * User-added custom assets per account: accountId -> CAIP-19 asset IDs.
-     * Used to ensure RPC polling refreshes custom assets even when they have
-     * no entry in `assetsBalance` yet (e.g. zero balance, first fetch failed).
+     * Not read by asset-scoped polls (`BalancePollingInput.assetIds`). Regular
+     * polls still include these so v5 supplemental/`customAssetsOnly` refreshes
+     * pins that have no `assetsBalance` row yet.
      */
     customAssets?: Record<string, string[]>;
 };

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

@@ -1 +1 @@
-{"version":3,"file":"state.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,mBAAkB;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,gCAAgC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,iDAAiD;IACjD,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACtC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,gDAAgD;IAChD,iBAAiB,EAAE,MAAM,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gEAAgE;IAChE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACjE;;;;OAIG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACzC,CAAC"}
+{"version":3,"file":"state.d.cts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,mBAAkB;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,gCAAgC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,iDAAiD;IACjD,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACtC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,gDAAgD;IAChD,iBAAiB,EAAE,MAAM,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gEAAgE;IAChE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACjE;;;;;OAKG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACzC,CAAC"}

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

@@ -50,8 +50,9 @@ export type AssetsBalanceState = {
     assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;
     /**
      * User-added custom assets per account: accountId -> CAIP-19 asset IDs.
-     * Used to ensure RPC polling refreshes custom assets even when they have
-     * no entry in `assetsBalance` yet (e.g. zero balance, first fetch failed).
+     * Not read by asset-scoped polls (`BalancePollingInput.assetIds`). Regular
+     * polls still include these so v5 supplemental/`customAssetsOnly` refreshes
+     * pins that have no `assetsBalance` row yet.
      */
     customAssets?: Record<string, string[]>;
 };

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

@@ -1 +1 @@
-{"version":3,"file":"state.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,mBAAkB;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,gCAAgC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,iDAAiD;IACjD,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACtC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,gDAAgD;IAChD,iBAAiB,EAAE,MAAM,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gEAAgE;IAChE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACjE;;;;OAIG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACzC,CAAC"}
+{"version":3,"file":"state.d.mts","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,mBAAkB;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,gCAAgC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAClC,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,iDAAiD;IACjD,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACtC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,gDAAgD;IAChD,iBAAiB,EAAE,MAAM,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,oCAAoC;IACpC,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gEAAgE;IAChE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACjE;;;;;OAKG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACzC,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"file":"state.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"","sourcesContent":["import type { ChainId } from './core.js';\n\n/**\n * Single token entry from token list.\n */\nexport type TokenListEntry = {\n  /** Contract address */\n  address: string;\n  /** Token symbol */\n  symbol: string;\n  /** Token name */\n  name: string;\n  /** Token decimals */\n  decimals: number;\n  /** Logo URL */\n  iconUrl?: string;\n  /** Aggregator sources */\n  aggregators?: string[];\n  /** Occurrence count in lists */\n  occurrences?: number;\n};\n\n/**\n * Cache entry for a chain's token list.\n */\nexport type TokenChainsCacheEntry = {\n  /** Timestamp when the cache was last updated */\n  timestamp: number;\n  /** Token list data: address -> TokenListEntry */\n  data: Record<string, TokenListEntry>;\n};\n\n/**\n * Token list state shape (from TokenListController).\n */\nexport type TokenListState = {\n  /** Map of chain ID to token list cache entry */\n  tokensChainsCache: Record<ChainId, TokenChainsCacheEntry>;\n};\n\n/**\n * Single asset balance entry.\n */\nexport type AssetBalanceEntry = {\n  /** Human-readable balance amount */\n  amount: string;\n};\n\n/**\n * Assets balance state shape (from AssetsController).\n * Maps accountId -> assetId (CAIP-19) -> balance entry.\n */\nexport type AssetsBalanceState = {\n  /** Balance data per account: accountId -> assetId -> balance */\n  assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;\n  /**\n   * User-added custom assets per account: accountId -> CAIP-19 asset IDs.\n   * Used to ensure RPC polling refreshes custom assets even when they have\n   * no entry in `assetsBalance` yet (e.g. zero balance, first fetch failed).\n   */\n  customAssets?: Record<string, string[]>;\n};\n"]}
+{"version":3,"file":"state.mjs","sourceRoot":"","sources":["../../../../src/data-sources/evm-rpc-services/types/state.ts"],"names":[],"mappings":"","sourcesContent":["import type { ChainId } from './core.js';\n\n/**\n * Single token entry from token list.\n */\nexport type TokenListEntry = {\n  /** Contract address */\n  address: string;\n  /** Token symbol */\n  symbol: string;\n  /** Token name */\n  name: string;\n  /** Token decimals */\n  decimals: number;\n  /** Logo URL */\n  iconUrl?: string;\n  /** Aggregator sources */\n  aggregators?: string[];\n  /** Occurrence count in lists */\n  occurrences?: number;\n};\n\n/**\n * Cache entry for a chain's token list.\n */\nexport type TokenChainsCacheEntry = {\n  /** Timestamp when the cache was last updated */\n  timestamp: number;\n  /** Token list data: address -> TokenListEntry */\n  data: Record<string, TokenListEntry>;\n};\n\n/**\n * Token list state shape (from TokenListController).\n */\nexport type TokenListState = {\n  /** Map of chain ID to token list cache entry */\n  tokensChainsCache: Record<ChainId, TokenChainsCacheEntry>;\n};\n\n/**\n * Single asset balance entry.\n */\nexport type AssetBalanceEntry = {\n  /** Human-readable balance amount */\n  amount: string;\n};\n\n/**\n * Assets balance state shape (from AssetsController).\n * Maps accountId -> assetId (CAIP-19) -> balance entry.\n */\nexport type AssetsBalanceState = {\n  /** Balance data per account: accountId -> assetId -> balance */\n  assetsBalance: Record<string, Record<string, AssetBalanceEntry>>;\n  /**\n   * User-added custom assets per account: accountId -> CAIP-19 asset IDs.\n   * Not read by asset-scoped polls (`BalancePollingInput.assetIds`). Regular\n   * polls still include these so v5 supplemental/`customAssetsOnly` refreshes\n   * pins that have no `assetsBalance` row yet.\n   */\n  customAssets?: Record<string, string[]>;\n};\n"]}

package/dist/middlewares/ParallelMiddleware.cjs

@@ -45,6 +45,15 @@ function mergeDataResponses(responses) {
                 ...response.errors,
             };
         }
+        if (response.unprocessedCustomAssets &&
+            response.unprocessedCustomAssets.length > 0) {
+            merged.unprocessedCustomAssets = [
+                ...new Set([
+                    ...(merged.unprocessedCustomAssets ?? []),
+                    ...response.unprocessedCustomAssets,
+                ]),
+            ];
+        }
         if (response.detectedAssets) {
             merged.detectedAssets = {
                 ...(merged.detectedAssets ?? {}),

package/dist/middlewares/ParallelMiddleware.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.cjs","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":";;;;;;AAAA,+DAA6B;AAW7B,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;;;;;GAMG;AACH,SAAgB,kBAAkB,CAAC,SAAyB;IAC1D,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,aAAa,KAApB,MAAM,CAAC,aAAa,GAAK,EAAE,EAAC;YAC5B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;gBACF,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;oBAChC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC1C,GAAG,eAAe;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,UAAU,GAAG;gBAClB,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;gBAC5B,GAAG,QAAQ,CAAC,UAAU;aACvB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,WAAW,GAAG;gBACnB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC7B,GAAG,QAAQ,CAAC,WAAW;aACxB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,GAAG;gBACd,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;gBACxB,GAAG,QAAQ,CAAC,MAAM;aACnB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC5B,MAAM,CAAC,cAAc,GAAG;gBACtB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;gBAChC,GAAG,QAAQ,CAAC,cAAc;aAC3B,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACnC,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC;QAC7B,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,OAAO;YAC/B,MAAM,CAAC,UAAU,KAAK,MAAM,EAC5B,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC;QAC9B,CAAC;QACD,IAAI,QAAQ,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,CAAC,2BAA2B,GAAG,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,MAAM,CAAC,UAAU,KAAjB,MAAM,CAAC,UAAU,GAAK,OAAO,EAAC;IAE9B,OAAO,MAAM,CAAC;AAChB,CAAC;AAtDD,gDAsDC;AAED,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAErE,kEAAkE;AAClE,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAS9B;;;;;;;;GAQG;AACH,SAAS,uBAAuB,CAC9B,OAAoB,EACpB,OAAwB;IAExB,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAW,CAAC;IAEpC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CACrC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAC/C,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAElD,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,eAAe;SAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACxB,QAAuB,EACvB,OAAqC;IAErC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAW,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC;QAChD,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,+BAA+B,CAAC,OAAwB;IAItE,OAAO;QACL,OAAO;YACL,OAAO,gCAAgC,CAAC;QAC1C,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,IAAA,iBAAM,EAAC,mBAAmB,CAAC,CAAC;YAE1C,uEAAuE;YACvE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACpB,QAAQ,EAAE,EAAE;oBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YACF,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEzD,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE,CAAC;gBACrD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;gBAChE,oBAAoB,CAAC,GAAG,CAAC;oBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;YAClD,CAAC;YAED,IAAI,cAAc,GAAG,kBAAkB,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,8EAA8E;YAC9E,MAAM,cAAc,GAAG,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAgB;oBACnC,GAAG,OAAO,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,GAAG,cAAc,CAAC;iBAC9B,CAAC;gBACF,MAAM,gBAAgB,GAAG,uBAAuB,CAC9C,eAAe,EACf,OAAO,CACR,CAAC;gBACF,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;wBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;wBAC5B,QAAQ,EAAE,EAAE;wBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AApHD,0EAoHC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,SAAgB,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,IAAA,iBAAM,EAAC,WAAW,CAAC,CAAC;YAElC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;oBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAxDD,4DAwDC","sourcesContent":["import pLimit from 'p-limit';\n\nimport type {\n  ChainId,\n  Context,\n  DataRequest,\n  DataResponse,\n  Middleware,\n  AssetsDataSource,\n} from '../types.js';\n\n// ============================================================================\n// MERGE HELPER\n// ============================================================================\n\n/**\n * Deep-merge multiple DataResponses into one.\n * Used when running balance data sources in parallel.\n *\n * @param responses - Array of DataResponse from each source.\n * @returns Single merged DataResponse.\n */\nexport function mergeDataResponses(responses: DataResponse[]): DataResponse {\n  const merged: DataResponse = {};\n\n  for (const response of responses) {\n    if (response.assetsBalance) {\n      merged.assetsBalance ??= {};\n      for (const [accountId, accountBalances] of Object.entries(\n        response.assetsBalance,\n      )) {\n        merged.assetsBalance[accountId] = {\n          ...(merged.assetsBalance[accountId] ?? {}),\n          ...accountBalances,\n        };\n      }\n    }\n    if (response.assetsInfo) {\n      merged.assetsInfo = {\n        ...(merged.assetsInfo ?? {}),\n        ...response.assetsInfo,\n      };\n    }\n    if (response.assetsPrice) {\n      merged.assetsPrice = {\n        ...(merged.assetsPrice ?? {}),\n        ...response.assetsPrice,\n      };\n    }\n    if (response.errors) {\n      merged.errors = {\n        ...(merged.errors ?? {}),\n        ...response.errors,\n      };\n    }\n    if (response.detectedAssets) {\n      merged.detectedAssets = {\n        ...(merged.detectedAssets ?? {}),\n        ...response.detectedAssets,\n      };\n    }\n    if (response.updateMode === 'full') {\n      merged.updateMode = 'full';\n    } else if (\n      response.updateMode === 'merge' &&\n      merged.updateMode !== 'full'\n    ) {\n      merged.updateMode = 'merge';\n    }\n    if (response.replaceCoveredChainBalances) {\n      merged.replaceCoveredChainBalances = true;\n    }\n  }\n  merged.updateMode ??= 'merge';\n\n  return merged;\n}\n\n// ============================================================================\n// PARALLEL BALANCE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_BALANCE_MIDDLEWARE_NAME = 'ParallelBalanceMiddleware';\n\n/** Max concurrent balance source calls (round 1 and fallback). */\nconst BALANCE_CONCURRENCY = 3;\n\nexport type BalanceSource = {\n  getName(): string;\n  /** Chains this source can fetch (e.g. from getActiveChainsSync()). Used to partition chains with no overlap. */\n  getActiveChainsSync(): ChainId[];\n  assetsMiddleware: Middleware;\n};\n\n/**\n * Partition request.chainIds so each chain is assigned to exactly one source\n * (by source order: first source that supports the chain gets it). Ensures no\n * chain overlap across data source calls.\n *\n * @param request - The data request with chainIds to partition.\n * @param sources - Balance sources in priority order (e.g. AccountsAPI, Snap, Rpc).\n * @returns Array of requests, one per source, each with only that source's assigned chainIds.\n */\nfunction partitionChainsBySource(\n  request: DataRequest,\n  sources: BalanceSource[],\n): DataRequest[] {\n  const { chainIds } = request;\n  const assigned = new Set<ChainId>();\n\n  return sources.map((source) => {\n    const supported = new Set(source.getActiveChainsSync());\n    const chainsForSource = chainIds.filter(\n      (id) => supported.has(id) && !assigned.has(id),\n    );\n    chainsForSource.forEach((id) => assigned.add(id));\n\n    return {\n      ...request,\n      chainIds: chainsForSource,\n    };\n  });\n}\n\n/**\n * Collect chain IDs that failed in the first round (present in response.errors).\n * Used to run a fallback round with remaining sources.\n *\n * @param requests - Partitioned requests, one per source (same order as results).\n * @param results - Results from each source; chain IDs in requests[i] that have errors in results[i].response.errors are considered failed.\n * @returns Set of chain IDs that had errors in the first round.\n */\nfunction getFailedChainIds(\n  requests: DataRequest[],\n  results: { response: DataResponse }[],\n): Set<ChainId> {\n  const failed = new Set<ChainId>();\n  for (let i = 0; i < results.length; i++) {\n    const errors = results[i].response.errors ?? {};\n    for (const chainId of requests[i].chainIds) {\n      if (errors[chainId]) {\n        failed.add(chainId);\n      }\n    }\n  }\n  return failed;\n}\n\n/**\n * Middleware that runs multiple balance data source middlewares in parallel,\n * with no chain overlap. Chains that fail (response.errors) are re-partitioned\n * and fetched again in a fallback round so lower-priority sources can try them.\n *\n * @param sources - Array of balance sources in priority order (each with getName(), getActiveChainsSync(), assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelBalanceMiddleware(sources: BalanceSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_BALANCE_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(BALANCE_CONCURRENCY);\n\n      // Round 1: partition chains (no overlap), run with limited concurrency\n      const requests = partitionChainsBySource(context.request, sources);\n      const round1Timed = await Promise.all(\n        sources.map((source, i) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n      const results = round1Timed.map((entry) => entry.result);\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of round1Timed) {\n        const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n        durationByDataSource[key] =\n          (durationByDataSource[key] ?? 0) + durationMs;\n      }\n\n      let mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      // Fallback: chains that failed (in errors) get re-partitioned and tried again\n      const failedChainIds = getFailedChainIds(requests, results);\n      if (failedChainIds.size > 0) {\n        const fallbackRequest: DataRequest = {\n          ...context.request,\n          chainIds: [...failedChainIds],\n        };\n        const fallbackRequests = partitionChainsBySource(\n          fallbackRequest,\n          sources,\n        );\n        const fallbackTimed = await Promise.all(\n          sources.map((source, i) =>\n            limit(async () => {\n              const start = Date.now();\n              const result = await source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\n                  getAssetsState: context.getAssetsState,\n                },\n                noopNext,\n              );\n              const durationMs = Date.now() - start;\n              return { result, sourceName: source.getName(), durationMs };\n            }),\n          ),\n        );\n        const fallbackResults = fallbackTimed.map((entry) => entry.result);\n        for (const { sourceName, durationMs } of fallbackTimed) {\n          const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n          durationByDataSource[key] =\n            (durationByDataSource[key] ?? 0) + durationMs;\n        }\n        const fallbackMerged = mergeDataResponses(\n          fallbackResults.map((result) => result.response),\n        );\n        mergedResponse = mergeDataResponses([mergedResponse, fallbackMerged]);\n        // Remove errors for chains we successfully got balance for in fallback\n        if (mergedResponse.errors && mergedResponse.assetsBalance) {\n          const chainsWithBalance = new Set<ChainId>();\n          for (const accountBalances of Object.values(\n            mergedResponse.assetsBalance,\n          )) {\n            for (const assetId of Object.keys(accountBalances)) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              chainsWithBalance.add(chainId);\n            }\n          }\n          for (const chainId of failedChainIds) {\n            if (chainsWithBalance.has(chainId)) {\n              delete mergedResponse.errors[chainId];\n            }\n          }\n        }\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n\n// ============================================================================\n// PARALLEL TOKEN/PRICE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_MIDDLEWARE_NAME = 'ParallelMiddleware';\n\n/** Max concurrent token/price source calls. */\nconst CONCURRENCY = 2;\n\n/**\n * Middleware that runs multiple data source middlewares (e.g. TokenDataSource,\n * PriceDataSource) in parallel with the same request. Responses are merged so\n * that assetsInfo (token metadata) and assetsPrice are combined. Use this to\n * fetch token and price data concurrently instead of sequentially.\n *\n * @param sources - Array of named middleware sources (getName + assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelMiddleware(sources: AssetsDataSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n\n      const results = timedResults.map((entry) => entry.result);\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of timedResults) {\n        durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =\n          durationMs;\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n"]}
+{"version":3,"file":"ParallelMiddleware.cjs","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":";;;;;;AAAA,+DAA6B;AAW7B,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;;;;;GAMG;AACH,SAAgB,kBAAkB,CAAC,SAAyB;IAC1D,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,aAAa,KAApB,MAAM,CAAC,aAAa,GAAK,EAAE,EAAC;YAC5B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;gBACF,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;oBAChC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC1C,GAAG,eAAe;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,UAAU,GAAG;gBAClB,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;gBAC5B,GAAG,QAAQ,CAAC,UAAU;aACvB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,WAAW,GAAG;gBACnB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC7B,GAAG,QAAQ,CAAC,WAAW;aACxB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,GAAG;gBACd,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;gBACxB,GAAG,QAAQ,CAAC,MAAM;aACnB,CAAC;QACJ,CAAC;QACD,IACE,QAAQ,CAAC,uBAAuB;YAChC,QAAQ,CAAC,uBAAuB,CAAC,MAAM,GAAG,CAAC,EAC3C,CAAC;YACD,MAAM,CAAC,uBAAuB,GAAG;gBAC/B,GAAG,IAAI,GAAG,CAAC;oBACT,GAAG,CAAC,MAAM,CAAC,uBAAuB,IAAI,EAAE,CAAC;oBACzC,GAAG,QAAQ,CAAC,uBAAuB;iBACpC,CAAC;aACH,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC5B,MAAM,CAAC,cAAc,GAAG;gBACtB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;gBAChC,GAAG,QAAQ,CAAC,cAAc;aAC3B,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACnC,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC;QAC7B,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,OAAO;YAC/B,MAAM,CAAC,UAAU,KAAK,MAAM,EAC5B,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC;QAC9B,CAAC;QACD,IAAI,QAAQ,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,CAAC,2BAA2B,GAAG,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,MAAM,CAAC,UAAU,KAAjB,MAAM,CAAC,UAAU,GAAK,OAAO,EAAC;IAE9B,OAAO,MAAM,CAAC;AAChB,CAAC;AAjED,gDAiEC;AAED,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAErE,kEAAkE;AAClE,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAS9B;;;;;;;;GAQG;AACH,SAAS,uBAAuB,CAC9B,OAAoB,EACpB,OAAwB;IAExB,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAW,CAAC;IAEpC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CACrC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAC/C,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAElD,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,eAAe;SAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACxB,QAAuB,EACvB,OAAqC;IAErC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAW,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC;QAChD,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,+BAA+B,CAAC,OAAwB;IAItE,OAAO;QACL,OAAO;YACL,OAAO,gCAAgC,CAAC;QAC1C,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,IAAA,iBAAM,EAAC,mBAAmB,CAAC,CAAC;YAE1C,uEAAuE;YACvE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACpB,QAAQ,EAAE,EAAE;oBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YACF,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEzD,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE,CAAC;gBACrD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;gBAChE,oBAAoB,CAAC,GAAG,CAAC;oBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;YAClD,CAAC;YAED,IAAI,cAAc,GAAG,kBAAkB,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,8EAA8E;YAC9E,MAAM,cAAc,GAAG,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAgB;oBACnC,GAAG,OAAO,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,GAAG,cAAc,CAAC;iBAC9B,CAAC;gBACF,MAAM,gBAAgB,GAAG,uBAAuB,CAC9C,eAAe,EACf,OAAO,CACR,CAAC;gBACF,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;wBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;wBAC5B,QAAQ,EAAE,EAAE;wBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AApHD,0EAoHC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,SAAgB,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,IAAA,iBAAM,EAAC,WAAW,CAAC,CAAC;YAElC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;oBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAxDD,4DAwDC","sourcesContent":["import pLimit from 'p-limit';\n\nimport type {\n  ChainId,\n  Context,\n  DataRequest,\n  DataResponse,\n  Middleware,\n  AssetsDataSource,\n} from '../types.js';\n\n// ============================================================================\n// MERGE HELPER\n// ============================================================================\n\n/**\n * Deep-merge multiple DataResponses into one.\n * Used when running balance data sources in parallel.\n *\n * @param responses - Array of DataResponse from each source.\n * @returns Single merged DataResponse.\n */\nexport function mergeDataResponses(responses: DataResponse[]): DataResponse {\n  const merged: DataResponse = {};\n\n  for (const response of responses) {\n    if (response.assetsBalance) {\n      merged.assetsBalance ??= {};\n      for (const [accountId, accountBalances] of Object.entries(\n        response.assetsBalance,\n      )) {\n        merged.assetsBalance[accountId] = {\n          ...(merged.assetsBalance[accountId] ?? {}),\n          ...accountBalances,\n        };\n      }\n    }\n    if (response.assetsInfo) {\n      merged.assetsInfo = {\n        ...(merged.assetsInfo ?? {}),\n        ...response.assetsInfo,\n      };\n    }\n    if (response.assetsPrice) {\n      merged.assetsPrice = {\n        ...(merged.assetsPrice ?? {}),\n        ...response.assetsPrice,\n      };\n    }\n    if (response.errors) {\n      merged.errors = {\n        ...(merged.errors ?? {}),\n        ...response.errors,\n      };\n    }\n    if (\n      response.unprocessedCustomAssets &&\n      response.unprocessedCustomAssets.length > 0\n    ) {\n      merged.unprocessedCustomAssets = [\n        ...new Set([\n          ...(merged.unprocessedCustomAssets ?? []),\n          ...response.unprocessedCustomAssets,\n        ]),\n      ];\n    }\n    if (response.detectedAssets) {\n      merged.detectedAssets = {\n        ...(merged.detectedAssets ?? {}),\n        ...response.detectedAssets,\n      };\n    }\n    if (response.updateMode === 'full') {\n      merged.updateMode = 'full';\n    } else if (\n      response.updateMode === 'merge' &&\n      merged.updateMode !== 'full'\n    ) {\n      merged.updateMode = 'merge';\n    }\n    if (response.replaceCoveredChainBalances) {\n      merged.replaceCoveredChainBalances = true;\n    }\n  }\n  merged.updateMode ??= 'merge';\n\n  return merged;\n}\n\n// ============================================================================\n// PARALLEL BALANCE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_BALANCE_MIDDLEWARE_NAME = 'ParallelBalanceMiddleware';\n\n/** Max concurrent balance source calls (round 1 and fallback). */\nconst BALANCE_CONCURRENCY = 3;\n\nexport type BalanceSource = {\n  getName(): string;\n  /** Chains this source can fetch (e.g. from getActiveChainsSync()). Used to partition chains with no overlap. */\n  getActiveChainsSync(): ChainId[];\n  assetsMiddleware: Middleware;\n};\n\n/**\n * Partition request.chainIds so each chain is assigned to exactly one source\n * (by source order: first source that supports the chain gets it). Ensures no\n * chain overlap across data source calls.\n *\n * @param request - The data request with chainIds to partition.\n * @param sources - Balance sources in priority order (e.g. AccountsAPI, Snap, Rpc).\n * @returns Array of requests, one per source, each with only that source's assigned chainIds.\n */\nfunction partitionChainsBySource(\n  request: DataRequest,\n  sources: BalanceSource[],\n): DataRequest[] {\n  const { chainIds } = request;\n  const assigned = new Set<ChainId>();\n\n  return sources.map((source) => {\n    const supported = new Set(source.getActiveChainsSync());\n    const chainsForSource = chainIds.filter(\n      (id) => supported.has(id) && !assigned.has(id),\n    );\n    chainsForSource.forEach((id) => assigned.add(id));\n\n    return {\n      ...request,\n      chainIds: chainsForSource,\n    };\n  });\n}\n\n/**\n * Collect chain IDs that failed in the first round (present in response.errors).\n * Used to run a fallback round with remaining sources.\n *\n * @param requests - Partitioned requests, one per source (same order as results).\n * @param results - Results from each source; chain IDs in requests[i] that have errors in results[i].response.errors are considered failed.\n * @returns Set of chain IDs that had errors in the first round.\n */\nfunction getFailedChainIds(\n  requests: DataRequest[],\n  results: { response: DataResponse }[],\n): Set<ChainId> {\n  const failed = new Set<ChainId>();\n  for (let i = 0; i < results.length; i++) {\n    const errors = results[i].response.errors ?? {};\n    for (const chainId of requests[i].chainIds) {\n      if (errors[chainId]) {\n        failed.add(chainId);\n      }\n    }\n  }\n  return failed;\n}\n\n/**\n * Middleware that runs multiple balance data source middlewares in parallel,\n * with no chain overlap. Chains that fail (response.errors) are re-partitioned\n * and fetched again in a fallback round so lower-priority sources can try them.\n *\n * @param sources - Array of balance sources in priority order (each with getName(), getActiveChainsSync(), assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelBalanceMiddleware(sources: BalanceSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_BALANCE_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(BALANCE_CONCURRENCY);\n\n      // Round 1: partition chains (no overlap), run with limited concurrency\n      const requests = partitionChainsBySource(context.request, sources);\n      const round1Timed = await Promise.all(\n        sources.map((source, i) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n      const results = round1Timed.map((entry) => entry.result);\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of round1Timed) {\n        const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n        durationByDataSource[key] =\n          (durationByDataSource[key] ?? 0) + durationMs;\n      }\n\n      let mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      // Fallback: chains that failed (in errors) get re-partitioned and tried again\n      const failedChainIds = getFailedChainIds(requests, results);\n      if (failedChainIds.size > 0) {\n        const fallbackRequest: DataRequest = {\n          ...context.request,\n          chainIds: [...failedChainIds],\n        };\n        const fallbackRequests = partitionChainsBySource(\n          fallbackRequest,\n          sources,\n        );\n        const fallbackTimed = await Promise.all(\n          sources.map((source, i) =>\n            limit(async () => {\n              const start = Date.now();\n              const result = await source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\n                  getAssetsState: context.getAssetsState,\n                },\n                noopNext,\n              );\n              const durationMs = Date.now() - start;\n              return { result, sourceName: source.getName(), durationMs };\n            }),\n          ),\n        );\n        const fallbackResults = fallbackTimed.map((entry) => entry.result);\n        for (const { sourceName, durationMs } of fallbackTimed) {\n          const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n          durationByDataSource[key] =\n            (durationByDataSource[key] ?? 0) + durationMs;\n        }\n        const fallbackMerged = mergeDataResponses(\n          fallbackResults.map((result) => result.response),\n        );\n        mergedResponse = mergeDataResponses([mergedResponse, fallbackMerged]);\n        // Remove errors for chains we successfully got balance for in fallback\n        if (mergedResponse.errors && mergedResponse.assetsBalance) {\n          const chainsWithBalance = new Set<ChainId>();\n          for (const accountBalances of Object.values(\n            mergedResponse.assetsBalance,\n          )) {\n            for (const assetId of Object.keys(accountBalances)) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              chainsWithBalance.add(chainId);\n            }\n          }\n          for (const chainId of failedChainIds) {\n            if (chainsWithBalance.has(chainId)) {\n              delete mergedResponse.errors[chainId];\n            }\n          }\n        }\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n\n// ============================================================================\n// PARALLEL TOKEN/PRICE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_MIDDLEWARE_NAME = 'ParallelMiddleware';\n\n/** Max concurrent token/price source calls. */\nconst CONCURRENCY = 2;\n\n/**\n * Middleware that runs multiple data source middlewares (e.g. TokenDataSource,\n * PriceDataSource) in parallel with the same request. Responses are merged so\n * that assetsInfo (token metadata) and assetsPrice are combined. Use this to\n * fetch token and price data concurrently instead of sequentially.\n *\n * @param sources - Array of named middleware sources (getName + assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelMiddleware(sources: AssetsDataSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n\n      const results = timedResults.map((entry) => entry.result);\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of timedResults) {\n        durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =\n          durationMs;\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n"]}

package/dist/middlewares/ParallelMiddleware.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.d.cts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,qBAAoB;AAMrB;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,YAAY,EAAE,GAAG,YAAY,CAsD1E;AAWD,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,IAAI,MAAM,CAAC;IAClB,gHAAgH;IAChH,mBAAmB,IAAI,OAAO,EAAE,CAAC;IACjC,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAAC;AAwDF;;;;;;;GAOG;AACH,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,aAAa,EAAE,GAAG;IACzE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAiHA;AAWD;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG;IACrE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAqDA"}
+{"version":3,"file":"ParallelMiddleware.d.cts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,qBAAoB;AAMrB;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,YAAY,EAAE,GAAG,YAAY,CAiE1E;AAWD,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,IAAI,MAAM,CAAC;IAClB,gHAAgH;IAChH,mBAAmB,IAAI,OAAO,EAAE,CAAC;IACjC,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAAC;AAwDF;;;;;;;GAOG;AACH,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,aAAa,EAAE,GAAG;IACzE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAiHA;AAWD;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG;IACrE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAqDA"}

package/dist/middlewares/ParallelMiddleware.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.d.mts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,qBAAoB;AAMrB;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,YAAY,EAAE,GAAG,YAAY,CAsD1E;AAWD,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,IAAI,MAAM,CAAC;IAClB,gHAAgH;IAChH,mBAAmB,IAAI,OAAO,EAAE,CAAC;IACjC,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAAC;AAwDF;;;;;;;GAOG;AACH,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,aAAa,EAAE,GAAG;IACzE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAiHA;AAWD;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG;IACrE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAqDA"}
+{"version":3,"file":"ParallelMiddleware.d.mts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,qBAAoB;AAMrB;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,YAAY,EAAE,GAAG,YAAY,CAiE1E;AAWD,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,IAAI,MAAM,CAAC;IAClB,gHAAgH;IAChH,mBAAmB,IAAI,OAAO,EAAE,CAAC;IACjC,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAAC;AAwDF;;;;;;;GAOG;AACH,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,aAAa,EAAE,GAAG;IACzE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAiHA;AAWD;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG;IACrE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAqDA"}

package/dist/middlewares/ParallelMiddleware.mjs

@@ -46,6 +46,15 @@ export function mergeDataResponses(responses) {
                 ...response.errors,
             };
         }
+        if (response.unprocessedCustomAssets &&
+            response.unprocessedCustomAssets.length > 0) {
+            merged.unprocessedCustomAssets = [
+                ...new Set([
+                    ...(merged.unprocessedCustomAssets ?? []),
+                    ...response.unprocessedCustomAssets,
+                ]),
+            ];
+        }
         if (response.detectedAssets) {
             merged.detectedAssets = {
                 ...(merged.detectedAssets ?? {}),

package/dist/middlewares/ParallelMiddleware.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,OAAM,yBAAgB;;AAW7B,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;;;;;GAMG;AACH,MAAM,UAAU,kBAAkB,CAAC,SAAyB;IAC1D,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,aAAa,KAApB,MAAM,CAAC,aAAa,GAAK,EAAE,EAAC;YAC5B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;gBACF,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;oBAChC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC1C,GAAG,eAAe;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,UAAU,GAAG;gBAClB,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;gBAC5B,GAAG,QAAQ,CAAC,UAAU;aACvB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,WAAW,GAAG;gBACnB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC7B,GAAG,QAAQ,CAAC,WAAW;aACxB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,GAAG;gBACd,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;gBACxB,GAAG,QAAQ,CAAC,MAAM;aACnB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC5B,MAAM,CAAC,cAAc,GAAG;gBACtB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;gBAChC,GAAG,QAAQ,CAAC,cAAc;aAC3B,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACnC,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC;QAC7B,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,OAAO;YAC/B,MAAM,CAAC,UAAU,KAAK,MAAM,EAC5B,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC;QAC9B,CAAC;QACD,IAAI,QAAQ,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,CAAC,2BAA2B,GAAG,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,MAAM,CAAC,UAAU,KAAjB,MAAM,CAAC,UAAU,GAAK,OAAO,EAAC;IAE9B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAErE,kEAAkE;AAClE,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAS9B;;;;;;;;GAQG;AACH,SAAS,uBAAuB,CAC9B,OAAoB,EACpB,OAAwB;IAExB,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAW,CAAC;IAEpC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CACrC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAC/C,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAElD,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,eAAe;SAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACxB,QAAuB,EACvB,OAAqC;IAErC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAW,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC;QAChD,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,+BAA+B,CAAC,OAAwB;IAItE,OAAO;QACL,OAAO;YACL,OAAO,gCAAgC,CAAC;QAC1C,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAE1C,uEAAuE;YACvE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACpB,QAAQ,EAAE,EAAE;oBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YACF,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEzD,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE,CAAC;gBACrD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;gBAChE,oBAAoB,CAAC,GAAG,CAAC;oBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;YAClD,CAAC;YAED,IAAI,cAAc,GAAG,kBAAkB,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,8EAA8E;YAC9E,MAAM,cAAc,GAAG,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAgB;oBACnC,GAAG,OAAO,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,GAAG,cAAc,CAAC;iBAC9B,CAAC;gBACF,MAAM,gBAAgB,GAAG,uBAAuB,CAC9C,eAAe,EACf,OAAO,CACR,CAAC;gBACF,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;wBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;wBAC5B,QAAQ,EAAE,EAAE;wBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;YAElC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;oBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["import pLimit from 'p-limit';\n\nimport type {\n  ChainId,\n  Context,\n  DataRequest,\n  DataResponse,\n  Middleware,\n  AssetsDataSource,\n} from '../types.js';\n\n// ============================================================================\n// MERGE HELPER\n// ============================================================================\n\n/**\n * Deep-merge multiple DataResponses into one.\n * Used when running balance data sources in parallel.\n *\n * @param responses - Array of DataResponse from each source.\n * @returns Single merged DataResponse.\n */\nexport function mergeDataResponses(responses: DataResponse[]): DataResponse {\n  const merged: DataResponse = {};\n\n  for (const response of responses) {\n    if (response.assetsBalance) {\n      merged.assetsBalance ??= {};\n      for (const [accountId, accountBalances] of Object.entries(\n        response.assetsBalance,\n      )) {\n        merged.assetsBalance[accountId] = {\n          ...(merged.assetsBalance[accountId] ?? {}),\n          ...accountBalances,\n        };\n      }\n    }\n    if (response.assetsInfo) {\n      merged.assetsInfo = {\n        ...(merged.assetsInfo ?? {}),\n        ...response.assetsInfo,\n      };\n    }\n    if (response.assetsPrice) {\n      merged.assetsPrice = {\n        ...(merged.assetsPrice ?? {}),\n        ...response.assetsPrice,\n      };\n    }\n    if (response.errors) {\n      merged.errors = {\n        ...(merged.errors ?? {}),\n        ...response.errors,\n      };\n    }\n    if (response.detectedAssets) {\n      merged.detectedAssets = {\n        ...(merged.detectedAssets ?? {}),\n        ...response.detectedAssets,\n      };\n    }\n    if (response.updateMode === 'full') {\n      merged.updateMode = 'full';\n    } else if (\n      response.updateMode === 'merge' &&\n      merged.updateMode !== 'full'\n    ) {\n      merged.updateMode = 'merge';\n    }\n    if (response.replaceCoveredChainBalances) {\n      merged.replaceCoveredChainBalances = true;\n    }\n  }\n  merged.updateMode ??= 'merge';\n\n  return merged;\n}\n\n// ============================================================================\n// PARALLEL BALANCE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_BALANCE_MIDDLEWARE_NAME = 'ParallelBalanceMiddleware';\n\n/** Max concurrent balance source calls (round 1 and fallback). */\nconst BALANCE_CONCURRENCY = 3;\n\nexport type BalanceSource = {\n  getName(): string;\n  /** Chains this source can fetch (e.g. from getActiveChainsSync()). Used to partition chains with no overlap. */\n  getActiveChainsSync(): ChainId[];\n  assetsMiddleware: Middleware;\n};\n\n/**\n * Partition request.chainIds so each chain is assigned to exactly one source\n * (by source order: first source that supports the chain gets it). Ensures no\n * chain overlap across data source calls.\n *\n * @param request - The data request with chainIds to partition.\n * @param sources - Balance sources in priority order (e.g. AccountsAPI, Snap, Rpc).\n * @returns Array of requests, one per source, each with only that source's assigned chainIds.\n */\nfunction partitionChainsBySource(\n  request: DataRequest,\n  sources: BalanceSource[],\n): DataRequest[] {\n  const { chainIds } = request;\n  const assigned = new Set<ChainId>();\n\n  return sources.map((source) => {\n    const supported = new Set(source.getActiveChainsSync());\n    const chainsForSource = chainIds.filter(\n      (id) => supported.has(id) && !assigned.has(id),\n    );\n    chainsForSource.forEach((id) => assigned.add(id));\n\n    return {\n      ...request,\n      chainIds: chainsForSource,\n    };\n  });\n}\n\n/**\n * Collect chain IDs that failed in the first round (present in response.errors).\n * Used to run a fallback round with remaining sources.\n *\n * @param requests - Partitioned requests, one per source (same order as results).\n * @param results - Results from each source; chain IDs in requests[i] that have errors in results[i].response.errors are considered failed.\n * @returns Set of chain IDs that had errors in the first round.\n */\nfunction getFailedChainIds(\n  requests: DataRequest[],\n  results: { response: DataResponse }[],\n): Set<ChainId> {\n  const failed = new Set<ChainId>();\n  for (let i = 0; i < results.length; i++) {\n    const errors = results[i].response.errors ?? {};\n    for (const chainId of requests[i].chainIds) {\n      if (errors[chainId]) {\n        failed.add(chainId);\n      }\n    }\n  }\n  return failed;\n}\n\n/**\n * Middleware that runs multiple balance data source middlewares in parallel,\n * with no chain overlap. Chains that fail (response.errors) are re-partitioned\n * and fetched again in a fallback round so lower-priority sources can try them.\n *\n * @param sources - Array of balance sources in priority order (each with getName(), getActiveChainsSync(), assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelBalanceMiddleware(sources: BalanceSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_BALANCE_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(BALANCE_CONCURRENCY);\n\n      // Round 1: partition chains (no overlap), run with limited concurrency\n      const requests = partitionChainsBySource(context.request, sources);\n      const round1Timed = await Promise.all(\n        sources.map((source, i) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n      const results = round1Timed.map((entry) => entry.result);\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of round1Timed) {\n        const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n        durationByDataSource[key] =\n          (durationByDataSource[key] ?? 0) + durationMs;\n      }\n\n      let mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      // Fallback: chains that failed (in errors) get re-partitioned and tried again\n      const failedChainIds = getFailedChainIds(requests, results);\n      if (failedChainIds.size > 0) {\n        const fallbackRequest: DataRequest = {\n          ...context.request,\n          chainIds: [...failedChainIds],\n        };\n        const fallbackRequests = partitionChainsBySource(\n          fallbackRequest,\n          sources,\n        );\n        const fallbackTimed = await Promise.all(\n          sources.map((source, i) =>\n            limit(async () => {\n              const start = Date.now();\n              const result = await source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\n                  getAssetsState: context.getAssetsState,\n                },\n                noopNext,\n              );\n              const durationMs = Date.now() - start;\n              return { result, sourceName: source.getName(), durationMs };\n            }),\n          ),\n        );\n        const fallbackResults = fallbackTimed.map((entry) => entry.result);\n        for (const { sourceName, durationMs } of fallbackTimed) {\n          const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n          durationByDataSource[key] =\n            (durationByDataSource[key] ?? 0) + durationMs;\n        }\n        const fallbackMerged = mergeDataResponses(\n          fallbackResults.map((result) => result.response),\n        );\n        mergedResponse = mergeDataResponses([mergedResponse, fallbackMerged]);\n        // Remove errors for chains we successfully got balance for in fallback\n        if (mergedResponse.errors && mergedResponse.assetsBalance) {\n          const chainsWithBalance = new Set<ChainId>();\n          for (const accountBalances of Object.values(\n            mergedResponse.assetsBalance,\n          )) {\n            for (const assetId of Object.keys(accountBalances)) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              chainsWithBalance.add(chainId);\n            }\n          }\n          for (const chainId of failedChainIds) {\n            if (chainsWithBalance.has(chainId)) {\n              delete mergedResponse.errors[chainId];\n            }\n          }\n        }\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n\n// ============================================================================\n// PARALLEL TOKEN/PRICE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_MIDDLEWARE_NAME = 'ParallelMiddleware';\n\n/** Max concurrent token/price source calls. */\nconst CONCURRENCY = 2;\n\n/**\n * Middleware that runs multiple data source middlewares (e.g. TokenDataSource,\n * PriceDataSource) in parallel with the same request. Responses are merged so\n * that assetsInfo (token metadata) and assetsPrice are combined. Use this to\n * fetch token and price data concurrently instead of sequentially.\n *\n * @param sources - Array of named middleware sources (getName + assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelMiddleware(sources: AssetsDataSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n\n      const results = timedResults.map((entry) => entry.result);\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of timedResults) {\n        durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =\n          durationMs;\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n"]}
+{"version":3,"file":"ParallelMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,OAAM,yBAAgB;;AAW7B,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;;;;;GAMG;AACH,MAAM,UAAU,kBAAkB,CAAC,SAAyB;IAC1D,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,aAAa,KAApB,MAAM,CAAC,aAAa,GAAK,EAAE,EAAC;YAC5B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;gBACF,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;oBAChC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC1C,GAAG,eAAe;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,UAAU,GAAG;gBAClB,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;gBAC5B,GAAG,QAAQ,CAAC,UAAU;aACvB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,WAAW,GAAG;gBACnB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC7B,GAAG,QAAQ,CAAC,WAAW;aACxB,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,GAAG;gBACd,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;gBACxB,GAAG,QAAQ,CAAC,MAAM;aACnB,CAAC;QACJ,CAAC;QACD,IACE,QAAQ,CAAC,uBAAuB;YAChC,QAAQ,CAAC,uBAAuB,CAAC,MAAM,GAAG,CAAC,EAC3C,CAAC;YACD,MAAM,CAAC,uBAAuB,GAAG;gBAC/B,GAAG,IAAI,GAAG,CAAC;oBACT,GAAG,CAAC,MAAM,CAAC,uBAAuB,IAAI,EAAE,CAAC;oBACzC,GAAG,QAAQ,CAAC,uBAAuB;iBACpC,CAAC;aACH,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC5B,MAAM,CAAC,cAAc,GAAG;gBACtB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;gBAChC,GAAG,QAAQ,CAAC,cAAc;aAC3B,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACnC,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC;QAC7B,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,OAAO;YAC/B,MAAM,CAAC,UAAU,KAAK,MAAM,EAC5B,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC;QAC9B,CAAC;QACD,IAAI,QAAQ,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,CAAC,2BAA2B,GAAG,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,MAAM,CAAC,UAAU,KAAjB,MAAM,CAAC,UAAU,GAAK,OAAO,EAAC;IAE9B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAErE,kEAAkE;AAClE,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAS9B;;;;;;;;GAQG;AACH,SAAS,uBAAuB,CAC9B,OAAoB,EACpB,OAAwB;IAExB,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAW,CAAC;IAEpC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CACrC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAC/C,CAAC;QACF,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAElD,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,eAAe;SAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACxB,QAAuB,EACvB,OAAqC;IAErC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAW,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC;QAChD,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,+BAA+B,CAAC,OAAwB;IAItE,OAAO;QACL,OAAO;YACL,OAAO,gCAAgC,CAAC;QAC1C,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAE1C,uEAAuE;YACvE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACpB,QAAQ,EAAE,EAAE;oBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YACF,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEzD,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE,CAAC;gBACrD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;gBAChE,oBAAoB,CAAC,GAAG,CAAC;oBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;YAClD,CAAC;YAED,IAAI,cAAc,GAAG,kBAAkB,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,8EAA8E;YAC9E,MAAM,cAAc,GAAG,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAgB;oBACnC,GAAG,OAAO,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,GAAG,cAAc,CAAC;iBAC9B,CAAC;gBACF,MAAM,gBAAgB,GAAG,uBAAuB,CAC9C,eAAe,EACf,OAAO,CACR,CAAC;gBACF,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACrC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;wBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;wBAC5B,QAAQ,EAAE,EAAE;wBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;YAElC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;oBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["import pLimit from 'p-limit';\n\nimport type {\n  ChainId,\n  Context,\n  DataRequest,\n  DataResponse,\n  Middleware,\n  AssetsDataSource,\n} from '../types.js';\n\n// ============================================================================\n// MERGE HELPER\n// ============================================================================\n\n/**\n * Deep-merge multiple DataResponses into one.\n * Used when running balance data sources in parallel.\n *\n * @param responses - Array of DataResponse from each source.\n * @returns Single merged DataResponse.\n */\nexport function mergeDataResponses(responses: DataResponse[]): DataResponse {\n  const merged: DataResponse = {};\n\n  for (const response of responses) {\n    if (response.assetsBalance) {\n      merged.assetsBalance ??= {};\n      for (const [accountId, accountBalances] of Object.entries(\n        response.assetsBalance,\n      )) {\n        merged.assetsBalance[accountId] = {\n          ...(merged.assetsBalance[accountId] ?? {}),\n          ...accountBalances,\n        };\n      }\n    }\n    if (response.assetsInfo) {\n      merged.assetsInfo = {\n        ...(merged.assetsInfo ?? {}),\n        ...response.assetsInfo,\n      };\n    }\n    if (response.assetsPrice) {\n      merged.assetsPrice = {\n        ...(merged.assetsPrice ?? {}),\n        ...response.assetsPrice,\n      };\n    }\n    if (response.errors) {\n      merged.errors = {\n        ...(merged.errors ?? {}),\n        ...response.errors,\n      };\n    }\n    if (\n      response.unprocessedCustomAssets &&\n      response.unprocessedCustomAssets.length > 0\n    ) {\n      merged.unprocessedCustomAssets = [\n        ...new Set([\n          ...(merged.unprocessedCustomAssets ?? []),\n          ...response.unprocessedCustomAssets,\n        ]),\n      ];\n    }\n    if (response.detectedAssets) {\n      merged.detectedAssets = {\n        ...(merged.detectedAssets ?? {}),\n        ...response.detectedAssets,\n      };\n    }\n    if (response.updateMode === 'full') {\n      merged.updateMode = 'full';\n    } else if (\n      response.updateMode === 'merge' &&\n      merged.updateMode !== 'full'\n    ) {\n      merged.updateMode = 'merge';\n    }\n    if (response.replaceCoveredChainBalances) {\n      merged.replaceCoveredChainBalances = true;\n    }\n  }\n  merged.updateMode ??= 'merge';\n\n  return merged;\n}\n\n// ============================================================================\n// PARALLEL BALANCE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_BALANCE_MIDDLEWARE_NAME = 'ParallelBalanceMiddleware';\n\n/** Max concurrent balance source calls (round 1 and fallback). */\nconst BALANCE_CONCURRENCY = 3;\n\nexport type BalanceSource = {\n  getName(): string;\n  /** Chains this source can fetch (e.g. from getActiveChainsSync()). Used to partition chains with no overlap. */\n  getActiveChainsSync(): ChainId[];\n  assetsMiddleware: Middleware;\n};\n\n/**\n * Partition request.chainIds so each chain is assigned to exactly one source\n * (by source order: first source that supports the chain gets it). Ensures no\n * chain overlap across data source calls.\n *\n * @param request - The data request with chainIds to partition.\n * @param sources - Balance sources in priority order (e.g. AccountsAPI, Snap, Rpc).\n * @returns Array of requests, one per source, each with only that source's assigned chainIds.\n */\nfunction partitionChainsBySource(\n  request: DataRequest,\n  sources: BalanceSource[],\n): DataRequest[] {\n  const { chainIds } = request;\n  const assigned = new Set<ChainId>();\n\n  return sources.map((source) => {\n    const supported = new Set(source.getActiveChainsSync());\n    const chainsForSource = chainIds.filter(\n      (id) => supported.has(id) && !assigned.has(id),\n    );\n    chainsForSource.forEach((id) => assigned.add(id));\n\n    return {\n      ...request,\n      chainIds: chainsForSource,\n    };\n  });\n}\n\n/**\n * Collect chain IDs that failed in the first round (present in response.errors).\n * Used to run a fallback round with remaining sources.\n *\n * @param requests - Partitioned requests, one per source (same order as results).\n * @param results - Results from each source; chain IDs in requests[i] that have errors in results[i].response.errors are considered failed.\n * @returns Set of chain IDs that had errors in the first round.\n */\nfunction getFailedChainIds(\n  requests: DataRequest[],\n  results: { response: DataResponse }[],\n): Set<ChainId> {\n  const failed = new Set<ChainId>();\n  for (let i = 0; i < results.length; i++) {\n    const errors = results[i].response.errors ?? {};\n    for (const chainId of requests[i].chainIds) {\n      if (errors[chainId]) {\n        failed.add(chainId);\n      }\n    }\n  }\n  return failed;\n}\n\n/**\n * Middleware that runs multiple balance data source middlewares in parallel,\n * with no chain overlap. Chains that fail (response.errors) are re-partitioned\n * and fetched again in a fallback round so lower-priority sources can try them.\n *\n * @param sources - Array of balance sources in priority order (each with getName(), getActiveChainsSync(), assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelBalanceMiddleware(sources: BalanceSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_BALANCE_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(BALANCE_CONCURRENCY);\n\n      // Round 1: partition chains (no overlap), run with limited concurrency\n      const requests = partitionChainsBySource(context.request, sources);\n      const round1Timed = await Promise.all(\n        sources.map((source, i) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n      const results = round1Timed.map((entry) => entry.result);\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of round1Timed) {\n        const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n        durationByDataSource[key] =\n          (durationByDataSource[key] ?? 0) + durationMs;\n      }\n\n      let mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      // Fallback: chains that failed (in errors) get re-partitioned and tried again\n      const failedChainIds = getFailedChainIds(requests, results);\n      if (failedChainIds.size > 0) {\n        const fallbackRequest: DataRequest = {\n          ...context.request,\n          chainIds: [...failedChainIds],\n        };\n        const fallbackRequests = partitionChainsBySource(\n          fallbackRequest,\n          sources,\n        );\n        const fallbackTimed = await Promise.all(\n          sources.map((source, i) =>\n            limit(async () => {\n              const start = Date.now();\n              const result = await source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\n                  getAssetsState: context.getAssetsState,\n                },\n                noopNext,\n              );\n              const durationMs = Date.now() - start;\n              return { result, sourceName: source.getName(), durationMs };\n            }),\n          ),\n        );\n        const fallbackResults = fallbackTimed.map((entry) => entry.result);\n        for (const { sourceName, durationMs } of fallbackTimed) {\n          const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;\n          durationByDataSource[key] =\n            (durationByDataSource[key] ?? 0) + durationMs;\n        }\n        const fallbackMerged = mergeDataResponses(\n          fallbackResults.map((result) => result.response),\n        );\n        mergedResponse = mergeDataResponses([mergedResponse, fallbackMerged]);\n        // Remove errors for chains we successfully got balance for in fallback\n        if (mergedResponse.errors && mergedResponse.assetsBalance) {\n          const chainsWithBalance = new Set<ChainId>();\n          for (const accountBalances of Object.values(\n            mergedResponse.assetsBalance,\n          )) {\n            for (const assetId of Object.keys(accountBalances)) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              chainsWithBalance.add(chainId);\n            }\n          }\n          for (const chainId of failedChainIds) {\n            if (chainsWithBalance.has(chainId)) {\n              delete mergedResponse.errors[chainId];\n            }\n          }\n        }\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n\n// ============================================================================\n// PARALLEL TOKEN/PRICE MIDDLEWARE\n// ============================================================================\n\nconst PARALLEL_MIDDLEWARE_NAME = 'ParallelMiddleware';\n\n/** Max concurrent token/price source calls. */\nconst CONCURRENCY = 2;\n\n/**\n * Middleware that runs multiple data source middlewares (e.g. TokenDataSource,\n * PriceDataSource) in parallel with the same request. Responses are merged so\n * that assetsInfo (token metadata) and assetsPrice are combined. Use this to\n * fetch token and price data concurrently instead of sequentially.\n *\n * @param sources - Array of named middleware sources (getName + assetsMiddleware).\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelMiddleware(sources: AssetsDataSource[]): {\n  getName(): string;\n  assetsMiddleware: Middleware;\n} {\n  return {\n    getName(): string {\n      return PARALLEL_MIDDLEWARE_NAME;\n    },\n\n    assetsMiddleware: async (context, next): Promise<Context> => {\n      if (sources.length === 0) {\n        return next(context);\n      }\n\n      const noopNext = async (ctx: typeof context): Promise<typeof context> =>\n        ctx;\n      const limit = pLimit(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          limit(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n\n      const results = timedResults.map((entry) => entry.result);\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of timedResults) {\n        durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =\n          durationMs;\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n"]}