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

package/dist/middlewares/DetectionMiddleware.cjs

@@ -0,0 +1,154 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DetectionMiddleware = void 0;
+const logger_js_1 = require("../logger.cjs");
+const types_js_1 = require("../types.cjs");
+const index_js_1 = require("../utils/index.cjs");
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+const CONTROLLER_NAME = 'DetectionMiddleware';
+// Logger for debugging
+(0, logger_js_1.createModuleLogger)(logger_js_1.projectLogger, CONTROLLER_NAME);
+// ============================================================================
+// DETECTION MIDDLEWARE
+// ============================================================================
+/**
+ * DetectionMiddleware builds the set of assets that downstream sources use for
+ * metadata and price fetching.
+ *
+ * An asset is included in `detectedAssets` only when it is genuinely new:
+ * - Assets from `response.assetsBalance` are included only if they are absent
+ *   from BOTH `state.assetsBalance` (never tracked before) AND `state.assetsInfo`
+ *   (no metadata yet). Assets already present in either collection are considered
+ *   known and are intentionally excluded — PriceDataSource's own subscription
+ *   handles periodic refreshes for those.
+ * - Each account's custom assets from state are always included because they
+ *   may have no balance yet and are explicitly managed by the user.
+ *
+ * Usage:
+ * ```typescript
+ * const detectionMiddleware = new DetectionMiddleware();
+ * const middleware = detectionMiddleware.assetsMiddleware;
+ * ```
+ */
+class DetectionMiddleware {
+    constructor() {
+        this.name = CONTROLLER_NAME;
+    }
+    getName() {
+        return this.name;
+    }
+    /**
+     * Get the middleware that builds detectedAssets for metadata and price fetching.
+     *
+     * This middleware:
+     * 1. Includes assets from response.assetsBalance that are absent from both
+     *    state.assetsBalance and state.assetsInfo (brand-new assets only)
+     * 2. Always includes each account's custom assets from state
+     * 3. Fills response.detectedAssets with the resulting asset IDs per account
+     * 4. Queues detected assets that lack a price in state on
+     *    request.assetsForPriceUpdate so PriceDataSource fetches them in the same
+     *    pipeline pass (including the background RPC detection path)
+     *
+     * @returns The middleware function for the assets pipeline.
+     */
+    get assetsMiddleware() {
+        return (0, types_js_1.forDataTypes)(['balance'], async (ctx, next) => {
+            const { request, response } = ctx;
+            // Get state for custom assets, existing balances, and existing metadata
+            const state = ctx.getAssetsState();
+            const { customAssets: stateCustomAssets, assetsBalance: stateAssetsBalance, assetsInfo: stateAssetsInfo, assetsPrice: stateAssetsPrice, } = state;
+            const detectedAssets = {};
+            // 1. From balance response: only include assets that are genuinely new —
+            //    not already present in state.assetsBalance or state.assetsInfo.
+            if (response.assetsBalance) {
+                for (const [accountId, accountBalances] of Object.entries(response.assetsBalance)) {
+                    const detected = [];
+                    const stateAccountBalances = stateAssetsBalance[accountId] ?? {};
+                    for (const assetId of Object.keys(accountBalances)) {
+                        const caipAssetId = assetId;
+                        // Skip if already tracked in state balances or already has metadata
+                        if (stateAccountBalances[caipAssetId] !== undefined ||
+                            stateAssetsInfo[caipAssetId] !== undefined) {
+                            continue;
+                        }
+                        detected.push(caipAssetId);
+                    }
+                    // Merge custom assets for this account, applying the same filter:
+                    // skip if already in state balance or already has metadata.
+                    const customForAccount = stateCustomAssets?.[accountId] ?? [];
+                    for (const assetId of customForAccount) {
+                        if (detected.includes(assetId)) {
+                            continue;
+                        }
+                        if (stateAccountBalances[assetId] !== undefined ||
+                            stateAssetsInfo[assetId] !== undefined) {
+                            continue;
+                        }
+                        detected.push(assetId);
+                    }
+                    if (detected.length > 0) {
+                        detectedAssets[accountId] = detected;
+                    }
+                }
+            }
+            // 2. Accounts in request that weren't in balance response: include their
+            //    custom assets that are not yet in state.
+            for (const { account } of request.accountsWithSupportedChains) {
+                const accountId = account.id;
+                if (detectedAssets[accountId]) {
+                    continue;
+                }
+                const stateAccountBalances = stateAssetsBalance[accountId] ?? {};
+                const customForAccount = stateCustomAssets?.[accountId] ?? [];
+                const newCustomAssets = customForAccount.filter((assetId) => {
+                    const inBalance = stateAccountBalances[assetId] !== undefined;
+                    const inInfo = stateAssetsInfo[assetId] !== undefined;
+                    return !inBalance && !inInfo;
+                });
+                if (newCustomAssets.length > 0) {
+                    detectedAssets[accountId] = newCustomAssets;
+                }
+            }
+            if (Object.keys(detectedAssets).length > 0) {
+                response.detectedAssets = detectedAssets;
+            }
+            // Queue prices for:
+            // 1) newly detected assets missing a price, and
+            // 2) assets in this balance response that already exist in state but still
+            //    lack a price (e.g. natives seeded before the first price poll).
+            // PriceDataSource filters non-priceable IDs before calling the API.
+            const prices = stateAssetsPrice;
+            const missingPriceAssets = new Set();
+            const maybeQueue = (assetId) => {
+                const normalizedAssetId = (0, index_js_1.normalizeAssetId)(assetId);
+                if (prices[normalizedAssetId] === undefined &&
+                    prices[assetId] === undefined) {
+                    missingPriceAssets.add(normalizedAssetId);
+                }
+            };
+            for (const accountAssetIds of Object.values(detectedAssets)) {
+                for (const assetId of accountAssetIds) {
+                    maybeQueue(assetId);
+                }
+            }
+            if (response.assetsBalance) {
+                for (const accountBalances of Object.values(response.assetsBalance)) {
+                    for (const assetId of Object.keys(accountBalances)) {
+                        maybeQueue(assetId);
+                    }
+                }
+            }
+            if (missingPriceAssets.size > 0) {
+                request.assetsForPriceUpdate = [
+                    ...(request.assetsForPriceUpdate ?? []),
+                    ...missingPriceAssets,
+                ];
+            }
+            return next(ctx);
+        });
+    }
+}
+exports.DetectionMiddleware = DetectionMiddleware;
+//# sourceMappingURL=DetectionMiddleware.cjs.map

