@metamask-previews/assets-controller 0.2.0-preview-c3cd77f → 0.2.0-preview-d01b2f93d

package/dist/data-sources/PriceDataSource.mjs

@@ -9,7 +9,7 @@ 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 _PriceDataSource_instances, _PriceDataSource_messenger, _PriceDataSource_currency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_registerActionHandlers, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
+var _PriceDataSource_instances, _PriceDataSource_currency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
 import { ApiPlatformClient } from "@metamask/core-backend";
 import { projectLogger, createModuleLogger } from "../logger.mjs";
 import { forDataTypes } from "../types.mjs";
@@ -65,44 +65,23 @@ function isValidMarketData(data) {
  * This data source:
  * - Fetches prices from Price API v3 spot-prices endpoint
  * - Supports one-time fetch and subscription-based polling
- * - In subscribe mode, automatically fetches prices for all assets in assetsBalance state
- * - Publishes price updates via messenger events
+ * - In subscribe mode, uses getAssetsState from SubscriptionRequest to read assetsBalance and fetch prices
  *
- * Usage:
- * ```typescript
- * // Create and initialize (registers messenger actions)
- * const priceDataSource = new PriceDataSource({ messenger });
- *
- * // One-time fetch for specific assets
- * const response = await messenger.call('PriceDataSource:fetch', {
- *   customAssets: ['eip155:1/erc20:0x...'],
- * });
- *
- * // Subscribe to price updates (polls all assets in balance state)
- * await messenger.call('PriceDataSource:subscribe', { request, subscriptionId });
- *
- * // Listen for updates
- * messenger.subscribe('PriceDataSource:assetsUpdated', (response) => {
- *   // Handle price updates
- * });
- * ```
+ * Usage: Create with queryApiClient; subscribe() requires getAssetsState in the request for balance-based pricing.
  */
 export class PriceDataSource {
     constructor(options) {
         _PriceDataSource_instances.add(this);
         this.name = CONTROLLER_NAME;
-        _PriceDataSource_messenger.set(this, void 0);
         _PriceDataSource_currency.set(this, void 0);
         _PriceDataSource_pollInterval.set(this, void 0);
         /** ApiPlatformClient for cached API calls */
         _PriceDataSource_apiClient.set(this, void 0);
         /** Active subscriptions by ID */
         _PriceDataSource_activeSubscriptions.set(this, new Map());
-        __classPrivateFieldSet(this, _PriceDataSource_messenger, options.messenger, "f");
         __classPrivateFieldSet(this, _PriceDataSource_currency, options.currency ?? 'usd', "f");
         __classPrivateFieldSet(this, _PriceDataSource_pollInterval, options.pollInterval ?? DEFAULT_POLL_INTERVAL, "f");
         __classPrivateFieldSet(this, _PriceDataSource_apiClient, options.queryApiClient, "f");
-        __classPrivateFieldGet(this, _PriceDataSource_instances, "m", _PriceDataSource_registerActionHandlers).call(this);
     }
     // ============================================================================
     // MIDDLEWARE
@@ -171,15 +150,16 @@ export class PriceDataSource {
     // ============================================================================
     /**
      * Fetch prices for assets held by the accounts and chains in the request.
-     * Gets asset IDs from balance state, filtered by request.accounts and request.chainIds.
+     * When getAssetsState is provided, gets asset IDs from balance state; otherwise returns empty.
      *
      * @param request - The data request specifying accounts and chains.
+     * @param getAssetsState - Optional state access (e.g. from SubscriptionRequest).
      * @returns DataResponse containing asset prices.
      */
-    async fetch(request) {
+    async fetch(request, getAssetsState) {
         const response = {};
-        // Get asset IDs from balance state, filtered by accounts and chains
-        const rawAssetIds = __classPrivateFieldGet(this, _PriceDataSource_instances, "m", _PriceDataSource_getAssetIdsFromBalanceState).call(this, request);
+        // Get asset IDs from balance state when state access is provided
+        const rawAssetIds = __classPrivateFieldGet(this, _PriceDataSource_instances, "m", _PriceDataSource_getAssetIdsFromBalanceState).call(this, request, getAssetsState);
         // Filter out non-priceable assets (e.g., Tron bandwidth/energy resources)
         const assetIds = rawAssetIds.filter(isPriceableAsset);
         if (assetIds.length === 0) {
@@ -227,19 +207,19 @@ export class PriceDataSource {
         // Clean up existing subscription
         await this.unsubscribe(subscriptionId);
         const pollInterval = request.updateInterval ?? __classPrivateFieldGet(this, _PriceDataSource_pollInterval, "f");
-        // Create poll function - fetches prices for all assets in balance state
+        // Create poll function - fetches prices using getAssetsState from subscription
         const pollFn = async () => {
             try {
                 const subscription = __classPrivateFieldGet(this, _PriceDataSource_activeSubscriptions, "f").get(subscriptionId);
                 if (!subscription) {
                     return;
                 }
-                // Fetch prices for all assets currently in balance state
-                const fetchResponse = await this.fetch(subscription.request);
+                // Fetch prices for all assets in balance state (uses subscription's getAssetsState)
+                const fetchResponse = await this.fetch(subscription.request, subscription.getAssetsState);
                 // Only report if we got prices
                 if (fetchResponse.assetsPrice &&
                     Object.keys(fetchResponse.assetsPrice).length > 0) {
-                    await __classPrivateFieldGet(this, _PriceDataSource_messenger, "f").call('AssetsController:assetsUpdate', fetchResponse, CONTROLLER_NAME);
+                    await subscription.onAssetsUpdate(fetchResponse);
                 }
             }
             catch (error) {
@@ -250,12 +230,14 @@ export class PriceDataSource {
         const timer = setInterval(() => {
             pollFn().catch(console.error);
         }, pollInterval);
-        // Store subscription
+        // Store subscription (getAssetsState from request for balance-based pricing)
         __classPrivateFieldGet(this, _PriceDataSource_activeSubscriptions, "f").set(subscriptionId, {
             cleanup: () => {
                 clearInterval(timer);
             },
             request,
+            onAssetsUpdate: subscriptionRequest.onAssetsUpdate,
+            getAssetsState: subscriptionRequest.getAssetsState,
         });
         // Initial fetch
         await pollFn();
@@ -282,14 +264,7 @@ export class PriceDataSource {
         __classPrivateFieldGet(this, _PriceDataSource_activeSubscriptions, "f").clear();
     }
 }
-_PriceDataSource_messenger = new WeakMap(), _PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_registerActionHandlers = function _PriceDataSource_registerActionHandlers() {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const messenger = __classPrivateFieldGet(this, _PriceDataSource_messenger, "f");
-    messenger.registerActionHandler('PriceDataSource:getAssetsMiddleware', () => this.assetsMiddleware);
-    messenger.registerActionHandler('PriceDataSource:fetch', (request) => this.fetch(request));
-    messenger.registerActionHandler('PriceDataSource:subscribe', (subscriptionRequest) => this.subscribe(subscriptionRequest));
-    messenger.registerActionHandler('PriceDataSource:unsubscribe', (subscriptionId) => this.unsubscribe(subscriptionId));
-}, _PriceDataSource_fetchSpotPrices = 
+_PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_fetchSpotPrices = 
 // ============================================================================
 // HELPERS
 // ============================================================================
@@ -304,11 +279,14 @@ async function _PriceDataSource_fetchSpotPrices(assetIds) {
         currency: __classPrivateFieldGet(this, _PriceDataSource_currency, "f"),
         includeMarketData: true,
     });
-}, _PriceDataSource_getAssetIdsFromBalanceState = function _PriceDataSource_getAssetIdsFromBalanceState(request) {
+}, _PriceDataSource_getAssetIdsFromBalanceState = function _PriceDataSource_getAssetIdsFromBalanceState(request, getAssetsState) {
+    if (!getAssetsState) {
+        return [];
+    }
     try {
-        const state = __classPrivateFieldGet(this, _PriceDataSource_messenger, "f").call('AssetsController:getState');
+        const state = getAssetsState();
         const assetIds = new Set();
-        const accountIds = request.accounts.map((a) => a.id);
+        const accountIds = request.accountsWithSupportedChains.map((a) => a.account.id);
         const accountFilter = accountIds.length > 0 ? new Set(accountIds) : undefined;
         const chainFilter = request.chainIds.length > 0 ? new Set(request.chainIds) : undefined;
         if (state?.assetsBalance) {

package/dist/data-sources/PriceDataSource.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"PriceDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAIA,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAI3D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAC9D,OAAO,EAAE,YAAY,EAAE,qBAAiB;AAYxC,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,iBAAiB,CAAC;AAC1C,MAAM,qBAAqB,GAAG,KAAM,CAAC,CAAC,6BAA6B;AAEnE,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAwG/D,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,4BAA4B,GAAG;IACnC,2DAA2D;IAC3D,2BAA2B;IAC3B,sBAAsB;IACtB,mBAAmB;IACnB,8BAA8B;IAC9B,2BAA2B;CAC5B,CAAC;AAEF;;;;;;GAMG;AACH,SAAS,gBAAgB,CAAC,OAAsB;IAC9C,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;AAChF,CAAC;AAKD;;;;;GAKG;AACH,SAAS,iBAAiB,CAAC,IAAa;IACtC,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;QACxB,IAAI,KAAK,IAAI;QACb,OAAO,IAAI,IAAI;QACf,OAAQ,IAA4B,CAAC,KAAK,KAAK,QAAQ,CACxD,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,MAAM,OAAO,eAAe;IAkB1B,YAAY,OAA+B;;QAjBlC,SAAI,GAAG,eAAe,CAAC;QAEvB,6CAAqC;QAErC,4CAA6B;QAE7B,gDAAsB;QAE/B,6CAA6C;QACpC,6CAA8B;QAEvC,iCAAiC;QACxB,+CAGL,IAAI,GAAG,EAAE,EAAC;QAGZ,uBAAA,IAAI,8BAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,6BAAa,OAAO,CAAC,QAAQ,IAAI,KAAK,MAAA,CAAC;QAC3C,uBAAA,IAAI,iCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,8BAAc,OAAO,CAAC,cAAc,MAAA,CAAC;QAEzC,uBAAA,IAAI,2EAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IA4BD,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB;QAClB,OAAO,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACjD,gCAAgC;YAChC,MAAM,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;YAEzB,kEAAkE;YAClE,mEAAmE;YACnE,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAiB,CAAC;YAClD,KAAK,MAAM,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACjE,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;oBAClC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;YAED,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,kCAAkC;YAClC,MAAM,iBAAiB,GAAG,CAAC,GAAG,gBAAgB,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAEzE,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,oEAAiB,MAArB,IAAI,EAAkB,iBAAiB,CAAC,CAAC;gBAErE,QAAQ,CAAC,WAAW,KAApB,QAAQ,CAAC,WAAW,GAAK,EAAE,EAAC;gBAE5B,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;oBAClE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAAC;wBACnC,SAAS;oBACX,CAAC;oBAED,MAAM,WAAW,GAAG,OAAwB,CAAC;oBAC7C,QAAQ,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG;wBAClC,GAAG,UAAU;wBACb,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;qBACxB,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1D,CAAC;YAED,0DAA0D;YAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAoED,+EAA+E;IAC/E,QAAQ;IACR,+EAA+E;IAE/E;;;;;;OAMG;IACH,KAAK,CAAC,KAAK,CAAC,OAAoB;QAC9B,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAElC,oEAAoE;QACpE,MAAM,WAAW,GAAG,uBAAA,IAAI,gFAA6B,MAAjC,IAAI,EAA8B,OAAO,CAAC,CAAC;QAE/D,0EAA0E;QAC1E,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAEtD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,oEAAiB,MAArB,IAAI,EAAkB,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC;YAEjE,QAAQ,CAAC,WAAW,GAAG,EAAE,CAAC;YAE1B,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClE,+EAA+E;gBAC/E,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAAC;oBACnC,SAAS;gBACX,CAAC;gBAED,MAAM,WAAW,GAAG,OAAwB,CAAC;gBAC7C,QAAQ,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG;oBAClC,GAAG,UAAU;oBACb,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;iBACxB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,+EAA+E;IAC/E,YAAY;IACZ,+EAA+E;IAE/E;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,uDAAuD;QACvD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC3B,OAAO;YACT,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,MAAM,YAAY,GAAG,OAAO,CAAC,cAAc,IAAI,uBAAA,IAAI,qCAAc,CAAC;QAElE,wEAAwE;QACxE,MAAM,MAAM,GAAG,KAAK,IAAmB,EAAE;YACvC,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACnE,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,OAAO;gBACT,CAAC;gBAED,yDAAyD;gBACzD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;gBAE7D,+BAA+B;gBAC/B,IACE,aAAa,CAAC,WAAW;oBACzB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,EACjD,CAAC;oBACD,MAAM,uBAAA,IAAI,kCAAW,CAAC,IAAI,CACxB,+BAA+B,EAC/B,aAAa,EACb,eAAe,CAChB,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,0BAA0B,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC;QAEF,iBAAiB;QACjB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7B,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,EAAE,YAAY,CAAC,CAAC;QAEjB,qBAAqB;QACrB,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,OAAO,EAAE,GAAG,EAAE;gBACZ,aAAa,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YACD,OAAO;SACR,CAAC,CAAC;QAEH,gBAAgB;QAChB,MAAM,MAAM,EAAE,CAAC;IACjB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,uBAAA,IAAI,4CAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,4CAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,uBAAA,IAAI,4CAAqB,CAAC,KAAK,EAAE,CAAC;IACpC,CAAC;CACF;;IAjTG,8DAA8D;IAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,kCAAkB,CAAC;IAEzC,SAAS,CAAC,qBAAqB,CAC7B,qCAAqC,EACrC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAC5B,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,uBAAuB,EACvB,CAAC,OAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAC9C,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,2BAA2B,EAC3B,CAAC,mBAAwC,EAAE,EAAE,CAC3C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACtC,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,6BAA6B,EAC7B,CAAC,cAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAC7D,CAAC;AACJ,CAAC;AA2ED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E;;;;;GAKG;AACH,KAAK,2CAAkB,QAAkB;IACvC,OAAO,uBAAA,IAAI,kCAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE;QACxD,QAAQ,EAAE,uBAAA,IAAI,iCAAU;QACxB,iBAAiB,EAAE,IAAI;KACxB,CAAC,CAAC;AACL,CAAC,uGAS4B,OAAoB;IAC/C,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,uBAAA,IAAI,kCAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAChE,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;QAE1C,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,aAAa,GACjB,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1D,MAAM,WAAW,GACf,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEtE,IAAI,KAAK,EAAE,aAAa,EAAE,CAAC;YACzB,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,KAAK,CAAC,aAAa,CACpB,EAAE,CAAC;gBACF,iCAAiC;gBACjC,IAAI,aAAa,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBACnD,SAAS;gBACX,CAAC;gBAED,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;oBACF,+BAA+B;oBAC/B,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;wBACjD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC9B,SAAS;wBACX,CAAC;oBACH,CAAC;oBACD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,QAAQ,CAAC,CAAC;IACvB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC","sourcesContent":["import type {\n  SupportedCurrency,\n  V3SpotPricesResponse,\n} from '@metamask/core-backend';\nimport { ApiPlatformClient } from '@metamask/core-backend';\nimport type { Messenger } from '@metamask/messenger';\n\nimport type { SubscriptionRequest } from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport { forDataTypes } from '../types';\nimport type {\n  Caip19AssetId,\n  AssetBalance,\n  AccountId,\n  ChainId,\n  DataRequest,\n  DataResponse,\n  FungibleAssetPrice,\n  Middleware,\n} from '../types';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'PriceDataSource';\nconst DEFAULT_POLL_INTERVAL = 60_000; // 1 minute for price updates\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\n/**\n * Action to get balance state (used to determine which assets need prices).\n */\ntype GetAssetsBalanceStateAction = {\n  type: 'AssetsController:getState';\n  handler: () => {\n    assetsBalance: Record<AccountId, Record<Caip19AssetId, AssetBalance>>;\n  };\n};\n\n/**\n * Action to get the PriceDataSource middleware.\n */\nexport type PriceDataSourceGetAssetsMiddlewareAction = {\n  type: `${typeof CONTROLLER_NAME}:getAssetsMiddleware`;\n  handler: () => Middleware;\n};\n\n/**\n * Action to fetch prices for assets.\n */\nexport type PriceDataSourceFetchAction = {\n  type: `${typeof CONTROLLER_NAME}:fetch`;\n  handler: (request: DataRequest) => Promise<DataResponse>;\n};\n\n/**\n * Action to subscribe to price updates.\n */\nexport type PriceDataSourceSubscribeAction = {\n  type: `${typeof CONTROLLER_NAME}:subscribe`;\n  handler: (request: SubscriptionRequest) => Promise<void>;\n};\n\n/**\n * Action to unsubscribe from price updates.\n */\nexport type PriceDataSourceUnsubscribeAction = {\n  type: `${typeof CONTROLLER_NAME}:unsubscribe`;\n  handler: (subscriptionId: string) => Promise<void>;\n};\n\n/**\n * All actions exposed by PriceDataSource.\n */\nexport type PriceDataSourceActions =\n  | PriceDataSourceGetAssetsMiddlewareAction\n  | PriceDataSourceFetchAction\n  | PriceDataSourceSubscribeAction\n  | PriceDataSourceUnsubscribeAction;\n\n/**\n * Event emitted when prices are updated.\n */\nexport type PriceDataSourceAssetsUpdatedEvent = {\n  type: `${typeof CONTROLLER_NAME}:assetsUpdated`;\n  payload: [DataResponse, string];\n};\n\n/**\n * All events exposed by PriceDataSource.\n */\nexport type PriceDataSourceEvents = PriceDataSourceAssetsUpdatedEvent;\n\n// Action to report assets updated to AssetsController\ntype AssetsControllerAssetsUpdateAction = {\n  type: 'AssetsController:assetsUpdate';\n  handler: (response: DataResponse, sourceId: string) => Promise<void>;\n};\n\n/**\n * External actions that PriceDataSource needs to call.\n * Note: Uses ApiPlatformClient directly, so no BackendApiClient actions needed.\n */\nexport type PriceDataSourceAllowedActions =\n  | GetAssetsBalanceStateAction\n  | AssetsControllerAssetsUpdateAction;\n\nexport type PriceDataSourceMessenger = Messenger<\n  typeof CONTROLLER_NAME,\n  PriceDataSourceAllowedActions | PriceDataSourceActions,\n  PriceDataSourceEvents\n>;\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type PriceDataSourceOptions = {\n  messenger: PriceDataSourceMessenger;\n  /** ApiPlatformClient for API calls with caching */\n  queryApiClient: ApiPlatformClient;\n  /** Currency to fetch prices in (default: 'usd') */\n  currency?: SupportedCurrency;\n  /** Polling interval in ms (default: 60000) */\n  pollInterval?: number;\n};\n\n// ============================================================================\n// HELPER FUNCTIONS\n// ============================================================================\n\n/**\n * Asset reference patterns that should NOT be sent to the Price API.\n * These are internal resource tracking values without market prices.\n */\nconst NON_PRICEABLE_ASSET_PATTERNS = [\n  // Tron resource assets (bandwidth, energy, staking states)\n  /\\/slip44:\\d+-staked-for-/u,\n  /\\/slip44:bandwidth$/u,\n  /\\/slip44:energy$/u,\n  /\\/slip44:maximum-bandwidth$/u,\n  /\\/slip44:maximum-energy$/u,\n];\n\n/**\n * Check if an asset ID represents a priceable asset.\n * Filters out internal resource tracking values that don't have market prices.\n *\n * @param assetId - The CAIP-19 asset ID to check.\n * @returns True if the asset has market price data.\n */\nfunction isPriceableAsset(assetId: Caip19AssetId): boolean {\n  return !NON_PRICEABLE_ASSET_PATTERNS.some((pattern) => pattern.test(assetId));\n}\n\n/** Market data item from spot prices response (same as FungibleAssetPrice without lastUpdated) */\ntype SpotPriceMarketData = Omit<FungibleAssetPrice, 'lastUpdated'>;\n\n/**\n * Type guard to check if market data has a valid price\n *\n * @param data - The data to check.\n * @returns True if data is valid SpotPriceMarketData.\n */\nfunction isValidMarketData(data: unknown): data is SpotPriceMarketData {\n  return (\n    typeof data === 'object' &&\n    data !== null &&\n    'price' in data &&\n    typeof (data as SpotPriceMarketData).price === 'number'\n  );\n}\n\n// ============================================================================\n// PRICE DATA SOURCE\n// ============================================================================\n\n/**\n * PriceDataSource fetches asset prices from the Price API.\n *\n * This data source:\n * - Fetches prices from Price API v3 spot-prices endpoint\n * - Supports one-time fetch and subscription-based polling\n * - In subscribe mode, automatically fetches prices for all assets in assetsBalance state\n * - Publishes price updates via messenger events\n *\n * Usage:\n * ```typescript\n * // Create and initialize (registers messenger actions)\n * const priceDataSource = new PriceDataSource({ messenger });\n *\n * // One-time fetch for specific assets\n * const response = await messenger.call('PriceDataSource:fetch', {\n *   customAssets: ['eip155:1/erc20:0x...'],\n * });\n *\n * // Subscribe to price updates (polls all assets in balance state)\n * await messenger.call('PriceDataSource:subscribe', { request, subscriptionId });\n *\n * // Listen for updates\n * messenger.subscribe('PriceDataSource:assetsUpdated', (response) => {\n *   // Handle price updates\n * });\n * ```\n */\nexport class PriceDataSource {\n  readonly name = CONTROLLER_NAME;\n\n  readonly #messenger: PriceDataSourceMessenger;\n\n  readonly #currency: SupportedCurrency;\n\n  readonly #pollInterval: number;\n\n  /** ApiPlatformClient for cached API calls */\n  readonly #apiClient: ApiPlatformClient;\n\n  /** Active subscriptions by ID */\n  readonly #activeSubscriptions: Map<\n    string,\n    { cleanup: () => void; request: DataRequest }\n  > = new Map();\n\n  constructor(options: PriceDataSourceOptions) {\n    this.#messenger = options.messenger;\n    this.#currency = options.currency ?? 'usd';\n    this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n    this.#apiClient = options.queryApiClient;\n\n    this.#registerActionHandlers();\n  }\n\n  #registerActionHandlers(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    messenger.registerActionHandler(\n      'PriceDataSource:getAssetsMiddleware',\n      () => this.assetsMiddleware,\n    );\n\n    messenger.registerActionHandler(\n      'PriceDataSource:fetch',\n      (request: DataRequest) => this.fetch(request),\n    );\n\n    messenger.registerActionHandler(\n      'PriceDataSource:subscribe',\n      (subscriptionRequest: SubscriptionRequest) =>\n        this.subscribe(subscriptionRequest),\n    );\n\n    messenger.registerActionHandler(\n      'PriceDataSource:unsubscribe',\n      (subscriptionId: string) => this.unsubscribe(subscriptionId),\n    );\n  }\n\n  // ============================================================================\n  // MIDDLEWARE\n  // ============================================================================\n\n  /**\n   * Get the middleware for enriching responses with price data.\n   *\n   * This middleware:\n   * 1. Extracts the response from context\n   * 2. Fetches prices for detected assets (assets without metadata)\n   * 3. Enriches the response with fetched prices\n   * 4. Calls next() at the end to continue the middleware chain\n   *\n   * Note: This middleware ONLY fetches prices for detected assets.\n   * For fetching prices for all assets, use the subscription mechanism\n   * which polls prices for all assets in the balance state.\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['price'], async (ctx, next) => {\n      // Extract response from context\n      const { response } = ctx;\n\n      // Only fetch prices for detected assets (assets without metadata)\n      // The subscription handles fetching prices for all existing assets\n      if (!response.detectedAssets) {\n        return next(ctx);\n      }\n\n      const detectedAssetIds = new Set<Caip19AssetId>();\n      for (const detectedIds of Object.values(response.detectedAssets)) {\n        for (const assetId of detectedIds) {\n          detectedAssetIds.add(assetId);\n        }\n      }\n\n      if (detectedAssetIds.size === 0) {\n        return next(ctx);\n      }\n\n      // Filter to only priceable assets\n      const priceableAssetIds = [...detectedAssetIds].filter(isPriceableAsset);\n\n      if (priceableAssetIds.length === 0) {\n        return next(ctx);\n      }\n\n      try {\n        const priceResponse = await this.#fetchSpotPrices(priceableAssetIds);\n\n        response.assetsPrice ??= {};\n\n        for (const [assetId, marketData] of Object.entries(priceResponse)) {\n          if (!isValidMarketData(marketData)) {\n            continue;\n          }\n\n          const caipAssetId = assetId as Caip19AssetId;\n          response.assetsPrice[caipAssetId] = {\n            ...marketData,\n            lastUpdated: Date.now(),\n          };\n        }\n      } catch (error) {\n        log('Failed to fetch prices via middleware', { error });\n      }\n\n      // Call next() at the end to continue the middleware chain\n      return next(ctx);\n    });\n  }\n\n  // ============================================================================\n  // HELPERS\n  // ============================================================================\n\n  /**\n   * Fetch spot prices with caching and deduplication via query service.\n   *\n   * @param assetIds - Array of CAIP-19 asset IDs\n   * @returns Spot prices response\n   */\n  async #fetchSpotPrices(assetIds: string[]): Promise<V3SpotPricesResponse> {\n    return this.#apiClient.prices.fetchV3SpotPrices(assetIds, {\n      currency: this.#currency,\n      includeMarketData: true,\n    });\n  }\n\n  /**\n   * Get unique asset IDs from the assetsBalance state.\n   * Filters by accounts and chains from the request.\n   *\n   * @param request - Data request with accounts and chainIds filters.\n   * @returns Array of CAIP-19 asset IDs from balance state.\n   */\n  #getAssetIdsFromBalanceState(request: DataRequest): Caip19AssetId[] {\n    try {\n      const state = this.#messenger.call('AssetsController:getState');\n      const assetIds = new Set<Caip19AssetId>();\n\n      const accountIds = request.accounts.map((a) => a.id);\n      const accountFilter =\n        accountIds.length > 0 ? new Set(accountIds) : undefined;\n      const chainFilter =\n        request.chainIds.length > 0 ? new Set(request.chainIds) : undefined;\n\n      if (state?.assetsBalance) {\n        for (const [accountId, accountBalances] of Object.entries(\n          state.assetsBalance,\n        )) {\n          // Filter by account if specified\n          if (accountFilter && !accountFilter.has(accountId)) {\n            continue;\n          }\n\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            // Filter by chain if specified\n            if (chainFilter) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              if (!chainFilter.has(chainId)) {\n                continue;\n              }\n            }\n            assetIds.add(assetId as Caip19AssetId);\n          }\n        }\n      }\n\n      return [...assetIds];\n    } catch (error) {\n      log('Failed to get asset IDs from balance state', { error });\n      return [];\n    }\n  }\n\n  // ============================================================================\n  // FETCH\n  // ============================================================================\n\n  /**\n   * Fetch prices for assets held by the accounts and chains in the request.\n   * Gets asset IDs from balance state, filtered by request.accounts and request.chainIds.\n   *\n   * @param request - The data request specifying accounts and chains.\n   * @returns DataResponse containing asset prices.\n   */\n  async fetch(request: DataRequest): Promise<DataResponse> {\n    const response: DataResponse = {};\n\n    // Get asset IDs from balance state, filtered by accounts and chains\n    const rawAssetIds = this.#getAssetIdsFromBalanceState(request);\n\n    // Filter out non-priceable assets (e.g., Tron bandwidth/energy resources)\n    const assetIds = rawAssetIds.filter(isPriceableAsset);\n\n    if (assetIds.length === 0) {\n      return response;\n    }\n\n    try {\n      const priceResponse = await this.#fetchSpotPrices([...assetIds]);\n\n      response.assetsPrice = {};\n\n      for (const [assetId, marketData] of Object.entries(priceResponse)) {\n        // Skip assets with invalid market data (API doesn't have price for this asset)\n        if (!isValidMarketData(marketData)) {\n          continue;\n        }\n\n        const caipAssetId = assetId as Caip19AssetId;\n        response.assetsPrice[caipAssetId] = {\n          ...marketData,\n          lastUpdated: Date.now(),\n        };\n      }\n    } catch (error) {\n      log('Failed to fetch prices', { error });\n    }\n\n    return response;\n  }\n\n  // ============================================================================\n  // SUBSCRIBE\n  // ============================================================================\n\n  /**\n   * Subscribe to price updates.\n   * Sets up polling that fetches prices for all assets in assetsBalance state.\n   *\n   * @param subscriptionRequest - The subscription request configuration.\n   */\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Handle subscription update - just update the request\n    if (isUpdate) {\n      const existing = this.#activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        existing.request = request;\n        return;\n      }\n    }\n\n    // Clean up existing subscription\n    await this.unsubscribe(subscriptionId);\n\n    const pollInterval = request.updateInterval ?? this.#pollInterval;\n\n    // Create poll function - fetches prices for all assets in balance state\n    const pollFn = async (): Promise<void> => {\n      try {\n        const subscription = this.#activeSubscriptions.get(subscriptionId);\n        if (!subscription) {\n          return;\n        }\n\n        // Fetch prices for all assets currently in balance state\n        const fetchResponse = await this.fetch(subscription.request);\n\n        // Only report if we got prices\n        if (\n          fetchResponse.assetsPrice &&\n          Object.keys(fetchResponse.assetsPrice).length > 0\n        ) {\n          await this.#messenger.call(\n            'AssetsController:assetsUpdate',\n            fetchResponse,\n            CONTROLLER_NAME,\n          );\n        }\n      } catch (error) {\n        log('Subscription poll failed', { subscriptionId, error });\n      }\n    };\n\n    // Set up polling\n    const timer = setInterval(() => {\n      pollFn().catch(console.error);\n    }, pollInterval);\n\n    // Store subscription\n    this.#activeSubscriptions.set(subscriptionId, {\n      cleanup: () => {\n        clearInterval(timer);\n      },\n      request,\n    });\n\n    // Initial fetch\n    await pollFn();\n  }\n\n  /**\n   * Unsubscribe from price updates.\n   *\n   * @param subscriptionId - The ID of the subscription to cancel.\n   */\n  async unsubscribe(subscriptionId: string): Promise<void> {\n    const subscription = this.#activeSubscriptions.get(subscriptionId);\n    if (subscription) {\n      subscription.cleanup();\n      this.#activeSubscriptions.delete(subscriptionId);\n    }\n  }\n\n  /**\n   * Destroy the data source and clean up all subscriptions.\n   */\n  destroy(): void {\n    for (const subscription of this.#activeSubscriptions.values()) {\n      subscription.cleanup();\n    }\n    this.#activeSubscriptions.clear();\n  }\n}\n"]}
+{"version":3,"file":"PriceDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAIA,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAG3D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAC9D,OAAO,EAAE,YAAY,EAAE,qBAAiB;AAWxC,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,iBAAiB,CAAC;AAC1C,MAAM,qBAAqB,GAAG,KAAM,CAAC,CAAC,6BAA6B;AAEnE,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAe/D,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,4BAA4B,GAAG;IACnC,2DAA2D;IAC3D,2BAA2B;IAC3B,sBAAsB;IACtB,mBAAmB;IACnB,8BAA8B;IAC9B,2BAA2B;CAC5B,CAAC;AAEF;;;;;;GAMG;AACH,SAAS,gBAAgB,CAAC,OAAsB;IAC9C,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;AAChF,CAAC;AAKD;;;;;GAKG;AACH,SAAS,iBAAiB,CAAC,IAAa;IACtC,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;QACxB,IAAI,KAAK,IAAI;QACb,OAAO,IAAI,IAAI;QACf,OAAQ,IAA4B,CAAC,KAAK,KAAK,QAAQ,CACxD,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E;;;;;;;;;GASG;AACH,MAAM,OAAO,eAAe;IAqB1B,YAAY,OAA+B;;QApBlC,SAAI,GAAG,eAAe,CAAC;QAEvB,4CAA6B;QAE7B,gDAAsB;QAE/B,6CAA6C;QACpC,6CAA8B;QAEvC,iCAAiC;QACxB,+CAQL,IAAI,GAAG,EAAE,EAAC;QAGZ,uBAAA,IAAI,6BAAa,OAAO,CAAC,QAAQ,IAAI,KAAK,MAAA,CAAC;QAC3C,uBAAA,IAAI,iCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,8BAAc,OAAO,CAAC,cAAc,MAAA,CAAC;IAC3C,CAAC;IAED,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB;QAClB,OAAO,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACjD,gCAAgC;YAChC,MAAM,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;YAEzB,kEAAkE;YAClE,mEAAmE;YACnE,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAiB,CAAC;YAClD,KAAK,MAAM,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACjE,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;oBAClC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;YAED,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,kCAAkC;YAClC,MAAM,iBAAiB,GAAG,CAAC,GAAG,gBAAgB,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAEzE,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,oEAAiB,MAArB,IAAI,EAAkB,iBAAiB,CAAC,CAAC;gBAErE,QAAQ,CAAC,WAAW,KAApB,QAAQ,CAAC,WAAW,GAAK,EAAE,EAAC;gBAE5B,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;oBAClE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAAC;wBACnC,SAAS;oBACX,CAAC;oBAED,MAAM,WAAW,GAAG,OAAwB,CAAC;oBAC7C,QAAQ,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG;wBAClC,GAAG,UAAU;wBACb,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;qBACxB,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1D,CAAC;YAED,0DAA0D;YAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IA6ED,+EAA+E;IAC/E,QAAQ;IACR,+EAA+E;IAE/E;;;;;;;OAOG;IACH,KAAK,CAAC,KAAK,CACT,OAAoB,EACpB,cAAoD;QAEpD,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAElC,iEAAiE;QACjE,MAAM,WAAW,GAAG,uBAAA,IAAI,gFAA6B,MAAjC,IAAI,EACtB,OAAO,EACP,cAAc,CACf,CAAC;QAEF,0EAA0E;QAC1E,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAEtD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,oEAAiB,MAArB,IAAI,EAAkB,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC;YAEjE,QAAQ,CAAC,WAAW,GAAG,EAAE,CAAC;YAE1B,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClE,+EAA+E;gBAC/E,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAAC;oBACnC,SAAS;gBACX,CAAC;gBAED,MAAM,WAAW,GAAG,OAAwB,CAAC;gBAC7C,QAAQ,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG;oBAClC,GAAG,UAAU;oBACb,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;iBACxB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,+EAA+E;IAC/E,YAAY;IACZ,+EAA+E;IAE/E;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,uDAAuD;QACvD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC3B,OAAO;YACT,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,MAAM,YAAY,GAAG,OAAO,CAAC,cAAc,IAAI,uBAAA,IAAI,qCAAc,CAAC;QAElE,+EAA+E;QAC/E,MAAM,MAAM,GAAG,KAAK,IAAmB,EAAE;YACvC,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACnE,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,OAAO;gBACT,CAAC;gBAED,oFAAoF;gBACpF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CACpC,YAAY,CAAC,OAAO,EACpB,YAAY,CAAC,cAAc,CAC5B,CAAC;gBAEF,+BAA+B;gBAC/B,IACE,aAAa,CAAC,WAAW;oBACzB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,EACjD,CAAC;oBACD,MAAM,YAAY,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,0BAA0B,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC;QAEF,iBAAiB;QACjB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7B,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,EAAE,YAAY,CAAC,CAAC;QAEjB,6EAA6E;QAC7E,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,OAAO,EAAE,GAAG,EAAE;gBACZ,aAAa,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YACD,OAAO;YACP,cAAc,EAAE,mBAAmB,CAAC,cAAc;YAClD,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,gBAAgB;QAChB,MAAM,MAAM,EAAE,CAAC;IACjB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,4CAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,uBAAA,IAAI,4CAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,4CAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,uBAAA,IAAI,4CAAqB,CAAC,KAAK,EAAE,CAAC;IACpC,CAAC;CACF;;AAhOC,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E;;;;;GAKG;AACH,KAAK,2CAAkB,QAAkB;IACvC,OAAO,uBAAA,IAAI,kCAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE;QACxD,QAAQ,EAAE,uBAAA,IAAI,iCAAU;QACxB,iBAAiB,EAAE,IAAI;KACxB,CAAC,CAAC;AACL,CAAC,uGAWC,OAAoB,EACpB,cAAoD;IAEpD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;QAE1C,MAAM,UAAU,GAAG,OAAO,CAAC,2BAA2B,CAAC,GAAG,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CACpB,CAAC;QACF,MAAM,aAAa,GACjB,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1D,MAAM,WAAW,GACf,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEtE,IAAI,KAAK,EAAE,aAAa,EAAE,CAAC;YACzB,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,KAAK,CAAC,aAAa,CACpB,EAAE,CAAC;gBACF,iCAAiC;gBACjC,IAAI,aAAa,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBACnD,SAAS;gBACX,CAAC;gBAED,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,eAA0C,CAC3C,EAAE,CAAC;oBACF,+BAA+B;oBAC/B,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAY,CAAC;wBACjD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC9B,SAAS;wBACX,CAAC;oBACH,CAAC;oBACD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,QAAQ,CAAC,CAAC;IACvB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC","sourcesContent":["import type {\n  SupportedCurrency,\n  V3SpotPricesResponse,\n} from '@metamask/core-backend';\nimport { ApiPlatformClient } from '@metamask/core-backend';\n\nimport type { SubscriptionRequest } from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport { forDataTypes } from '../types';\nimport type {\n  Caip19AssetId,\n  ChainId,\n  DataRequest,\n  DataResponse,\n  FungibleAssetPrice,\n  Middleware,\n  AssetsControllerStateInternal,\n} from '../types';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'PriceDataSource';\nconst DEFAULT_POLL_INTERVAL = 60_000; // 1 minute for price updates\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type PriceDataSourceOptions = {\n  /** ApiPlatformClient for API calls with caching */\n  queryApiClient: ApiPlatformClient;\n  /** Currency to fetch prices in (default: 'usd') */\n  currency?: SupportedCurrency;\n  /** Polling interval in ms (default: 60000) */\n  pollInterval?: number;\n};\n\n// ============================================================================\n// HELPER FUNCTIONS\n// ============================================================================\n\n/**\n * Asset reference patterns that should NOT be sent to the Price API.\n * These are internal resource tracking values without market prices.\n */\nconst NON_PRICEABLE_ASSET_PATTERNS = [\n  // Tron resource assets (bandwidth, energy, staking states)\n  /\\/slip44:\\d+-staked-for-/u,\n  /\\/slip44:bandwidth$/u,\n  /\\/slip44:energy$/u,\n  /\\/slip44:maximum-bandwidth$/u,\n  /\\/slip44:maximum-energy$/u,\n];\n\n/**\n * Check if an asset ID represents a priceable asset.\n * Filters out internal resource tracking values that don't have market prices.\n *\n * @param assetId - The CAIP-19 asset ID to check.\n * @returns True if the asset has market price data.\n */\nfunction isPriceableAsset(assetId: Caip19AssetId): boolean {\n  return !NON_PRICEABLE_ASSET_PATTERNS.some((pattern) => pattern.test(assetId));\n}\n\n/** Market data item from spot prices response (same as FungibleAssetPrice without lastUpdated) */\ntype SpotPriceMarketData = Omit<FungibleAssetPrice, 'lastUpdated'>;\n\n/**\n * Type guard to check if market data has a valid price\n *\n * @param data - The data to check.\n * @returns True if data is valid SpotPriceMarketData.\n */\nfunction isValidMarketData(data: unknown): data is SpotPriceMarketData {\n  return (\n    typeof data === 'object' &&\n    data !== null &&\n    'price' in data &&\n    typeof (data as SpotPriceMarketData).price === 'number'\n  );\n}\n\n// ============================================================================\n// PRICE DATA SOURCE\n// ============================================================================\n\n/**\n * PriceDataSource fetches asset prices from the Price API.\n *\n * This data source:\n * - Fetches prices from Price API v3 spot-prices endpoint\n * - Supports one-time fetch and subscription-based polling\n * - In subscribe mode, uses getAssetsState from SubscriptionRequest to read assetsBalance and fetch prices\n *\n * Usage: Create with queryApiClient; subscribe() requires getAssetsState in the request for balance-based pricing.\n */\nexport class PriceDataSource {\n  readonly name = CONTROLLER_NAME;\n\n  readonly #currency: SupportedCurrency;\n\n  readonly #pollInterval: number;\n\n  /** ApiPlatformClient for cached API calls */\n  readonly #apiClient: ApiPlatformClient;\n\n  /** Active subscriptions by ID */\n  readonly #activeSubscriptions: Map<\n    string,\n    {\n      cleanup: () => void;\n      request: DataRequest;\n      onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\n      getAssetsState?: () => AssetsControllerStateInternal;\n    }\n  > = new Map();\n\n  constructor(options: PriceDataSourceOptions) {\n    this.#currency = options.currency ?? 'usd';\n    this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n    this.#apiClient = options.queryApiClient;\n  }\n\n  // ============================================================================\n  // MIDDLEWARE\n  // ============================================================================\n\n  /**\n   * Get the middleware for enriching responses with price data.\n   *\n   * This middleware:\n   * 1. Extracts the response from context\n   * 2. Fetches prices for detected assets (assets without metadata)\n   * 3. Enriches the response with fetched prices\n   * 4. Calls next() at the end to continue the middleware chain\n   *\n   * Note: This middleware ONLY fetches prices for detected assets.\n   * For fetching prices for all assets, use the subscription mechanism\n   * which polls prices for all assets in the balance state.\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return forDataTypes(['price'], async (ctx, next) => {\n      // Extract response from context\n      const { response } = ctx;\n\n      // Only fetch prices for detected assets (assets without metadata)\n      // The subscription handles fetching prices for all existing assets\n      if (!response.detectedAssets) {\n        return next(ctx);\n      }\n\n      const detectedAssetIds = new Set<Caip19AssetId>();\n      for (const detectedIds of Object.values(response.detectedAssets)) {\n        for (const assetId of detectedIds) {\n          detectedAssetIds.add(assetId);\n        }\n      }\n\n      if (detectedAssetIds.size === 0) {\n        return next(ctx);\n      }\n\n      // Filter to only priceable assets\n      const priceableAssetIds = [...detectedAssetIds].filter(isPriceableAsset);\n\n      if (priceableAssetIds.length === 0) {\n        return next(ctx);\n      }\n\n      try {\n        const priceResponse = await this.#fetchSpotPrices(priceableAssetIds);\n\n        response.assetsPrice ??= {};\n\n        for (const [assetId, marketData] of Object.entries(priceResponse)) {\n          if (!isValidMarketData(marketData)) {\n            continue;\n          }\n\n          const caipAssetId = assetId as Caip19AssetId;\n          response.assetsPrice[caipAssetId] = {\n            ...marketData,\n            lastUpdated: Date.now(),\n          };\n        }\n      } catch (error) {\n        log('Failed to fetch prices via middleware', { error });\n      }\n\n      // Call next() at the end to continue the middleware chain\n      return next(ctx);\n    });\n  }\n\n  // ============================================================================\n  // HELPERS\n  // ============================================================================\n\n  /**\n   * Fetch spot prices with caching and deduplication via query service.\n   *\n   * @param assetIds - Array of CAIP-19 asset IDs\n   * @returns Spot prices response\n   */\n  async #fetchSpotPrices(assetIds: string[]): Promise<V3SpotPricesResponse> {\n    return this.#apiClient.prices.fetchV3SpotPrices(assetIds, {\n      currency: this.#currency,\n      includeMarketData: true,\n    });\n  }\n\n  /**\n   * Get unique asset IDs from the assetsBalance state.\n   * Filters by accounts and chains from the request.\n   *\n   * @param request - Data request with accounts and chainIds filters.\n   * @param getAssetsState - State access; when omitted, returns [].\n   * @returns Array of CAIP-19 asset IDs from balance state.\n   */\n  #getAssetIdsFromBalanceState(\n    request: DataRequest,\n    getAssetsState?: () => AssetsControllerStateInternal,\n  ): Caip19AssetId[] {\n    if (!getAssetsState) {\n      return [];\n    }\n    try {\n      const state = getAssetsState();\n      const assetIds = new Set<Caip19AssetId>();\n\n      const accountIds = request.accountsWithSupportedChains.map(\n        (a) => a.account.id,\n      );\n      const accountFilter =\n        accountIds.length > 0 ? new Set(accountIds) : undefined;\n      const chainFilter =\n        request.chainIds.length > 0 ? new Set(request.chainIds) : undefined;\n\n      if (state?.assetsBalance) {\n        for (const [accountId, accountBalances] of Object.entries(\n          state.assetsBalance,\n        )) {\n          // Filter by account if specified\n          if (accountFilter && !accountFilter.has(accountId)) {\n            continue;\n          }\n\n          for (const assetId of Object.keys(\n            accountBalances as Record<string, unknown>,\n          )) {\n            // Filter by chain if specified\n            if (chainFilter) {\n              const chainId = assetId.split('/')[0] as ChainId;\n              if (!chainFilter.has(chainId)) {\n                continue;\n              }\n            }\n            assetIds.add(assetId as Caip19AssetId);\n          }\n        }\n      }\n\n      return [...assetIds];\n    } catch (error) {\n      log('Failed to get asset IDs from balance state', { error });\n      return [];\n    }\n  }\n\n  // ============================================================================\n  // FETCH\n  // ============================================================================\n\n  /**\n   * Fetch prices for assets held by the accounts and chains in the request.\n   * When getAssetsState is provided, gets asset IDs from balance state; otherwise returns empty.\n   *\n   * @param request - The data request specifying accounts and chains.\n   * @param getAssetsState - Optional state access (e.g. from SubscriptionRequest).\n   * @returns DataResponse containing asset prices.\n   */\n  async fetch(\n    request: DataRequest,\n    getAssetsState?: () => AssetsControllerStateInternal,\n  ): Promise<DataResponse> {\n    const response: DataResponse = {};\n\n    // Get asset IDs from balance state when state access is provided\n    const rawAssetIds = this.#getAssetIdsFromBalanceState(\n      request,\n      getAssetsState,\n    );\n\n    // Filter out non-priceable assets (e.g., Tron bandwidth/energy resources)\n    const assetIds = rawAssetIds.filter(isPriceableAsset);\n\n    if (assetIds.length === 0) {\n      return response;\n    }\n\n    try {\n      const priceResponse = await this.#fetchSpotPrices([...assetIds]);\n\n      response.assetsPrice = {};\n\n      for (const [assetId, marketData] of Object.entries(priceResponse)) {\n        // Skip assets with invalid market data (API doesn't have price for this asset)\n        if (!isValidMarketData(marketData)) {\n          continue;\n        }\n\n        const caipAssetId = assetId as Caip19AssetId;\n        response.assetsPrice[caipAssetId] = {\n          ...marketData,\n          lastUpdated: Date.now(),\n        };\n      }\n    } catch (error) {\n      log('Failed to fetch prices', { error });\n    }\n\n    return response;\n  }\n\n  // ============================================================================\n  // SUBSCRIBE\n  // ============================================================================\n\n  /**\n   * Subscribe to price updates.\n   * Sets up polling that fetches prices for all assets in assetsBalance state.\n   *\n   * @param subscriptionRequest - The subscription request configuration.\n   */\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Handle subscription update - just update the request\n    if (isUpdate) {\n      const existing = this.#activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        existing.request = request;\n        return;\n      }\n    }\n\n    // Clean up existing subscription\n    await this.unsubscribe(subscriptionId);\n\n    const pollInterval = request.updateInterval ?? this.#pollInterval;\n\n    // Create poll function - fetches prices using getAssetsState from subscription\n    const pollFn = async (): Promise<void> => {\n      try {\n        const subscription = this.#activeSubscriptions.get(subscriptionId);\n        if (!subscription) {\n          return;\n        }\n\n        // Fetch prices for all assets in balance state (uses subscription's getAssetsState)\n        const fetchResponse = await this.fetch(\n          subscription.request,\n          subscription.getAssetsState,\n        );\n\n        // Only report if we got prices\n        if (\n          fetchResponse.assetsPrice &&\n          Object.keys(fetchResponse.assetsPrice).length > 0\n        ) {\n          await subscription.onAssetsUpdate(fetchResponse);\n        }\n      } catch (error) {\n        log('Subscription poll failed', { subscriptionId, error });\n      }\n    };\n\n    // Set up polling\n    const timer = setInterval(() => {\n      pollFn().catch(console.error);\n    }, pollInterval);\n\n    // Store subscription (getAssetsState from request for balance-based pricing)\n    this.#activeSubscriptions.set(subscriptionId, {\n      cleanup: () => {\n        clearInterval(timer);\n      },\n      request,\n      onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n      getAssetsState: subscriptionRequest.getAssetsState,\n    });\n\n    // Initial fetch\n    await pollFn();\n  }\n\n  /**\n   * Unsubscribe from price updates.\n   *\n   * @param subscriptionId - The ID of the subscription to cancel.\n   */\n  async unsubscribe(subscriptionId: string): Promise<void> {\n    const subscription = this.#activeSubscriptions.get(subscriptionId);\n    if (subscription) {\n      subscription.cleanup();\n      this.#activeSubscriptions.delete(subscriptionId);\n    }\n  }\n\n  /**\n   * Destroy the data source and clean up all subscriptions.\n   */\n  destroy(): void {\n    for (const subscription of this.#activeSubscriptions.values()) {\n      subscription.cleanup();\n    }\n    this.#activeSubscriptions.clear();\n  }\n}\n"]}

package/dist/data-sources/RpcDataSource.cjs

@@ -13,14 +13,14 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
-var _RpcDataSource_instances, _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_registerActionHandlers, _RpcDataSource_subscribeToNetworkController, _RpcDataSource_initializeFromNetworkController, _RpcDataSource_updateFromNetworkState, _RpcDataSource_getProvider, _RpcDataSource_getMulticallProvider, _RpcDataSource_clearProviderCache, _RpcDataSource_accountSupportsChain, _RpcDataSource_buildNativeAssetId, _RpcDataSource_getExistingAssetsMetadata, _RpcDataSource_getTokenMetadataFromTokenList;
+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;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createRpcDataSource = exports.RpcDataSource = exports.caipChainIdToHex = void 0;
 const providers_1 = require("@ethersproject/providers");
-const base_controller_1 = require("@metamask/base-controller");
 const controller_utils_1 = require("@metamask/controller-utils");
 const utils_1 = require("@metamask/utils");
 const bignumber_js_1 = __importDefault(require("bignumber.js"));
+const AbstractDataSource_1 = require("./AbstractDataSource.cjs");
 const evm_rpc_services_1 = require("./evm-rpc-services/index.cjs");
 const logger_1 = require("../logger.cjs");
 const CONTROLLER_NAME = 'RpcDataSource';
@@ -61,15 +61,12 @@ exports.caipChainIdToHex = caipChainIdToHex;
  * - RpcDataSource:activeChainsUpdated
  * - RpcDataSource:assetsUpdated
  */
-class RpcDataSource extends base_controller_1.BaseController {
+class RpcDataSource extends AbstractDataSource_1.AbstractDataSource {
     constructor(options) {
-        super({
-            name: CONTROLLER_NAME,
-            metadata: {},
-            state: {},
-            messenger: options.messenger,
-        });
+        super(CONTROLLER_NAME, { activeChains: [] });
         _RpcDataSource_instances.add(this);
+        _RpcDataSource_messenger.set(this, void 0);
+        _RpcDataSource_onActiveChainsUpdated.set(this, void 0);
         _RpcDataSource_timeout.set(this, void 0);
         _RpcDataSource_tokenDetectionEnabled.set(this, void 0);
         /** Currently active chains */
@@ -84,6 +81,8 @@ class RpcDataSource extends base_controller_1.BaseController {
         _RpcDataSource_multicallClient.set(this, void 0);
         _RpcDataSource_balanceFetcher.set(this, void 0);
         _RpcDataSource_tokenDetector.set(this, void 0);
+        __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");
         const balanceInterval = options.balanceInterval ?? DEFAULT_BALANCE_INTERVAL;
@@ -101,7 +100,7 @@ class RpcDataSource extends base_controller_1.BaseController {
         // Create messenger adapters for BalanceFetcher and TokenDetector
         const balanceFetcherMessenger = {
             call: (_action) => {
-                const state = this.messenger.call('AssetsController:getState');
+                const state = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('AssetsController:getState');
                 return {
                     assetsBalance: (state.assetsBalance ?? {}),
                 };
@@ -109,7 +108,7 @@ class RpcDataSource extends base_controller_1.BaseController {
         };
         const tokenDetectorMessenger = {
             call: (_action) => {
-                return this.messenger.call('TokenListController:getState');
+                return __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('TokenListController:getState');
             },
         };
         // Initialize BalanceFetcher with polling interval
@@ -118,7 +117,6 @@ class RpcDataSource extends base_controller_1.BaseController {
         // Initialize TokenDetector with polling interval
         __classPrivateFieldSet(this, _RpcDataSource_tokenDetector, new evm_rpc_services_1.TokenDetector(__classPrivateFieldGet(this, _RpcDataSource_multicallClient, "f"), tokenDetectorMessenger, { pollingInterval: detectionInterval }), "f");
         __classPrivateFieldGet(this, _RpcDataSource_tokenDetector, "f").setOnDetectionUpdate(__classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_handleDetectionUpdate).bind(this));
-        __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_registerActionHandlers).call(this);
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_subscribeToNetworkController).call(this);
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_initializeFromNetworkController).call(this);
     }
@@ -127,17 +125,6 @@ class RpcDataSource extends base_controller_1.BaseController {
      *
      * @returns The name of this data source.
      */
-    getName() {
-        return this.name;
-    }
-    /**
-     * Get currently active chains.
-     *
-     * @returns Promise resolving to array of active chain IDs.
-     */
-    async getActiveChains() {
-        return __classPrivateFieldGet(this, _RpcDataSource_activeChains, "f");
-    }
     /**
      * Get the status of all configured chains.
      *
@@ -193,7 +180,7 @@ class RpcDataSource extends base_controller_1.BaseController {
         const response = {};
         const chainsToFetch = request.chainIds.filter((chainId) => __classPrivateFieldGet(this, _RpcDataSource_activeChains, "f").includes(chainId));
         log('Fetch requested', {
-            accounts: request.accounts.map((a) => a.id),
+            accounts: request.accountsWithSupportedChains.map((a) => a.account.id),
             requestedChains: request.chainIds,
             chainsToFetch,
         });
@@ -204,14 +191,15 @@ class RpcDataSource extends base_controller_1.BaseController {
         const assetsBalance = {};
         const assetsMetadata = {};
         const failedChains = [];
-        // Fetch balances for each chain using BalanceFetcher
-        for (const chainId of chainsToFetch) {
-            const hexChainId = (0, exports.caipChainIdToHex)(chainId);
-            for (const account of request.accounts) {
-                if (!__classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_accountSupportsChain).call(this, account, chainId)) {
-                    continue;
-                }
-                const { address, id: accountId } = account;
+        // Fetch balances for each account and its supported chains (pre-computed in request)
+        for (const { account, supportedChains, } of request.accountsWithSupportedChains) {
+            const chainsForAccount = chainsToFetch.filter((chain) => supportedChains.includes(chain));
+            if (chainsForAccount.length === 0) {
+                continue;
+            }
+            const { address, id: accountId } = account;
+            for (const chainId of chainsForAccount) {
+                const hexChainId = (0, exports.caipChainIdToHex)(chainId);
                 try {
                     // Use BalanceFetcher for batched balance fetching
                     const result = await __classPrivateFieldGet(this, _RpcDataSource_balanceFetcher, "f").fetchBalancesForTokens(hexChainId, accountId, address, [], // Empty array means just native token
@@ -359,7 +347,7 @@ class RpcDataSource extends base_controller_1.BaseController {
             let successfullyHandledChains = [];
             log('Middleware fetching', {
                 chains: supportedChains,
-                accounts: request.accounts.map((a) => a.id),
+                accounts: request.accountsWithSupportedChains.map((a) => a.account.id),
             });
             const response = await this.fetch({
                 ...request,
@@ -409,7 +397,7 @@ class RpcDataSource extends base_controller_1.BaseController {
         log('Subscribe requested', {
             subscriptionId,
             isUpdate,
-            accounts: request.accounts.map((a) => a.id),
+            accounts: request.accountsWithSupportedChains.map((a) => a.account.id),
             chainsToSubscribe,
         });
         if (chainsToSubscribe.length === 0) {
@@ -430,16 +418,17 @@ class RpcDataSource extends base_controller_1.BaseController {
         }
         // Clean up existing subscription (stops old polling)
         await this.unsubscribe(subscriptionId);
-        // Start polling through BalanceFetcher and TokenDetector
+        // Start polling through BalanceFetcher and TokenDetector (use pre-computed supportedChains per account)
         const balancePollingTokens = [];
         const detectionPollingTokens = [];
-        for (const chainId of chainsToSubscribe) {
-            const hexChainId = (0, exports.caipChainIdToHex)(chainId);
-            for (const account of request.accounts) {
-                if (!__classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_accountSupportsChain).call(this, account, chainId)) {
-                    continue;
-                }
-                const { address, id: accountId } = account;
+        for (const { account, supportedChains, } of request.accountsWithSupportedChains) {
+            const chainsForAccount = chainsToSubscribe.filter((chain) => supportedChains.includes(chain));
+            if (chainsForAccount.length === 0) {
+                continue;
+            }
+            const { address, id: accountId } = account;
+            for (const chainId of chainsForAccount) {
+                const hexChainId = (0, exports.caipChainIdToHex)(chainId);
                 // Start balance polling
                 const balanceInput = {
                     chainId: hexChainId,
@@ -461,11 +450,13 @@ class RpcDataSource extends base_controller_1.BaseController {
             }
         }
         // Store subscription data
+        const accounts = request.accountsWithSupportedChains.map((entry) => entry.account);
         __classPrivateFieldGet(this, _RpcDataSource_activeSubscriptions, "f").set(subscriptionId, {
             balancePollingTokens,
             detectionPollingTokens,
             chains: chainsToSubscribe,
-            accounts: request.accounts,
+            accounts,
+            onAssetsUpdate: subscriptionRequest.onAssetsUpdate,
         });
         log('Subscription SUCCESS', {
             subscriptionId,
@@ -509,7 +500,7 @@ class RpcDataSource extends base_controller_1.BaseController {
     }
 }
 exports.RpcDataSource = RpcDataSource;
-_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_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) {
     const rawAmount = new bignumber_js_1.default(rawBalance);
     const divisor = new bignumber_js_1.default(10).pow(decimals);
     return rawAmount.dividedBy(divisor).toString();
@@ -585,11 +576,11 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
         accountId: result.accountId,
         newBalanceCount: Object.keys(newBalances).length,
     });
-    this.messenger
-        .call('AssetsController:assetsUpdate', response, CONTROLLER_NAME)
-        .catch((error) => {
-        log('Failed to update assets', { error });
-    });
+    for (const subscription of __classPrivateFieldGet(this, _RpcDataSource_activeSubscriptions, "f").values()) {
+        subscription.onAssetsUpdate(response)?.catch((error) => {
+            log('Failed to update assets', { error });
+        });
+    }
 }, _RpcDataSource_handleDetectionUpdate = function _RpcDataSource_handleDetectionUpdate(result) {
     log('Detected new tokens', {
         count: result.detectedAssets.length,
@@ -635,24 +626,13 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
             [result.accountId]: newBalances,
         },
     };
-    this.messenger
-        .call('AssetsController:assetsUpdate', response, CONTROLLER_NAME)
-        .catch((error) => {
-        log('Failed to update detected assets', { error });
-    });
-}, _RpcDataSource_registerActionHandlers = function _RpcDataSource_registerActionHandlers() {
-    const getAssetsMiddlewareHandler = () => this.assetsMiddleware;
-    const getActiveChainsHandler = async () => this.getActiveChains();
-    const fetchHandler = async (request) => this.fetch(request);
-    const subscribeHandler = async (request) => this.subscribe(request);
-    const unsubscribeHandler = async (subscriptionId) => this.unsubscribe(subscriptionId);
-    this.messenger.registerActionHandler('RpcDataSource:getAssetsMiddleware', getAssetsMiddlewareHandler);
-    this.messenger.registerActionHandler('RpcDataSource:getActiveChains', getActiveChainsHandler);
-    this.messenger.registerActionHandler('RpcDataSource:fetch', fetchHandler);
-    this.messenger.registerActionHandler('RpcDataSource:subscribe', subscribeHandler);
-    this.messenger.registerActionHandler('RpcDataSource:unsubscribe', unsubscribeHandler);
+    for (const subscription of __classPrivateFieldGet(this, _RpcDataSource_activeSubscriptions, "f").values()) {
+        subscription.onAssetsUpdate(response)?.catch((error) => {
+            log('Failed to update detected assets', { error });
+        });
+    }
 }, _RpcDataSource_subscribeToNetworkController = function _RpcDataSource_subscribeToNetworkController() {
-    this.messenger.subscribe('NetworkController:stateChange', (networkState) => {
+    __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").subscribe('NetworkController:stateChange', (networkState) => {
         log('NetworkController state changed');
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_clearProviderCache).call(this);
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_updateFromNetworkState).call(this, networkState);
@@ -660,7 +640,7 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
 }, _RpcDataSource_initializeFromNetworkController = function _RpcDataSource_initializeFromNetworkController() {
     log('Initializing from NetworkController');
     try {
-        const networkState = this.messenger.call('NetworkController:getState');
+        const networkState = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('NetworkController:getState');
         __classPrivateFieldGet(this, _RpcDataSource_instances, "m", _RpcDataSource_updateFromNetworkState).call(this, networkState);
     }
     catch (error) {
@@ -703,8 +683,9 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
     __classPrivateFieldSet(this, _RpcDataSource_chainStatuses, chainStatuses, "f");
     __classPrivateFieldSet(this, _RpcDataSource_activeChains, activeChains, "f");
     if (hasChanges) {
-        this.messenger.call('AssetsController:activeChainsUpdate', CONTROLLER_NAME, activeChains);
+        __classPrivateFieldGet(this, _RpcDataSource_onActiveChainsUpdated, "f").call(this, activeChains);
     }
+    this.state.activeChains = activeChains;
 }, _RpcDataSource_getProvider = function _RpcDataSource_getProvider(chainId) {
     const cached = __classPrivateFieldGet(this, _RpcDataSource_providerCache, "f").get(chainId);
     if (cached) {
@@ -715,7 +696,10 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
         return undefined;
     }
     try {
-        const networkClient = this.messenger.call('NetworkController:getNetworkClientById', chainStatus.networkClientId);
+        const networkClient = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('NetworkController:getNetworkClientById', chainStatus.networkClientId);
+        if (!networkClient?.provider) {
+            return undefined;
+        }
         const web3Provider = new providers_1.Web3Provider(networkClient.provider);
         __classPrivateFieldGet(this, _RpcDataSource_providerCache, "f").set(chainId, web3Provider);
         return web3Provider;
@@ -745,38 +729,13 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
     };
 }, _RpcDataSource_clearProviderCache = function _RpcDataSource_clearProviderCache() {
     __classPrivateFieldGet(this, _RpcDataSource_providerCache, "f").clear();
-}, _RpcDataSource_accountSupportsChain = function _RpcDataSource_accountSupportsChain(account, chainId) {
-    const scopes = account.scopes ?? [];
-    if (scopes.length === 0) {
-        return true;
-    }
-    const [chainNamespace, chainReference] = chainId.split(':');
-    for (const scope of scopes) {
-        const [scopeNamespace, scopeReference] = scope.split(':');
-        if (scopeNamespace !== chainNamespace) {
-            continue;
-        }
-        // Wildcard scope (e.g., eip155:0) matches all chains in the namespace
-        if (scopeReference === '0') {
-            return true;
-        }
-        // RpcDataSource only handles eip155 (EVM) chains
-        // Normalize hex chain references (e.g., 0x1 -> 1) for comparison
-        const normalizedScopeRef = scopeReference?.startsWith('0x')
-            ? parseInt(scopeReference, 16).toString()
-            : scopeReference;
-        if (normalizedScopeRef === chainReference) {
-            return true;
-        }
-    }
-    return false;
 }, _RpcDataSource_buildNativeAssetId = function _RpcDataSource_buildNativeAssetId(chainId) {
-    const { nativeAssetIdentifiers } = this.messenger.call('NetworkEnablementController:getState');
+    const { nativeAssetIdentifiers } = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('NetworkEnablementController:getState');
     return (nativeAssetIdentifiers[chainId] ??
         `${chainId}/slip44:60`);
 }, _RpcDataSource_getExistingAssetsMetadata = function _RpcDataSource_getExistingAssetsMetadata() {
     try {
-        const state = this.messenger.call('AssetsController:getState');
+        const state = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('AssetsController:getState');
         return (state.assetsMetadata ?? {});
     }
     catch {
@@ -794,8 +753,8 @@ _RpcDataSource_timeout = new WeakMap(), _RpcDataSource_tokenDetectionEnabled = n
         const tokenAddress = assetPart.slice(6); // Remove 'erc20:' prefix
         const chainIdDecimal = chainPart.split(':')[1];
         const hexChainId = `0x${parseInt(chainIdDecimal, 10).toString(16)}`;
-        const tokenListState = this.messenger.call('TokenListController:getState');
-        const chainCacheEntry = tokenListState.tokensChainsCache[hexChainId];
+        const tokenListState = __classPrivateFieldGet(this, _RpcDataSource_messenger, "f").call('TokenListController:getState');
+        const chainCacheEntry = tokenListState?.tokensChainsCache?.[hexChainId];
         const chainTokenList = chainCacheEntry?.data;
         if (!chainTokenList) {
             return undefined;