@metamask/assets-controller 1.0.0 → 2.0.1

package/dist/data-sources/RpcDataSource.mjs

@@ -9,10 +9,10 @@ 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 _RpcDataSource_instances, _RpcDataSource_messenger, _RpcDataSource_onActiveChainsUpdated, _RpcDataSource_timeout, _RpcDataSource_tokenDetectionEnabled, _RpcDataSource_activeChains, _RpcDataSource_chainStatuses, _RpcDataSource_providerCache, _RpcDataSource_activeSubscriptions, _RpcDataSource_multicallClient, _RpcDataSource_balanceFetcher, _RpcDataSource_tokenDetector, _RpcDataSource_convertToHumanReadable, _RpcDataSource_collectMetadataForBalances, _RpcDataSource_handleBalanceUpdate, _RpcDataSource_handleDetectionUpdate, _RpcDataSource_subscribeToNetworkController, _RpcDataSource_initializeFromNetworkController, _RpcDataSource_updateFromNetworkState, _RpcDataSource_getProvider, _RpcDataSource_getMulticallProvider, _RpcDataSource_clearProviderCache, _RpcDataSource_buildNativeAssetId, _RpcDataSource_getExistingAssetsMetadata, _RpcDataSource_getTokenMetadataFromTokenList;
+var _RpcDataSource_instances, _RpcDataSource_messenger, _RpcDataSource_onActiveChainsUpdated, _RpcDataSource_timeout, _RpcDataSource_tokenDetectionEnabled, _RpcDataSource_useExternalService, _RpcDataSource_activeChains, _RpcDataSource_chainStatuses, _RpcDataSource_providerCache, _RpcDataSource_activeSubscriptions, _RpcDataSource_unsubscribeTransactionConfirmed, _RpcDataSource_unsubscribeIncomingTransactions, _RpcDataSource_multicallClient, _RpcDataSource_balanceFetcher, _RpcDataSource_tokenDetector, _RpcDataSource_convertToHumanReadable, _RpcDataSource_collectMetadataForBalances, _RpcDataSource_handleBalanceUpdate, _RpcDataSource_handleDetectionUpdate, _RpcDataSource_subscribeToNetworkController, _RpcDataSource_subscribeToTransactionEvents, _RpcDataSource_onTransactionConfirmed, _RpcDataSource_onIncomingTransactions, _RpcDataSource_refreshBalanceForChains, _RpcDataSource_initializeFromNetworkController, _RpcDataSource_updateFromNetworkState, _RpcDataSource_getProvider, _RpcDataSource_getMulticallProvider, _RpcDataSource_clearProviderCache, _RpcDataSource_buildNativeAssetId, _RpcDataSource_getExistingAssetsMetadata, _RpcDataSource_getTokenMetadataFromTokenList;
 import { Web3Provider } from "@ethersproject/providers";
 import { toHex } from "@metamask/controller-utils";
-import { isStrictHexString, isCaipChainId, parseCaipChainId } from "@metamask/utils";
+import { isStrictHexString, isCaipChainId, numberToHex, parseCaipAssetType, parseCaipChainId } from "@metamask/utils";
 import BigNumberJS from "bignumber.js";
 import { AbstractDataSource } from "./AbstractDataSource.mjs";
 import { BalanceFetcher, MulticallClient, TokenDetector } from "./evm-rpc-services/index.mjs";
