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

package/dist/middlewares/RpcFallbackMiddleware.mjs

@@ -9,130 +9,190 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _RpcFallbackMiddleware_rpcDataSource;
+var _RpcFallbackMiddleware_instances, _RpcFallbackMiddleware_rpcDataSource, _RpcFallbackMiddleware_isBalanceV6Enabled, _RpcFallbackMiddleware_recoverV6, _RpcFallbackMiddleware_recoverV5, _RpcFallbackMiddleware_recoverErroredChains, _RpcFallbackMiddleware_recoverUnprocessedAssets;
 import { KnownCaipNamespace } from "@metamask/utils";
 import { isStakingContractAssetId } from "../data-sources/evm-rpc-services/index.mjs";
 import { projectLogger, createModuleLogger } from "../logger.mjs";
 import { forDataTypes } from "../types.mjs";
+import { normalizeAssetId } from "../utils/index.mjs";
 import { mergeDataResponses } from "./ParallelMiddleware.mjs";
 const CONTROLLER_NAME = 'RpcFallbackMiddleware';
 const log = createModuleLogger(projectLogger, CONTROLLER_NAME);
+const noopNext = async (ctx) => ctx;
 /**
- * RpcFallbackMiddleware re-reads balances from the RPC data source in two
- * cases:
+ * RpcFallbackMiddleware recovers what upstream sources left outstanding on RPC.
  *
- * 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 v5: chains in `response.errors`, plus EVM assets tracked in
+ * state that this response left empty (the API omits tokens it does not index).
  *
- * Place this immediately after `createParallelBalanceMiddleware` in the fast
- * pipeline, after `CustomAssetGraduationMiddleware` so the custom assets RPC
- * carries never trigger graduation.
+ * Accounts API v6: chains in `response.errors`, plus pins in
+ * `response.unprocessedCustomAssets`.
  */
 export class RpcFallbackMiddleware {
     constructor(options) {
+        _RpcFallbackMiddleware_instances.add(this);
         this.name = CONTROLLER_NAME;
         _RpcFallbackMiddleware_rpcDataSource.set(this, void 0);
+        _RpcFallbackMiddleware_isBalanceV6Enabled.set(this, void 0);
         __classPrivateFieldSet(this, _RpcFallbackMiddleware_rpcDataSource, options.rpcDataSource, "f");
+        __classPrivateFieldSet(this, _RpcFallbackMiddleware_isBalanceV6Enabled, options.isBalanceV6Enabled ?? (() => false), "f");
     }
     getName() {
         return this.name;
     }
     get assetsMiddleware() {
         return forDataTypes(['balance'], async (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);
+            if (__classPrivateFieldGet(this, _RpcFallbackMiddleware_isBalanceV6Enabled, "f").call(this)) {
+                return __classPrivateFieldGet(this, _RpcFallbackMiddleware_instances, "m", _RpcFallbackMiddleware_recoverV6).call(this, ctx, next);
             }
-            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 __classPrivateFieldGet(this, _RpcFallbackMiddleware_rpcDataSource, "f").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];
-                    }
-                }
-            }
-            return next({ ...ctx, response: merged });
+            return __classPrivateFieldGet(this, _RpcFallbackMiddleware_instances, "m", _RpcFallbackMiddleware_recoverV5).call(this, ctx, next);
         });
     }
 }
