@metamask-previews/assets-controller 2.0.2-preview-980f677 → 2.0.2-preview-3d4d0d0ef

package/dist/middlewares/ParallelMiddleware.mjs

@@ -1,214 +0,0 @@
-function $importDefault(module) {
-    if (module?.__esModule) {
-        return module.default;
-    }
-    return module;
-}
-import $pLimit from "p-limit/index.js";
-const pLimit = $importDefault($pLimit);
-// ============================================================================
-// MERGE HELPER
-// ============================================================================
-/**
- * Deep-merge multiple DataResponses into one.
- * Used when running balance data sources in parallel.
- *
- * @param responses - Array of DataResponse from each source.
- * @returns Single merged DataResponse.
- */
-export function mergeDataResponses(responses) {
-    const merged = {};
-    for (const response of responses) {
-        if (response.assetsBalance) {
-            merged.assetsBalance ?? (merged.assetsBalance = {});
-            for (const [accountId, accountBalances] of Object.entries(response.assetsBalance)) {
-                merged.assetsBalance[accountId] = {
-                    ...(merged.assetsBalance[accountId] ?? {}),
-                    ...accountBalances,
-                };
-            }
-        }
-        if (response.assetsInfo) {
-            merged.assetsInfo = {
-                ...(merged.assetsInfo ?? {}),
-                ...response.assetsInfo,
-            };
-        }
-        if (response.assetsPrice) {
-            merged.assetsPrice = {
-                ...(merged.assetsPrice ?? {}),
-                ...response.assetsPrice,
-            };
-        }
-        if (response.errors) {
-            merged.errors = {
-                ...(merged.errors ?? {}),
-                ...response.errors,
-            };
-        }
-        if (response.detectedAssets) {
-            merged.detectedAssets = {
-                ...(merged.detectedAssets ?? {}),
-                ...response.detectedAssets,
-            };
-        }
-        if (response.updateMode === 'full') {
-            merged.updateMode = 'full';
-        }
-    }
-    merged.updateMode ?? (merged.updateMode = 'merge');
-    return merged;
-}
-// ============================================================================
-// PARALLEL BALANCE MIDDLEWARE
-// ============================================================================
-const PARALLEL_BALANCE_MIDDLEWARE_NAME = 'ParallelBalanceMiddleware';
-/** Max concurrent balance source calls (round 1 and fallback). */
-const BALANCE_CONCURRENCY = 3;
-/**
- * Partition request.chainIds so each chain is assigned to exactly one source
- * (by source order: first source that supports the chain gets it). Ensures no
- * chain overlap across data source calls.
- *
- * @param request - The data request with chainIds to partition.
- * @param sources - Balance sources in priority order (e.g. AccountsAPI, Snap, Rpc).
- * @returns Array of requests, one per source, each with only that source's assigned chainIds.
- */
-function partitionChainsBySource(request, sources) {
-    const { chainIds } = request;
-    const assigned = new Set();
-    return sources.map((source) => {
-        const supported = new Set(source.getActiveChainsSync());
-        const chainsForSource = chainIds.filter((id) => supported.has(id) && !assigned.has(id));
-        chainsForSource.forEach((id) => assigned.add(id));
-        return {
-            ...request,
-            chainIds: chainsForSource,
-        };
-    });
-}
-/**
- * Collect chain IDs that failed in the first round (present in response.errors).
- * Used to run a fallback round with remaining sources.
- *
- * @param requests - Partitioned requests, one per source (same order as results).
- * @param results - Results from each source; chain IDs in requests[i] that have errors in results[i].response.errors are considered failed.
- * @returns Set of chain IDs that had errors in the first round.
- */
-function getFailedChainIds(requests, results) {
-    const failed = new Set();
-    for (let i = 0; i < results.length; i++) {
-        const errors = results[i].response.errors ?? {};
-        for (const chainId of requests[i].chainIds) {
-            if (errors[chainId]) {
-                failed.add(chainId);
-            }
-        }
-    }
-    return failed;
-}
-/**
- * Middleware that runs multiple balance data source middlewares in parallel,
- * with no chain overlap. Chains that fail (response.errors) are re-partitioned
- * and fetched again in a fallback round so lower-priority sources can try them.
- *
- * @param sources - Array of balance sources in priority order (each with getName(), getActiveChainsSync(), assetsMiddleware).
- * @returns A single middleware that runs all sources in parallel and merges responses.
- */
-export function createParallelBalanceMiddleware(sources) {
-    return {
-        getName() {
-            return PARALLEL_BALANCE_MIDDLEWARE_NAME;
-        },
-        assetsMiddleware: async (context, next) => {
-            if (sources.length === 0) {
-                return next(context);
-            }
-            const noopNext = async (ctx) => ctx;
-            const limit = pLimit(BALANCE_CONCURRENCY);
-            // Round 1: partition chains (no overlap), run with limited concurrency
-            const requests = partitionChainsBySource(context.request, sources);
-            const results = await Promise.all(sources.map((source, i) => limit(() => source.assetsMiddleware({
-                request: requests[i],
-                response: {},
-                getAssetsState: context.getAssetsState,
-            }, noopNext))));
-            let mergedResponse = mergeDataResponses(results.map((result) => result.response));
-            // Fallback: chains that failed (in errors) get re-partitioned and tried again
-            const failedChainIds = getFailedChainIds(requests, results);
-            if (failedChainIds.size > 0) {
-                const fallbackRequest = {
-                    ...context.request,
-                    chainIds: [...failedChainIds],
-                };
-                const fallbackRequests = partitionChainsBySource(fallbackRequest, sources);
-                const fallbackResults = await Promise.all(sources.map((source, i) => limit(() => source.assetsMiddleware({
-                    request: fallbackRequests[i],
-                    response: {},
-                    getAssetsState: context.getAssetsState,
-                }, noopNext))));
-                const fallbackMerged = mergeDataResponses(fallbackResults.map((result) => result.response));
-                mergedResponse = mergeDataResponses([mergedResponse, fallbackMerged]);
-                // Remove errors for chains we successfully got balance for in fallback
-                if (mergedResponse.errors && mergedResponse.assetsBalance) {
-                    const chainsWithBalance = new Set();
-                    for (const accountBalances of Object.values(mergedResponse.assetsBalance)) {
-                        for (const assetId of Object.keys(accountBalances)) {
-                            const chainId = assetId.split('/')[0];
-                            chainsWithBalance.add(chainId);
-                        }
-                    }
-                    for (const chainId of failedChainIds) {
-                        if (chainsWithBalance.has(chainId)) {
-                            delete mergedResponse.errors[chainId];
-                        }
-                    }
-                }
-            }
-            return next({
-                ...context,
-                response: mergeDataResponses([context.response, mergedResponse]),
-            });
-        },
-    };
-}
-// ============================================================================
-// PARALLEL TOKEN/PRICE MIDDLEWARE
-// ============================================================================
-const PARALLEL_MIDDLEWARE_NAME = 'ParallelMiddleware';
-/** Max concurrent token/price source calls. */
-const CONCURRENCY = 2;
-/**
- * Middleware that runs multiple data source middlewares (e.g. TokenDataSource,
- * PriceDataSource) in parallel with the same request. Responses are merged so
- * that assetsInfo (token metadata) and assetsPrice are combined. Use this to
- * fetch token and price data concurrently instead of sequentially.
- *
- * @param sources - Array of sources with getName() and assetsMiddleware.
- * @returns A single middleware that runs all sources in parallel and merges responses.
- */
-export function createParallelMiddleware(sources) {
-    return {
-        getName() {
-            return PARALLEL_MIDDLEWARE_NAME;
-        },
-        assetsMiddleware: async (context, next) => {
-            if (sources.length === 0) {
-                return next(context);
-            }
-            const noopNext = async (ctx) => ctx;
-            const limit = pLimit(CONCURRENCY);
-            const results = await Promise.all(sources.map((source) => limit(() => source.assetsMiddleware({
-                request: context.request,
-                response: { ...context.response },
-                getAssetsState: context.getAssetsState,
-            }, noopNext))));
-            const mergedResponse = mergeDataResponses(results.map((result) => result.response));
-            return next({
-                ...context,
-                response: mergeDataResponses([context.response, mergedResponse]),
-            });
-        },
-    };
-}
-//# sourceMappingURL=ParallelMiddleware.mjs.map

