@metamask-previews/assets-controller 16.1.0-preview-fb486948f → 16.1.0-preview-aecbce4ff

package/dist/middlewares/ParallelMiddleware.js.map

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

package/dist/middlewares/RpcFallbackMiddleware.d.ts

@@ -2,22 +2,24 @@ import type { AssetsDataSource, Middleware } from '../types.js';
 export type RpcFallbackMiddlewareOptions = {
     /** The RPC data source to use as a fallback. */
     rpcDataSource: AssetsDataSource;
-    /**
-     * When true, recover `unprocessedCustomAssets` (Accounts API v6). When false,
-     * re-read stale tracked assets the v5 API omitted (legacy path).
-     */
-    isBalanceV6Enabled?: () => boolean;
 };
 /**
- * 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. `customAssets` is the union of
- * `response.unprocessedCustomAssets` (trust the API) and every tracked
- * asset on chains in `response.errors` (so a `full` update cannot drop
- * imported or previously detected tokens on a failed chain).
+ * 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,gBAAgB,EAMhB,UAAU,EAEX,MAAM,aAAa,CAAC;AAQrB,MAAM,MAAM,4BAA4B,GAAG;IACzC,gDAAgD;IAChD,aAAa,EAAE,gBAAgB,CAAC;IAChC;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,OAAO,CAAC;CACpC,CAAC;AAIF;;;;;;;;;;GAUG;AACH,qBAAa,qBAAqB;;IAChC,QAAQ,CAAC,IAAI,2BAAmB;IAMhC,YAAY,OAAO,EAAE,4BAA4B,EAIhD;IAED,OAAO,IAAI,MAAM,CAEhB;IAED,IAAI,gBAAgB,IAAI,UAAU,CAOjC;CAuJF"}
+{"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,204 +2,124 @@ 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 { normalizeAssetId } from '../utils/index.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. `customAssets` is the union of
- * `response.unprocessedCustomAssets` (trust the API) and every tracked
- * asset on chains in `response.errors` (so a `full` update cannot drop
- * imported or previously detected tokens on a failed chain).
+ * 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;
     constructor(options) {
         this.#rpcDataSource = options.rpcDataSource;
-        this.#isBalanceV6Enabled =
-            options.isBalanceV6Enabled ?? (() => false);
     }
     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 unprocessedAssetsPins = [
-            ...new Set(ctx.response.unprocessedCustomAssets ?? []),
-        ];
-        const assetsOnErroredChains = collectTrackedAssetsOnChains(ctx, erroredChains);
-        const customAssets = [
-            ...new Set([...unprocessedAssetsPins, ...assetsOnErroredChains]),
-        ];
-        const chainsToFetch = [
-            ...new Set([
-                ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),
-                ...customAssets.map((assetId) => chainIdOfAsset(assetId)),
-            ]),
-        ];
-        if (chainsToFetch.length === 0) {
-            return next(ctx);
-        }
-        const rpcRequest = {
-            ...ctx.request,
-            chainIds: chainsToFetch,
-            ...(customAssets.length > 0 ? { customAssets } : {}),
-        };
-        const rpcResult = await this.#rpcDataSource.assetsMiddleware({ ...ctx, request: rpcRequest, response: {} }, noopNext);
-        const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
-        const rpcAssetsBalance = filterOutChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
-        let 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];
+                    }
                 }
             }
-        }
-        merged = clearRecoveredAssetIds(merged);
-        return next({ ...ctx, response: merged });
-    }
-    async #recoverV5(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)),
-                ...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);
-        const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
-        const rpcAssetsBalance = filterOutChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
-        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,
-            },
-        ]);
-        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];
-                }
-            }
-        }
-        return next({ ...ctx, response: merged });
     }
 }
-function chainIdOfAsset(assetId) {
-    return assetId.split('/')[0];
-}
 /**
- * Every EVM asset already in state on `chains` — balances and pins — so an RPC
- * retry of a failed chain re-reads the whole slice, not natives only. Hidden
- * assets are left out, matching the `excludeAssetIds` the Accounts API is sent.
- * Staking vaults belong to `StakedBalanceDataSource` and are skipped.
+ * Remove all balances that belong to the given chains.
  *
- * @param ctx - Pipeline context, for request accounts and assets state.
- * @param chains - Chains being recovered on RPC.
- * @returns Deduplicated asset IDs on those 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 collectTrackedAssetsOnChains(ctx, chains) {
-    if (chains.size === 0) {
-        return [];
-    }
-    const { assetsBalance, customAssets: statePins, assetPreferences, } = ctx.getAssetsState();
-    const tracked = new Set();
-    for (const { account } of ctx.request.accountsWithSupportedChains) {
-        const balanceIds = Object.keys(assetsBalance[account.id] ?? {});
-        const pinIds = statePins?.[account.id] ?? [];
-        for (const assetId of [...balanceIds, ...pinIds]) {
-            if (chains.has(chainIdOfAsset(assetId)) &&
-                assetId.startsWith(`${KnownCaipNamespace.Eip155}:`) &&
-                !isStakingContractAssetId(assetId) &&
-                !assetPreferences?.[normalizeAssetId(assetId)]?.hidden) {
-                tracked.add(assetId);
-            }
-        }
-    }
-    return [...tracked];
-}
-function clearRecoveredAssetIds(response) {
-    if (!response.unprocessedCustomAssets ||
-        response.unprocessedCustomAssets.length === 0) {
-        return response;
-    }
-    const recovered = new Set();
-    for (const accountBalances of Object.values(response.assetsBalance ?? {})) {
-        for (const assetId of Object.keys(accountBalances)) {
-            recovered.add(normalizeAssetId(assetId));
-        }
-    }
-    const stillUnprocessed = response.unprocessedCustomAssets.filter((assetId) => !recovered.has(normalizeAssetId(assetId)));
-    if (stillUnprocessed.length === response.unprocessedCustomAssets.length) {
-        return response;
-    }
-    const next = { ...response };
-    if (stillUnprocessed.length === 0) {
-        delete next.unprocessedCustomAssets;
-    }
-    else {
-        next.unprocessedCustomAssets = stillUnprocessed;
-    }
-    return next;
-}
 function filterOutChainBalances(assetsBalance, chainIds) {
     if (!assetsBalance || chainIds.size === 0) {
         return assetsBalance;
@@ -213,6 +133,17 @@ function filterOutChainBalances(assetsBalance, chainIds) {
     }
     return filtered;
 }
+/**
+ * EVM assets tracked in state that this balance response left empty and RPC
+ * should re-read.
+ *
+ * Limited to chains both requested and supported by the owning account:
+ * RpcDataSource fetches per account and skips chains outside its supported
+ * set, so anything else would be queued and then silently dropped.
+ *
+ * @param ctx - Pipeline context.
+ * @returns Asset IDs to hand to the RPC data source.
+ */
 function collectStaleTrackedAssets(ctx) {
     const { assetsBalance: stateAssetsBalance, customAssets: stateCustomAssets } = ctx.getAssetsState();
     const staleAssets = new Set();
@@ -226,6 +157,8 @@ function collectStaleTrackedAssets(ctx) {
         const chainsForAccount = supportedChains.filter((chainId) => ctx.request.chainIds.includes(chainId));
         for (const assetId of trackedAssetIds) {
             if (isEvmAssetOnChains(assetId, chainsForAccount) &&
+                // Staked vault balances belong to StakedBalanceDataSource; an RPC
+                // ERC-20 read of the share token would clobber them.
                 !isStakingContractAssetId(assetId) &&
                 isBalanceEmpty(ctx.response.assetsBalance?.[accountId], assetId)) {
                 staleAssets.add(assetId);
@@ -234,6 +167,16 @@ function collectStaleTrackedAssets(ctx) {
     }
     return [...staleAssets];
 }
+/**
+ * Whether a balance response carries no positive amount for an asset. A
+ * returned `0` cannot be distinguished from "not indexed", so both count as
+ * empty. Asset IDs are matched case-insensitively: state keys ERC-20 assets by
+ * checksummed address, while some data sources return them lower-cased.
+ *
+ * @param balances - Balance map for a single account from the response.
+ * @param assetId - Asset ID to check.
+ * @returns True when the response holds no positive amount for the asset.
+ */
 function isBalanceEmpty(balances, assetId) {
     let amount = balances?.[assetId]?.amount;
     if (amount === undefined && balances) {
@@ -242,6 +185,13 @@ function isBalanceEmpty(balances, assetId) {
     }
     return !(Number(amount) > 0);
 }
+/**
+ * Whether an asset is an EVM asset on one of the given chains.
+ *
+ * @param assetId - CAIP-19 asset ID.
+ * @param chainIds - Chains to match against.
+ * @returns True for EVM assets whose chain is in `chainIds`.
+ */
 function isEvmAssetOnChains(assetId, chainIds) {
     if (!assetId.startsWith(`${KnownCaipNamespace.Eip155}:`)) {
         return false;

package/dist/middlewares/RpcFallbackMiddleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"RpcFallbackMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAY3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,MAAM,eAAe,GAAG,uBAAuB,CAAC;AAEhD,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAY/D,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAY,EAAoB,EAAE,CAAC,GAAG,CAAC;AAE/D;;;;;;;;;;GAUG;AACH,MAAM,OAAO,qBAAqB;IACvB,IAAI,GAAG,eAAe,CAAC;IAEvB,cAAc,CAAmB;IAEjC,mBAAmB,CAAgB;IAE5C,YAAY,OAAqC;QAC/C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;QAC5C,IAAI,CAAC,mBAAmB;YACtB,OAAO,CAAC,kBAAkB,IAAI,CAAC,GAAY,EAAE,CAAC,KAAK,CAAC,CAAC;IACzD,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,gBAAgB;QAClB,OAAO,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACnD,IAAI,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;gBAC/B,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACpC,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAY,EAAE,IAAkB;QAC/C,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CACpD,CAAC;QACF,MAAM,qBAAqB,GAAG;YAC5B,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,uBAAuB,IAAI,EAAE,CAAC;SACvD,CAAC;QACF,MAAM,qBAAqB,GAAG,4BAA4B,CACxD,GAAG,EACH,aAAa,CACd,CAAC;QACF,MAAM,YAAY,GAAG;YACnB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,qBAAqB,EAAE,GAAG,qBAAqB,CAAC,CAAC;SACjE,CAAC;QACF,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,GAAG,CAAC;gBACT,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7D,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;aAC1D,CAAC;SACH,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,UAAU,GAAgB;YAC9B,GAAG,GAAG,CAAC,OAAO;YACd,QAAQ,EAAE,aAAa;YACvB,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACrD,CAAC;QACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAC1D,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,EAC7C,QAAQ,CACT,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CAC1D,CAAC;QACF,MAAM,gBAAgB,GAAG,sBAAsB,CAC7C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAChC,eAAe,CAChB,CAAC;QAEF,IAAI,MAAM,GAAG,kBAAkB,CAAC;YAC9B,GAAG,CAAC,QAAQ;YACZ;gBACE,GAAG,SAAS,CAAC,QAAQ;gBACrB,aAAa,EAAE,gBAAgB;aAChC;SACF,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;YACtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC9D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;gBACpC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBACtC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;QAExC,OAAO,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAY,EAAE,IAAkB;QAC/C,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CACpD,CAAC;QACF,MAAM,WAAW,GAAG,yBAAyB,CAAC,GAAG,CAAC,CAAC;QAEnD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,GAAG,CAAC;gBACT,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7D,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;aAClE,CAAC;SACH,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,GAAG,CAAC,4BAA4B,EAAE;YAChC,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC;YACjC,WAAW;YACX,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG;YACtB,GAAG,GAAG,CAAC,OAAO;YACd,QAAQ,EAAE,aAAa;YACvB,YAAY,EAAE;gBACZ,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC;aAClE;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAC1D;YACE,GAAG,GAAG;YACN,OAAO,EAAE,eAAe;YACxB,QAAQ,EAAE,EAAE;SACb,EACD,QAAQ,CACT,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CAC1D,CAAC;QACF,MAAM,gBAAgB,GAAG,sBAAsB,CAC7C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAChC,eAAe,CAChB,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAClC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CACnE,aAAa,CAAC,GAAG,CAAC,OAAkB,CAAC,CACtC,CACF,CAAC;QAEF,MAAM,MAAM,GAAiB,kBAAkB,CAAC;YAC9C,GAAG,CAAC,QAAQ;YACZ;gBACE,GAAG,SAAS,CAAC,QAAQ;gBACrB,aAAa,EAAE,gBAAgB;gBAC/B,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;YACtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC9D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;gBACpC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBACtC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IAC5C,CAAC;CACF;AAED,SAAS,cAAc,CAAC,OAAsB;IAC5C,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;AAC1C,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,4BAA4B,CACnC,GAAY,EACZ,MAAoB;IAEpB,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,EACJ,aAAa,EACb,YAAY,EAAE,SAAS,EACvB,gBAAgB,GACjB,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IACzB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAiB,CAAC;IAEzC,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,GAAG,CAAC,OAAO,CAAC,2BAA2B,EAAE,CAAC;QAClE,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAC5B,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CACb,CAAC;QACrB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7C,KAAK,MAAM,OAAO,IAAI,CAAC,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;YACjD,IACE,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;gBACnC,OAAO,CAAC,UAAU,CAAC,GAAG,kBAAkB,CAAC,MAAM,GAAG,CAAC;gBACnD,CAAC,wBAAwB,CAAC,OAAO,CAAC;gBAClC,CAAC,gBAAgB,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EACtD,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAsB;IACpD,IACE,CAAC,QAAQ,CAAC,uBAAuB;QACjC,QAAQ,CAAC,uBAAuB,CAAC,MAAM,KAAK,CAAC,EAC7C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,GAAG,EAAiB,CAAC;IAC3C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,CAAC;QAC1E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACnD,SAAS,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAwB,CAAC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,MAAM,gBAAgB,GAAG,QAAQ,CAAC,uBAAuB,CAAC,MAAM,CAC9D,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CACvD,CAAC;IAEF,IAAI,gBAAgB,CAAC,MAAM,KAAK,QAAQ,CAAC,uBAAuB,CAAC,MAAM,EAAE,CAAC;QACxE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,IAAI,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;IAC7B,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,uBAAuB,GAAG,gBAAgB,CAAC;IAClD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,sBAAsB,CAC7B,aAA4C,EAC5C,QAAsB;IAEtB,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,MAAM,QAAQ,GAA+C,EAAE,CAAC;IAChE,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QACzE,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAC7B,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,MAAM,CACpC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC,CAC/D,CACqC,CAAC;QACzC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,QAAQ,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,yBAAyB,CAAC,GAAY;IAC7C,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,YAAY,EAAE,iBAAiB,EAAE,GAC1E,GAAG,CAAC,cAAc,EAAE,CAAC;IAEvB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAiB,CAAC;IAE7C,KAAK,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,GAAG,CAAC,OAAO;SACnD,2BAA2B,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,IAAI,GAAG,CAAgB;YAC7C,GAAI,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAqB;YACxE,GAAG,CAAC,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;SAC1C,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CACvC,CAAC;QAEF,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;YACtC,IACE,kBAAkB,CAAC,OAAO,EAAE,gBAAgB,CAAC;gBAC7C,CAAC,wBAAwB,CAAC,OAAO,CAAC;gBAClC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,EAChE,CAAC;gBACD,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,WAAW,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,cAAc,CACrB,QAAkD,EAClD,OAAsB;IAEtB,IAAI,MAAM,GAAG,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IACzC,IAAI,MAAM,KAAK,SAAS,IAAI,QAAQ,EAAE,CAAC;QACrC,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAChD,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CACpC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,iBAAiB,CACjD,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;IACjB,CAAC;IACD,OAAO,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,kBAAkB,CACzB,OAAsB,EACtB,QAAmB;IAEnB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC,CAAC;AAC7D,CAAC","sourcesContent":["import { KnownCaipNamespace } from '@metamask/utils';\n\nimport { isStakingContractAssetId } from '../data-sources/evm-rpc-services/index.js';\nimport { projectLogger, createModuleLogger } from '../logger.js';\nimport { forDataTypes } from '../types.js';\nimport type {\n  AssetBalance,\n  AssetsDataSource,\n  Caip19AssetId,\n  ChainId,\n  Context,\n  DataRequest,\n  DataResponse,\n  Middleware,\n  NextFunction,\n} from '../types.js';\nimport { normalizeAssetId } from '../utils/index.js';\nimport { mergeDataResponses } from './ParallelMiddleware.js';\n\nconst CONTROLLER_NAME = 'RpcFallbackMiddleware';\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\nexport type RpcFallbackMiddlewareOptions = {\n  /** The RPC data source to use as a fallback. */\n  rpcDataSource: AssetsDataSource;\n  /**\n   * When true, recover `unprocessedCustomAssets` (Accounts API v6). When false,\n   * re-read stale tracked assets the v5 API omitted (legacy path).\n   */\n  isBalanceV6Enabled?: () => boolean;\n};\n\nconst noopNext = async (ctx: Context): Promise<Context> => ctx;\n\n/**\n * RpcFallbackMiddleware recovers what upstream sources left outstanding on RPC.\n *\n * Accounts API v5: chains in `response.errors`, plus EVM assets tracked in\n * state that this response left empty (the API omits tokens it does not index).\n *\n * Accounts API v6: one RPC retry. `customAssets` is the union of\n * `response.unprocessedCustomAssets` (trust the API) and every tracked\n * asset on chains in `response.errors` (so a `full` update cannot drop\n * imported or previously detected tokens on a failed chain).\n */\nexport class RpcFallbackMiddleware {\n  readonly name = CONTROLLER_NAME;\n\n  readonly #rpcDataSource: AssetsDataSource;\n\n  readonly #isBalanceV6Enabled: () => boolean;\n\n  constructor(options: RpcFallbackMiddlewareOptions) {\n    this.#rpcDataSource = options.rpcDataSource;\n    this.#isBalanceV6Enabled =\n      options.isBalanceV6Enabled ?? ((): boolean => false);\n  }\n\n  getName(): string {\n    return this.name;\n  }\n\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['balance'], async (ctx, next) => {\n      if (this.#isBalanceV6Enabled()) {\n        return this.#recoverV6(ctx, next);\n      }\n      return this.#recoverV5(ctx, next);\n    });\n  }\n\n  async #recoverV6(ctx: Context, next: NextFunction): Promise<Context> {\n    const erroredChains = new Set<ChainId>(\n      Object.keys(ctx.response.errors ?? {}) as ChainId[],\n    );\n    const unprocessedAssetsPins = [\n      ...new Set(ctx.response.unprocessedCustomAssets ?? []),\n    ];\n    const assetsOnErroredChains = collectTrackedAssetsOnChains(\n      ctx,\n      erroredChains,\n    );\n    const customAssets = [\n      ...new Set([...unprocessedAssetsPins, ...assetsOnErroredChains]),\n    ];\n    const chainsToFetch = [\n      ...new Set([\n        ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),\n        ...customAssets.map((assetId) => chainIdOfAsset(assetId)),\n      ]),\n    ];\n\n    if (chainsToFetch.length === 0) {\n      return next(ctx);\n    }\n\n    const rpcRequest: DataRequest = {\n      ...ctx.request,\n      chainIds: chainsToFetch,\n      ...(customAssets.length > 0 ? { customAssets } : {}),\n    };\n    const rpcResult = await this.#rpcDataSource.assetsMiddleware(\n      { ...ctx, request: rpcRequest, response: {} },\n      noopNext,\n    );\n\n    const rpcFailedChains = new Set<ChainId>(\n      Object.keys(rpcResult.response.errors ?? {}) as ChainId[],\n    );\n    const rpcAssetsBalance = filterOutChainBalances(\n      rpcResult.response.assetsBalance,\n      rpcFailedChains,\n    );\n\n    let merged = mergeDataResponses([\n      ctx.response,\n      {\n        ...rpcResult.response,\n        assetsBalance: rpcAssetsBalance,\n      },\n    ]);\n\n    if (merged.errors && rpcAssetsBalance) {\n      const chainsRecoveredByRpc = new Set<string>();\n      for (const accountBalances of Object.values(rpcAssetsBalance)) {\n        for (const assetId of Object.keys(accountBalances)) {\n          chainsRecoveredByRpc.add(assetId.split('/')[0]);\n        }\n      }\n      for (const chainId of erroredChains) {\n        if (chainsRecoveredByRpc.has(chainId)) {\n          delete merged.errors[chainId];\n        }\n      }\n    }\n\n    merged = clearRecoveredAssetIds(merged);\n\n    return next({ ...ctx, response: merged });\n  }\n\n  async #recoverV5(ctx: Context, next: NextFunction): Promise<Context> {\n    const erroredChains = new Set<ChainId>(\n      Object.keys(ctx.response.errors ?? {}) as ChainId[],\n    );\n    const staleAssets = collectStaleTrackedAssets(ctx);\n\n    const chainsToFetch = [\n      ...new Set([\n        ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),\n        ...staleAssets.map((assetId) => assetId.split('/')[0] as ChainId),\n      ]),\n    ];\n\n    if (chainsToFetch.length === 0) {\n      return next(ctx);\n    }\n\n    log('Re-reading balances on RPC', {\n      erroredChains: [...erroredChains],\n      staleAssets,\n      chains: chainsToFetch,\n    });\n\n    const filteredRequest = {\n      ...ctx.request,\n      chainIds: chainsToFetch,\n      customAssets: [\n        ...new Set([...(ctx.request.customAssets ?? []), ...staleAssets]),\n      ],\n    };\n\n    const rpcResult = await this.#rpcDataSource.assetsMiddleware(\n      {\n        ...ctx,\n        request: filteredRequest,\n        response: {},\n      },\n      noopNext,\n    );\n\n    const rpcFailedChains = new Set<ChainId>(\n      Object.keys(rpcResult.response.errors ?? {}) as ChainId[],\n    );\n    const rpcAssetsBalance = filterOutChainBalances(\n      rpcResult.response.assetsBalance,\n      rpcFailedChains,\n    );\n\n    const rpcErrors = Object.fromEntries(\n      Object.entries(rpcResult.response.errors ?? {}).filter(([chainId]) =>\n        erroredChains.has(chainId as ChainId),\n      ),\n    );\n\n    const merged: DataResponse = mergeDataResponses([\n      ctx.response,\n      {\n        ...rpcResult.response,\n        assetsBalance: rpcAssetsBalance,\n        errors: rpcErrors,\n      },\n    ]);\n\n    if (merged.errors && rpcAssetsBalance) {\n      const chainsRecoveredByRpc = new Set<string>();\n      for (const accountBalances of Object.values(rpcAssetsBalance)) {\n        for (const assetId of Object.keys(accountBalances)) {\n          chainsRecoveredByRpc.add(assetId.split('/')[0]);\n        }\n      }\n      for (const chainId of erroredChains) {\n        if (chainsRecoveredByRpc.has(chainId)) {\n          delete merged.errors[chainId];\n        }\n      }\n    }\n\n    return next({ ...ctx, response: merged });\n  }\n}\n\nfunction chainIdOfAsset(assetId: Caip19AssetId): ChainId {\n  return assetId.split('/')[0] as ChainId;\n}\n\n/**\n * Every EVM asset already in state on `chains` — balances and pins — so an RPC\n * retry of a failed chain re-reads the whole slice, not natives only. Hidden\n * assets are left out, matching the `excludeAssetIds` the Accounts API is sent.\n * Staking vaults belong to `StakedBalanceDataSource` and are skipped.\n *\n * @param ctx - Pipeline context, for request accounts and assets state.\n * @param chains - Chains being recovered on RPC.\n * @returns Deduplicated asset IDs on those chains.\n */\nfunction collectTrackedAssetsOnChains(\n  ctx: Context,\n  chains: Set<ChainId>,\n): Caip19AssetId[] {\n  if (chains.size === 0) {\n    return [];\n  }\n\n  const {\n    assetsBalance,\n    customAssets: statePins,\n    assetPreferences,\n  } = ctx.getAssetsState();\n  const tracked = new Set<Caip19AssetId>();\n\n  for (const { account } of ctx.request.accountsWithSupportedChains) {\n    const balanceIds = Object.keys(\n      assetsBalance[account.id] ?? {},\n    ) as Caip19AssetId[];\n    const pinIds = statePins?.[account.id] ?? [];\n    for (const assetId of [...balanceIds, ...pinIds]) {\n      if (\n        chains.has(chainIdOfAsset(assetId)) &&\n        assetId.startsWith(`${KnownCaipNamespace.Eip155}:`) &&\n        !isStakingContractAssetId(assetId) &&\n        !assetPreferences?.[normalizeAssetId(assetId)]?.hidden\n      ) {\n        tracked.add(assetId);\n      }\n    }\n  }\n\n  return [...tracked];\n}\n\nfunction clearRecoveredAssetIds(response: DataResponse): DataResponse {\n  if (\n    !response.unprocessedCustomAssets ||\n    response.unprocessedCustomAssets.length === 0\n  ) {\n    return response;\n  }\n\n  const recovered = new Set<Caip19AssetId>();\n  for (const accountBalances of Object.values(response.assetsBalance ?? {})) {\n    for (const assetId of Object.keys(accountBalances)) {\n      recovered.add(normalizeAssetId(assetId as Caip19AssetId));\n    }\n  }\n\n  const stillUnprocessed = response.unprocessedCustomAssets.filter(\n    (assetId) => !recovered.has(normalizeAssetId(assetId)),\n  );\n\n  if (stillUnprocessed.length === response.unprocessedCustomAssets.length) {\n    return response;\n  }\n\n  const next = { ...response };\n  if (stillUnprocessed.length === 0) {\n    delete next.unprocessedCustomAssets;\n  } else {\n    next.unprocessedCustomAssets = stillUnprocessed;\n  }\n  return next;\n}\n\nfunction filterOutChainBalances(\n  assetsBalance: DataResponse['assetsBalance'],\n  chainIds: Set<ChainId>,\n): DataResponse['assetsBalance'] {\n  if (!assetsBalance || chainIds.size === 0) {\n    return assetsBalance;\n  }\n\n  const filtered: NonNullable<DataResponse['assetsBalance']> = {};\n  for (const [accountId, accountBalances] of Object.entries(assetsBalance)) {\n    const kept = Object.fromEntries(\n      Object.entries(accountBalances).filter(\n        ([assetId]) => !chainIds.has(assetId.split('/')[0] as ChainId),\n      ),\n    ) as Record<Caip19AssetId, AssetBalance>;\n    if (Object.keys(kept).length > 0) {\n      filtered[accountId] = kept;\n    }\n  }\n  return filtered;\n}\n\nfunction collectStaleTrackedAssets(ctx: Context): Caip19AssetId[] {\n  const { assetsBalance: stateAssetsBalance, customAssets: stateCustomAssets } =\n    ctx.getAssetsState();\n\n  const staleAssets = new Set<Caip19AssetId>();\n\n  for (const { account, supportedChains } of ctx.request\n    .accountsWithSupportedChains) {\n    const accountId = account.id;\n    const trackedAssetIds = new Set<Caip19AssetId>([\n      ...(Object.keys(stateAssetsBalance[accountId] ?? {}) as Caip19AssetId[]),\n      ...(stateCustomAssets?.[accountId] ?? []),\n    ]);\n\n    const chainsForAccount = supportedChains.filter((chainId) =>\n      ctx.request.chainIds.includes(chainId),\n    );\n\n    for (const assetId of trackedAssetIds) {\n      if (\n        isEvmAssetOnChains(assetId, chainsForAccount) &&\n        !isStakingContractAssetId(assetId) &&\n        isBalanceEmpty(ctx.response.assetsBalance?.[accountId], assetId)\n      ) {\n        staleAssets.add(assetId);\n      }\n    }\n  }\n\n  return [...staleAssets];\n}\n\nfunction isBalanceEmpty(\n  balances: Record<string, AssetBalance> | undefined,\n  assetId: Caip19AssetId,\n): boolean {\n  let amount = balances?.[assetId]?.amount;\n  if (amount === undefined && balances) {\n    const lowerCasedAssetId = assetId.toLowerCase();\n    amount = Object.entries(balances).find(\n      ([id]) => id.toLowerCase() === lowerCasedAssetId,\n    )?.[1]?.amount;\n  }\n  return !(Number(amount) > 0);\n}\n\nfunction isEvmAssetOnChains(\n  assetId: Caip19AssetId,\n  chainIds: ChainId[],\n): boolean {\n  if (!assetId.startsWith(`${KnownCaipNamespace.Eip155}:`)) {\n    return false;\n  }\n  return chainIds.includes(assetId.split('/')[0] as ChainId);\n}\n"]}
+{"version":3,"file":"RpcFallbackMiddleware.js","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,OAAO,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAU3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,MAAM,eAAe,GAAG,uBAAuB,CAAC;AAEhD,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAO/D;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,qBAAqB;IACvB,IAAI,GAAG,eAAe,CAAC;IAEvB,cAAc,CAAmB;IAE1C,YAAY,OAAqC;QAC/C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;IAC9C,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,gBAAgB;QAClB,OAAO,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACnD,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CACpD,CAAC;YACF,MAAM,WAAW,GAAG,yBAAyB,CAAC,GAAG,CAAC,CAAC;YAEnD,MAAM,aAAa,GAAG;gBACpB,GAAG,IAAI,GAAG,CAAC;oBACT,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAC7D,iEAAiE;oBACjE,oEAAoE;oBACpE,2BAA2B;oBAC3B,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;iBAClE,CAAC;aACH,CAAC;YAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,GAAG,CAAC,4BAA4B,EAAE;gBAChC,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC;gBACjC,WAAW;gBACX,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG;gBACtB,GAAG,GAAG,CAAC,OAAO;gBACd,QAAQ,EAAE,aAAa;gBACvB,YAAY,EAAE;oBACZ,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC;iBAClE;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,KAAK,EAAE,KAAiB,EAAuB,EAAE,CAAC,KAAK,CAAC;YACzE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAC1D;gBACE,GAAG,GAAG;gBACN,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE,EAAE;aACb,EACD,QAAQ,CACT,CAAC;YAEF,oEAAoE;YACpE,qEAAqE;YACrE,mEAAmE;YACnE,uEAAuE;YACvE,0DAA0D;YAC1D,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CAC1D,CAAC;YACF,MAAM,gBAAgB,GAAG,sBAAsB,CAC7C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAChC,eAAe,CAChB,CAAC;YAEF,gEAAgE;YAChE,mEAAmE;YACnE,wEAAwE;YACxE,6DAA6D;YAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAClC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CACnE,aAAa,CAAC,GAAG,CAAC,OAAkB,CAAC,CACtC,CACF,CAAC;YAEF,MAAM,MAAM,GAAiB,kBAAkB,CAAC;gBAC9C,GAAG,CAAC,QAAQ;gBACZ;oBACE,GAAG,SAAS,CAAC,QAAQ;oBACrB,aAAa,EAAE,gBAAgB;oBAC/B,MAAM,EAAE,SAAS;iBAClB;aACF,CAAC,CAAC;YAEH,qEAAqE;YACrE,qEAAqE;YACrE,yEAAyE;YACzE,kEAAkE;YAClE,sDAAsD;YACtD,iEAAiE;YACjE,mEAAmE;YACnE,IAAI,MAAM,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;gBACtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;gBAC/C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBAC9D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;wBACnD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;gBACD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;oBACpC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;wBACtC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;;;;;;;GASG;AACH,SAAS,sBAAsB,CAC7B,aAA4C,EAC5C,QAAsB;IAEtB,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,MAAM,QAAQ,GAA+C,EAAE,CAAC;IAChE,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QACzE,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAC7B,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,MAAM,CACpC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC,CAC/D,CACqC,CAAC;QACzC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,QAAQ,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,yBAAyB,CAAC,GAAY;IAC7C,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,YAAY,EAAE,iBAAiB,EAAE,GAC1E,GAAG,CAAC,cAAc,EAAE,CAAC;IAEvB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAiB,CAAC;IAE7C,KAAK,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,GAAG,CAAC,OAAO;SACnD,2BAA2B,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,IAAI,GAAG,CAAgB;YAC7C,GAAI,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAqB;YACxE,GAAG,CAAC,iBAAiB,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;SAC1C,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CACvC,CAAC;QAEF,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;YACtC,IACE,kBAAkB,CAAC,OAAO,EAAE,gBAAgB,CAAC;gBAC7C,kEAAkE;gBAClE,qDAAqD;gBACrD,CAAC,wBAAwB,CAAC,OAAO,CAAC;gBAClC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,EAChE,CAAC;gBACD,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,WAAW,CAAC,CAAC;AAC1B,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,cAAc,CACrB,QAAkD,EAClD,OAAsB;IAEtB,IAAI,MAAM,GAAG,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IACzC,IAAI,MAAM,KAAK,SAAS,IAAI,QAAQ,EAAE,CAAC;QACrC,MAAM,iBAAiB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAChD,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CACpC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,iBAAiB,CACjD,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;IACjB,CAAC;IACD,OAAO,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACH,SAAS,kBAAkB,CACzB,OAAsB,EACtB,QAAmB;IAEnB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC,CAAC;AAC7D,CAAC","sourcesContent":["import { KnownCaipNamespace } from '@metamask/utils';\n\nimport { isStakingContractAssetId } from '../data-sources/evm-rpc-services/index.js';\nimport { projectLogger, createModuleLogger } from '../logger.js';\nimport { forDataTypes } from '../types.js';\nimport type {\n  AssetBalance,\n  AssetsDataSource,\n  Caip19AssetId,\n  ChainId,\n  Context,\n  DataResponse,\n  Middleware,\n} from '../types.js';\nimport { mergeDataResponses } from './ParallelMiddleware.js';\n\nconst CONTROLLER_NAME = 'RpcFallbackMiddleware';\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\nexport type RpcFallbackMiddlewareOptions = {\n  /** The RPC data source to use as a fallback. */\n  rpcDataSource: AssetsDataSource;\n};\n\n/**\n * RpcFallbackMiddleware re-reads balances from the RPC data source in two\n * cases:\n *\n * 1. Chains present in `response.errors` (network error, unprocessedNetworks,\n *    timeout, …). Successful RPC results are merged into the response and\n *    their entries are cleared from `response.errors`.\n * 2. EVM assets tracked in state (`state.assetsBalance` or\n *    `state.customAssets`) that this balance response left empty. The Accounts\n *    API omits tokens it does not index, and a returned `0` is\n *    indistinguishable from \"not indexed\", so the `merge` state update would\n *    otherwise keep a stale amount forever. They are passed to RPC as\n *    `customAssets` so the balance fetcher includes them in its multicall.\n *\n * Place this immediately after `createParallelBalanceMiddleware` in the fast\n * pipeline, after `CustomAssetGraduationMiddleware` so the custom assets RPC\n * carries never trigger graduation.\n */\nexport class RpcFallbackMiddleware {\n  readonly name = CONTROLLER_NAME;\n\n  readonly #rpcDataSource: AssetsDataSource;\n\n  constructor(options: RpcFallbackMiddlewareOptions) {\n    this.#rpcDataSource = options.rpcDataSource;\n  }\n\n  getName(): string {\n    return this.name;\n  }\n\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['balance'], async (ctx, next) => {\n      const erroredChains = new Set<ChainId>(\n        Object.keys(ctx.response.errors ?? {}) as ChainId[],\n      );\n      const staleAssets = collectStaleTrackedAssets(ctx);\n\n      const chainsToFetch = [\n        ...new Set([\n          ...ctx.request.chainIds.filter((id) => erroredChains.has(id)),\n          // Already restricted to requested chains. Their chain may not be\n          // errored: the Accounts API can answer for a chain while omitting a\n          // token it does not index.\n          ...staleAssets.map((assetId) => assetId.split('/')[0] as ChainId),\n        ]),\n      ];\n\n      if (chainsToFetch.length === 0) {\n        return next(ctx);\n      }\n\n      log('Re-reading balances on RPC', {\n        erroredChains: [...erroredChains],\n        staleAssets,\n        chains: chainsToFetch,\n      });\n\n      const filteredRequest = {\n        ...ctx.request,\n        chainIds: chainsToFetch,\n        customAssets: [\n          ...new Set([...(ctx.request.customAssets ?? []), ...staleAssets]),\n        ],\n      };\n\n      const noopNext = async (inner: typeof ctx): Promise<typeof ctx> => inner;\n      const rpcResult = await this.#rpcDataSource.assetsMiddleware(\n        {\n          ...ctx,\n          request: filteredRequest,\n          response: {},\n        },\n        noopNext,\n      );\n\n      // A chain RPC itself failed on contributed nothing trustworthy: its\n      // balances are failure stubs (native 0) that would overwrite correct\n      // upstream amounts and, with replaceCoveredChainBalances, wipe the\n      // chain's token slice from state. Drop them before merging — this also\n      // keeps failed chains from counting as \"recovered\" below.\n      const rpcFailedChains = new Set<ChainId>(\n        Object.keys(rpcResult.response.errors ?? {}) as ChainId[],\n      );\n      const rpcAssetsBalance = filterOutChainBalances(\n        rpcResult.response.assetsBalance,\n        rpcFailedChains,\n      );\n\n      // RPC errors are kept only for chains that were already errored\n      // upstream. For chains fetched solely for stale tracked assets the\n      // upstream response succeeded and stays authoritative — the stale asset\n      // keeps its previous amount and is retried on the next pass.\n      const rpcErrors = Object.fromEntries(\n        Object.entries(rpcResult.response.errors ?? {}).filter(([chainId]) =>\n          erroredChains.has(chainId as ChainId),\n        ),\n      );\n\n      const merged: DataResponse = mergeDataResponses([\n        ctx.response,\n        {\n          ...rpcResult.response,\n          assetsBalance: rpcAssetsBalance,\n          errors: rpcErrors,\n        },\n      ]);\n\n      // Clear errors only for chains RPC actually recovered a balance for.\n      // We must inspect the (filtered) RPC balances — NOT merged — because\n      // merged also contains balances from the upstream sources (AccountsApi /\n      // Websocket / Staked). If those sources returned partial data for\n      // a chain that they also flagged as errored (e.g. via\n      // unprocessedNetworks), and RPC then failed for that same chain,\n      // looking at merged would incorrectly mark the error as recovered.\n      if (merged.errors && rpcAssetsBalance) {\n        const chainsRecoveredByRpc = new Set<string>();\n        for (const accountBalances of Object.values(rpcAssetsBalance)) {\n          for (const assetId of Object.keys(accountBalances)) {\n            chainsRecoveredByRpc.add(assetId.split('/')[0]);\n          }\n        }\n        for (const chainId of erroredChains) {\n          if (chainsRecoveredByRpc.has(chainId)) {\n            delete merged.errors[chainId];\n          }\n        }\n      }\n\n      return next({ ...ctx, response: merged });\n    });\n  }\n}\n\n/**\n * Remove all balances that belong to the given chains.\n *\n * Used to discard results for chains the RPC source itself failed on, whose\n * entries are failure stubs (native `0`) rather than real readings.\n *\n * @param assetsBalance - Balances by account from the RPC response.\n * @param chainIds - Chains whose balances should be dropped.\n * @returns The filtered balance map, without accounts left empty.\n */\nfunction filterOutChainBalances(\n  assetsBalance: DataResponse['assetsBalance'],\n  chainIds: Set<ChainId>,\n): DataResponse['assetsBalance'] {\n  if (!assetsBalance || chainIds.size === 0) {\n    return assetsBalance;\n  }\n\n  const filtered: NonNullable<DataResponse['assetsBalance']> = {};\n  for (const [accountId, accountBalances] of Object.entries(assetsBalance)) {\n    const kept = Object.fromEntries(\n      Object.entries(accountBalances).filter(\n        ([assetId]) => !chainIds.has(assetId.split('/')[0] as ChainId),\n      ),\n    ) as Record<Caip19AssetId, AssetBalance>;\n    if (Object.keys(kept).length > 0) {\n      filtered[accountId] = kept;\n    }\n  }\n  return filtered;\n}\n\n/**\n * EVM assets tracked in state that this balance response left empty and RPC\n * should re-read.\n *\n * Limited to chains both requested and supported by the owning account:\n * RpcDataSource fetches per account and skips chains outside its supported\n * set, so anything else would be queued and then silently dropped.\n *\n * @param ctx - Pipeline context.\n * @returns Asset IDs to hand to the RPC data source.\n */\nfunction collectStaleTrackedAssets(ctx: Context): Caip19AssetId[] {\n  const { assetsBalance: stateAssetsBalance, customAssets: stateCustomAssets } =\n    ctx.getAssetsState();\n\n  const staleAssets = new Set<Caip19AssetId>();\n\n  for (const { account, supportedChains } of ctx.request\n    .accountsWithSupportedChains) {\n    const accountId = account.id;\n    const trackedAssetIds = new Set<Caip19AssetId>([\n      ...(Object.keys(stateAssetsBalance[accountId] ?? {}) as Caip19AssetId[]),\n      ...(stateCustomAssets?.[accountId] ?? []),\n    ]);\n\n    const chainsForAccount = supportedChains.filter((chainId) =>\n      ctx.request.chainIds.includes(chainId),\n    );\n\n    for (const assetId of trackedAssetIds) {\n      if (\n        isEvmAssetOnChains(assetId, chainsForAccount) &&\n        // Staked vault balances belong to StakedBalanceDataSource; an RPC\n        // ERC-20 read of the share token would clobber them.\n        !isStakingContractAssetId(assetId) &&\n        isBalanceEmpty(ctx.response.assetsBalance?.[accountId], assetId)\n      ) {\n        staleAssets.add(assetId);\n      }\n    }\n  }\n\n  return [...staleAssets];\n}\n\n/**\n * Whether a balance response carries no positive amount for an asset. A\n * returned `0` cannot be distinguished from \"not indexed\", so both count as\n * empty. Asset IDs are matched case-insensitively: state keys ERC-20 assets by\n * checksummed address, while some data sources return them lower-cased.\n *\n * @param balances - Balance map for a single account from the response.\n * @param assetId - Asset ID to check.\n * @returns True when the response holds no positive amount for the asset.\n */\nfunction isBalanceEmpty(\n  balances: Record<string, AssetBalance> | undefined,\n  assetId: Caip19AssetId,\n): boolean {\n  let amount = balances?.[assetId]?.amount;\n  if (amount === undefined && balances) {\n    const lowerCasedAssetId = assetId.toLowerCase();\n    amount = Object.entries(balances).find(\n      ([id]) => id.toLowerCase() === lowerCasedAssetId,\n    )?.[1]?.amount;\n  }\n  return !(Number(amount) > 0);\n}\n\n/**\n * Whether an asset is an EVM asset on one of the given chains.\n *\n * @param assetId - CAIP-19 asset ID.\n * @param chainIds - Chains to match against.\n * @returns True for EVM assets whose chain is in `chainIds`.\n */\nfunction isEvmAssetOnChains(\n  assetId: Caip19AssetId,\n  chainIds: ChainId[],\n): boolean {\n  if (!assetId.startsWith(`${KnownCaipNamespace.Eip155}:`)) {\n    return false;\n  }\n  return chainIds.includes(assetId.split('/')[0] as ChainId);\n}\n"]}

package/dist/pipeline/buildFastFetchSources.d.ts

@@ -34,14 +34,9 @@ export type FastFetchSources = {
  * @param options - Lane options.
  * @param options.isBasicFunctionality - When false, only the staking balance and
  * detection run; no network-backed source is used.
- * @param options.includeCustomAssetGraduation - `true` on the Accounts API v5
- * lane. The v6 lane never graduates custom assets: it sends the pins to the
- * endpoint as `includeAssetIds` and keeps the ones it could not resolve as
- * `unprocessedCustomAssets`.
  * @returns The composed source list, ready for `executeAssetsPipeline`.
  */
 export declare function buildFastFetchSources(sources: FastFetchSources, options: {
     isBasicFunctionality: boolean;
-    includeCustomAssetGraduation: boolean;
 }): AssetsDataSource[];
 //# sourceMappingURL=buildFastFetchSources.d.ts.map

package/dist/pipeline/buildFastFetchSources.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"buildFastFetchSources.d.ts","sourceRoot":"","sources":["../../src/pipeline/buildFastFetchSources.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AAC1E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAEpD;;;;;;GAMG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,qBAAqB,EAAE,aAAa,CAAC;IACrC,uBAAuB,EAAE,aAAa,CAAC;IACvC,+BAA+B,EAAE,gBAAgB,CAAC;IAClD,qBAAqB,EAAE,gBAAgB,CAAC;IACxC,mBAAmB,EAAE,gBAAgB,CAAC;IACtC,eAAe,EAAE,gBAAgB,CAAC;IAClC,eAAe,EAAE,gBAAgB,CAAC;CACnC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,gBAAgB,EACzB,OAAO,EAAE;IACP,oBAAoB,EAAE,OAAO,CAAC;IAC9B,4BAA4B,EAAE,OAAO,CAAC;CACvC,GACA,gBAAgB,EAAE,CA2BpB"}
+{"version":3,"file":"buildFastFetchSources.d.ts","sourceRoot":"","sources":["../../src/pipeline/buildFastFetchSources.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AAC1E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAEpD;;;;;;GAMG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,qBAAqB,EAAE,aAAa,CAAC;IACrC,uBAAuB,EAAE,aAAa,CAAC;IACvC,+BAA+B,EAAE,gBAAgB,CAAC;IAClD,qBAAqB,EAAE,gBAAgB,CAAC;IACxC,mBAAmB,EAAE,gBAAgB,CAAC;IACtC,eAAe,EAAE,gBAAgB,CAAC;IAClC,eAAe,EAAE,gBAAgB,CAAC;CACnC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,gBAAgB,EACzB,OAAO,EAAE;IAAE,oBAAoB,EAAE,OAAO,CAAA;CAAE,GACzC,gBAAgB,EAAE,CAyBpB"}

package/dist/pipeline/buildFastFetchSources.js

@@ -17,10 +17,6 @@ import { createParallelBalanceMiddleware, createParallelMiddleware, } from '../m
  * @param options - Lane options.
  * @param options.isBasicFunctionality - When false, only the staking balance and
  * detection run; no network-backed source is used.
- * @param options.includeCustomAssetGraduation - `true` on the Accounts API v5
- * lane. The v6 lane never graduates custom assets: it sends the pins to the
- * endpoint as `includeAssetIds` and keeps the ones it could not resolve as
- * `unprocessedCustomAssets`.
  * @returns The composed source list, ready for `executeAssetsPipeline`.
  */
 export function buildFastFetchSources(sources, options) {
@@ -33,9 +29,7 @@ export function buildFastFetchSources(sources, options) {
             accountsApiDataSource,
             stakedBalanceDataSource,
         ]),
-        ...(options.includeCustomAssetGraduation
-            ? [customAssetGraduationMiddleware]
-            : []),
+        customAssetGraduationMiddleware,
         rpcFallbackMiddleware,
         detectionMiddleware,
         createParallelMiddleware([tokenDataSource, priceDataSource]),