-_RpcFallbackMiddleware_rpcDataSource = new WeakMap();
-/**
- * Remove all balances that belong to the given chains.
- *
- * Used to discard results for chains the RPC source itself failed on, whose
- * entries are failure stubs (native `0`) rather than real readings.
- *
- * @param assetsBalance - Balances by account from the RPC response.
- * @param chainIds - Chains whose balances should be dropped.
- * @returns The filtered balance map, without accounts left empty.
- */
+_RpcFallbackMiddleware_rpcDataSource = new WeakMap(), _RpcFallbackMiddleware_isBalanceV6Enabled = new WeakMap(), _RpcFallbackMiddleware_instances = new WeakSet(), _RpcFallbackMiddleware_recoverV6 = async function _RpcFallbackMiddleware_recoverV6(ctx, next) {
+    const erroredChains = new Set(Object.keys(ctx.response.errors ?? {}));
+    const unprocessedCustomAssets = [
+        ...new Set(ctx.response.unprocessedCustomAssets ?? []),
+    ];
+    if (erroredChains.size === 0 && unprocessedCustomAssets.length === 0) {
+        return next(ctx);
+    }
+    let merged = ctx.response;
+    if (erroredChains.size > 0) {
+        merged = await __classPrivateFieldGet(this, _RpcFallbackMiddleware_instances, "m", _RpcFallbackMiddleware_recoverErroredChains).call(this, ctx, merged, erroredChains);
+    }
+    const assetsToRecover = unprocessedCustomAssets.filter((assetId) => !erroredChains.has(chainIdOfAsset(assetId)));
+    if (assetsToRecover.length > 0) {
+        merged = await __classPrivateFieldGet(this, _RpcFallbackMiddleware_instances, "m", _RpcFallbackMiddleware_recoverUnprocessedAssets).call(this, ctx, merged, assetsToRecover);
+    }
+    merged = clearRecoveredAssetIds(merged);
+    return next({ ...ctx, response: merged });
+}, _RpcFallbackMiddleware_recoverV5 = async function _RpcFallbackMiddleware_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,
+    });
+    const filteredRequest = {
+        ...ctx.request,
+        chainIds: chainsToFetch,
+        customAssets: [
+            ...new Set([...(ctx.request.customAssets ?? []), ...staleAssets]),
+        ],
+    };
+    const rpcResult = await __classPrivateFieldGet(this, _RpcFallbackMiddleware_rpcDataSource, "f").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 });
+}, _RpcFallbackMiddleware_recoverErroredChains = async function _RpcFallbackMiddleware_recoverErroredChains(ctx, currentResponse, erroredChains) {
+    log('Retrying failed chains on RPC', { chains: [...erroredChains] });
+    const chainRequest = {
+        ...ctx.request,
+        chainIds: ctx.request.chainIds.filter((id) => erroredChains.has(id)),
+    };
+    const rpcResult = await __classPrivateFieldGet(this, _RpcFallbackMiddleware_rpcDataSource, "f").assetsMiddleware({ ...ctx, request: chainRequest, response: {} }, noopNext);
+    const rpcFailedChains = new Set(Object.keys(rpcResult.response.errors ?? {}));
+    const rpcAssetsBalance = filterOutChainBalances(rpcResult.response.assetsBalance, rpcFailedChains);
+    const merged = mergeDataResponses([
+        currentResponse,
+        {
+            ...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]);
+            }
+        }
+        for (const chainId of erroredChains) {
+            if (chainsRecoveredByRpc.has(chainId)) {
+                delete merged.errors[chainId];
+            }
+        }
+    }
+    return merged;
+}, _RpcFallbackMiddleware_recoverUnprocessedAssets = async function _RpcFallbackMiddleware_recoverUnprocessedAssets(ctx, currentResponse, assetsToRecover) {
+    const assetChains = [
+        ...new Set(assetsToRecover.map((assetId) => chainIdOfAsset(assetId))),
+    ];
+    log('Recovering unprocessed pinned assets on RPC', {
+        assetIds: assetsToRecover,
+    });
+    const assetRequest = {
+        ...ctx.request,
+        chainIds: assetChains,
+        customAssets: assetsToRecover,
+    };
+    const rpcResult = await __classPrivateFieldGet(this, _RpcFallbackMiddleware_rpcDataSource, "f").assetsMiddleware({ ...ctx, request: assetRequest, response: {} }, noopNext);
+    return mergeDataResponses([currentResponse, rpcResult.response]);
+};
+function chainIdOfAsset(assetId) {
+    return assetId.split('/')[0];
+}
+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;
@@ -146,17 +206,6 @@ 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();
@@ -170,8 +219,6 @@ 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);
@@ -180,16 +227,6 @@ 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) {
@@ -198,13 +235,6 @@ 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.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"RpcFallbackMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAErD,OAAO,EAAE,wBAAwB,EAAE,mDAAkD;AACrF,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAqB;AACjE,OAAO,EAAE,YAAY,EAAE,qBAAoB;AAU3C,OAAO,EAAE,kBAAkB,EAAE,iCAAgC;AAE7D,MAAM,eAAe,GAAG,uBAAuB,CAAC;AAEhD,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAO/D;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,qBAAqB;IAKhC,YAAY,OAAqC;QAJxC,SAAI,GAAG,eAAe,CAAC;QAEvB,uDAAiC;QAGxC,uBAAA,IAAI,wCAAkB,OAAO,CAAC,aAAa,MAAA,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,uBAAA,IAAI,4CAAe,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"]}
+{"version":3,"file":"RpcFallbackMiddleware.mjs","sourceRoot":"","sources":["../../src/middlewares/RpcFallbackMiddleware.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAErD,OAAO,EAAE,wBAAwB,EAAE,mDAAkD;AACrF,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAqB;AACjE,OAAO,EAAE,YAAY,EAAE,qBAAoB;AAY3C,OAAO,EAAE,gBAAgB,EAAE,2BAA0B;AACrD,OAAO,EAAE,kBAAkB,EAAE,iCAAgC;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;;;;;;;;GAQG;AACH,MAAM,OAAO,qBAAqB;IAOhC,YAAY,OAAqC;;QANxC,SAAI,GAAG,eAAe,CAAC;QAEvB,uDAAiC;QAEjC,4DAAmC;QAG1C,uBAAA,IAAI,wCAAkB,OAAO,CAAC,aAAa,MAAA,CAAC;QAC5C,uBAAA,IAAI,6CACF,OAAO,CAAC,kBAAkB,IAAI,CAAC,GAAY,EAAE,CAAC,KAAK,CAAC,MAAA,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,uBAAA,IAAI,iDAAoB,MAAxB,IAAI,CAAsB,EAAE,CAAC;gBAC/B,OAAO,uBAAA,IAAI,0EAAW,MAAf,IAAI,EAAY,GAAG,EAAE,IAAI,CAAC,CAAC;YACpC,CAAC;YACD,OAAO,uBAAA,IAAI,0EAAW,MAAf,IAAI,EAAY,GAAG,EAAE,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;CA8LF;sMA5LC,KAAK,2CAAY,GAAY,EAAE,IAAkB;IAC/C,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CACpD,CAAC;IACF,MAAM,uBAAuB,GAAG;QAC9B,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,uBAAuB,IAAI,EAAE,CAAC;KACvD,CAAC;IAEF,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC,IAAI,uBAAuB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,IAAI,MAAM,GAAiB,GAAG,CAAC,QAAQ,CAAC;IAExC,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,MAAM,uBAAA,IAAI,qFAAsB,MAA1B,IAAI,EAAuB,GAAG,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,eAAe,GAAG,uBAAuB,CAAC,MAAM,CACpD,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CACzD,CAAC;IACF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,GAAG,MAAM,uBAAA,IAAI,yFAA0B,MAA9B,IAAI,EACjB,GAAG,EACH,MAAM,EACN,eAAe,CAChB,CAAC;IACJ,CAAC;IAED,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAExC,OAAO,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC,qCAED,KAAK,2CAAY,GAAY,EAAE,IAAkB;IAC/C,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CACpD,CAAC;IACF,MAAM,WAAW,GAAG,yBAAyB,CAAC,GAAG,CAAC,CAAC;IAEnD,MAAM,aAAa,GAAG;QACpB,GAAG,IAAI,GAAG,CAAC;YACT,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7D,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;SAClE,CAAC;KACH,CAAC;IAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,GAAG,CAAC,4BAA4B,EAAE;QAChC,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC;QACjC,WAAW;QACX,MAAM,EAAE,aAAa;KACtB,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG;QACtB,GAAG,GAAG,CAAC,OAAO;QACd,QAAQ,EAAE,aAAa;QACvB,YAAY,EAAE;YACZ,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC;SAClE;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,4CAAe,CAAC,gBAAgB,CAC1D;QACE,GAAG,GAAG;QACN,OAAO,EAAE,eAAe;QACxB,QAAQ,EAAE,EAAE;KACb,EACD,QAAQ,CACT,CAAC;IAEF,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CAC1D,CAAC;IACF,MAAM,gBAAgB,GAAG,sBAAsB,CAC7C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAChC,eAAe,CAChB,CAAC;IAEF,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;IAEF,MAAM,MAAM,GAAiB,kBAAkB,CAAC;QAC9C,GAAG,CAAC,QAAQ;QACZ;YACE,GAAG,SAAS,CAAC,QAAQ;YACrB,aAAa,EAAE,gBAAgB;YAC/B,MAAM,EAAE,SAAS;SAClB;KACF,CAAC,CAAC;IAEH,IAAI,MAAM,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;QACtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC9D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;gBACnD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;YACpC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC,gDAED,KAAK,sDACH,GAAY,EACZ,eAA6B,EAC7B,aAA2B;IAE3B,GAAG,CAAC,+BAA+B,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;IAErE,MAAM,YAAY,GAAgB;QAChC,GAAG,GAAG,CAAC,OAAO;QACd,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;KACrE,CAAC;IACF,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,4CAAe,CAAC,gBAAgB,CAC1D,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,EAC/C,QAAQ,CACT,CAAC;IAEF,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAc,CAC1D,CAAC;IACF,MAAM,gBAAgB,GAAG,sBAAsB,CAC7C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAChC,eAAe,CAChB,CAAC;IAEF,MAAM,MAAM,GAAG,kBAAkB,CAAC;QAChC,eAAe;QACf;YACE,GAAG,SAAS,CAAC,QAAQ;YACrB,aAAa,EAAE,gBAAgB;SAChC;KACF,CAAC,CAAC;IAEH,IAAI,MAAM,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;QACtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC9D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;gBACnD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;YACpC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtC,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC,oDAED,KAAK,0DACH,GAAY,EACZ,eAA6B,EAC7B,eAAgC;IAEhC,MAAM,WAAW,GAAG;QAClB,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC;KACtE,CAAC;IAEF,GAAG,CAAC,6CAA6C,EAAE;QACjD,QAAQ,EAAE,eAAe;KAC1B,CAAC,CAAC;IAEH,MAAM,YAAY,GAAgB;QAChC,GAAG,GAAG,CAAC,OAAO;QACd,QAAQ,EAAE,WAAW;QACrB,YAAY,EAAE,eAAe;KAC9B,CAAC;IACF,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,4CAAe,CAAC,gBAAgB,CAC1D,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,EAC/C,QAAQ,CACT,CAAC;IAEF,OAAO,kBAAkB,CAAC,CAAC,eAAe,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;AACnE,CAAC;AAGH,SAAS,cAAc,CAAC,OAAsB;IAC5C,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;AAC1C,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: chains in `response.errors`, plus pins in\n * `response.unprocessedCustomAssets`.\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 unprocessedCustomAssets = [\n      ...new Set(ctx.response.unprocessedCustomAssets ?? []),\n    ];\n\n    if (erroredChains.size === 0 && unprocessedCustomAssets.length === 0) {\n      return next(ctx);\n    }\n\n    let merged: DataResponse = ctx.response;\n\n    if (erroredChains.size > 0) {\n      merged = await this.#recoverErroredChains(ctx, merged, erroredChains);\n    }\n\n    const assetsToRecover = unprocessedCustomAssets.filter(\n      (assetId) => !erroredChains.has(chainIdOfAsset(assetId)),\n    );\n    if (assetsToRecover.length > 0) {\n      merged = await this.#recoverUnprocessedAssets(\n        ctx,\n        merged,\n        assetsToRecover,\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  async #recoverErroredChains(\n    ctx: Context,\n    currentResponse: DataResponse,\n    erroredChains: Set<ChainId>,\n  ): Promise<DataResponse> {\n    log('Retrying failed chains on RPC', { chains: [...erroredChains] });\n\n    const chainRequest: DataRequest = {\n      ...ctx.request,\n      chainIds: ctx.request.chainIds.filter((id) => erroredChains.has(id)),\n    };\n    const rpcResult = await this.#rpcDataSource.assetsMiddleware(\n      { ...ctx, request: chainRequest, 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    const merged = mergeDataResponses([\n      currentResponse,\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    return merged;\n  }\n\n  async #recoverUnprocessedAssets(\n    ctx: Context,\n    currentResponse: DataResponse,\n    assetsToRecover: Caip19AssetId[],\n  ): Promise<DataResponse> {\n    const assetChains = [\n      ...new Set(assetsToRecover.map((assetId) => chainIdOfAsset(assetId))),\n    ];\n\n    log('Recovering unprocessed pinned assets on RPC', {\n      assetIds: assetsToRecover,\n    });\n\n    const assetRequest: DataRequest = {\n      ...ctx.request,\n      chainIds: assetChains,\n      customAssets: assetsToRecover,\n    };\n    const rpcResult = await this.#rpcDataSource.assetsMiddleware(\n      { ...ctx, request: assetRequest, response: {} },\n      noopNext,\n    );\n\n    return mergeDataResponses([currentResponse, rpcResult.response]);\n  }\n}\n\nfunction chainIdOfAsset(assetId: Caip19AssetId): ChainId {\n  return assetId.split('/')[0] as ChainId;\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"]}

package/dist/selectors/balance.cjs

@@ -135,18 +135,13 @@ function mergeBalancesIntoMap(args) {
         if (assetPreferences?.[typedAssetId]?.hidden) {
             continue;
         }
-        // Balances without a matching `assetsInfo` entry are not renderable
-        // (no symbol/name/decimals), so they are excluded from aggregation.
-        const meta = metadata[typedAssetId];
-        if (!meta) {
-            continue;
-        }
         const info = getAssetInfo(assetInfoCache, typedAssetId);
         if (!isChainEnabledByMap(enabledNetworkMap, info.chainId)) {
             continue;
         }
         const amountStr = getAmountFromBalance(accountBalances[typedAssetId]);
         const amountBn = toBigNumberOrZero(amountStr);
+        const meta = metadata[typedAssetId];
         if (amountBn.isZero()) {
             continue; // skip zeros early to reduce map pressure
         }
@@ -157,9 +152,9 @@ function mergeBalancesIntoMap(args) {
         }
         out.set(typedAssetId, {
             amount: amountBn,
-            decimals: meta.decimals,
-            symbol: meta.symbol,
-            name: meta.name,
+            decimals: meta?.decimals,
+            symbol: meta?.symbol,
+            name: meta?.name,
         });
     }
 }