package/dist/middlewares/ParallelMiddleware.mjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"ParallelMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,OAAM,yBAAgB;;AAU7B,+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;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,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,GAAG,EAAE,CACT,MAAM,CAAC,gBAAgB,CACrB;gBACE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;gBACpB,QAAQ,EAAE,EAAE;gBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;aACvC,EACD,QAAQ,CACT,CACF,CACF,CACF,CAAC;YAEF,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,eAAe,GAAG,MAAM,OAAO,CAAC,GAAG,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CACxB,KAAK,CAAC,GAAG,EAAE,CACT,MAAM,CAAC,gBAAgB,CACrB;oBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;oBAC5B,QAAQ,EAAE,EAAE;oBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CACF,CACF,CACF,CAAC;gBACF,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;aACjE,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;AAOtB;;;;;;;;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,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,KAAK,CAAC,GAAG,EAAE,CACT,MAAM,CAAC,gBAAgB,CACrB;gBACE,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;gBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;aACvC,EACD,QAAQ,CACT,CACF,CACF,CACF,CAAC;YAEF,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;aACjE,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} from '../types';\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    }\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 results = await Promise.all(\n        sources.map((source, i) =>\n          limit(() =>\n            source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            ),\n          ),\n        ),\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 fallbackResults = await Promise.all(\n          sources.map((source, i) =>\n            limit(() =>\n              source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\n                  getAssetsState: context.getAssetsState,\n                },\n                noopNext,\n              ),\n            ),\n          ),\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      });\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\nexport type TokenPriceSource = {\n  getName(): string;\n  assetsMiddleware: Middleware;\n};\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 sources with getName() and assetsMiddleware.\n * @returns A single middleware that runs all sources in parallel and merges responses.\n */\nexport function createParallelMiddleware(sources: TokenPriceSource[]): {\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 results = await Promise.all(\n        sources.map((source) =>\n          limit(() =>\n            source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n                getAssetsState: context.getAssetsState,\n              },\n              noopNext,\n            ),\n          ),\n        ),\n      );\n\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n      });\n    },\n  };\n}\n"]}