@metamask-previews/assets-controller 16.1.0-preview-b1e33e9f1 → 16.1.0-preview-d2ea3cc

package/dist/middlewares/DetectionMiddleware.d.ts

@@ -1,4 +1,4 @@
-import type { AssetsControllerState, Middleware } from '../types.js';
+import type { Middleware } from '../types.js';
 /**
  * DetectionMiddleware builds the set of assets that downstream sources use for
  * metadata and price fetching.
@@ -14,20 +14,12 @@ import type { AssetsControllerState, Middleware } from '../types.js';
  *
  * Usage:
  * ```typescript
- * const detectionMiddleware = new DetectionMiddleware({
- *   getAssetsState: () => this.state,
- * });
+ * const detectionMiddleware = new DetectionMiddleware();
  * const middleware = detectionMiddleware.assetsMiddleware;
  * ```
  */
-export type DetectionMiddlewareOptions = {
-    /** Current AssetsController state. Used to distinguish newly detected assets. */
-    getAssetsState: () => AssetsControllerState;
-};
 export declare class DetectionMiddleware {
-    #private;
     readonly name = "DetectionMiddleware";
-    constructor(options: DetectionMiddlewareOptions);
     getName(): string;
     /**
      * Get the middleware that builds detectedAssets for metadata and price fetching.

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

@@ -1 +1 @@
-{"version":3,"file":"DetectionMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAEV,qBAAqB,EAErB,UAAU,EACX,MAAM,aAAa,CAAC;AAgBrB;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,MAAM,0BAA0B,GAAG;IACvC,iFAAiF;IACjF,cAAc,EAAE,MAAM,qBAAqB,CAAC;CAC7C,CAAC;AAEF,qBAAa,mBAAmB;;IAC9B,QAAQ,CAAC,IAAI,yBAAmB;IAIhC,YAAY,OAAO,EAAE,0BAA0B,EAE9C;IAED,OAAO,IAAI,MAAM,CAEhB;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA+HjC;CACF"}
+{"version":3,"file":"DetectionMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAA4B,UAAU,EAAE,MAAM,aAAa,CAAC;AAgBxE;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,mBAAmB;IAC9B,QAAQ,CAAC,IAAI,yBAAmB;IAEhC,OAAO,IAAI,MAAM,CAEhB;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA+HjC;CACF"}

package/dist/middlewares/DetectionMiddleware.js

@@ -7,12 +7,30 @@ import { normalizeAssetId } from '../utils/index.js';
 const CONTROLLER_NAME = 'DetectionMiddleware';
 // Logger for debugging
 createModuleLogger(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;
+ * ```
+ */
 export class DetectionMiddleware {
     name = CONTROLLER_NAME;
-    #getAssetsState;
-    constructor(options) {
-        this.#getAssetsState = options.getAssetsState;
-    }
     getName() {
         return this.name;
     }
@@ -34,7 +52,7 @@ export class DetectionMiddleware {
         return forDataTypes(['balance'], async (ctx, next) => {
             const { request, response } = ctx;
             // Get state for custom assets, existing balances, and existing metadata
-            const state = this.#getAssetsState();
+            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 —

package/dist/middlewares/DetectionMiddleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"DetectionMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;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;AAgCnD,MAAM,OAAO,mBAAmB;IACrB,IAAI,GAAG,eAAe,CAAC;IAEvB,eAAe,CAA8B;IAEtD,YAAY,OAAmC;QAC7C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,cAAc,CAAC;IAChD,CAAC;IAED,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,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,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 {\n  AccountId,\n  AssetsControllerState,\n  Caip19AssetId,\n  Middleware,\n} 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 *   getAssetsState: () => this.state,\n * });\n * const middleware = detectionMiddleware.assetsMiddleware;\n * ```\n */\nexport type DetectionMiddlewareOptions = {\n  /** Current AssetsController state. Used to distinguish newly detected assets. */\n  getAssetsState: () => AssetsControllerState;\n};\n\nexport class DetectionMiddleware {\n  readonly name = CONTROLLER_NAME;\n\n  readonly #getAssetsState: () => AssetsControllerState;\n\n  constructor(options: DetectionMiddlewareOptions) {\n    this.#getAssetsState = options.getAssetsState;\n  }\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 = this.#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"]}
+{"version":3,"file":"DetectionMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/DetectionMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;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;IACrB,IAAI,GAAG,eAAe,CAAC;IAEhC,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.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,MAAM,aAAa,CAAC;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,CA+GA;AAWD;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG;IACrE,OAAO,IAAI,MAAM,CAAC;IAClB,gBAAgB,EAAE,UAAU,CAAC;CAC9B,CAoDA"}
+{"version":3,"file":"ParallelMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,OAAO,EAGP,YAAY,EACZ,UAAU,EACV,gBAAgB,EACjB,MAAM,aAAa,CAAC;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"}

package/dist/middlewares/ParallelMiddleware.js

@@ -133,6 +133,7 @@ export function createParallelBalanceMiddleware(sources) {
                 const result = await source.assetsMiddleware({
                     request: requests[i],
                     response: {},
+                    getAssetsState: context.getAssetsState,
                 }, noopNext);
                 const durationMs = Date.now() - start;
                 return { result, sourceName: source.getName(), durationMs };
@@ -160,6 +161,7 @@ export function createParallelBalanceMiddleware(sources) {
                     const result = await source.assetsMiddleware({
                         request: fallbackRequests[i],
                         response: {},
+                        getAssetsState: context.getAssetsState,
                     }, noopNext);
                     const durationMs = Date.now() - start;
                     return { result, sourceName: source.getName(), durationMs };
@@ -227,6 +229,7 @@ export function createParallelMiddleware(sources) {
                 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 };

package/dist/middlewares/ParallelMiddleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAWxC,+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,KAAK,EAAE,CAAC;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,KAAK,OAAO,CAAC;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,SAAS,GAAG,IAAI,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAErD,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,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;gBAChC,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;iBACb,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,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,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;qBACb,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC;YAE7C,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;gBAChC,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;iBAClC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["import { Semaphore } from 'async-mutex';\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 semaphore = new Semaphore(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          semaphore.runExclusive(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: requests[i],\n                response: {},\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            semaphore.runExclusive(async () => {\n              const start = Date.now();\n              const result = await source.assetsMiddleware(\n                {\n                  request: fallbackRequests[i],\n                  response: {},\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 semaphore = new Semaphore(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          semaphore.runExclusive(async () => {\n            const start = Date.now();\n            const result = await source.assetsMiddleware(\n              {\n                request: context.request,\n                response: { ...context.response },\n              },\n              noopNext,\n            );\n            const durationMs = Date.now() - start;\n            return { result, sourceName: source.getName(), durationMs };\n          }),\n        ),\n      );\n\n      const results = timedResults.map((entry) => entry.result);\n      const mergedResponse = mergeDataResponses(\n        results.map((result) => result.response),\n      );\n\n      const durationByDataSource: Record<string, number> = {\n        ...context.durationByDataSource,\n      };\n      for (const { sourceName, durationMs } of timedResults) {\n        durationByDataSource[`${PARALLEL_MIDDLEWARE_NAME}.${sourceName}`] =\n          durationMs;\n      }\n\n      return next({\n        ...context,\n        response: mergeDataResponses([context.response, mergedResponse]),\n        durationByDataSource,\n      });\n    },\n  };\n}\n"]}
+{"version":3,"file":"ParallelMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/ParallelMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAWxC,+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,KAAK,EAAE,CAAC;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,KAAK,OAAO,CAAC;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,SAAS,GAAG,IAAI,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAErD,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,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;gBAChC,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,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;wBACE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;wBAC5B,QAAQ,EAAE,EAAE;wBACZ,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,EACD,QAAQ,CACT,CAAC;oBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;oBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;gBAC9D,CAAC,CAAC,CACH,CACF,CAAC;gBACF,MAAM,eAAe,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBACnE,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,aAAa,EAAE,CAAC;oBACvD,MAAM,GAAG,GAAG,GAAG,gCAAgC,IAAI,UAAU,EAAE,CAAC;oBAChE,oBAAoB,CAAC,GAAG,CAAC;wBACvB,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAClD,CAAC;gBACD,MAAM,cAAc,GAAG,kBAAkB,CACvC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACjD,CAAC;gBACF,cAAc,GAAG,kBAAkB,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC;gBACtE,uEAAuE;gBACvE,IAAI,cAAc,CAAC,MAAM,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;oBAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAW,CAAC;oBAC7C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CACzC,cAAc,CAAC,aAAa,CAC7B,EAAE,CAAC;wBACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;4BACnD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;4BACjD,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACjC,CAAC;oBACH,CAAC;oBACD,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;wBACrC,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BACnC,OAAO,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBACxC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,kCAAkC;AAClC,+EAA+E;AAE/E,MAAM,wBAAwB,GAAG,oBAAoB,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;;;;GAQG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAA2B;IAIlE,OAAO;QACL,OAAO;YACL,OAAO,wBAAwB,CAAC;QAClC,CAAC;QAED,gBAAgB,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAoB,EAAE;YAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAmB,EAA2B,EAAE,CACtE,GAAG,CAAC;YACN,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC;YAE7C,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,SAAS,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE;gBAChC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAC1C;oBACE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE;oBACjC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,EACD,QAAQ,CACT,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBACtC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC;YAC9D,CAAC,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,kBAAkB,CACvC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;YAEF,MAAM,oBAAoB,GAA2B;gBACnD,GAAG,OAAO,CAAC,oBAAoB;aAChC,CAAC;YACF,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,YAAY,EAAE,CAAC;gBACtD,oBAAoB,CAAC,GAAG,wBAAwB,IAAI,UAAU,EAAE,CAAC;oBAC/D,UAAU,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,OAAO;gBACV,QAAQ,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBAChE,oBAAoB;aACrB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["import { Semaphore } from 'async-mutex';\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 semaphore = new Semaphore(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          semaphore.runExclusive(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            semaphore.runExclusive(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 semaphore = new Semaphore(CONCURRENCY);\n\n      const timedResults = await Promise.all(\n        sources.map((source) =>\n          semaphore.runExclusive(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/RpcFallbackMiddleware.d.ts

@@ -1,26 +1,25 @@
-import type { AssetsControllerState, AssetsDataSource, Middleware } from '../types.js';
+import type { AssetsDataSource, Middleware } from '../types.js';
 export type RpcFallbackMiddlewareOptions = {
     /** The RPC data source to use as a fallback. */
     rpcDataSource: AssetsDataSource;
-    /**
-     * When true, recover failed and unprocessed chains with a full RPC refetch
-     * (Accounts API v6). When false, re-read stale tracked assets the v5 API
-     * omitted (legacy path).
-     */
-    isBalanceV6Enabled?: () => boolean;
-    /** Current AssetsController state. Used to find tracked assets the response left empty. */
-    getAssetsState: () => AssetsControllerState;
 };
 /**
- * RpcFallbackMiddleware recovers what upstream sources left outstanding on RPC.
+ * RpcFallbackMiddleware re-reads balances from the RPC data source in two
+ * cases:
  *
- * Accounts API v5: chains in `response.errors`, plus EVM assets tracked in
- * state that this response left empty (the API omits tokens it does not index).
+ * 1. Chains present in `response.errors` (network error, unprocessedNetworks,
+ *    timeout, …). Successful RPC results are merged into the response and
+ *    their entries are cleared from `response.errors`.
+ * 2. EVM assets tracked in state (`state.assetsBalance` or
+ *    `state.customAssets`) that this balance response left empty. The Accounts
+ *    API omits tokens it does not index, and a returned `0` is
+ *    indistinguishable from "not indexed", so the `merge` state update would
+ *    otherwise keep a stale amount forever. They are passed to RPC as
+ *    `customAssets` so the balance fetcher includes them in its multicall.
  *
- * Accounts API v6: one RPC retry on the chain axis. Chains in
- * `response.errors` are refetched in full (native + existing balances + pins +
- * default tracked assets). RPC reads those from state; the request is not
- * scoped via `customAssets`.
+ * Place this immediately after `createParallelBalanceMiddleware` in the fast
+ * pipeline, after `CustomAssetGraduationMiddleware` so the custom assets RPC
+ * carries never trigger graduation.
  */
 export declare class RpcFallbackMiddleware {
     #private;

package/dist/middlewares/RpcFallbackMiddleware.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RpcFallbackMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,qBAAqB,EACrB,gBAAgB,EAKhB,UAAU,EAEX,MAAM,aAAa,CAAC;AAQrB,MAAM,MAAM,4BAA4B,GAAG;IACzC,gDAAgD;IAChD,aAAa,EAAE,gBAAgB,CAAC;IAChC;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,MAAM,OAAO,CAAC;IACnC,2FAA2F;IAC3F,cAAc,EAAE,MAAM,qBAAqB,CAAC;CAC7C,CAAC;AAIF;;;;;;;;;;GAUG;AACH,qBAAa,qBAAqB;;IAChC,QAAQ,CAAC,IAAI,2BAAmB;IAQhC,YAAY,OAAO,EAAE,4BAA4B,EAKhD;IAED,OAAO,IAAI,MAAM,CAEhB;IAED,IAAI,gBAAgB,IAAI,UAAU,CAOjC;CA0JF"}
+{"version":3,"file":"RpcFallbackMiddleware.d.ts","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,gBAAgB,EAKhB,UAAU,EACX,MAAM,aAAa,CAAC;AAOrB,MAAM,MAAM,4BAA4B,GAAG;IACzC,gDAAgD;IAChD,aAAa,EAAE,gBAAgB,CAAC;CACjC,CAAC;AAEF;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,qBAAqB;;IAChC,QAAQ,CAAC,IAAI,2BAAmB;IAIhC,YAAY,OAAO,EAAE,4BAA4B,EAEhD;IAED,OAAO,IAAI,MAAM,CAEhB;IAED,IAAI,gBAAgB,IAAI,UAAU,CAoGjC;CACF"}

package/dist/middlewares/RpcFallbackMiddleware.js

@@ -2,153 +2,136 @@ import { KnownCaipNamespace } from '@metamask/utils';
 import { isStakingContractAssetId } from '../data-sources/evm-rpc-services/index.js';
 import { projectLogger, createModuleLogger } from '../logger.js';
 import { forDataTypes } from '../types.js';
-import { filterFailedChainBalances } from '../utils/filterFailedChainBalances.js';
 import { mergeDataResponses } from './ParallelMiddleware.js';
 const CONTROLLER_NAME = 'RpcFallbackMiddleware';
 const log = createModuleLogger(projectLogger, CONTROLLER_NAME);
-const noopNext = async (ctx) => ctx;
 /**
- * RpcFallbackMiddleware recovers what upstream sources left outstanding on RPC.
+ * RpcFallbackMiddleware re-reads balances from the RPC data source in two
+ * cases:
  *
- * Accounts API v5: chains in `response.errors`, plus EVM assets tracked in
- * state that this response left empty (the API omits tokens it does not index).
+ * 1. Chains present in `response.errors` (network error, unprocessedNetworks,
+ *    timeout, …). Successful RPC results are merged into the response and
+ *    their entries are cleared from `response.errors`.
+ * 2. EVM assets tracked in state (`state.assetsBalance` or
+ *    `state.customAssets`) that this balance response left empty. The Accounts
+ *    API omits tokens it does not index, and a returned `0` is
+ *    indistinguishable from "not indexed", so the `merge` state update would
+ *    otherwise keep a stale amount forever. They are passed to RPC as
+ *    `customAssets` so the balance fetcher includes them in its multicall.
  *
- * Accounts API v6: one RPC retry on the chain axis. Chains in
- * `response.errors` are refetched in full (native + existing balances + pins +
- * default tracked assets). RPC reads those from state; the request is not
- * scoped via `customAssets`.
+ * Place this immediately after `createParallelBalanceMiddleware` in the fast
+ * pipeline, after `CustomAssetGraduationMiddleware` so the custom assets RPC
+ * carries never trigger graduation.
  */
 export class RpcFallbackMiddleware {
     name = CONTROLLER_NAME;
     #rpcDataSource;
-    #isBalanceV6Enabled;
-    #getAssetsState;
     constructor(options) {
         this.#rpcDataSource = options.rpcDataSource;
-        this.#isBalanceV6Enabled =
-            options.isBalanceV6Enabled ?? (() => false);
-        this.#getAssetsState = options.getAssetsState;
     }
     getName() {
         return this.name;
     }
     get assetsMiddleware() {
         return forDataTypes(['balance'], async (ctx, next) => {
-            if (this.#isBalanceV6Enabled()) {
-                return this.#recoverV6(ctx, next);
+            const erroredChains = new Set(Object.keys(ctx.response.errors ?? {}));
+            const staleAssets = collectStaleTrackedAssets(ctx);
+            const chainsToFetch = [
+                ...new Set([
+                    ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),
+                    // Already restricted to requested chains. Their chain may not be
+                    // errored: the Accounts API can answer for a chain while omitting a
+                    // token it does not index.
+                    ...staleAssets.map((assetId) => assetId.split('/')[0]),
+                ]),
+            ];
+            if (chainsToFetch.length === 0) {
+                return next(ctx);
             }
-            return this.#recoverV5(ctx, next);
-        });
-    }
-    async #recoverV6(ctx, next) {
-        const erroredChains = new Set(Object.keys(ctx.response.errors ?? {}));
-        const chainsToFetch = ctx.request.chainIds.filter((id) => erroredChains.has(id));
-        if (chainsToFetch.length === 0) {
-            return next(ctx);
-        }
-        const rpcResult = await this.#rpcDataSource.assetsMiddleware({
-            ...ctx,
-            request: { ...ctx.request, chainIds: chainsToFetch },
-            response: {},
-        }, noopNext);
-        const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
-        const rpcAssetsBalance = filterFailedChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
-        const merged = mergeDataResponses([
-            ctx.response,
-            {
-                ...rpcResult.response,
-                assetsBalance: rpcAssetsBalance,
-            },
-        ]);
-        if (merged.errors && rpcAssetsBalance) {
-            const chainsRecoveredByRpc = new Set();
-            for (const accountBalances of Object.values(rpcAssetsBalance)) {
-                for (const assetId of Object.keys(accountBalances)) {
-                    chainsRecoveredByRpc.add(assetId.split('/')[0]);
+            log('Re-reading balances on RPC', {
+                erroredChains: [...erroredChains],
+                staleAssets,
+                chains: chainsToFetch,
+            });
+            const filteredRequest = {
+                ...ctx.request,
+                chainIds: chainsToFetch,
+                customAssets: [
+                    ...new Set([...(ctx.request.customAssets ?? []), ...staleAssets]),
+                ],
+            };
+            const noopNext = async (inner) => inner;
+            const rpcResult = await this.#rpcDataSource.assetsMiddleware({
+                ...ctx,
+                request: filteredRequest,
+                response: {},
+            }, noopNext);
+            // A chain RPC itself failed on contributed nothing trustworthy: its
+            // balances are failure stubs (native 0) that would overwrite correct
+            // upstream amounts and, with replaceCoveredChainBalances, wipe the
+            // chain's token slice from state. Drop them before merging — this also
+            // keeps failed chains from counting as "recovered" below.
+            const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
+            const rpcAssetsBalance = filterOutChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
+            // RPC errors are kept only for chains that were already errored
+            // upstream. For chains fetched solely for stale tracked assets the
+            // upstream response succeeded and stays authoritative — the stale asset
+            // keeps its previous amount and is retried on the next pass.
+            const rpcErrors = Object.fromEntries(Object.entries(rpcResult.response.errors ?? {}).filter(([chainId]) => erroredChains.has(chainId)));
+            const merged = mergeDataResponses([
+                ctx.response,
+                {
+                    ...rpcResult.response,
+                    assetsBalance: rpcAssetsBalance,
+                    errors: rpcErrors,
+                },
+            ]);
+            // Clear errors only for chains RPC actually recovered a balance for.
+            // We must inspect the (filtered) RPC balances — NOT merged — because
+            // merged also contains balances from the upstream sources (AccountsApi /
+            // Websocket / Staked). If those sources returned partial data for
+            // a chain that they also flagged as errored (e.g. via
+            // unprocessedNetworks), and RPC then failed for that same chain,
+            // looking at merged would incorrectly mark the error as recovered.
+            if (merged.errors && rpcAssetsBalance) {
+                const chainsRecoveredByRpc = new Set();
+                for (const accountBalances of Object.values(rpcAssetsBalance)) {
+                    for (const assetId of Object.keys(accountBalances)) {
+                        chainsRecoveredByRpc.add(assetId.split('/')[0]);
+                    }
                 }
-            }
-            for (const chainId of erroredChains) {
-                if (chainsRecoveredByRpc.has(chainId)) {
-                    delete merged.errors[chainId];
+                for (const chainId of erroredChains) {
+                    if (chainsRecoveredByRpc.has(chainId)) {
+                        delete merged.errors[chainId];
+                    }
                 }
             }
-        }
-        return next({ ...ctx, response: merged });
-    }
-    async #recoverV5(ctx, next) {
-        const erroredChains = new Set(Object.keys(ctx.response.errors ?? {}));
-        const staleAssets = collectStaleTrackedAssets(ctx, this.#getAssetsState());
-        const chainsToFetch = [
-            ...new Set([
-                ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),
-                // Already restricted to requested chains. Their chain may not be
-                // errored: the Accounts API can answer for a chain while omitting a
-                // token it does not index.
-                ...staleAssets.map((assetId) => assetId.split('/')[0]),
-            ]),
-        ];
-        if (chainsToFetch.length === 0) {
-            return next(ctx);
-        }
-        log('Re-reading balances on RPC', {
-            erroredChains: [...erroredChains],
-            staleAssets,
-            chains: chainsToFetch,
+            return next({ ...ctx, response: merged });
         });
-        const filteredRequest = {
-            ...ctx.request,
-            chainIds: chainsToFetch,
-            customAssets: [
-                ...new Set([...(ctx.request.customAssets ?? []), ...staleAssets]),
-            ],
-        };
-        const rpcResult = await this.#rpcDataSource.assetsMiddleware({
-            ...ctx,
-            request: filteredRequest,
-            response: {},
-        }, noopNext);
-        // A chain RPC itself failed on contributed nothing trustworthy: its
-        // balances are failure stubs (native 0) that would overwrite correct
-        // upstream amounts and, with replaceCoveredChainBalances, wipe the
-        // chain's token slice from state. Drop them before merging — this also
-        // keeps failed chains from counting as "recovered" below.
-        const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
-        const rpcAssetsBalance = filterFailedChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
-        // RPC errors are kept only for chains that were already errored
-        // upstream. For chains fetched solely for stale tracked assets the
-        // upstream response succeeded and stays authoritative — the stale asset
-        // keeps its previous amount and is retried on the next pass.
-        const rpcErrors = Object.fromEntries(Object.entries(rpcResult.response.errors ?? {}).filter(([chainId]) => erroredChains.has(chainId)));
-        const merged = mergeDataResponses([
-            ctx.response,
-            {
-                ...rpcResult.response,
-                assetsBalance: rpcAssetsBalance,
-                errors: rpcErrors,
-            },
-        ]);
-        // Clear errors only for chains RPC actually recovered a balance for.
-        // We must inspect the (filtered) RPC balances — NOT merged — because
-        // merged also contains balances from the upstream sources (AccountsApi /
-        // Websocket / Staked). If those sources returned partial data for
-        // a chain that they also flagged as errored (e.g. via
-        // unprocessedNetworks), and RPC then failed for that same chain,
-        // looking at merged would incorrectly mark the error as recovered.
-        if (merged.errors && rpcAssetsBalance) {
-            const chainsRecoveredByRpc = new Set();
-            for (const accountBalances of Object.values(rpcAssetsBalance)) {
-                for (const assetId of Object.keys(accountBalances)) {
-                    chainsRecoveredByRpc.add(assetId.split('/')[0]);
-                }
-            }
-            for (const chainId of erroredChains) {
-                if (chainsRecoveredByRpc.has(chainId)) {
-                    delete merged.errors[chainId];
-                }
-            }
+    }
+}
+/**
+ * Remove all balances that belong to the given chains.
+ *
+ * Used to discard results for chains the RPC source itself failed on, whose
+ * entries are failure stubs (native `0`) rather than real readings.
+ *
+ * @param assetsBalance - Balances by account from the RPC response.
+ * @param chainIds - Chains whose balances should be dropped.
+ * @returns The filtered balance map, without accounts left empty.
+ */
+function filterOutChainBalances(assetsBalance, chainIds) {
+    if (!assetsBalance || chainIds.size === 0) {
+        return assetsBalance;
+    }
+    const filtered = {};
+    for (const [accountId, accountBalances] of Object.entries(assetsBalance)) {
+        const kept = Object.fromEntries(Object.entries(accountBalances).filter(([assetId]) => !chainIds.has(assetId.split('/')[0])));
+        if (Object.keys(kept).length > 0) {
+            filtered[accountId] = kept;
         }
-        return next({ ...ctx, response: merged });
     }
+    return filtered;
 }
 /**
  * EVM assets tracked in state that this balance response left empty and RPC
@@ -159,11 +142,10 @@ export class RpcFallbackMiddleware {
  * set, so anything else would be queued and then silently dropped.
  *
  * @param ctx - Pipeline context.
- * @param state - Current AssetsController state.
  * @returns Asset IDs to hand to the RPC data source.
  */
-function collectStaleTrackedAssets(ctx, state) {
-    const { assetsBalance: stateAssetsBalance, customAssets: stateCustomAssets } = state;
+function collectStaleTrackedAssets(ctx) {
+    const { assetsBalance: stateAssetsBalance, customAssets: stateCustomAssets } = ctx.getAssetsState();
     const staleAssets = new Set();
     for (const { account, supportedChains } of ctx.request
         .accountsWithSupportedChains) {