package/dist/middlewares/DetectionMiddleware.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"DetectionMiddleware.cjs","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":";;;AAAA,6CAAiE;AACjE,2CAA2C;AAE3C,iDAAqD;AAErD,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,qBAAqB,CAAC;AAE9C,uBAAuB;AACvB,IAAA,8BAAkB,EAAC,yBAAa,EAAE,eAAe,CAAC,CAAC;AAEnD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAa,mBAAmB;IAAhC;QACW,SAAI,GAAG,eAAe,CAAC;IAoJlC,CAAC;IAlJC,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB;QAClB,OAAO,IAAA,uBAAY,EAAC,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACnD,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;YAElC,wEAAwE;YACxE,MAAM,KAAK,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YACnC,MAAM,EACJ,YAAY,EAAE,iBAAiB,EAC/B,aAAa,EAAE,kBAAkB,EACjC,UAAU,EAAE,eAAe,EAC3B,WAAW,EAAE,gBAAgB,GAC9B,GAAG,KAAK,CAAC;YAEV,MAAM,cAAc,GAAuC,EAAE,CAAC;YAE9D,yEAAyE;YACzE,qEAAqE;YACrE,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;oBACF,MAAM,QAAQ,GAAoB,EAAE,CAAC;oBAErC,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAEjE,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;wBACF,MAAM,WAAW,GAAG,OAAwB,CAAC;wBAC7C,oEAAoE;wBACpE,IACE,oBAAoB,CAAC,WAAW,CAAC,KAAK,SAAS;4BAC/C,eAAe,CAAC,WAAW,CAAC,KAAK,SAAS,EAC1C,CAAC;4BACD,SAAS;wBACX,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBAC7B,CAAC;oBAED,kEAAkE;oBAClE,4DAA4D;oBAC5D,MAAM,gBAAgB,GAAG,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC9D,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;wBACvC,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC/B,SAAS;wBACX,CAAC;wBACD,IACE,oBAAoB,CAAC,OAAO,CAAC,KAAK,SAAS;4BAC3C,eAAe,CAAC,OAAO,CAAC,KAAK,SAAS,EACtC,CAAC;4BACD,SAAS;wBACX,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACzB,CAAC;oBAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxB,cAAc,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;oBACvC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,yEAAyE;YACzE,8CAA8C;YAC9C,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;gBAC9D,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;gBAC7B,IAAI,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,SAAS;gBACX,CAAC;gBACD,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACjE,MAAM,gBAAgB,GAAG,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC9D,MAAM,eAAe,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC1D,MAAM,SAAS,GAAG,oBAAoB,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC;oBAC9D,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC;oBACtD,OAAO,CAAC,SAAS,IAAI,CAAC,MAAM,CAAC;gBAC/B,CAAC,CAAC,CAAC;gBACH,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,cAAc,CAAC,SAAS,CAAC,GAAG,eAAe,CAAC;gBAC9C,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3C,QAAQ,CAAC,cAAc,GAAG,cAAc,CAAC;YAC3C,CAAC;YAED,oBAAoB;YACpB,gDAAgD;YAChD,2EAA2E;YAC3E,qEAAqE;YACrE,oEAAoE;YACpE,MAAM,MAAM,GAAG,gBAA2C,CAAC;YAC3D,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAiB,CAAC;YAEpD,MAAM,UAAU,GAAG,CAAC,OAAsB,EAAQ,EAAE;gBAClD,MAAM,iBAAiB,GAAG,IAAA,2BAAgB,EAAC,OAAO,CAAC,CAAC;gBACpD,IACE,MAAM,CAAC,iBAAiB,CAAC,KAAK,SAAS;oBACvC,MAAM,CAAC,OAAO,CAAC,KAAK,SAAS,EAC7B,CAAC;oBACD,kBAAkB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC;YAEF,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC5D,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;oBACtC,UAAU,CAAC,OAAO,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;oBACpE,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;wBACF,UAAU,CAAC,OAAwB,CAAC,CAAC;oBACvC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,kBAAkB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAChC,OAAO,CAAC,oBAAoB,GAAG;oBAC7B,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oBACvC,GAAG,kBAAkB;iBACtB,CAAC;YACJ,CAAC;YAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AArJD,kDAqJC","sourcesContent":["import { projectLogger, createModuleLogger } from '../logger.js';\nimport { forDataTypes } from '../types.js';\nimport type { AccountId, Caip19AssetId, Middleware } from '../types.js';\nimport { normalizeAssetId } from '../utils/index.js';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'DetectionMiddleware';\n\n// Logger for debugging\ncreateModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// DETECTION MIDDLEWARE\n// ============================================================================\n\n/**\n * DetectionMiddleware builds the set of assets that downstream sources use for\n * metadata and price fetching.\n *\n * An asset is included in `detectedAssets` only when it is genuinely new:\n * - Assets from `response.assetsBalance` are included only if they are absent\n *   from BOTH `state.assetsBalance` (never tracked before) AND `state.assetsInfo`\n *   (no metadata yet). Assets already present in either collection are considered\n *   known and are intentionally excluded — PriceDataSource's own subscription\n *   handles periodic refreshes for those.\n * - Each account's custom assets from state are always included because they\n *   may have no balance yet and are explicitly managed by the user.\n *\n * Usage:\n * ```typescript\n * const detectionMiddleware = new DetectionMiddleware();\n * const middleware = detectionMiddleware.assetsMiddleware;\n * ```\n */\nexport class DetectionMiddleware {\n  readonly name = CONTROLLER_NAME;\n\n  getName(): string {\n    return this.name;\n  }\n\n  /**\n   * Get the middleware that builds detectedAssets for metadata and price fetching.\n   *\n   * This middleware:\n   * 1. Includes assets from response.assetsBalance that are absent from both\n   *    state.assetsBalance and state.assetsInfo (brand-new assets only)\n   * 2. Always includes each account's custom assets from state\n   * 3. Fills response.detectedAssets with the resulting asset IDs per account\n   * 4. Queues detected assets that lack a price in state on\n   *    request.assetsForPriceUpdate so PriceDataSource fetches them in the same\n   *    pipeline pass (including the background RPC detection path)\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['balance'], async (ctx, next) => {\n      const { request, response } = ctx;\n\n      // Get state for custom assets, existing balances, and existing metadata\n      const state = ctx.getAssetsState();\n      const {\n        customAssets: stateCustomAssets,\n        assetsBalance: stateAssetsBalance,\n        assetsInfo: stateAssetsInfo,\n        assetsPrice: stateAssetsPrice,\n      } = state;\n\n      const detectedAssets: Record<AccountId, Caip19AssetId[]> = {};\n\n      // 1. From balance response: only include assets that are genuinely new —\n      //    not already present in state.assetsBalance or state.assetsInfo.\n      if (response.assetsBalance) {\n        for (const [accountId, accountBalances] of Object.entries(\n          response.assetsBalance,\n        )) {\n          const detected: Caip19AssetId[] = [];\n\n          const stateAccountBalances = stateAssetsBalance[accountId] ?? {};\n\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            const caipAssetId = assetId as Caip19AssetId;\n            // Skip if already tracked in state balances or already has metadata\n            if (\n              stateAccountBalances[caipAssetId] !== undefined ||\n              stateAssetsInfo[caipAssetId] !== undefined\n            ) {\n              continue;\n            }\n            detected.push(caipAssetId);\n          }\n\n          // Merge custom assets for this account, applying the same filter:\n          // skip if already in state balance or already has metadata.\n          const customForAccount = stateCustomAssets?.[accountId] ?? [];\n          for (const assetId of customForAccount) {\n            if (detected.includes(assetId)) {\n              continue;\n            }\n            if (\n              stateAccountBalances[assetId] !== undefined ||\n              stateAssetsInfo[assetId] !== undefined\n            ) {\n              continue;\n            }\n            detected.push(assetId);\n          }\n\n          if (detected.length > 0) {\n            detectedAssets[accountId] = detected;\n          }\n        }\n      }\n\n      // 2. Accounts in request that weren't in balance response: include their\n      //    custom assets that are not yet in state.\n      for (const { account } of request.accountsWithSupportedChains) {\n        const accountId = account.id;\n        if (detectedAssets[accountId]) {\n          continue;\n        }\n        const stateAccountBalances = stateAssetsBalance[accountId] ?? {};\n        const customForAccount = stateCustomAssets?.[accountId] ?? [];\n        const newCustomAssets = customForAccount.filter((assetId) => {\n          const inBalance = stateAccountBalances[assetId] !== undefined;\n          const inInfo = stateAssetsInfo[assetId] !== undefined;\n          return !inBalance && !inInfo;\n        });\n        if (newCustomAssets.length > 0) {\n          detectedAssets[accountId] = newCustomAssets;\n        }\n      }\n\n      if (Object.keys(detectedAssets).length > 0) {\n        response.detectedAssets = detectedAssets;\n      }\n\n      // Queue prices for:\n      // 1) newly detected assets missing a price, and\n      // 2) assets in this balance response that already exist in state but still\n      //    lack a price (e.g. natives seeded before the first price poll).\n      // PriceDataSource filters non-priceable IDs before calling the API.\n      const prices = stateAssetsPrice as Record<string, unknown>;\n      const missingPriceAssets = new Set<Caip19AssetId>();\n\n      const maybeQueue = (assetId: Caip19AssetId): void => {\n        const normalizedAssetId = normalizeAssetId(assetId);\n        if (\n          prices[normalizedAssetId] === undefined &&\n          prices[assetId] === undefined\n        ) {\n          missingPriceAssets.add(normalizedAssetId);\n        }\n      };\n\n      for (const accountAssetIds of Object.values(detectedAssets)) {\n        for (const assetId of accountAssetIds) {\n          maybeQueue(assetId);\n        }\n      }\n\n      if (response.assetsBalance) {\n        for (const accountBalances of Object.values(response.assetsBalance)) {\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            maybeQueue(assetId as Caip19AssetId);\n          }\n        }\n      }\n\n      if (missingPriceAssets.size > 0) {\n        request.assetsForPriceUpdate = [\n          ...(request.assetsForPriceUpdate ?? []),\n          ...missingPriceAssets,\n        ];\n      }\n\n      return next(ctx);\n    });\n  }\n}\n"]}