@@ -62,6 +62,7 @@ export class RpcDataSource extends AbstractDataSource {
         _RpcDataSource_onActiveChainsUpdated.set(this, void 0);
         _RpcDataSource_timeout.set(this, void 0);
         _RpcDataSource_tokenDetectionEnabled.set(this, void 0);
+        _RpcDataSource_useExternalService.set(this, void 0);
         /** Currently active chains */
         _RpcDataSource_activeChains.set(this, []);
         /** Network status for each active chain */
@@ -70,6 +71,8 @@ export class RpcDataSource extends AbstractDataSource {
         _RpcDataSource_providerCache.set(this, new Map());
         /** Active subscriptions by ID */
         _RpcDataSource_activeSubscriptions.set(this, new Map());
+        _RpcDataSource_unsubscribeTransactionConfirmed.set(this, undefined);
+        _RpcDataSource_unsubscribeIncomingTransactions.set(this, undefined);
         // Rpc-datasource components
         _RpcDataSource_multicallClient.set(this, void 0);
         _RpcDataSource_balanceFetcher.set(this, void 0);
@@ -77,14 +80,16 @@ export class RpcDataSource extends AbstractDataSource {
         __classPrivateFieldSet(this, _RpcDataSource_messenger, options.messenger, "f");
         __classPrivateFieldSet(this, _RpcDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
         __classPrivateFieldSet(this, _RpcDataSource_timeout, options.timeout ?? 10000, "f");
-        __classPrivateFieldSet(this, _RpcDataSource_tokenDetectionEnabled, options.tokenDetectionEnabled ?? false, "f");
+        __classPrivateFieldSet(this, _RpcDataSource_tokenDetectionEnabled, options.tokenDetectionEnabled ?? (() => true), "f");
+        __classPrivateFieldSet(this, _RpcDataSource_useExternalService, options.useExternalService ?? (() => true), "f");
         const balanceInterval = options.balanceInterval ?? DEFAULT_BALANCE_INTERVAL;
         const detectionInterval = options.detectionInterval ?? DEFAULT_DETECTION_INTERVAL;
         log('Initializing RpcDataSource', {
             timeout: __classPrivateFieldGet(this, _RpcDataSource_timeout, "f"),
             balanceInterval,
             detectionInterval,
-            tokenDetectionEnabled: __classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f"),
+            tokenDetectionEnabled: __classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f").call(this),
+            useExternalService: __classPrivateFieldGet(this, _RpcDataSource_useExternalService, "f").call(this),
         });
         // Initialize MulticallClient with a provider getter
         __classPrivateFieldSet(this, _RpcDataSource_multicallClient, new MulticallClient((hexChainId) => {
@@ -108,9 +113,14 @@ export class RpcDataSource extends AbstractDataSource {
         __classPrivateFieldSet(this, _RpcDataSource_balanceFetcher, new BalanceFetcher(__classPrivateFieldGet(this, _RpcDataSource_multicallClient, "f"), balanceFetcherMessenger, { pollingInterval: balanceInterval }), "f");
         __classPrivateFieldGet(this, _RpcDataSource_balanceFetcher, "f").setOnBalanceUpdate(__classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_handleBalanceUpdate).bind(this));
         // Initialize TokenDetector with polling interval
-        __classPrivateFieldSet(this, _RpcDataSource_tokenDetector, new TokenDetector(__classPrivateFieldGet(this, _RpcDataSource_multicallClient, "f"), tokenDetectorMessenger, { pollingInterval: detectionInterval }), "f");
+        __classPrivateFieldSet(this, _RpcDataSource_tokenDetector, new TokenDetector(__classPrivateFieldGet(this, _RpcDataSource_multicallClient, "f"), tokenDetectorMessenger, {
+            pollingInterval: detectionInterval,
+            tokenDetectionEnabled: __classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f"),
+            useExternalService: __classPrivateFieldGet(this, _RpcDataSource_useExternalService, "f"),
+        }), "f");
         __classPrivateFieldGet(this, _RpcDataSource_tokenDetector, "f").setOnDetectionUpdate(__classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_handleDetectionUpdate).bind(this));
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_subscribeToNetworkController).call(this);
+        __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_subscribeToTransactionEvents).call(this);
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_initializeFromNetworkController).call(this);
     }
     /**
@@ -182,7 +192,7 @@ export class RpcDataSource extends AbstractDataSource {
             return response;
         }
         const assetsBalance = {};
-        const assetsMetadata = {};
+        const assetsInfo = {};
         const failedChains = [];
         // Fetch balances for each account and its supported chains (pre-computed in request)
         for (const { account, supportedChains, } of request.accountsWithSupportedChains) {
@@ -202,10 +212,10 @@ export class RpcDataSource extends AbstractDataSource {
                     }
                     // Collect metadata for all balances
                     const balanceMetadata = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_collectMetadataForBalances).call(this, result.balances, chainId);
-                    Object.assign(assetsMetadata, balanceMetadata);
+                    Object.assign(assetsInfo, balanceMetadata);
                     // Convert balances to human-readable format
                     for (const balance of result.balances) {
-                        const metadata = assetsMetadata[balance.assetId];
+                        const metadata = assetsInfo[balance.assetId];
                         // Default to 18 decimals (ERC20 standard) for consistent human-readable format
                         const decimals = metadata?.decimals ?? 18;
                         const humanReadableAmount = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_convertToHumanReadable).call(this, balance.balance, decimals);
@@ -224,7 +234,7 @@ export class RpcDataSource extends AbstractDataSource {
                     // Even on error, include native token metadata
                     const chainStatus = __classPrivateFieldGet(this, _RpcDataSource_chainStatuses, "f")[chainId];
                     if (chainStatus) {
-                        assetsMetadata[nativeAssetId] = {
+                        assetsInfo[nativeAssetId] = {
                             type: 'native',
                             symbol: chainStatus.nativeCurrency,
                             name: chainStatus.nativeCurrency,
@@ -255,8 +265,8 @@ export class RpcDataSource extends AbstractDataSource {
         }
         response.assetsBalance = assetsBalance;
         // Include metadata for native tokens if we have any
-        if (Object.keys(assetsMetadata).length > 0) {
-            response.assetsMetadata = assetsMetadata;
+        if (Object.keys(assetsInfo).length > 0) {
+            response.assetsInfo = assetsInfo;
         }
         return response;
     }
@@ -268,14 +278,17 @@ export class RpcDataSource extends AbstractDataSource {
      * @returns Promise resolving to a DataResponse with detected assets.
      */
     async detectTokens(chainId, account) {
-        if (!__classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f")) {
+        if (!__classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f").call(this) || !__classPrivateFieldGet(this, _RpcDataSource_useExternalService, "f").call(this)) {
             return {};
         }
         const hexChainId = caipChainIdToHex(chainId);
         const { address, id: accountId } = account;
         log('Running token detection', { chainId, accountId });
         try {
-            const result = await __classPrivateFieldGet(this, _RpcDataSource_tokenDetector, "f").detectTokens(hexChainId, accountId, address);
+            const result = await __classPrivateFieldGet(this, _RpcDataSource_tokenDetector, "f").detectTokens(hexChainId, accountId, address, {
+                tokenDetectionEnabled: __classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f").call(this),
+                useExternalService: __classPrivateFieldGet(this, _RpcDataSource_useExternalService, "f").call(this),
+            });
             if (result.detectedAssets.length === 0) {
                 log('No new tokens detected');
                 return {};
@@ -287,11 +300,11 @@ export class RpcDataSource extends AbstractDataSource {
             });
             // Convert detected assets to DataResponse format
             const balances = {};
-            const assetsMetadata = {};
+            const assetsInfo = {};
             // Build metadata from detected assets
             for (const asset of result.detectedAssets) {
                 if (asset.symbol && asset.decimals !== undefined) {
-                    assetsMetadata[asset.assetId] = {
+                    assetsInfo[asset.assetId] = {
                         type: 'erc20',
                         symbol: asset.symbol,
                         name: asset.name ?? asset.symbol,
@@ -319,8 +332,8 @@ export class RpcDataSource extends AbstractDataSource {
                 },
             };
             // Include metadata if we have any
-            if (Object.keys(assetsMetadata).length > 0) {
-                response.assetsMetadata = assetsMetadata;
+            if (Object.keys(assetsInfo).length > 0) {
+                response.assetsInfo = assetsInfo;
             }
             return response;
         }
@@ -356,11 +369,11 @@ export class RpcDataSource extends AbstractDataSource {
                     };
                 }
             }
-            if (response.assetsMetadata) {
-                (_c = context.response).assetsMetadata ?? (_c.assetsMetadata = {});
-                context.response.assetsMetadata = {
-                    ...context.response.assetsMetadata,
-                    ...response.assetsMetadata,
+            if (response.assetsInfo) {
+                (_c = context.response).assetsInfo ?? (_c.assetsInfo = {});
+                context.response.assetsInfo = {
+                    ...context.response.assetsInfo,
+                    ...response.assetsInfo,
                 };
             }
             const failedChains = new Set(Object.keys(response.errors ?? {}));
@@ -430,8 +443,8 @@ export class RpcDataSource extends AbstractDataSource {
                 };
                 const balanceToken = __classPrivateFieldGet(this, _RpcDataSource_balanceFetcher, "f").startPolling(balanceInput);
                 balancePollingTokens.push(balanceToken);
-                // Start detection polling if enabled
-                if (__classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f")) {
+                // Start detection polling if enabled and external services allowed
+                if (__classPrivateFieldGet(this, _RpcDataSource_tokenDetectionEnabled, "f").call(this) && __classPrivateFieldGet(this, _RpcDataSource_useExternalService, "f").call(this)) {
                     const detectionInput = {
                         chainId: hexChainId,
                         accountId,
@@ -483,6 +496,8 @@ export class RpcDataSource extends AbstractDataSource {
      */
     destroy() {
         log('Destroying RpcDataSource');
+        __classPrivateFieldGet(this, _RpcDataSource_unsubscribeTransactionConfirmed, "f")?.call(this);
+        __classPrivateFieldGet(this, _RpcDataSource_unsubscribeIncomingTransactions, "f")?.call(this);
         // Stop all polling
         __classPrivateFieldGet(this, _RpcDataSource_balanceFetcher, "f").stopAllPolling();
         __classPrivateFieldGet(this, _RpcDataSource_tokenDetector, "f").stopAllPolling();
@@ -492,19 +507,19 @@ export class RpcDataSource extends AbstractDataSource {
         __classPrivateFieldGet(this, _RpcDataSource_providerCache, "f").clear();
     }
 }
-_RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated = new WeakMap(), _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = new WeakMap(), _RpcDataSource_activeChains = new WeakMap(), _RpcDataSource_chainStatuses = new WeakMap(), _RpcDataSource_providerCache = new WeakMap(), _RpcDataSource_activeSubscriptions = new WeakMap(), _RpcDataSource_multicallClient = new WeakMap(), _RpcDataSource_balanceFetcher = new WeakMap(), _RpcDataSource_tokenDetector = new WeakMap(), _RpcDataSource_instances = new WeakSet(), _RpcDataSource_convertToHumanReadable = function _RpcDataSource_convertToHumanReadable(rawBalance, decimals) {
+_RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated = new WeakMap(), _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = new WeakMap(), _RpcDataSource_useExternalService = new WeakMap(), _RpcDataSource_activeChains = new WeakMap(), _RpcDataSource_chainStatuses = new WeakMap(), _RpcDataSource_providerCache = new WeakMap(), _RpcDataSource_activeSubscriptions = new WeakMap(), _RpcDataSource_unsubscribeTransactionConfirmed = new WeakMap(), _RpcDataSource_unsubscribeIncomingTransactions = new WeakMap(), _RpcDataSource_multicallClient = new WeakMap(), _RpcDataSource_balanceFetcher = new WeakMap(), _RpcDataSource_tokenDetector = new WeakMap(), _RpcDataSource_instances = new WeakSet(), _RpcDataSource_convertToHumanReadable = function _RpcDataSource_convertToHumanReadable(rawBalance, decimals) {
     const rawAmount = new BigNumberJS(rawBalance);
     const divisor = new BigNumberJS(10).pow(decimals);
     return rawAmount.dividedBy(divisor).toString();
 }, _RpcDataSource_collectMetadataForBalances = function _RpcDataSource_collectMetadataForBalances(balances, chainId) {
-    const assetsMetadata = {};
+    const assetsInfo = {};
     const existingMetadata = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_getExistingAssetsMetadata).call(this);
     for (const balance of balances) {
         const isNative = balance.assetId.includes('/slip44:');
         if (isNative) {
             const chainStatus = __classPrivateFieldGet(this, _RpcDataSource_chainStatuses, "f")[chainId];
             if (chainStatus) {
-                assetsMetadata[balance.assetId] = {
+                assetsInfo[balance.assetId] = {
                     type: 'native',
                     symbol: chainStatus.nativeCurrency,
                     name: chainStatus.nativeCurrency,
@@ -516,19 +531,19 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
             // For ERC20 tokens, try existing metadata from state first
             const existingMeta = existingMetadata[balance.assetId];
             if (existingMeta) {
-                assetsMetadata[balance.assetId] = existingMeta;
+                assetsInfo[balance.assetId] = existingMeta;
             }
             else {
                 // Fallback to token list if not in state
                 const tokenListMeta = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_getTokenMetadataFromTokenList).call(this, balance.assetId);
                 if (tokenListMeta) {
-                    assetsMetadata[balance.assetId] = tokenListMeta;
+                    assetsInfo[balance.assetId] = tokenListMeta;
                 }
                 else {
                     // Default metadata for unknown ERC20 tokens.
                     // Use 18 decimals (the standard for most ERC20 tokens)
                     // to ensure consistent human-readable balance format.
-                    assetsMetadata[balance.assetId] = {
+                    assetsInfo[balance.assetId] = {
                         type: 'erc20',
                         symbol: '',
                         name: '',
@@ -538,17 +553,17 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
             }
         }
     }
-    return assetsMetadata;
+    return assetsInfo;
 }, _RpcDataSource_handleBalanceUpdate = function _RpcDataSource_handleBalanceUpdate(result) {
     const newBalances = {};
     // Convert hex chain ID to CAIP-2 format
     const chainIdDecimal = parseInt(result.chainId, 16);
     const caipChainId = `eip155:${chainIdDecimal}`;
     // Collect metadata for all balances
-    const assetsMetadata = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_collectMetadataForBalances).call(this, result.balances, caipChainId);
+    const assetsInfo = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_collectMetadataForBalances).call(this, result.balances, caipChainId);
     // Convert balances to human-readable format using metadata
     for (const balance of result.balances) {
-        const metadata = assetsMetadata[balance.assetId];
+        const metadata = assetsInfo[balance.assetId];
         // Default to 18 decimals (ERC20 standard) for consistent human-readable format
         const decimals = metadata?.decimals ?? 18;
         const humanReadableAmount = __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_convertToHumanReadable).call(this, balance.balance, decimals);
@@ -562,7 +577,7 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
         assetsBalance: {
             [result.accountId]: newBalances,
         },
-        assetsMetadata,
+        assetsInfo,
     };
     log('Balance update response', {
         accountId: result.accountId,
@@ -613,7 +628,7 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
         detectedAssets: {
             [result.accountId]: result.detectedAssets.map((asset) => asset.assetId),
         },
-        assetsMetadata: newMetadata,
+        assetsInfo: newMetadata,
         assetsBalance: {
             [result.accountId]: newBalances,
         },
@@ -629,6 +644,73 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_clearProviderCache).call(this);
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_updateFromNetworkState).call(this, networkState);
     });
+}, _RpcDataSource_subscribeToTransactionEvents = function _RpcDataSource_subscribeToTransactionEvents() {
+    const unsubConfirmed = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").subscribe('TransactionController:transactionConfirmed', __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_onTransactionConfirmed).bind(this));
+    __classPrivateFieldSet(this, _RpcDataSource_unsubscribeTransactionConfirmed, typeof unsubConfirmed === 'function' ? unsubConfirmed : undefined, "f");
+    const unsubIncoming = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").subscribe('TransactionController:incomingTransactionsReceived', __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_onIncomingTransactions).bind(this));
+    __classPrivateFieldSet(this, _RpcDataSource_unsubscribeIncomingTransactions, typeof unsubIncoming === 'function' ? unsubIncoming : undefined, "f");
+}, _RpcDataSource_onTransactionConfirmed = function _RpcDataSource_onTransactionConfirmed(payload) {
+    const hexChainId = payload?.chainId;
+    if (!hexChainId) {
+        return;
+    }
+    const caipChainId = `eip155:${parseInt(hexChainId, 16)}`;
+    __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_refreshBalanceForChains).call(this, [caipChainId]).catch((error) => {
+        log('Failed to refresh balance after transaction confirmed', { error });
+    });
+}, _RpcDataSource_onIncomingTransactions = function _RpcDataSource_onIncomingTransactions(payload) {
+    const chainIds = Array.from(new Set((payload ?? [])
+        .map((item) => item?.chainId)
+        .filter((id) => Boolean(id))));
+    const caipChainIds = chainIds.map((hexChainId) => `eip155:${parseInt(hexChainId, 16)}`);
+    const toRefresh = caipChainIds.length > 0 ? caipChainIds : [...__classPrivateFieldGet(this, _RpcDataSource_activeChains, "f")];
+    __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_refreshBalanceForChains).call(this, toRefresh).catch((error) => {
+        log('Failed to refresh balance after incoming transactions', { error });
+    });
+}, _RpcDataSource_refreshBalanceForChains = 
+/**
+ * Fetch balances for the given chains across all active subscriptions and
+ * push updates to the controller.
+ *
+ * @param chainIds - CAIP-2 chain IDs to refresh.
+ */
+async function _RpcDataSource_refreshBalanceForChains(chainIds) {
+    const chainIdsSet = new Set(chainIds);
+    const chainsToFetch = chainIds.filter((chainId) => __classPrivateFieldGet(this, _RpcDataSource_activeChains, "f").includes(chainId));
+    if (chainsToFetch.length === 0) {
+        return;
+    }
+    for (const subscription of __classPrivateFieldGet(this, _RpcDataSource_activeSubscriptions, "f").values()) {
+        const subscriptionChains = subscription.chains.filter((chainId) => chainIdsSet.has(chainId));
+        if (subscriptionChains.length === 0) {
+            continue;
+        }
+        const request = {
+            accountsWithSupportedChains: subscription.accounts.map((account) => ({
+                account,
+                supportedChains: subscriptionChains,
+            })),
+            chainIds: subscriptionChains,
+            dataTypes: ['balance'],
+        };
+        try {
+            const response = await this.fetch(request);
+            if (response.assetsBalance &&
+                Object.keys(response.assetsBalance).length > 0) {
+                subscription.onAssetsUpdate(response)?.catch((error) => {
+                    log('Failed to report balance update after transaction', {
+                        error,
+                    });
+                });
+            }
+        }
+        catch (error) {
+            log('Failed to fetch balance after transaction', {
+                chains: subscriptionChains,
+                error,
+            });
+        }
+    }
 }, _RpcDataSource_initializeFromNetworkController = function _RpcDataSource_initializeFromNetworkController() {
     log('Initializing from NetworkController');
     try {
@@ -726,12 +808,11 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
     __classPrivateFieldGet(this, _RpcDataSource_providerCache, "f").clear();
 }, _RpcDataSource_buildNativeAssetId = function _RpcDataSource_buildNativeAssetId(chainId) {
     const { nativeAssetIdentifiers } = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('NetworkEnablementController:getState');
-    return (nativeAssetIdentifiers[chainId] ??
-        `${chainId}/slip44:60`);
+    return nativeAssetIdentifiers[chainId] ?? `${chainId}/slip44:60`;
 }, _RpcDataSource_getExistingAssetsMetadata = function _RpcDataSource_getExistingAssetsMetadata() {
     try {
         const state = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('AssetsController:getState');
-        return (state.assetsMetadata ?? {});
+        return state.assetsInfo ?? {};
     }
     catch {
         // If AssetsController:getState fails, return empty metadata
@@ -739,15 +820,13 @@ _RpcDataSource_messenger = new WeakMap(), _RpcDataSource_onActiveChainsUpdated =
     }
 }, _RpcDataSource_getTokenMetadataFromTokenList = function _RpcDataSource_getTokenMetadataFromTokenList(assetId) {
     try {
-        // Parse asset ID to get chain and token address
-        // Format: eip155:{chainId}/erc20:{address}
-        const [chainPart, assetPart] = assetId.split('/');
-        if (!assetPart?.startsWith('erc20:')) {
+        const parsed = parseCaipAssetType(assetId);
+        if (parsed.assetNamespace !== 'erc20') {
             return undefined;
         }
-        const tokenAddress = assetPart.slice(6); // Remove 'erc20:' prefix
-        const chainIdDecimal = chainPart.split(':')[1];
-        const hexChainId = `0x${parseInt(chainIdDecimal, 10).toString(16)}`;
+        const tokenAddress = parsed.assetReference;
+        const { reference } = parseCaipChainId(parsed.chainId);
+        const hexChainId = numberToHex(parseInt(reference, 10));
         const tokenListState = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('TokenListController:getState');
         const chainCacheEntry = tokenListState?.tokensChainsCache?.[hexChainId];
         const chainTokenList = chainCacheEntry?.data;