package/dist/middlewares/{DetectionMiddleware.d.ts → DetectionMiddleware.d.cts}

@@ -1,4 +1,4 @@
-import type { Middleware } from '../types.js';
+import type { Middleware } from "../types.cjs";
 /**
  * DetectionMiddleware builds the set of assets that downstream sources use for
  * metadata and price fetching.
@@ -37,4 +37,4 @@ export declare class DetectionMiddleware {
      */
     get assetsMiddleware(): Middleware;
 }
-//# sourceMappingURL=DetectionMiddleware.d.ts.map
+//# sourceMappingURL=DetectionMiddleware.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"DetectionMiddleware.d.cts","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAA4B,UAAU,EAAE,qBAAoB;AAgBxE;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,mBAAmB;IAC9B,QAAQ,CAAC,IAAI,yBAAmB;IAEhC,OAAO,IAAI,MAAM;IAIjB;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA+HjC;CACF"}

package/dist/middlewares/DetectionMiddleware.d.mts

@@ -0,0 +1,40 @@
+import type { Middleware } from "../types.mjs";
+/**
+ * DetectionMiddleware builds the set of assets that downstream sources use for
+ * metadata and price fetching.
+ *
+ * An asset is included in `detectedAssets` only when it is genuinely new:
+ * - Assets from `response.assetsBalance` are included only if they are absent
+ *   from BOTH `state.assetsBalance` (never tracked before) AND `state.assetsInfo`
+ *   (no metadata yet). Assets already present in either collection are considered
+ *   known and are intentionally excluded — PriceDataSource's own subscription
+ *   handles periodic refreshes for those.
+ * - Each account's custom assets from state are always included because they
+ *   may have no balance yet and are explicitly managed by the user.
+ *
+ * Usage:
+ * ```typescript
+ * const detectionMiddleware = new DetectionMiddleware();
+ * const middleware = detectionMiddleware.assetsMiddleware;
+ * ```
+ */
+export declare class DetectionMiddleware {
+    readonly name = "DetectionMiddleware";
+    getName(): string;
+    /**
+     * Get the middleware that builds detectedAssets for metadata and price fetching.
+     *
+     * This middleware:
+     * 1. Includes assets from response.assetsBalance that are absent from both
+     *    state.assetsBalance and state.assetsInfo (brand-new assets only)
+     * 2. Always includes each account's custom assets from state
+     * 3. Fills response.detectedAssets with the resulting asset IDs per account
+     * 4. Queues detected assets that lack a price in state on
+     *    request.assetsForPriceUpdate so PriceDataSource fetches them in the same
+     *    pipeline pass (including the background RPC detection path)
+     *
+     * @returns The middleware function for the assets pipeline.
+     */
+    get assetsMiddleware(): Middleware;
+}
+//# sourceMappingURL=DetectionMiddleware.d.mts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"DetectionMiddleware.d.mts","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAA4B,UAAU,EAAE,qBAAoB;AAgBxE;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,mBAAmB;IAC9B,QAAQ,CAAC,IAAI,yBAAmB;IAEhC,OAAO,IAAI,MAAM;IAIjB;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA+HjC;CACF"}

package/dist/middlewares/{DetectionMiddleware.js → DetectionMiddleware.mjs}

@@ -1,6 +1,6 @@
-import { projectLogger, createModuleLogger } from '../logger.js';
-import { forDataTypes } from '../types.js';
-import { normalizeAssetId } from '../utils/index.js';
+import { projectLogger, createModuleLogger } from "../logger.mjs";
+import { forDataTypes } from "../types.mjs";
+import { normalizeAssetId } from "../utils/index.mjs";
 // ============================================================================
 // CONSTANTS
 // ============================================================================
@@ -147,4 +147,4 @@ export class DetectionMiddleware {
         });
     }
 }
-//# sourceMappingURL=DetectionMiddleware.js.map
+//# sourceMappingURL=DetectionMiddleware.mjs.map

package/dist/middlewares/DetectionMiddleware.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"DetectionMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAqB;AACjE,OAAO,EAAE,YAAY,EAAE,qBAAoB;AAE3C,OAAO,EAAE,gBAAgB,EAAE,2BAA0B;AAErD,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,qBAAqB,CAAC;AAE9C,uBAAuB;AACvB,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAEnD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,OAAO,mBAAmB;IAAhC;QACW,SAAI,GAAG,eAAe,CAAC;IAoJlC,CAAC;IAlJC,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB;QAClB,OAAO,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACnD,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;YAElC,wEAAwE;YACxE,MAAM,KAAK,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YACnC,MAAM,EACJ,YAAY,EAAE,iBAAiB,EAC/B,aAAa,EAAE,kBAAkB,EACjC,UAAU,EAAE,eAAe,EAC3B,WAAW,EAAE,gBAAgB,GAC9B,GAAG,KAAK,CAAC;YAEV,MAAM,cAAc,GAAuC,EAAE,CAAC;YAE9D,yEAAyE;YACzE,qEAAqE;YACrE,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;oBACF,MAAM,QAAQ,GAAoB,EAAE,CAAC;oBAErC,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAEjE,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;wBACF,MAAM,WAAW,GAAG,OAAwB,CAAC;wBAC7C,oEAAoE;wBACpE,IACE,oBAAoB,CAAC,WAAW,CAAC,KAAK,SAAS;4BAC/C,eAAe,CAAC,WAAW,CAAC,KAAK,SAAS,EAC1C,CAAC;4BACD,SAAS;wBACX,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBAC7B,CAAC;oBAED,kEAAkE;oBAClE,4DAA4D;oBAC5D,MAAM,gBAAgB,GAAG,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC9D,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;wBACvC,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC/B,SAAS;wBACX,CAAC;wBACD,IACE,oBAAoB,CAAC,OAAO,CAAC,KAAK,SAAS;4BAC3C,eAAe,CAAC,OAAO,CAAC,KAAK,SAAS,EACtC,CAAC;4BACD,SAAS;wBACX,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACzB,CAAC;oBAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxB,cAAc,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;oBACvC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,yEAAyE;YACzE,8CAA8C;YAC9C,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;gBAC9D,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;gBAC7B,IAAI,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,SAAS;gBACX,CAAC;gBACD,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACjE,MAAM,gBAAgB,GAAG,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC9D,MAAM,eAAe,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC1D,MAAM,SAAS,GAAG,oBAAoB,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC;oBAC9D,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC;oBACtD,OAAO,CAAC,SAAS,IAAI,CAAC,MAAM,CAAC;gBAC/B,CAAC,CAAC,CAAC;gBACH,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,cAAc,CAAC,SAAS,CAAC,GAAG,eAAe,CAAC;gBAC9C,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3C,QAAQ,CAAC,cAAc,GAAG,cAAc,CAAC;YAC3C,CAAC;YAED,oBAAoB;YACpB,gDAAgD;YAChD,2EAA2E;YAC3E,qEAAqE;YACrE,oEAAoE;YACpE,MAAM,MAAM,GAAG,gBAA2C,CAAC;YAC3D,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAiB,CAAC;YAEpD,MAAM,UAAU,GAAG,CAAC,OAAsB,EAAQ,EAAE;gBAClD,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBACpD,IACE,MAAM,CAAC,iBAAiB,CAAC,KAAK,SAAS;oBACvC,MAAM,CAAC,OAAO,CAAC,KAAK,SAAS,EAC7B,CAAC;oBACD,kBAAkB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC;YAEF,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC5D,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;oBACtC,UAAU,CAAC,OAAO,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;oBACpE,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;wBACF,UAAU,CAAC,OAAwB,CAAC,CAAC;oBACvC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,kBAAkB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAChC,OAAO,CAAC,oBAAoB,GAAG;oBAC7B,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oBACvC,GAAG,kBAAkB;iBACtB,CAAC;YACJ,CAAC;YAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["import { projectLogger, createModuleLogger } from '../logger.js';\nimport { forDataTypes } from '../types.js';\nimport type { AccountId, Caip19AssetId, Middleware } from '../types.js';\nimport { normalizeAssetId } from '../utils/index.js';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'DetectionMiddleware';\n\n// Logger for debugging\ncreateModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// DETECTION MIDDLEWARE\n// ============================================================================\n\n/**\n * DetectionMiddleware builds the set of assets that downstream sources use for\n * metadata and price fetching.\n *\n * An asset is included in `detectedAssets` only when it is genuinely new:\n * - Assets from `response.assetsBalance` are included only if they are absent\n *   from BOTH `state.assetsBalance` (never tracked before) AND `state.assetsInfo`\n *   (no metadata yet). Assets already present in either collection are considered\n *   known and are intentionally excluded — PriceDataSource's own subscription\n *   handles periodic refreshes for those.\n * - Each account's custom assets from state are always included because they\n *   may have no balance yet and are explicitly managed by the user.\n *\n * Usage:\n * ```typescript\n * const detectionMiddleware = new DetectionMiddleware();\n * const middleware = detectionMiddleware.assetsMiddleware;\n * ```\n */\nexport class DetectionMiddleware {\n  readonly name = CONTROLLER_NAME;\n\n  getName(): string {\n    return this.name;\n  }\n\n  /**\n   * Get the middleware that builds detectedAssets for metadata and price fetching.\n   *\n   * This middleware:\n   * 1. Includes assets from response.assetsBalance that are absent from both\n   *    state.assetsBalance and state.assetsInfo (brand-new assets only)\n   * 2. Always includes each account's custom assets from state\n   * 3. Fills response.detectedAssets with the resulting asset IDs per account\n   * 4. Queues detected assets that lack a price in state on\n   *    request.assetsForPriceUpdate so PriceDataSource fetches them in the same\n   *    pipeline pass (including the background RPC detection path)\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['balance'], async (ctx, next) => {\n      const { request, response } = ctx;\n\n      // Get state for custom assets, existing balances, and existing metadata\n      const state = ctx.getAssetsState();\n      const {\n        customAssets: stateCustomAssets,\n        assetsBalance: stateAssetsBalance,\n        assetsInfo: stateAssetsInfo,\n        assetsPrice: stateAssetsPrice,\n      } = state;\n\n      const detectedAssets: Record<AccountId, Caip19AssetId[]> = {};\n\n      // 1. From balance response: only include assets that are genuinely new —\n      //    not already present in state.assetsBalance or state.assetsInfo.\n      if (response.assetsBalance) {\n        for (const [accountId, accountBalances] of Object.entries(\n          response.assetsBalance,\n        )) {\n          const detected: Caip19AssetId[] = [];\n\n          const stateAccountBalances = stateAssetsBalance[accountId] ?? {};\n\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            const caipAssetId = assetId as Caip19AssetId;\n            // Skip if already tracked in state balances or already has metadata\n            if (\n              stateAccountBalances[caipAssetId] !== undefined ||\n              stateAssetsInfo[caipAssetId] !== undefined\n            ) {\n              continue;\n            }\n            detected.push(caipAssetId);\n          }\n\n          // Merge custom assets for this account, applying the same filter:\n          // skip if already in state balance or already has metadata.\n          const customForAccount = stateCustomAssets?.[accountId] ?? [];\n          for (const assetId of customForAccount) {\n            if (detected.includes(assetId)) {\n              continue;\n            }\n            if (\n              stateAccountBalances[assetId] !== undefined ||\n              stateAssetsInfo[assetId] !== undefined\n            ) {\n              continue;\n            }\n            detected.push(assetId);\n          }\n\n          if (detected.length > 0) {\n            detectedAssets[accountId] = detected;\n          }\n        }\n      }\n\n      // 2. Accounts in request that weren't in balance response: include their\n      //    custom assets that are not yet in state.\n      for (const { account } of request.accountsWithSupportedChains) {\n        const accountId = account.id;\n        if (detectedAssets[accountId]) {\n          continue;\n        }\n        const stateAccountBalances = stateAssetsBalance[accountId] ?? {};\n        const customForAccount = stateCustomAssets?.[accountId] ?? [];\n        const newCustomAssets = customForAccount.filter((assetId) => {\n          const inBalance = stateAccountBalances[assetId] !== undefined;\n          const inInfo = stateAssetsInfo[assetId] !== undefined;\n          return !inBalance && !inInfo;\n        });\n        if (newCustomAssets.length > 0) {\n          detectedAssets[accountId] = newCustomAssets;\n        }\n      }\n\n      if (Object.keys(detectedAssets).length > 0) {\n        response.detectedAssets = detectedAssets;\n      }\n\n      // Queue prices for:\n      // 1) newly detected assets missing a price, and\n      // 2) assets in this balance response that already exist in state but still\n      //    lack a price (e.g. natives seeded before the first price poll).\n      // PriceDataSource filters non-priceable IDs before calling the API.\n      const prices = stateAssetsPrice as Record<string, unknown>;\n      const missingPriceAssets = new Set<Caip19AssetId>();\n\n      const maybeQueue = (assetId: Caip19AssetId): void => {\n        const normalizedAssetId = normalizeAssetId(assetId);\n        if (\n          prices[normalizedAssetId] === undefined &&\n          prices[assetId] === undefined\n        ) {\n          missingPriceAssets.add(normalizedAssetId);\n        }\n      };\n\n      for (const accountAssetIds of Object.values(detectedAssets)) {\n        for (const assetId of accountAssetIds) {\n          maybeQueue(assetId);\n        }\n      }\n\n      if (response.assetsBalance) {\n        for (const accountBalances of Object.values(response.assetsBalance)) {\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            maybeQueue(assetId as Caip19AssetId);\n          }\n        }\n      }\n\n      if (missingPriceAssets.size > 0) {\n        request.assetsForPriceUpdate = [\n          ...(request.assetsForPriceUpdate ?? []),\n          ...missingPriceAssets,\n        ];\n      }\n\n      return next(ctx);\n    });\n  }\n}\n"]}

package/dist/middlewares/ParallelMiddleware.cjs

@@ -0,0 +1,263 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createParallelMiddleware = exports.createParallelBalanceMiddleware = exports.mergeDataResponses = void 0;
+const p_limit_1 = __importDefault(require("p-limit/index.js"));
+// ============================================================================
+// 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.
+ */
+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';
+        }
+        else if (response.updateMode === 'merge' &&
+            merged.updateMode !== 'full') {
+            merged.updateMode = 'merge';
+        }
+        if (response.replaceCoveredChainBalances) {
+            merged.replaceCoveredChainBalances = true;
+        }
+    }
+    merged.updateMode ?? (merged.updateMode = 'merge');
+    return merged;
+}
+exports.mergeDataResponses = mergeDataResponses;
+// ============================================================================
+// 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.
+ */
+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 = (0, p_limit_1.default)(BALANCE_CONCURRENCY);
+            // Round 1: partition chains (no overlap), run with limited concurrency
+            const requests = partitionChainsBySource(context.request, sources);
+            const round1Timed = await Promise.all(sources.map((source, i) => limit(async () => {
+                const start = Date.now();
+                const result = await source.assetsMiddleware({
+                    request: requests[i],
+                    response: {},
+                    getAssetsState: context.getAssetsState,
+                }, noopNext);
+                const durationMs = Date.now() - start;
+                return { result, sourceName: source.getName(), durationMs };
+            })));
+            const results = round1Timed.map((entry) => entry.result);
+            const durationByDataSource = {
+                ...context.durationByDataSource,
+            };
+            for (const { sourceName, durationMs } of round1Timed) {
+                const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;
+                durationByDataSource[key] =
+                    (durationByDataSource[key] ?? 0) + durationMs;
+            }
+            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 fallbackTimed = await Promise.all(sources.map((source, i) => limit(async () => {
+                    const start = Date.now();
+                    const result = await source.assetsMiddleware({
+                        request: fallbackRequests[i],
+                        response: {},
+                        getAssetsState: context.getAssetsState,
+                    }, noopNext);
+                    const durationMs = Date.now() - start;
+                    return { result, sourceName: source.getName(), durationMs };
+                })));
+                const fallbackResults = fallbackTimed.map((entry) => entry.result);
+                for (const { sourceName, durationMs } of fallbackTimed) {
+                    const key = `${PARALLEL_BALANCE_MIDDLEWARE_NAME}.${sourceName}`;
+                    durationByDataSource[key] =
+                        (durationByDataSource[key] ?? 0) + durationMs;
+                }
+                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]),
+                durationByDataSource,
+            });
+        },
+    };
+}
+exports.createParallelBalanceMiddleware = createParallelBalanceMiddleware;
+// ============================================================================
+// 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 named middleware sources (getName + assetsMiddleware).
+ * @returns A single middleware that runs all sources in parallel and merges responses.
+ */
+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 = (0, p_limit_1.default)(CONCURRENCY);
+            const timedResults = await Promise.all(sources.map((source) => limit(async () => {
+                const start = Date.now();
+                const result = await source.assetsMiddleware({
+                    request: context.request,
+                    response: { ...context.response },
+                    getAssetsState: context.getAssetsState,
+                }, noopNext);
+                const durationMs = Date.now() - start;
+                return { result, sourceName: source.getName(), durationMs };
+            })));
+            const results = timedResults.map((entry) => entry.result);
+            const mergedResponse = mergeDataResponses(results.map((result) => result.response));
+            const durationByDataSource = {
+                ...context.durationByDataSource,
+            };
+            for (const { sourceName, durationMs } of timedResults) {
+                durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =
+                    durationMs;
+            }
+            return next({
+                ...context,
+                response: mergeDataResponses([context.response, mergedResponse]),
+                durationByDataSource,
+            });
+        },
+    };
+}
+exports.createParallelMiddleware = createParallelMiddleware;
+//# sourceMappingURL=ParallelMiddleware.cjs.map

package/dist/middlewares/ParallelMiddleware.cjs.map

@@ -0,0 +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"]}

package/dist/middlewares/{ParallelMiddleware.d.ts → ParallelMiddleware.d.cts}

@@ -1,4 +1,4 @@
-import type { ChainId, DataResponse, Middleware, AssetsDataSource } from '../types.js';
+import type { ChainId, DataResponse, Middleware, AssetsDataSource } from "../types.cjs";
 /**
  * Deep-merge multiple DataResponses into one.
  * Used when running balance data sources in parallel.
@@ -38,4 +38,4 @@ export declare function createParallelMiddleware(sources: AssetsDataSource[]): {
     getName(): string;
     assetsMiddleware: Middleware;
 };
-//# sourceMappingURL=ParallelMiddleware.d.ts.map
+//# sourceMappingURL=ParallelMiddleware.d.cts.map

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

@@ -0,0 +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"}