@metamask-previews/assets-controller 2.0.2-preview-3d4d0d0ef → 2.0.2-preview-835c0b0

package/dist/data-sources/BackendWebsocketDataSource.mjs

@@ -11,6 +11,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
 };
 var _BackendWebsocketDataSource_instances, _BackendWebsocketDataSource_messenger, _BackendWebsocketDataSource_apiClient, _BackendWebsocketDataSource_onActiveChainsUpdated, _BackendWebsocketDataSource_chainsRefreshTimer, _BackendWebsocketDataSource_wsSubscriptions, _BackendWebsocketDataSource_pendingSubscriptions, _BackendWebsocketDataSource_subscriptionRequests, _BackendWebsocketDataSource_initializeActiveChains, _BackendWebsocketDataSource_refreshActiveChains, _BackendWebsocketDataSource_fetchActiveChains, _BackendWebsocketDataSource_subscribeToEvents, _BackendWebsocketDataSource_handleDisconnect, _BackendWebsocketDataSource_processPendingSubscriptions, _BackendWebsocketDataSource_handleNotification, _BackendWebsocketDataSource_processBalanceUpdates;
 import { isCaipChainId, KnownCaipNamespace, toCaipChainId } from "@metamask/utils";
+import BigNumberJS from "bignumber.js";
 import { AbstractDataSource } from "./AbstractDataSource.mjs";
 import { projectLogger, createModuleLogger } from "../logger.mjs";
 // ============================================================================
@@ -442,12 +443,20 @@ async function _BackendWebsocketDataSource_processPendingSubscriptions() {
         // Determine token type from asset type string
         const isNative = asset.type.includes('/slip44:');
         const tokenType = isNative ? 'native' : 'erc20';
-        // Parse balance amount (already in hex format like "0xc350")
-        const balanceAmount = postBalance.amount.startsWith('0x')
+        // We assume decimals are always present; skip malformed updates
+        if (asset.decimals === undefined) {
+            continue;
+        }
+        // Parse raw balance (hex like "0x26f0e5" or decimal string)
+        const rawBalanceStr = postBalance.amount.startsWith('0x')
             ? BigInt(postBalance.amount).toString()
             : postBalance.amount;
+        // Convert to human-readable using asset decimals (match RpcDataSource / pipeline format)
+        const humanReadableAmount = new BigNumberJS(rawBalanceStr)
+            .dividedBy(new BigNumberJS(10).pow(asset.decimals))
+            .toString();
         assetsBalance[accountId][assetId] = {
-            amount: balanceAmount,
+            amount: humanReadableAmount,
         };
         assetsMetadata[assetId] = {
             type: tokenType,
@@ -456,7 +465,7 @@ async function _BackendWebsocketDataSource_processPendingSubscriptions() {
             decimals: asset.decimals,
         };
     }
-    const response = {};
+    const response = { updateMode: 'merge' };
     if (Object.keys(assetsBalance[accountId]).length > 0) {
         response.assetsBalance = assetsBalance;
         response.assetsInfo = assetsMetadata;

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

@@ -1 +1 @@
-{"version":3,"file":"BackendWebsocketDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/BackendWebsocketDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAUA,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,aAAa,EACd,wBAAwB;AAEzB,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAM1D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAS9D,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,4BAA4B,CAAC;AACrD,MAAM,YAAY,GAAG,qBAAqB,CAAC;AAE3C,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAoB/D,MAAM,YAAY,GAAoC;IACpD,YAAY,EAAE,EAAE;CACjB,CAAC;AAoBF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;GAMG;AACH,SAAS,gBAAgB,CAAC,OAAgB;IACxC,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,6EAA6E;AAC7E,MAAM,2BAA2B,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAU,CAAC;AAElE;;;;;;;GAOG;AACH,SAAS,+BAA+B,CAAC,QAAmB;IAC1D,MAAM,UAAU,GAAG,IAAI,GAAG,CAAS,2BAA2B,CAAC,CAAC;IAChE,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,UAAU,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AAChC,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,4BAA4B,CACnC,OAA0C,EAC1C,SAAiB;IAEjB,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACpE,CAAC;IACD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACpE,CAAC;IACD,uFAAuF;IACvF,MAAM,UAAU,GAAG,GAAG,SAAS,GAAG,CAAC;IACnC,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;AACtE,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,2BAA2B,CAClC,SAAiB,EACjB,OAAe;IAEf,MAAM,SAAS,GAAG,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC3E,OAAO,GAAG,YAAY,IAAI,SAAS,MAAM,SAAS,EAAE,CAAC;AACvD,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,SAAS,CAAC,gBAAiC;IAClD,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;QACzC,IAAI,aAAa,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACpC,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,OAAO,aAAa,CAAC,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,aAAa,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,gGAAgG;AAEhG,+EAA+E;AAC/E,gCAAgC;AAChC,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,kCAAkC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAExE,MAAM,OAAO,0BAA2B,SAAQ,kBAG/C;IAuBC,YAAY,OAA0C;QACpD,KAAK,CAAC,eAAe,EAAE;YACrB,GAAG,YAAY;YACf,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QA1BI,wDAAsC;QAEtC,wDAA8B;QAE9B,oEAIC;QAEV,2BAA2B;QAC3B,yDAA6D,IAAI,EAAC;QAElE,8DAA8D;QACrD,sDAAuD,IAAI,GAAG,EAAE,EAAC;QAE1E,uEAAuE;QAC9D,2DAA0D,IAAI,GAAG,EAAE,EAAC;QAE7E,4DAA4D;QACnD,2DAA0D,IAAI,GAAG,EAAE,EAAC;QAQ3E,uBAAA,IAAI,yCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,yCAAc,OAAO,CAAC,cAAc,MAAA,CAAC;QACzC,uBAAA,IAAI,qDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAE5D,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1B,uBAAA,IAAI,iGAAwB,MAA5B,IAAI,CAA0B,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAyED;;;;;;OAMG;IACH,8BAA8B,CAAC,MAAiB;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC;IAwDD,+EAA+E;IAC/E,gBAAgB;IAChB,+EAA+E;IAE/E;;;;OAIG;IACH,qBAAqB,CAAC,MAAiB;QACrC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,YAAY;IACZ,+EAA+E;IAE/E,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,+BAA+B;QAC/B,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;QAEF,MAAM,SAAS,GAAG,OAAO,CAAC,2BAA2B,CAAC,GAAG,CACvD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzB,CAAC;QAEF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,uBAAA,IAAI,6CAAW,CAAC,IAAI,CACzC,2CAA2C,CAC5C,CAAC;YACF,IAAI,cAAc,CAAC,KAAK,KAAM,WAA8B,EAAE,CAAC;gBAC7D,oEAAoE;gBACpE,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;YACvD,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;YACpE,OAAO;QACT,CAAC;QAED,gEAAgE;QAChE,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAElD,6BAA6B;QAC7B,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9D,IAAI,QAAQ,EAAE,CAAC;gBACb,mFAAmF;gBACnF,MAAM,iBAAiB,GAAG,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;gBACnD,MAAM,gBAAgB,GACpB,SAAS,CAAC,MAAM,KAAK,iBAAiB,CAAC,MAAM;oBAC7C,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAE9D,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACtB,sDAAsD;oBACtD,QAAQ,CAAC,MAAM,GAAG,iBAAiB,CAAC;oBACpC,OAAO;gBACT,CAAC;gBACD,oEAAoE;YACtE,CAAC;QACH,CAAC;QAED,wCAAwC;QACxC,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,oGAAoG;QACpG,MAAM,UAAU,GAAG,+BAA+B,CAAC,iBAAiB,CAAC,CAAC;QAEtE,qGAAqG;QACrG,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;gBAC9D,MAAM,OAAO,GAAG,4BAA4B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACjE,IAAI,OAAO,EAAE,CAAC;oBACZ,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,gCAAgC;YAChC,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,6CAAW,CAAC,IAAI,CAC/C,mCAAmC,EACnC;gBACE,QAAQ;gBACR,WAAW,EAAE,YAAY;gBACzB,QAAQ,EAAE,CAAC,YAAuC,EAAE,EAAE;oBACpD,uBAAA,IAAI,6FAAoB,MAAxB,IAAI,EAAqB,YAAY,EAAE,cAAc,CAAC,CAAC;gBACzD,CAAC;aACF,CACF,CAAC;YAEF,+BAA+B;YAC/B,uBAAA,IAAI,mDAAiB,CAAC,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;YAE1D,mCAAmC;YACnC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;gBAC3C,OAAO,EAAE,GAAG,EAAE;oBACZ,MAAM,KAAK,GAAG,uBAAA,IAAI,mDAAiB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;oBACxD,IAAI,KAAK,EAAE,CAAC;wBACV,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,QAAiB,EAAE,EAAE;4BAC9C,GAAG,CAAC,qBAAqB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAClE,CAAC,CAAC,CAAC;wBACH,uBAAA,IAAI,mDAAiB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;oBAC/C,CAAC;oBACD,mCAAmC;oBACnC,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACpD,CAAC;gBACD,MAAM,EAAE,iBAAiB;gBACzB,SAAS;gBACT,cAAc,EAAE,mBAAmB,CAAC,cAAc;aACnD,CAAC,CAAC;YAEH,0CAA0C;YAC1C,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+BAA+B,EAAE;gBACnC,cAAc;gBACd,KAAK;gBACL,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAkHD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,IAAI,uBAAA,IAAI,sDAAoB,EAAE,CAAC;YAC7B,aAAa,CAAC,uBAAA,IAAI,sDAAoB,CAAC,CAAC;YACxC,uBAAA,IAAI,kDAAuB,IAAI,MAAA,CAAC;QAClC,CAAC;QAED,mCAAmC;QACnC,iEAAiE;QACjE,MAAM,aAAa,GAAG,CAAC,GAAG,uBAAA,IAAI,mDAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1D,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,2CAA2C;gBAC3C,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;oBAC7B,+BAA+B;gBACjC,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,mDAAiB,CAAC,KAAK,EAAE,CAAC;QAE9B,oCAAoC;QACpC,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;;AA5ZC,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,KAAK;IACH,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC/C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QAEF,uBAAA,IAAI,kDAAuB,WAAW,CAAC,GAAG,EAAE;YAC1C,uBAAA,IAAI,8FAAqB,MAAzB,IAAI,CAAuB,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,EAAE,kCAAkC,CAAC,MAAA,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC,oDAED,KAAK;IACH,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC/C,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAElC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,CAC/C,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CACjC,CAAC;QAEF,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;AACH,CAAC,kDAED,KAAK;IACH,MAAM,QAAQ,GAAG,MAAM,uBAAA,IAAI,6CAAW,CAAC,QAAQ,CAAC,wBAAwB,EAAE,CAAC;IAC3E,OAAO,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;IAOC,uFAAuF;IAErF,uBAAA,IAAI,6CAGL,CAAC,SAAS,CACT,gDAAgD,EAChD,CAAC,cAAsC,EAAE,EAAE;QACzC,IAAI,cAAc,CAAC,KAAK,KAAM,WAA8B,EAAE,CAAC;YAC7D,uBAAA,IAAI,sGAA6B,MAAjC,IAAI,CAA+B,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;aAAM,IACL,cAAc,CAAC,KAAK,KAAM,cAAiC,EAC3D,CAAC;YACD,uBAAA,IAAI,2FAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC3B,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC;IAkBC,GAAG,CAAC,gEAAgE,EAAE;QACpE,uBAAuB,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI;QACtD,mBAAmB,EAAE,uBAAA,IAAI,mDAAiB,CAAC,IAAI;KAChD,CAAC,CAAC;IAEH,2DAA2D;IAC3D,KAAK,MAAM,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACxD,MAAM,eAAe,GAAG,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACvE,IAAI,eAAe,EAAE,CAAC;YACpB,gDAAgD;YAChD,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE;gBAC7C,GAAG,eAAe;gBAClB,QAAQ,EAAE,KAAK,EAAE,oDAAoD;aACtE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,8DAA8D;IAC9D,uBAAA,IAAI,mDAAiB,CAAC,KAAK,EAAE,CAAC;IAE9B,uDAAuD;IACvD,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,KAAK;IACH,IAAI,uBAAA,IAAI,wDAAsB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IAED,oCAAoC;IACpC,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,uBAAA,IAAI,wDAAsB,CAAC,OAAO,EAAE,CAAC,CAAC;IAExE,KAAK,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,cAAc,EAAE,CAAC;QACvD,IAAI,CAAC;YACH,+DAA+D;YAC/D,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAClD,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,wCAAwC,EAAE;gBAC5C,cAAc;gBACd,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC,2GA+IC,YAAuC,EACvC,cAAsB;IAEtB,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,OAAO,GAAG,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;QACxE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,eAAe,GACnB,YAAY,CAAC,IAAyC,CAAC;QACzD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC;QAEjD,IAAI,CAAC,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QAED,yEAAyE;QACzE,MAAM,OAAO,GAAG,EAAE,CAAC,KAAgB,CAAC;QAEpC,wFAAwF;QACxF,MAAM,OAAO,GAAG,OAAO,CAAC,2BAA2B;aAChD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACV,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;YACxB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE;YACnD,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,CAC1B,CAAC;QACJ,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAE7B,wDAAwD;QACxD,MAAM,QAAQ,GAAG,uBAAA,IAAI,gGAAuB,MAA3B,IAAI,EAAwB,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAE1E,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,EAAE,CAAC;YACrD,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAC1D,OAAO,CAAC,KAAK,CACd,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC,iHAYC,OAAwB,EACxB,QAAiB,EACjB,SAAiB;IAEjB,MAAM,aAAa,GAAwD;QACzE,CAAC,SAAS,CAAC,EAAE,EAAE;KAChB,CAAC;IACF,MAAM,cAAc,GAAyC,EAAE,CAAC;IAEhE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAEtC,IAAI,CAAC,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;YAC3B,SAAS;QACX,CAAC;QAED,+EAA+E;QAC/E,yCAAyC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAqB,CAAC;QAE5C,8CAA8C;QAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACjD,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;QAEhD,6DAA6D;QAC7D,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;YACvD,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;YACvC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC;QAEvB,aAAa,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG;YAClC,MAAM,EAAE,aAAa;SACtB,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,GAAG;YACxB,IAAI,EAAE,SAAS;YACf,MAAM,EAAE,KAAK,CAAC,IAAI;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,2DAA2D;YAC7E,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAiB,EAAE,CAAC;IAClC,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACvC,QAAQ,CAAC,UAAU,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAgCH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,gCAAgC,CAC9C,OAA0C;IAE1C,OAAO,IAAI,0BAA0B,CAAC,OAAO,CAAC,CAAC;AACjD,CAAC","sourcesContent":["import type {\n  BackendWebSocketServiceActions,\n  BackendWebSocketServiceEvents,\n  ServerNotificationMessage,\n  WebSocketSubscription,\n  WebSocketState,\n  AccountActivityMessage,\n  BalanceUpdate,\n} from '@metamask/core-backend';\nimport type { ApiPlatformClient } from '@metamask/core-backend';\nimport {\n  isCaipChainId,\n  KnownCaipNamespace,\n  toCaipChainId,\n} from '@metamask/utils';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  ChainId,\n  Caip19AssetId,\n  AssetMetadata,\n  AssetBalance,\n  DataResponse,\n} from '../types';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'BackendWebsocketDataSource';\nconst CHANNEL_TYPE = 'account-activity.v1';\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\n// Allowed actions that BackendWebsocketDataSource can call\nexport type BackendWebsocketDataSourceAllowedActions =\n  BackendWebSocketServiceActions;\n\n// Allowed events that BackendWebsocketDataSource can subscribe to\nexport type BackendWebsocketDataSourceAllowedEvents =\n  BackendWebSocketServiceEvents;\n\n// ============================================================================\n// STATE\n// ============================================================================\n\nexport type BackendWebsocketDataSourceState = DataSourceState;\n\nconst defaultState: BackendWebsocketDataSourceState = {\n  activeChains: [],\n};\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type BackendWebsocketDataSourceOptions = {\n  /** The AssetsController messenger (shared by all data sources). */\n  messenger: AssetsControllerMessenger;\n  /** ApiPlatformClient for fetching supported networks at init (same as AccountsApiDataSource). */\n  queryApiClient: ApiPlatformClient;\n  /** Called when active chains are updated. Pass dataSourceName so the controller knows the source. */\n  onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n  state?: Partial<BackendWebsocketDataSourceState>;\n};\n\n// ============================================================================\n// HELPER FUNCTIONS\n// ============================================================================\n\n/**\n * Extract namespace from a CAIP-2 chain ID.\n * E.g., \"eip155:1\" -> \"eip155\", \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\" -> \"solana\"\n *\n * @param chainId - The CAIP-2 chain ID to extract namespace from.\n * @returns The namespace portion of the chain ID.\n */\nfunction extractNamespace(chainId: ChainId): string {\n  const [namespace] = chainId.split(':');\n  return namespace;\n}\n\n/** Namespaces we always subscribe to for account activity (EVM + Solana). */\nconst ACCOUNT_ACTIVITY_NAMESPACES = ['eip155', 'solana'] as const;\n\n/**\n * Get unique namespaces for account-activity subscriptions.\n * Always includes eip155 and solana so we subscribe to both EVM and Solana account activity,\n * plus any additional namespaces from the requested chain IDs.\n *\n * @param chainIds - Array of CAIP-2 chain IDs (from the subscription request).\n * @returns Array of unique namespaces (at least eip155 and solana).\n */\nfunction getNamespacesForAccountActivity(chainIds: ChainId[]): string[] {\n  const namespaces = new Set<string>(ACCOUNT_ACTIVITY_NAMESPACES);\n  for (const chainId of chainIds) {\n    namespaces.add(extractNamespace(chainId));\n  }\n  return Array.from(namespaces);\n}\n\n/**\n * Returns the address to use for account-activity subscription in the given namespace.\n * EIP-155 accounts use hex (0x...) address; Solana accounts use base58.\n * Returns null if this account type does not have an address in that namespace.\n *\n * @param account - Internal account (type + address).\n * @param account.type - Account type (e.g. \"eip155:eoa\", \"solana:data-account\").\n * @param account.address - Account address (hex for eip155, base58 for solana).\n * @param namespace - The chain namespace (e.g., \"eip155\", \"solana\").\n * @returns The address for that namespace, or null if the account does not support the namespace.\n */\nfunction getAddressForAccountActivity(\n  account: { type: string; address: string },\n  namespace: string,\n): string | null {\n  if (namespace === 'eip155') {\n    return account.type.startsWith('eip155') ? account.address : null;\n  }\n  if (namespace === 'solana') {\n    return account.type.startsWith('solana') ? account.address : null;\n  }\n  // Other namespaces (e.g. from chainIds): use address if account type matches namespace\n  const typePrefix = `${namespace}:`;\n  return account.type.startsWith(typePrefix) ? account.address : null;\n}\n\n/**\n * Build WebSocket channel name for account activity using CAIP-10 wildcard format.\n * Uses 0 as the chain reference to subscribe to all chains in the namespace.\n * EIP-155 addresses are lowercased (hex); Solana addresses are left as-is (base58).\n *\n * @param namespace - The chain namespace (e.g., \"eip155\", \"solana\").\n * @param address - The account address (hex for eip155, base58 for solana).\n * @returns The WebSocket channel name.\n */\nfunction buildAccountActivityChannel(\n  namespace: string,\n  address: string,\n): string {\n  const formatted = namespace === 'eip155' ? address.toLowerCase() : address;\n  return `${CHANNEL_TYPE}.${namespace}:0:${formatted}`;\n}\n\n/**\n * Normalize API chain identifier to CAIP-2 ChainId.\n * Passes through strings already in CAIP-2 form (e.g. eip155:1, solana:5eykt...).\n * Converts bare decimals to eip155:decimal.\n * Uses @metamask/utils for CAIP parsing.\n *\n * @param chainIdOrDecimal - Chain ID string (CAIP-2 or decimal) or decimal number.\n * @returns CAIP-2 ChainId.\n */\nfunction toChainId(chainIdOrDecimal: number | string): ChainId {\n  if (typeof chainIdOrDecimal === 'string') {\n    if (isCaipChainId(chainIdOrDecimal)) {\n      return chainIdOrDecimal;\n    }\n    return toCaipChainId(KnownCaipNamespace.Eip155, chainIdOrDecimal);\n  }\n  return toCaipChainId(KnownCaipNamespace.Eip155, String(chainIdOrDecimal));\n}\n\n// Note: AccountActivityMessage and BalanceUpdate types are imported from @metamask/core-backend\n\n// ============================================================================\n// BACKEND WEBSOCKET DATA SOURCE\n// ============================================================================\n\n/**\n * Data source for receiving real-time balance updates via WebSocket.\n *\n * This data source connects directly to BackendWebSocketService to receive\n * push notifications for account balance changes. Unlike AccountsApiDataSource\n * which polls for data, this provides instant updates.\n *\n * Uses Messenger pattern for all interactions:\n * - Calls BackendWebSocketService methods via messenger actions\n * - Exposes its own actions for AssetsController to call\n * - Publishes events for AssetsController to subscribe to\n *\n * Actions exposed:\n * - BackendWebsocketDataSource:getActiveChains\n * - BackendWebsocketDataSource:subscribe\n * - BackendWebsocketDataSource:unsubscribe\n *\n * Events published:\n * - BackendWebsocketDataSource:activeChainsUpdated\n * - BackendWebsocketDataSource:assetsUpdated\n *\n * Actions called (from BackendWebSocketService):\n * - BackendWebSocketService:subscribe\n * - BackendWebSocketService:getConnectionInfo\n * - BackendWebSocketService:findSubscriptionsByChannelPrefix\n */\nconst DEFAULT_CHAINS_REFRESH_INTERVAL_MS = 20 * 60 * 1000; // 20 minutes\n\nexport class BackendWebsocketDataSource extends AbstractDataSource<\n  typeof CONTROLLER_NAME,\n  BackendWebsocketDataSourceState\n> {\n  readonly #messenger: AssetsControllerMessenger;\n\n  readonly #apiClient: ApiPlatformClient;\n\n  readonly #onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n\n  /** Chains refresh timer */\n  #chainsRefreshTimer: ReturnType<typeof setInterval> | null = null;\n\n  /** WebSocket subscriptions by our internal subscription ID */\n  readonly #wsSubscriptions: Map<string, WebSocketSubscription> = new Map();\n\n  /** Pending subscription requests to process when WebSocket connects */\n  readonly #pendingSubscriptions: Map<string, SubscriptionRequest> = new Map();\n\n  /** Store original subscription requests for reconnection */\n  readonly #subscriptionRequests: Map<string, SubscriptionRequest> = new Map();\n\n  constructor(options: BackendWebsocketDataSourceOptions) {\n    super(CONTROLLER_NAME, {\n      ...defaultState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n    this.#apiClient = options.queryApiClient;\n    this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n\n    this.#subscribeToEvents();\n    this.#initializeActiveChains().catch(console.error);\n  }\n\n  // ============================================================================\n  // INITIALIZATION\n  // ============================================================================\n\n  async #initializeActiveChains(): Promise<void> {\n    try {\n      const chains = await this.#fetchActiveChains();\n      const previous = [...this.state.activeChains];\n      this.updateActiveChains(chains, (updatedChains) =>\n        this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n      );\n\n      this.#chainsRefreshTimer = setInterval(() => {\n        this.#refreshActiveChains().catch(console.error);\n      }, DEFAULT_CHAINS_REFRESH_INTERVAL_MS);\n    } catch (error) {\n      log('Failed to fetch active chains', error);\n    }\n  }\n\n  async #refreshActiveChains(): Promise<void> {\n    try {\n      const chains = await this.#fetchActiveChains();\n      const previousChains = new Set(this.state.activeChains);\n      const newChains = new Set(chains);\n\n      const added = chains.filter((chain) => !previousChains.has(chain));\n      const removed = Array.from(previousChains).filter(\n        (chain) => !newChains.has(chain),\n      );\n\n      if (added.length > 0 || removed.length > 0) {\n        const previous = [...this.state.activeChains];\n        this.updateActiveChains(chains, (updatedChains) =>\n          this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n        );\n      }\n    } catch (error) {\n      log('Failed to refresh active chains', error);\n    }\n  }\n\n  async #fetchActiveChains(): Promise<ChainId[]> {\n    const response = await this.#apiClient.accounts.fetchV2SupportedNetworks();\n    return response.fullSupport.map(toChainId);\n  }\n\n  #subscribeToEvents(): void {\n    type ConnectionStatePayload = {\n      state: WebSocketState;\n      [key: string]: unknown;\n    };\n    // Listen for WebSocket connection state changes (event not in AssetsControllerEvents).\n    (\n      this.#messenger as unknown as {\n        subscribe: (e: string, h: (p: ConnectionStatePayload) => void) => void;\n      }\n    ).subscribe(\n      'BackendWebSocketService:connectionStateChanged',\n      (connectionInfo: ConnectionStatePayload) => {\n        if (connectionInfo.state === ('connected' as WebSocketState)) {\n          this.#processPendingSubscriptions().catch(console.error);\n        } else if (\n          connectionInfo.state === ('disconnected' as WebSocketState)\n        ) {\n          this.#handleDisconnect();\n        }\n      },\n    );\n  }\n\n  /**\n   * Sync active chains from AccountsApiDataSource.\n   * Called by AssetsController.handleActiveChainsUpdate when the callback is\n   * invoked for BackendWebsocketDataSource (no messenger call; controller already updated).\n   *\n   * @param chains - Updated active chain IDs from AccountsApiDataSource.\n   */\n  setActiveChainsFromAccountsApi(chains: ChainId[]): void {\n    this.updateActiveChains(chains, () => undefined);\n  }\n\n  /**\n   * Handle WebSocket disconnection.\n   * Moves all active subscriptions to pending for re-subscription on reconnect.\n   */\n  #handleDisconnect(): void {\n    log('WebSocket disconnected, preserving subscriptions for reconnect', {\n      activeSubscriptionCount: this.activeSubscriptions.size,\n      wsSubscriptionCount: this.#wsSubscriptions.size,\n    });\n\n    // Move active subscriptions to pending for re-subscription\n    for (const [subscriptionId] of this.activeSubscriptions) {\n      const originalRequest = this.#subscriptionRequests.get(subscriptionId);\n      if (originalRequest) {\n        // Mark as update since it was previously active\n        this.#pendingSubscriptions.set(subscriptionId, {\n          ...originalRequest,\n          isUpdate: false, // Treat as new subscription since server cleared it\n        });\n      }\n    }\n\n    // Clear WebSocket subscriptions (server-side already cleared)\n    this.#wsSubscriptions.clear();\n\n    // Clear active subscriptions (they're no longer valid)\n    this.activeSubscriptions.clear();\n  }\n\n  /**\n   * Process any pending subscriptions that were queued while WebSocket was disconnected.\n   */\n  async #processPendingSubscriptions(): Promise<void> {\n    if (this.#pendingSubscriptions.size === 0) {\n      return;\n    }\n\n    // Process all pending subscriptions\n    const pendingEntries = Array.from(this.#pendingSubscriptions.entries());\n\n    for (const [subscriptionId, request] of pendingEntries) {\n      try {\n        // Remove from pending before processing to avoid infinite loop\n        this.#pendingSubscriptions.delete(subscriptionId);\n        await this.subscribe(request);\n      } catch (error) {\n        log('Failed to process pending subscription', {\n          subscriptionId,\n          error,\n        });\n      }\n    }\n  }\n\n  // ============================================================================\n  // ACTIVE CHAINS\n  // ============================================================================\n\n  /**\n   * Update active chains when AccountsApiDataSource reports new supported chains.\n   *\n   * @param chains - Array of supported chain IDs.\n   */\n  updateSupportedChains(chains: ChainId[]): void {\n    const previous = [...this.state.activeChains];\n    this.updateActiveChains(chains, (updatedChains) =>\n      this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n    );\n  }\n\n  // ============================================================================\n  // SUBSCRIBE\n  // ============================================================================\n\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Filter to active chains only\n    const chainsToSubscribe = request.chainIds.filter((chainId) =>\n      this.state.activeChains.includes(chainId),\n    );\n\n    const addresses = request.accountsWithSupportedChains.map(\n      (a) => a.account.address,\n    );\n\n    if (addresses.length === 0) {\n      return;\n    }\n\n    // Check WebSocket connection status\n    try {\n      const connectionInfo = this.#messenger.call(\n        'BackendWebSocketService:getConnectionInfo',\n      );\n      if (connectionInfo.state !== ('connected' as WebSocketState)) {\n        // Store the subscription request to process when WebSocket connects\n        this.#pendingSubscriptions.set(subscriptionId, subscriptionRequest);\n        return;\n      }\n    } catch {\n      // Store anyway - will be processed when we can connect\n      this.#pendingSubscriptions.set(subscriptionId, subscriptionRequest);\n      return;\n    }\n\n    // Remove from pending if it was there (we're processing it now)\n    this.#pendingSubscriptions.delete(subscriptionId);\n\n    // Handle subscription update\n    if (isUpdate) {\n      const existing = this.activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        // Check if accounts changed - if so, we need to re-subscribe to different channels\n        const existingAddresses = existing.addresses ?? [];\n        const addressesChanged =\n          addresses.length !== existingAddresses.length ||\n          addresses.some((addr) => !existingAddresses.includes(addr));\n\n        if (!addressesChanged) {\n          // Only chains changed - just update chains and return\n          existing.chains = chainsToSubscribe;\n          return;\n        }\n        // Accounts changed - fall through to re-subscribe with new channels\n      }\n    }\n\n    // Clean up existing subscription if any\n    await this.unsubscribe(subscriptionId);\n\n    // Always subscribe to eip155 and solana account activity, plus any namespaces from requested chains\n    const namespaces = getNamespacesForAccountActivity(chainsToSubscribe);\n\n    // Build channel names: use namespace-appropriate address per account (eip155 = hex, solana = base58)\n    const channels: string[] = [];\n    for (const namespace of namespaces) {\n      for (const { account } of request.accountsWithSupportedChains) {\n        const address = getAddressForAccountActivity(account, namespace);\n        if (address) {\n          channels.push(buildAccountActivityChannel(namespace, address));\n        }\n      }\n    }\n\n    try {\n      // Create WebSocket subscription\n      const wsSubscription = await this.#messenger.call(\n        'BackendWebSocketService:subscribe',\n        {\n          channels,\n          channelType: CHANNEL_TYPE,\n          callback: (notification: ServerNotificationMessage) => {\n            this.#handleNotification(notification, subscriptionId);\n          },\n        },\n      );\n\n      // Store WebSocket subscription\n      this.#wsSubscriptions.set(subscriptionId, wsSubscription);\n\n      // Store in abstract class tracking\n      this.activeSubscriptions.set(subscriptionId, {\n        cleanup: () => {\n          const wsSub = this.#wsSubscriptions.get(subscriptionId);\n          if (wsSub) {\n            wsSub.unsubscribe().catch((unsubErr: unknown) => {\n              log('Error unsubscribing', { subscriptionId, error: unsubErr });\n            });\n            this.#wsSubscriptions.delete(subscriptionId);\n          }\n          // Also clean up the stored request\n          this.#subscriptionRequests.delete(subscriptionId);\n        },\n        chains: chainsToSubscribe,\n        addresses,\n        onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n      });\n\n      // Store original request for reconnection\n      this.#subscriptionRequests.set(subscriptionId, subscriptionRequest);\n    } catch (error) {\n      log('WebSocket subscription FAILED', {\n        subscriptionId,\n        error,\n        chains: chainsToSubscribe,\n      });\n    }\n  }\n\n  // ============================================================================\n  // NOTIFICATION HANDLING\n  // ============================================================================\n\n  #handleNotification(\n    notification: ServerNotificationMessage,\n    subscriptionId: string,\n  ): void {\n    try {\n      const subscription = this.activeSubscriptions.get(subscriptionId);\n      const request = this.#subscriptionRequests.get(subscriptionId)?.request;\n      if (!request) {\n        return;\n      }\n\n      const activityMessage =\n        notification.data as unknown as AccountActivityMessage;\n      const { address, tx, updates } = activityMessage;\n\n      if (!address || !tx || !updates) {\n        return;\n      }\n\n      // Extract chain ID from transaction (CAIP-2 format, e.g., \"eip155:8453\")\n      const chainId = tx.chain as ChainId;\n\n      // Find matching account in request (eip155: case-insensitive hex; solana: exact base58)\n      const account = request.accountsWithSupportedChains\n        .map((entry) => entry.account)\n        .find((a) =>\n          a.address.startsWith('0x')\n            ? a.address.toLowerCase() === address.toLowerCase()\n            : a.address === address,\n        );\n      if (!account) {\n        return;\n      }\n      const accountId = account.id;\n\n      // Process all balance updates from the activity message\n      const response = this.#processBalanceUpdates(updates, chainId, accountId);\n\n      if (Object.keys(response).length > 0 && subscription) {\n        Promise.resolve(subscription.onAssetsUpdate(response)).catch(\n          console.error,\n        );\n      }\n    } catch (error) {\n      log('Error handling notification', error);\n    }\n  }\n\n  /**\n   * Process balance updates from AccountActivityMessage.\n   * Each update contains asset info, post-transaction balance, and transfer details.\n   *\n   * @param updates - Array of balance updates from the activity message.\n   * @param _chainId - The chain ID (unused but kept for context).\n   * @param accountId - The account ID to process updates for.\n   * @returns DataResponse containing processed balance and metadata.\n   */\n  #processBalanceUpdates(\n    updates: BalanceUpdate[],\n    _chainId: ChainId,\n    accountId: string,\n  ): DataResponse {\n    const assetsBalance: Record<string, Record<Caip19AssetId, AssetBalance>> = {\n      [accountId]: {},\n    };\n    const assetsMetadata: Record<Caip19AssetId, AssetMetadata> = {};\n\n    for (const update of updates) {\n      const { asset, postBalance } = update;\n\n      if (!asset || !postBalance) {\n        continue;\n      }\n\n      // Asset type is in CAIP format: \"eip155:1/erc20:0x...\" or \"eip155:1/slip44:60\"\n      // We can use it directly as the asset ID\n      const assetId = asset.type as Caip19AssetId;\n\n      // Determine token type from asset type string\n      const isNative = asset.type.includes('/slip44:');\n      const tokenType = isNative ? 'native' : 'erc20';\n\n      // Parse balance amount (already in hex format like \"0xc350\")\n      const balanceAmount = postBalance.amount.startsWith('0x')\n        ? BigInt(postBalance.amount).toString()\n        : postBalance.amount;\n\n      assetsBalance[accountId][assetId] = {\n        amount: balanceAmount,\n      };\n\n      assetsMetadata[assetId] = {\n        type: tokenType,\n        symbol: asset.unit,\n        name: asset.unit, // Use unit as name (actual name may not be in the message)\n        decimals: asset.decimals,\n      };\n    }\n\n    const response: DataResponse = {};\n    if (Object.keys(assetsBalance[accountId]).length > 0) {\n      response.assetsBalance = assetsBalance;\n      response.assetsInfo = assetsMetadata;\n    }\n\n    return response;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    if (this.#chainsRefreshTimer) {\n      clearInterval(this.#chainsRefreshTimer);\n      this.#chainsRefreshTimer = null;\n    }\n\n    // Clean up WebSocket subscriptions\n    // Convert to array first to avoid modifying map during iteration\n    const subscriptions = [...this.#wsSubscriptions.values()];\n    for (const wsSub of subscriptions) {\n      try {\n        // Fire and forget - don't await in destroy\n        wsSub.unsubscribe().catch(() => {\n          // Ignore errors during cleanup\n        });\n      } catch {\n        // Ignore errors during cleanup\n      }\n    }\n    this.#wsSubscriptions.clear();\n\n    // Clean up base class subscriptions\n    super.destroy();\n  }\n}\n\n// ============================================================================\n// FACTORY FUNCTION\n// ============================================================================\n\n/**\n * Creates a BackendWebsocketDataSource instance.\n *\n * @param options - Configuration options for the data source.\n * @returns A new BackendWebsocketDataSource instance.\n */\nexport function createBackendWebsocketDataSource(\n  options: BackendWebsocketDataSourceOptions,\n): BackendWebsocketDataSource {\n  return new BackendWebsocketDataSource(options);\n}\n"]}
+{"version":3,"file":"BackendWebsocketDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/BackendWebsocketDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAUA,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,aAAa,EACd,wBAAwB;AACzB,OAAO,WAAW,qBAAqB;AAEvC,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAM1D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAS9D,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,4BAA4B,CAAC;AACrD,MAAM,YAAY,GAAG,qBAAqB,CAAC;AAE3C,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AAoB/D,MAAM,YAAY,GAAoC;IACpD,YAAY,EAAE,EAAE;CACjB,CAAC;AAoBF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;GAMG;AACH,SAAS,gBAAgB,CAAC,OAAgB;IACxC,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,6EAA6E;AAC7E,MAAM,2BAA2B,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAU,CAAC;AAElE;;;;;;;GAOG;AACH,SAAS,+BAA+B,CAAC,QAAmB;IAC1D,MAAM,UAAU,GAAG,IAAI,GAAG,CAAS,2BAA2B,CAAC,CAAC;IAChE,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,UAAU,CAAC,GAAG,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AAChC,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,4BAA4B,CACnC,OAA0C,EAC1C,SAAiB;IAEjB,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACpE,CAAC;IACD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACpE,CAAC;IACD,uFAAuF;IACvF,MAAM,UAAU,GAAG,GAAG,SAAS,GAAG,CAAC;IACnC,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;AACtE,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,2BAA2B,CAClC,SAAiB,EACjB,OAAe;IAEf,MAAM,SAAS,GAAG,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC3E,OAAO,GAAG,YAAY,IAAI,SAAS,MAAM,SAAS,EAAE,CAAC;AACvD,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,SAAS,CAAC,gBAAiC;IAClD,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;QACzC,IAAI,aAAa,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACpC,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,OAAO,aAAa,CAAC,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,aAAa,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,gGAAgG;AAEhG,+EAA+E;AAC/E,gCAAgC;AAChC,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,kCAAkC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAExE,MAAM,OAAO,0BAA2B,SAAQ,kBAG/C;IAuBC,YAAY,OAA0C;QACpD,KAAK,CAAC,eAAe,EAAE;YACrB,GAAG,YAAY;YACf,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QA1BI,wDAAsC;QAEtC,wDAA8B;QAE9B,oEAIC;QAEV,2BAA2B;QAC3B,yDAA6D,IAAI,EAAC;QAElE,8DAA8D;QACrD,sDAAuD,IAAI,GAAG,EAAE,EAAC;QAE1E,uEAAuE;QAC9D,2DAA0D,IAAI,GAAG,EAAE,EAAC;QAE7E,4DAA4D;QACnD,2DAA0D,IAAI,GAAG,EAAE,EAAC;QAQ3E,uBAAA,IAAI,yCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,yCAAc,OAAO,CAAC,cAAc,MAAA,CAAC;QACzC,uBAAA,IAAI,qDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAE5D,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1B,uBAAA,IAAI,iGAAwB,MAA5B,IAAI,CAA0B,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAyED;;;;;;OAMG;IACH,8BAA8B,CAAC,MAAiB;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC;IAwDD,+EAA+E;IAC/E,gBAAgB;IAChB,+EAA+E;IAE/E;;;;OAIG;IACH,qBAAqB,CAAC,MAAiB;QACrC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,YAAY;IACZ,+EAA+E;IAE/E,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,+BAA+B;QAC/B,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;QAEF,MAAM,SAAS,GAAG,OAAO,CAAC,2BAA2B,CAAC,GAAG,CACvD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzB,CAAC;QAEF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,uBAAA,IAAI,6CAAW,CAAC,IAAI,CACzC,2CAA2C,CAC5C,CAAC;YACF,IAAI,cAAc,CAAC,KAAK,KAAM,WAA8B,EAAE,CAAC;gBAC7D,oEAAoE;gBACpE,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;YACvD,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;YACpE,OAAO;QACT,CAAC;QAED,gEAAgE;QAChE,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAElD,6BAA6B;QAC7B,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9D,IAAI,QAAQ,EAAE,CAAC;gBACb,mFAAmF;gBACnF,MAAM,iBAAiB,GAAG,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;gBACnD,MAAM,gBAAgB,GACpB,SAAS,CAAC,MAAM,KAAK,iBAAiB,CAAC,MAAM;oBAC7C,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAE9D,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACtB,sDAAsD;oBACtD,QAAQ,CAAC,MAAM,GAAG,iBAAiB,CAAC;oBACpC,OAAO;gBACT,CAAC;gBACD,oEAAoE;YACtE,CAAC;QACH,CAAC;QAED,wCAAwC;QACxC,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,oGAAoG;QACpG,MAAM,UAAU,GAAG,+BAA+B,CAAC,iBAAiB,CAAC,CAAC;QAEtE,qGAAqG;QACrG,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;gBAC9D,MAAM,OAAO,GAAG,4BAA4B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACjE,IAAI,OAAO,EAAE,CAAC;oBACZ,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,gCAAgC;YAChC,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,6CAAW,CAAC,IAAI,CAC/C,mCAAmC,EACnC;gBACE,QAAQ;gBACR,WAAW,EAAE,YAAY;gBACzB,QAAQ,EAAE,CAAC,YAAuC,EAAE,EAAE;oBACpD,uBAAA,IAAI,6FAAoB,MAAxB,IAAI,EAAqB,YAAY,EAAE,cAAc,CAAC,CAAC;gBACzD,CAAC;aACF,CACF,CAAC;YAEF,+BAA+B;YAC/B,uBAAA,IAAI,mDAAiB,CAAC,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;YAE1D,mCAAmC;YACnC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;gBAC3C,OAAO,EAAE,GAAG,EAAE;oBACZ,MAAM,KAAK,GAAG,uBAAA,IAAI,mDAAiB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;oBACxD,IAAI,KAAK,EAAE,CAAC;wBACV,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,QAAiB,EAAE,EAAE;4BAC9C,GAAG,CAAC,qBAAqB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAClE,CAAC,CAAC,CAAC;wBACH,uBAAA,IAAI,mDAAiB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;oBAC/C,CAAC;oBACD,mCAAmC;oBACnC,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACpD,CAAC;gBACD,MAAM,EAAE,iBAAiB;gBACzB,SAAS;gBACT,cAAc,EAAE,mBAAmB,CAAC,cAAc;aACnD,CAAC,CAAC;YAEH,0CAA0C;YAC1C,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+BAA+B,EAAE;gBACnC,cAAc;gBACd,KAAK;gBACL,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IA4HD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,IAAI,uBAAA,IAAI,sDAAoB,EAAE,CAAC;YAC7B,aAAa,CAAC,uBAAA,IAAI,sDAAoB,CAAC,CAAC;YACxC,uBAAA,IAAI,kDAAuB,IAAI,MAAA,CAAC;QAClC,CAAC;QAED,mCAAmC;QACnC,iEAAiE;QACjE,MAAM,aAAa,GAAG,CAAC,GAAG,uBAAA,IAAI,mDAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1D,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,2CAA2C;gBAC3C,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;oBAC7B,+BAA+B;gBACjC,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,mDAAiB,CAAC,KAAK,EAAE,CAAC;QAE9B,oCAAoC;QACpC,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;;AAtaC,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,KAAK;IACH,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC/C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QAEF,uBAAA,IAAI,kDAAuB,WAAW,CAAC,GAAG,EAAE;YAC1C,uBAAA,IAAI,8FAAqB,MAAzB,IAAI,CAAuB,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,EAAE,kCAAkC,CAAC,MAAA,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC,oDAED,KAAK;IACH,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC/C,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAElC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,CAC/C,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CACjC,CAAC;QAEF,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,yDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;AACH,CAAC,kDAED,KAAK;IACH,MAAM,QAAQ,GAAG,MAAM,uBAAA,IAAI,6CAAW,CAAC,QAAQ,CAAC,wBAAwB,EAAE,CAAC;IAC3E,OAAO,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;IAOC,uFAAuF;IAErF,uBAAA,IAAI,6CAGL,CAAC,SAAS,CACT,gDAAgD,EAChD,CAAC,cAAsC,EAAE,EAAE;QACzC,IAAI,cAAc,CAAC,KAAK,KAAM,WAA8B,EAAE,CAAC;YAC7D,uBAAA,IAAI,sGAA6B,MAAjC,IAAI,CAA+B,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;aAAM,IACL,cAAc,CAAC,KAAK,KAAM,cAAiC,EAC3D,CAAC;YACD,uBAAA,IAAI,2FAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC3B,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC;IAkBC,GAAG,CAAC,gEAAgE,EAAE;QACpE,uBAAuB,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI;QACtD,mBAAmB,EAAE,uBAAA,IAAI,mDAAiB,CAAC,IAAI;KAChD,CAAC,CAAC;IAEH,2DAA2D;IAC3D,KAAK,MAAM,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACxD,MAAM,eAAe,GAAG,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACvE,IAAI,eAAe,EAAE,CAAC;YACpB,gDAAgD;YAChD,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,EAAE;gBAC7C,GAAG,eAAe;gBAClB,QAAQ,EAAE,KAAK,EAAE,oDAAoD;aACtE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,8DAA8D;IAC9D,uBAAA,IAAI,mDAAiB,CAAC,KAAK,EAAE,CAAC;IAE9B,uDAAuD;IACvD,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,KAAK;IACH,IAAI,uBAAA,IAAI,wDAAsB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IAED,oCAAoC;IACpC,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,uBAAA,IAAI,wDAAsB,CAAC,OAAO,EAAE,CAAC,CAAC;IAExE,KAAK,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,cAAc,EAAE,CAAC;QACvD,IAAI,CAAC;YACH,+DAA+D;YAC/D,uBAAA,IAAI,wDAAsB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAClD,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,wCAAwC,EAAE;gBAC5C,cAAc;gBACd,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC,2GA+IC,YAAuC,EACvC,cAAsB;IAEtB,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,OAAO,GAAG,uBAAA,IAAI,wDAAsB,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;QACxE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,eAAe,GACnB,YAAY,CAAC,IAAyC,CAAC;QACzD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC;QAEjD,IAAI,CAAC,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QAED,yEAAyE;QACzE,MAAM,OAAO,GAAG,EAAE,CAAC,KAAgB,CAAC;QAEpC,wFAAwF;QACxF,MAAM,OAAO,GAAG,OAAO,CAAC,2BAA2B;aAChD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACV,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;YACxB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE;YACnD,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,CAC1B,CAAC;QACJ,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAE7B,wDAAwD;QACxD,MAAM,QAAQ,GAAG,uBAAA,IAAI,gGAAuB,MAA3B,IAAI,EAAwB,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAE1E,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,EAAE,CAAC;YACrD,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAC1D,OAAO,CAAC,KAAK,CACd,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC,iHAYC,OAAwB,EACxB,QAAiB,EACjB,SAAiB;IAEjB,MAAM,aAAa,GAAwD;QACzE,CAAC,SAAS,CAAC,EAAE,EAAE;KAChB,CAAC;IACF,MAAM,cAAc,GAAyC,EAAE,CAAC;IAEhE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAEtC,IAAI,CAAC,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;YAC3B,SAAS;QACX,CAAC;QAED,+EAA+E;QAC/E,yCAAyC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAqB,CAAC;QAE5C,8CAA8C;QAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACjD,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;QAEhD,gEAAgE;QAChE,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACjC,SAAS;QACX,CAAC;QAED,4DAA4D;QAC5D,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;YACvD,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;YACvC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC;QAEvB,yFAAyF;QACzF,MAAM,mBAAmB,GAAG,IAAI,WAAW,CAAC,aAAa,CAAC;aACvD,SAAS,CAAC,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;aAClD,QAAQ,EAAE,CAAC;QAEd,aAAa,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG;YAClC,MAAM,EAAE,mBAAmB;SAC5B,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,GAAG;YACxB,IAAI,EAAE,SAAS;YACf,MAAM,EAAE,KAAK,CAAC,IAAI;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,2DAA2D;YAC7E,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IACvD,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACvC,QAAQ,CAAC,UAAU,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAgCH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,gCAAgC,CAC9C,OAA0C;IAE1C,OAAO,IAAI,0BAA0B,CAAC,OAAO,CAAC,CAAC;AACjD,CAAC","sourcesContent":["import type {\n  BackendWebSocketServiceActions,\n  BackendWebSocketServiceEvents,\n  ServerNotificationMessage,\n  WebSocketSubscription,\n  WebSocketState,\n  AccountActivityMessage,\n  BalanceUpdate,\n} from '@metamask/core-backend';\nimport type { ApiPlatformClient } from '@metamask/core-backend';\nimport {\n  isCaipChainId,\n  KnownCaipNamespace,\n  toCaipChainId,\n} from '@metamask/utils';\nimport BigNumberJS from 'bignumber.js';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  ChainId,\n  Caip19AssetId,\n  AssetMetadata,\n  AssetBalance,\n  DataResponse,\n} from '../types';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'BackendWebsocketDataSource';\nconst CHANNEL_TYPE = 'account-activity.v1';\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\n// Allowed actions that BackendWebsocketDataSource can call\nexport type BackendWebsocketDataSourceAllowedActions =\n  BackendWebSocketServiceActions;\n\n// Allowed events that BackendWebsocketDataSource can subscribe to\nexport type BackendWebsocketDataSourceAllowedEvents =\n  BackendWebSocketServiceEvents;\n\n// ============================================================================\n// STATE\n// ============================================================================\n\nexport type BackendWebsocketDataSourceState = DataSourceState;\n\nconst defaultState: BackendWebsocketDataSourceState = {\n  activeChains: [],\n};\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type BackendWebsocketDataSourceOptions = {\n  /** The AssetsController messenger (shared by all data sources). */\n  messenger: AssetsControllerMessenger;\n  /** ApiPlatformClient for fetching supported networks at init (same as AccountsApiDataSource). */\n  queryApiClient: ApiPlatformClient;\n  /** Called when active chains are updated. Pass dataSourceName so the controller knows the source. */\n  onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n  state?: Partial<BackendWebsocketDataSourceState>;\n};\n\n// ============================================================================\n// HELPER FUNCTIONS\n// ============================================================================\n\n/**\n * Extract namespace from a CAIP-2 chain ID.\n * E.g., \"eip155:1\" -> \"eip155\", \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\" -> \"solana\"\n *\n * @param chainId - The CAIP-2 chain ID to extract namespace from.\n * @returns The namespace portion of the chain ID.\n */\nfunction extractNamespace(chainId: ChainId): string {\n  const [namespace] = chainId.split(':');\n  return namespace;\n}\n\n/** Namespaces we always subscribe to for account activity (EVM + Solana). */\nconst ACCOUNT_ACTIVITY_NAMESPACES = ['eip155', 'solana'] as const;\n\n/**\n * Get unique namespaces for account-activity subscriptions.\n * Always includes eip155 and solana so we subscribe to both EVM and Solana account activity,\n * plus any additional namespaces from the requested chain IDs.\n *\n * @param chainIds - Array of CAIP-2 chain IDs (from the subscription request).\n * @returns Array of unique namespaces (at least eip155 and solana).\n */\nfunction getNamespacesForAccountActivity(chainIds: ChainId[]): string[] {\n  const namespaces = new Set<string>(ACCOUNT_ACTIVITY_NAMESPACES);\n  for (const chainId of chainIds) {\n    namespaces.add(extractNamespace(chainId));\n  }\n  return Array.from(namespaces);\n}\n\n/**\n * Returns the address to use for account-activity subscription in the given namespace.\n * EIP-155 accounts use hex (0x...) address; Solana accounts use base58.\n * Returns null if this account type does not have an address in that namespace.\n *\n * @param account - Internal account (type + address).\n * @param account.type - Account type (e.g. \"eip155:eoa\", \"solana:data-account\").\n * @param account.address - Account address (hex for eip155, base58 for solana).\n * @param namespace - The chain namespace (e.g., \"eip155\", \"solana\").\n * @returns The address for that namespace, or null if the account does not support the namespace.\n */\nfunction getAddressForAccountActivity(\n  account: { type: string; address: string },\n  namespace: string,\n): string | null {\n  if (namespace === 'eip155') {\n    return account.type.startsWith('eip155') ? account.address : null;\n  }\n  if (namespace === 'solana') {\n    return account.type.startsWith('solana') ? account.address : null;\n  }\n  // Other namespaces (e.g. from chainIds): use address if account type matches namespace\n  const typePrefix = `${namespace}:`;\n  return account.type.startsWith(typePrefix) ? account.address : null;\n}\n\n/**\n * Build WebSocket channel name for account activity using CAIP-10 wildcard format.\n * Uses 0 as the chain reference to subscribe to all chains in the namespace.\n * EIP-155 addresses are lowercased (hex); Solana addresses are left as-is (base58).\n *\n * @param namespace - The chain namespace (e.g., \"eip155\", \"solana\").\n * @param address - The account address (hex for eip155, base58 for solana).\n * @returns The WebSocket channel name.\n */\nfunction buildAccountActivityChannel(\n  namespace: string,\n  address: string,\n): string {\n  const formatted = namespace === 'eip155' ? address.toLowerCase() : address;\n  return `${CHANNEL_TYPE}.${namespace}:0:${formatted}`;\n}\n\n/**\n * Normalize API chain identifier to CAIP-2 ChainId.\n * Passes through strings already in CAIP-2 form (e.g. eip155:1, solana:5eykt...).\n * Converts bare decimals to eip155:decimal.\n * Uses @metamask/utils for CAIP parsing.\n *\n * @param chainIdOrDecimal - Chain ID string (CAIP-2 or decimal) or decimal number.\n * @returns CAIP-2 ChainId.\n */\nfunction toChainId(chainIdOrDecimal: number | string): ChainId {\n  if (typeof chainIdOrDecimal === 'string') {\n    if (isCaipChainId(chainIdOrDecimal)) {\n      return chainIdOrDecimal;\n    }\n    return toCaipChainId(KnownCaipNamespace.Eip155, chainIdOrDecimal);\n  }\n  return toCaipChainId(KnownCaipNamespace.Eip155, String(chainIdOrDecimal));\n}\n\n// Note: AccountActivityMessage and BalanceUpdate types are imported from @metamask/core-backend\n\n// ============================================================================\n// BACKEND WEBSOCKET DATA SOURCE\n// ============================================================================\n\n/**\n * Data source for receiving real-time balance updates via WebSocket.\n *\n * This data source connects directly to BackendWebSocketService to receive\n * push notifications for account balance changes. Unlike AccountsApiDataSource\n * which polls for data, this provides instant updates.\n *\n * Uses Messenger pattern for all interactions:\n * - Calls BackendWebSocketService methods via messenger actions\n * - Exposes its own actions for AssetsController to call\n * - Publishes events for AssetsController to subscribe to\n *\n * Actions exposed:\n * - BackendWebsocketDataSource:getActiveChains\n * - BackendWebsocketDataSource:subscribe\n * - BackendWebsocketDataSource:unsubscribe\n *\n * Events published:\n * - BackendWebsocketDataSource:activeChainsUpdated\n * - BackendWebsocketDataSource:assetsUpdated\n *\n * Actions called (from BackendWebSocketService):\n * - BackendWebSocketService:subscribe\n * - BackendWebSocketService:getConnectionInfo\n * - BackendWebSocketService:findSubscriptionsByChannelPrefix\n */\nconst DEFAULT_CHAINS_REFRESH_INTERVAL_MS = 20 * 60 * 1000; // 20 minutes\n\nexport class BackendWebsocketDataSource extends AbstractDataSource<\n  typeof CONTROLLER_NAME,\n  BackendWebsocketDataSourceState\n> {\n  readonly #messenger: AssetsControllerMessenger;\n\n  readonly #apiClient: ApiPlatformClient;\n\n  readonly #onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n\n  /** Chains refresh timer */\n  #chainsRefreshTimer: ReturnType<typeof setInterval> | null = null;\n\n  /** WebSocket subscriptions by our internal subscription ID */\n  readonly #wsSubscriptions: Map<string, WebSocketSubscription> = new Map();\n\n  /** Pending subscription requests to process when WebSocket connects */\n  readonly #pendingSubscriptions: Map<string, SubscriptionRequest> = new Map();\n\n  /** Store original subscription requests for reconnection */\n  readonly #subscriptionRequests: Map<string, SubscriptionRequest> = new Map();\n\n  constructor(options: BackendWebsocketDataSourceOptions) {\n    super(CONTROLLER_NAME, {\n      ...defaultState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n    this.#apiClient = options.queryApiClient;\n    this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n\n    this.#subscribeToEvents();\n    this.#initializeActiveChains().catch(console.error);\n  }\n\n  // ============================================================================\n  // INITIALIZATION\n  // ============================================================================\n\n  async #initializeActiveChains(): Promise<void> {\n    try {\n      const chains = await this.#fetchActiveChains();\n      const previous = [...this.state.activeChains];\n      this.updateActiveChains(chains, (updatedChains) =>\n        this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n      );\n\n      this.#chainsRefreshTimer = setInterval(() => {\n        this.#refreshActiveChains().catch(console.error);\n      }, DEFAULT_CHAINS_REFRESH_INTERVAL_MS);\n    } catch (error) {\n      log('Failed to fetch active chains', error);\n    }\n  }\n\n  async #refreshActiveChains(): Promise<void> {\n    try {\n      const chains = await this.#fetchActiveChains();\n      const previousChains = new Set(this.state.activeChains);\n      const newChains = new Set(chains);\n\n      const added = chains.filter((chain) => !previousChains.has(chain));\n      const removed = Array.from(previousChains).filter(\n        (chain) => !newChains.has(chain),\n      );\n\n      if (added.length > 0 || removed.length > 0) {\n        const previous = [...this.state.activeChains];\n        this.updateActiveChains(chains, (updatedChains) =>\n          this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n        );\n      }\n    } catch (error) {\n      log('Failed to refresh active chains', error);\n    }\n  }\n\n  async #fetchActiveChains(): Promise<ChainId[]> {\n    const response = await this.#apiClient.accounts.fetchV2SupportedNetworks();\n    return response.fullSupport.map(toChainId);\n  }\n\n  #subscribeToEvents(): void {\n    type ConnectionStatePayload = {\n      state: WebSocketState;\n      [key: string]: unknown;\n    };\n    // Listen for WebSocket connection state changes (event not in AssetsControllerEvents).\n    (\n      this.#messenger as unknown as {\n        subscribe: (e: string, h: (p: ConnectionStatePayload) => void) => void;\n      }\n    ).subscribe(\n      'BackendWebSocketService:connectionStateChanged',\n      (connectionInfo: ConnectionStatePayload) => {\n        if (connectionInfo.state === ('connected' as WebSocketState)) {\n          this.#processPendingSubscriptions().catch(console.error);\n        } else if (\n          connectionInfo.state === ('disconnected' as WebSocketState)\n        ) {\n          this.#handleDisconnect();\n        }\n      },\n    );\n  }\n\n  /**\n   * Sync active chains from AccountsApiDataSource.\n   * Called by AssetsController.handleActiveChainsUpdate when the callback is\n   * invoked for BackendWebsocketDataSource (no messenger call; controller already updated).\n   *\n   * @param chains - Updated active chain IDs from AccountsApiDataSource.\n   */\n  setActiveChainsFromAccountsApi(chains: ChainId[]): void {\n    this.updateActiveChains(chains, () => undefined);\n  }\n\n  /**\n   * Handle WebSocket disconnection.\n   * Moves all active subscriptions to pending for re-subscription on reconnect.\n   */\n  #handleDisconnect(): void {\n    log('WebSocket disconnected, preserving subscriptions for reconnect', {\n      activeSubscriptionCount: this.activeSubscriptions.size,\n      wsSubscriptionCount: this.#wsSubscriptions.size,\n    });\n\n    // Move active subscriptions to pending for re-subscription\n    for (const [subscriptionId] of this.activeSubscriptions) {\n      const originalRequest = this.#subscriptionRequests.get(subscriptionId);\n      if (originalRequest) {\n        // Mark as update since it was previously active\n        this.#pendingSubscriptions.set(subscriptionId, {\n          ...originalRequest,\n          isUpdate: false, // Treat as new subscription since server cleared it\n        });\n      }\n    }\n\n    // Clear WebSocket subscriptions (server-side already cleared)\n    this.#wsSubscriptions.clear();\n\n    // Clear active subscriptions (they're no longer valid)\n    this.activeSubscriptions.clear();\n  }\n\n  /**\n   * Process any pending subscriptions that were queued while WebSocket was disconnected.\n   */\n  async #processPendingSubscriptions(): Promise<void> {\n    if (this.#pendingSubscriptions.size === 0) {\n      return;\n    }\n\n    // Process all pending subscriptions\n    const pendingEntries = Array.from(this.#pendingSubscriptions.entries());\n\n    for (const [subscriptionId, request] of pendingEntries) {\n      try {\n        // Remove from pending before processing to avoid infinite loop\n        this.#pendingSubscriptions.delete(subscriptionId);\n        await this.subscribe(request);\n      } catch (error) {\n        log('Failed to process pending subscription', {\n          subscriptionId,\n          error,\n        });\n      }\n    }\n  }\n\n  // ============================================================================\n  // ACTIVE CHAINS\n  // ============================================================================\n\n  /**\n   * Update active chains when AccountsApiDataSource reports new supported chains.\n   *\n   * @param chains - Array of supported chain IDs.\n   */\n  updateSupportedChains(chains: ChainId[]): void {\n    const previous = [...this.state.activeChains];\n    this.updateActiveChains(chains, (updatedChains) =>\n      this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n    );\n  }\n\n  // ============================================================================\n  // SUBSCRIBE\n  // ============================================================================\n\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Filter to active chains only\n    const chainsToSubscribe = request.chainIds.filter((chainId) =>\n      this.state.activeChains.includes(chainId),\n    );\n\n    const addresses = request.accountsWithSupportedChains.map(\n      (a) => a.account.address,\n    );\n\n    if (addresses.length === 0) {\n      return;\n    }\n\n    // Check WebSocket connection status\n    try {\n      const connectionInfo = this.#messenger.call(\n        'BackendWebSocketService:getConnectionInfo',\n      );\n      if (connectionInfo.state !== ('connected' as WebSocketState)) {\n        // Store the subscription request to process when WebSocket connects\n        this.#pendingSubscriptions.set(subscriptionId, subscriptionRequest);\n        return;\n      }\n    } catch {\n      // Store anyway - will be processed when we can connect\n      this.#pendingSubscriptions.set(subscriptionId, subscriptionRequest);\n      return;\n    }\n\n    // Remove from pending if it was there (we're processing it now)\n    this.#pendingSubscriptions.delete(subscriptionId);\n\n    // Handle subscription update\n    if (isUpdate) {\n      const existing = this.activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        // Check if accounts changed - if so, we need to re-subscribe to different channels\n        const existingAddresses = existing.addresses ?? [];\n        const addressesChanged =\n          addresses.length !== existingAddresses.length ||\n          addresses.some((addr) => !existingAddresses.includes(addr));\n\n        if (!addressesChanged) {\n          // Only chains changed - just update chains and return\n          existing.chains = chainsToSubscribe;\n          return;\n        }\n        // Accounts changed - fall through to re-subscribe with new channels\n      }\n    }\n\n    // Clean up existing subscription if any\n    await this.unsubscribe(subscriptionId);\n\n    // Always subscribe to eip155 and solana account activity, plus any namespaces from requested chains\n    const namespaces = getNamespacesForAccountActivity(chainsToSubscribe);\n\n    // Build channel names: use namespace-appropriate address per account (eip155 = hex, solana = base58)\n    const channels: string[] = [];\n    for (const namespace of namespaces) {\n      for (const { account } of request.accountsWithSupportedChains) {\n        const address = getAddressForAccountActivity(account, namespace);\n        if (address) {\n          channels.push(buildAccountActivityChannel(namespace, address));\n        }\n      }\n    }\n\n    try {\n      // Create WebSocket subscription\n      const wsSubscription = await this.#messenger.call(\n        'BackendWebSocketService:subscribe',\n        {\n          channels,\n          channelType: CHANNEL_TYPE,\n          callback: (notification: ServerNotificationMessage) => {\n            this.#handleNotification(notification, subscriptionId);\n          },\n        },\n      );\n\n      // Store WebSocket subscription\n      this.#wsSubscriptions.set(subscriptionId, wsSubscription);\n\n      // Store in abstract class tracking\n      this.activeSubscriptions.set(subscriptionId, {\n        cleanup: () => {\n          const wsSub = this.#wsSubscriptions.get(subscriptionId);\n          if (wsSub) {\n            wsSub.unsubscribe().catch((unsubErr: unknown) => {\n              log('Error unsubscribing', { subscriptionId, error: unsubErr });\n            });\n            this.#wsSubscriptions.delete(subscriptionId);\n          }\n          // Also clean up the stored request\n          this.#subscriptionRequests.delete(subscriptionId);\n        },\n        chains: chainsToSubscribe,\n        addresses,\n        onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n      });\n\n      // Store original request for reconnection\n      this.#subscriptionRequests.set(subscriptionId, subscriptionRequest);\n    } catch (error) {\n      log('WebSocket subscription FAILED', {\n        subscriptionId,\n        error,\n        chains: chainsToSubscribe,\n      });\n    }\n  }\n\n  // ============================================================================\n  // NOTIFICATION HANDLING\n  // ============================================================================\n\n  #handleNotification(\n    notification: ServerNotificationMessage,\n    subscriptionId: string,\n  ): void {\n    try {\n      const subscription = this.activeSubscriptions.get(subscriptionId);\n      const request = this.#subscriptionRequests.get(subscriptionId)?.request;\n      if (!request) {\n        return;\n      }\n\n      const activityMessage =\n        notification.data as unknown as AccountActivityMessage;\n      const { address, tx, updates } = activityMessage;\n\n      if (!address || !tx || !updates) {\n        return;\n      }\n\n      // Extract chain ID from transaction (CAIP-2 format, e.g., \"eip155:8453\")\n      const chainId = tx.chain as ChainId;\n\n      // Find matching account in request (eip155: case-insensitive hex; solana: exact base58)\n      const account = request.accountsWithSupportedChains\n        .map((entry) => entry.account)\n        .find((a) =>\n          a.address.startsWith('0x')\n            ? a.address.toLowerCase() === address.toLowerCase()\n            : a.address === address,\n        );\n      if (!account) {\n        return;\n      }\n      const accountId = account.id;\n\n      // Process all balance updates from the activity message\n      const response = this.#processBalanceUpdates(updates, chainId, accountId);\n\n      if (Object.keys(response).length > 0 && subscription) {\n        Promise.resolve(subscription.onAssetsUpdate(response)).catch(\n          console.error,\n        );\n      }\n    } catch (error) {\n      log('Error handling notification', error);\n    }\n  }\n\n  /**\n   * Process balance updates from AccountActivityMessage.\n   * Each update contains asset info, post-transaction balance, and transfer details.\n   *\n   * @param updates - Array of balance updates from the activity message.\n   * @param _chainId - The chain ID (unused but kept for context).\n   * @param accountId - The account ID to process updates for.\n   * @returns DataResponse containing processed balance and metadata.\n   */\n  #processBalanceUpdates(\n    updates: BalanceUpdate[],\n    _chainId: ChainId,\n    accountId: string,\n  ): DataResponse {\n    const assetsBalance: Record<string, Record<Caip19AssetId, AssetBalance>> = {\n      [accountId]: {},\n    };\n    const assetsMetadata: Record<Caip19AssetId, AssetMetadata> = {};\n\n    for (const update of updates) {\n      const { asset, postBalance } = update;\n\n      if (!asset || !postBalance) {\n        continue;\n      }\n\n      // Asset type is in CAIP format: \"eip155:1/erc20:0x...\" or \"eip155:1/slip44:60\"\n      // We can use it directly as the asset ID\n      const assetId = asset.type as Caip19AssetId;\n\n      // Determine token type from asset type string\n      const isNative = asset.type.includes('/slip44:');\n      const tokenType = isNative ? 'native' : 'erc20';\n\n      // We assume decimals are always present; skip malformed updates\n      if (asset.decimals === undefined) {\n        continue;\n      }\n\n      // Parse raw balance (hex like \"0x26f0e5\" or decimal string)\n      const rawBalanceStr = postBalance.amount.startsWith('0x')\n        ? BigInt(postBalance.amount).toString()\n        : postBalance.amount;\n\n      // Convert to human-readable using asset decimals (match RpcDataSource / pipeline format)\n      const humanReadableAmount = new BigNumberJS(rawBalanceStr)\n        .dividedBy(new BigNumberJS(10).pow(asset.decimals))\n        .toString();\n\n      assetsBalance[accountId][assetId] = {\n        amount: humanReadableAmount,\n      };\n\n      assetsMetadata[assetId] = {\n        type: tokenType,\n        symbol: asset.unit,\n        name: asset.unit, // Use unit as name (actual name may not be in the message)\n        decimals: asset.decimals,\n      };\n    }\n\n    const response: DataResponse = { updateMode: 'merge' };\n    if (Object.keys(assetsBalance[accountId]).length > 0) {\n      response.assetsBalance = assetsBalance;\n      response.assetsInfo = assetsMetadata;\n    }\n\n    return response;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    if (this.#chainsRefreshTimer) {\n      clearInterval(this.#chainsRefreshTimer);\n      this.#chainsRefreshTimer = null;\n    }\n\n    // Clean up WebSocket subscriptions\n    // Convert to array first to avoid modifying map during iteration\n    const subscriptions = [...this.#wsSubscriptions.values()];\n    for (const wsSub of subscriptions) {\n      try {\n        // Fire and forget - don't await in destroy\n        wsSub.unsubscribe().catch(() => {\n          // Ignore errors during cleanup\n        });\n      } catch {\n        // Ignore errors during cleanup\n      }\n    }\n    this.#wsSubscriptions.clear();\n\n    // Clean up base class subscriptions\n    super.destroy();\n  }\n}\n\n// ============================================================================\n// FACTORY FUNCTION\n// ============================================================================\n\n/**\n * Creates a BackendWebsocketDataSource instance.\n *\n * @param options - Configuration options for the data source.\n * @returns A new BackendWebsocketDataSource instance.\n */\nexport function createBackendWebsocketDataSource(\n  options: BackendWebsocketDataSourceOptions,\n): BackendWebsocketDataSource {\n  return new BackendWebsocketDataSource(options);\n}\n"]}

package/dist/data-sources/PriceDataSource.cjs

@@ -10,7 +10,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_currency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
+var _PriceDataSource_instances, _PriceDataSource_getSelectedCurrency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PriceDataSource = void 0;
 const core_backend_1 = require("@metamask/core-backend");
@@ -79,13 +79,13 @@ class PriceDataSource {
     }
     constructor(options) {
         _PriceDataSource_instances.add(this);
-        _PriceDataSource_currency.set(this, void 0);
+        _PriceDataSource_getSelectedCurrency.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_currency, options.currency ?? 'usd', "f");
+        __classPrivateFieldSet(this, _PriceDataSource_getSelectedCurrency, options.getSelectedCurrency, "f");
         __classPrivateFieldSet(this, _PriceDataSource_pollInterval, options.pollInterval ?? DEFAULT_POLL_INTERVAL, "f");
         __classPrivateFieldSet(this, _PriceDataSource_apiClient, options.queryApiClient, "f");
     }
@@ -110,23 +110,26 @@ class PriceDataSource {
     get assetsMiddleware() {
         return (0, types_1.forDataTypes)(['price'], async (ctx, next) => {
             // Extract response from context
-            const { response } = ctx;
+            const { response, request } = ctx;
             // Only fetch prices for detected assets (assets without metadata)
             // The subscription handles fetching prices for all existing assets
-            if (!response.detectedAssets) {
+            if (!response.detectedAssets && !request.assetsForPriceUpdate?.length) {
                 return next(ctx);
             }
-            const detectedAssetIds = new Set();
-            for (const detectedIds of Object.values(response.detectedAssets)) {
-                for (const assetId of detectedIds) {
-                    detectedAssetIds.add(assetId);
+            const assetIds = new Set();
+            for (const detectedAccountAssets of Object.values(response.detectedAssets ?? {})) {
+                for (const assetId of detectedAccountAssets) {
+                    assetIds.add(assetId);
                 }
             }
-            if (detectedAssetIds.size === 0) {
+            for (const assetId of request.assetsForPriceUpdate ?? []) {
+                assetIds.add(assetId);
+            }
+            if (assetIds.size === 0) {
                 return next(ctx);
             }
             // Filter to only priceable assets
-            const priceableAssetIds = [...detectedAssetIds].filter(isPriceableAsset);
+            const priceableAssetIds = [...assetIds].filter(isPriceableAsset);
             if (priceableAssetIds.length === 0) {
                 return next(ctx);
             }
@@ -225,7 +228,10 @@ class PriceDataSource {
                 // Only report if we got prices
                 if (fetchResponse.assetsPrice &&
                     Object.keys(fetchResponse.assetsPrice).length > 0) {
-                    await subscription.onAssetsUpdate(fetchResponse);
+                    await subscription.onAssetsUpdate({
+                        ...fetchResponse,
+                        updateMode: 'merge',
+                    });
                 }
             }
             catch (error) {
@@ -271,7 +277,7 @@ class PriceDataSource {
     }
 }
 exports.PriceDataSource = PriceDataSource;
-_PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_fetchSpotPrices = 
+_PriceDataSource_getSelectedCurrency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_fetchSpotPrices = 
 // ============================================================================
 // HELPERS
 // ============================================================================
@@ -283,7 +289,7 @@ _PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new W
  */
 async function _PriceDataSource_fetchSpotPrices(assetIds) {
     return __classPrivateFieldGet(this, _PriceDataSource_apiClient, "f").prices.fetchV3SpotPrices(assetIds, {
-        currency: __classPrivateFieldGet(this, _PriceDataSource_currency, "f"),
+        currency: __classPrivateFieldGet(this, _PriceDataSource_getSelectedCurrency, "f").call(this),
         includeMarketData: true,
     });
 }, _PriceDataSource_getAssetIdsFromBalanceState = function _PriceDataSource_getAssetIdsFromBalanceState(request, getAssetsState) {

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

@@ -1 +1 @@
-{"version":3,"file":"PriceDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAIA,yDAA2D;AAC3D,2CAAqD;AAGrD,0CAA8D;AAC9D,wCAAwC;AAUxC,+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,IAAA,2BAAkB,EAAC,sBAAa,EAAE,eAAe,CAAC,CAAC;AAmB/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,MAAa,eAAe;IAG1B,OAAO;QACL,OAAO,eAAe,CAAC,cAAc,CAAC;IACxC,CAAC;IAoBD,YAAY,OAA+B;;QAlBlC,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,IAAA,oBAAY,EAAC,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;IAuFD,+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;;AAjVH,0CAkVC;;AA1OC,+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,6EAA6E;oBAC7E,IAAI,WAAW,EAAE,CAAC;wBAChB,IAAI,CAAC;4BACH,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,0BAAkB,EACpC,OAAwB,CACzB,CAAC;4BACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gCAC9B,SAAS;4BACX,CAAC;wBACH,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,GAAG,CAAC,8CAA8C,EAAE;gCAClD,OAAO;gCACP,KAAK;6BACN,CAAC,CAAC;4BACH,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;AA1Le,8BAAc,GAAG,eAAe,AAAlB,CAAmB","sourcesContent":["import type {\n  SupportedCurrency,\n  V3SpotPricesResponse,\n} from '@metamask/core-backend';\nimport { ApiPlatformClient } from '@metamask/core-backend';\nimport { parseCaipAssetType } from '@metamask/utils';\n\nimport type { SubscriptionRequest } from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport { forDataTypes } from '../types';\nimport type {\n  Caip19AssetId,\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\n/** Optional configuration for PriceDataSource. */\nexport type PriceDataSourceConfig = {\n  /** Polling interval in ms (default: 60000) */\n  pollInterval?: number;\n};\n\nexport type PriceDataSourceOptions = PriceDataSourceConfig & {\n  /** ApiPlatformClient for API calls with caching */\n  queryApiClient: ApiPlatformClient;\n  /** Currency to fetch prices in (default: 'usd') */\n  currency?: SupportedCurrency;\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  static readonly controllerName = CONTROLLER_NAME;\n\n  getName(): string {\n    return PriceDataSource.controllerName;\n  }\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; skip malformed asset IDs for this entry only\n            if (chainFilter) {\n              try {\n                const { chainId } = parseCaipAssetType(\n                  assetId as Caip19AssetId,\n                );\n                if (!chainFilter.has(chainId)) {\n                  continue;\n                }\n              } catch (error) {\n                log('Skipping malformed asset ID in balance state', {\n                  assetId,\n                  error,\n                });\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"]}
+{"version":3,"file":"PriceDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAIA,yDAA2D;AAC3D,2CAAqD;AAGrD,0CAA8D;AAC9D,wCAAwC;AAUxC,+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,IAAA,2BAAkB,EAAC,sBAAa,EAAE,eAAe,CAAC,CAAC;AAmB/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,MAAa,eAAe;IAG1B,OAAO;QACL,OAAO,eAAe,CAAC,cAAc,CAAC;IACxC,CAAC;IAoBD,YAAY,OAA+B;;QAlBlC,uDAA8C;QAE9C,gDAAsB;QAE/B,6CAA6C;QACpC,6CAA8B;QAEvC,iCAAiC;QACxB,+CAQL,IAAI,GAAG,EAAE,EAAC;QAGZ,uBAAA,IAAI,wCAAwB,OAAO,CAAC,mBAAmB,MAAA,CAAC;QACxD,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,IAAA,oBAAY,EAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACjD,gCAAgC;YAChC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,GAAG,CAAC;YAElC,kEAAkE;YAClE,mEAAmE;YACnE,IAAI,CAAC,QAAQ,CAAC,cAAc,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,MAAM,EAAE,CAAC;gBACtE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;YAC1C,KAAK,MAAM,qBAAqB,IAAI,MAAM,CAAC,MAAM,CAC/C,QAAQ,CAAC,cAAc,IAAI,EAAE,CAC9B,EAAE,CAAC;gBACF,KAAK,MAAM,OAAO,IAAI,qBAAqB,EAAE,CAAC;oBAC5C,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YAED,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,oBAAoB,IAAI,EAAE,EAAE,CAAC;gBACzD,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;YAED,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACxB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,kCAAkC;YAClC,MAAM,iBAAiB,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAEjE,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;IAuFD,+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;wBAChC,GAAG,aAAa;wBAChB,UAAU,EAAE,OAAO;qBACpB,CAAC,CAAC;gBACL,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;;AA1VH,0CA2VC;;AA7OC,+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,4CAAqB,MAAzB,IAAI,CAAuB;QACrC,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,6EAA6E;oBAC7E,IAAI,WAAW,EAAE,CAAC;wBAChB,IAAI,CAAC;4BACH,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,0BAAkB,EACpC,OAAwB,CACzB,CAAC;4BACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gCAC9B,SAAS;4BACX,CAAC;wBACH,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,GAAG,CAAC,8CAA8C,EAAE;gCAClD,OAAO;gCACP,KAAK;6BACN,CAAC,CAAC;4BACH,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;AAhMe,8BAAc,GAAG,eAAe,AAAlB,CAAmB","sourcesContent":["import type {\n  SupportedCurrency,\n  V3SpotPricesResponse,\n} from '@metamask/core-backend';\nimport { ApiPlatformClient } from '@metamask/core-backend';\nimport { parseCaipAssetType } from '@metamask/utils';\n\nimport type { SubscriptionRequest } from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport { forDataTypes } from '../types';\nimport type {\n  Caip19AssetId,\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\n/** Optional configuration for PriceDataSource. */\nexport type PriceDataSourceConfig = {\n  /** Polling interval in ms (default: 60000) */\n  pollInterval?: number;\n};\n\nexport type PriceDataSourceOptions = PriceDataSourceConfig & {\n  /** ApiPlatformClient for API calls with caching */\n  queryApiClient: ApiPlatformClient;\n  /** Function returning the currently-active ISO 4217 currency code */\n  getSelectedCurrency: () => SupportedCurrency;\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  static readonly controllerName = CONTROLLER_NAME;\n\n  getName(): string {\n    return PriceDataSource.controllerName;\n  }\n\n  readonly #getSelectedCurrency: () => 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.#getSelectedCurrency = options.getSelectedCurrency;\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, request } = 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 && !request.assetsForPriceUpdate?.length) {\n        return next(ctx);\n      }\n\n      const assetIds = new Set<Caip19AssetId>();\n      for (const detectedAccountAssets of Object.values(\n        response.detectedAssets ?? {},\n      )) {\n        for (const assetId of detectedAccountAssets) {\n          assetIds.add(assetId);\n        }\n      }\n\n      for (const assetId of request.assetsForPriceUpdate ?? []) {\n        assetIds.add(assetId);\n      }\n\n      if (assetIds.size === 0) {\n        return next(ctx);\n      }\n\n      // Filter to only priceable assets\n      const priceableAssetIds = [...assetIds].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.#getSelectedCurrency(),\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; skip malformed asset IDs for this entry only\n            if (chainFilter) {\n              try {\n                const { chainId } = parseCaipAssetType(\n                  assetId as Caip19AssetId,\n                );\n                if (!chainFilter.has(chainId)) {\n                  continue;\n                }\n              } catch (error) {\n                log('Skipping malformed asset ID in balance state', {\n                  assetId,\n                  error,\n                });\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({\n            ...fetchResponse,\n            updateMode: 'merge',\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 (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/PriceDataSource.d.cts

@@ -10,8 +10,8 @@ export type PriceDataSourceConfig = {
 export type PriceDataSourceOptions = PriceDataSourceConfig & {
     /** ApiPlatformClient for API calls with caching */
     queryApiClient: ApiPlatformClient;
-    /** Currency to fetch prices in (default: 'usd') */
-    currency?: SupportedCurrency;
+    /** Function returning the currently-active ISO 4217 currency code */
+    getSelectedCurrency: () => SupportedCurrency;
 };
 /**
  * PriceDataSource fetches asset prices from the Price API.

package/dist/data-sources/PriceDataSource.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"PriceDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EAElB,+BAA+B;AAChC,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAG3D,OAAO,KAAK,EAAE,mBAAmB,EAAE,iCAA6B;AAGhE,OAAO,KAAK,EAEV,WAAW,EACX,YAAY,EAEZ,UAAU,EACV,6BAA6B,EAC9B,qBAAiB;AAelB,kDAAkD;AAClD,MAAM,MAAM,qBAAqB,GAAG;IAClC,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG,qBAAqB,GAAG;IAC3D,mDAAmD;IACnD,cAAc,EAAE,iBAAiB,CAAC;IAClC,mDAAmD;IACnD,QAAQ,CAAC,EAAE,iBAAiB,CAAC;CAC9B,CAAC;AAoDF;;;;;;;;;GASG;AACH,qBAAa,eAAe;;IAC1B,MAAM,CAAC,QAAQ,CAAC,cAAc,qBAAmB;IAEjD,OAAO,IAAI,MAAM;gBAsBL,OAAO,EAAE,sBAAsB;IAU3C;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB,IAAI,UAAU,CAoDjC;IA2FD;;;;;;;OAOG;IACG,KAAK,CACT,OAAO,EAAE,WAAW,EACpB,cAAc,CAAC,EAAE,MAAM,6BAA6B,GACnD,OAAO,CAAC,YAAY,CAAC;IA4CxB;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8DxE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}
+{"version":3,"file":"PriceDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EAElB,+BAA+B;AAChC,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAG3D,OAAO,KAAK,EAAE,mBAAmB,EAAE,iCAA6B;AAGhE,OAAO,KAAK,EAEV,WAAW,EACX,YAAY,EAEZ,UAAU,EACV,6BAA6B,EAC9B,qBAAiB;AAelB,kDAAkD;AAClD,MAAM,MAAM,qBAAqB,GAAG;IAClC,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG,qBAAqB,GAAG;IAC3D,mDAAmD;IACnD,cAAc,EAAE,iBAAiB,CAAC;IAClC,qEAAqE;IACrE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;CAC9C,CAAC;AAoDF;;;;;;;;;GASG;AACH,qBAAa,eAAe;;IAC1B,MAAM,CAAC,QAAQ,CAAC,cAAc,qBAAmB;IAEjD,OAAO,IAAI,MAAM;gBAsBL,OAAO,EAAE,sBAAsB;IAU3C;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA0DjC;IA2FD;;;;;;;OAOG;IACG,KAAK,CACT,OAAO,EAAE,WAAW,EACpB,cAAc,CAAC,EAAE,MAAM,6BAA6B,GACnD,OAAO,CAAC,YAAY,CAAC;IA4CxB;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAiExE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}

package/dist/data-sources/PriceDataSource.d.mts

@@ -10,8 +10,8 @@ export type PriceDataSourceConfig = {
 export type PriceDataSourceOptions = PriceDataSourceConfig & {
     /** ApiPlatformClient for API calls with caching */
     queryApiClient: ApiPlatformClient;
-    /** Currency to fetch prices in (default: 'usd') */
-    currency?: SupportedCurrency;
+    /** Function returning the currently-active ISO 4217 currency code */
+    getSelectedCurrency: () => SupportedCurrency;
 };
 /**
  * PriceDataSource fetches asset prices from the Price API.

package/dist/data-sources/PriceDataSource.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"PriceDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EAElB,+BAA+B;AAChC,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAG3D,OAAO,KAAK,EAAE,mBAAmB,EAAE,iCAA6B;AAGhE,OAAO,KAAK,EAEV,WAAW,EACX,YAAY,EAEZ,UAAU,EACV,6BAA6B,EAC9B,qBAAiB;AAelB,kDAAkD;AAClD,MAAM,MAAM,qBAAqB,GAAG;IAClC,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG,qBAAqB,GAAG;IAC3D,mDAAmD;IACnD,cAAc,EAAE,iBAAiB,CAAC;IAClC,mDAAmD;IACnD,QAAQ,CAAC,EAAE,iBAAiB,CAAC;CAC9B,CAAC;AAoDF;;;;;;;;;GASG;AACH,qBAAa,eAAe;;IAC1B,MAAM,CAAC,QAAQ,CAAC,cAAc,qBAAmB;IAEjD,OAAO,IAAI,MAAM;gBAsBL,OAAO,EAAE,sBAAsB;IAU3C;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB,IAAI,UAAU,CAoDjC;IA2FD;;;;;;;OAOG;IACG,KAAK,CACT,OAAO,EAAE,WAAW,EACpB,cAAc,CAAC,EAAE,MAAM,6BAA6B,GACnD,OAAO,CAAC,YAAY,CAAC;IA4CxB;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8DxE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}
+{"version":3,"file":"PriceDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/PriceDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EAElB,+BAA+B;AAChC,OAAO,EAAE,iBAAiB,EAAE,+BAA+B;AAG3D,OAAO,KAAK,EAAE,mBAAmB,EAAE,iCAA6B;AAGhE,OAAO,KAAK,EAEV,WAAW,EACX,YAAY,EAEZ,UAAU,EACV,6BAA6B,EAC9B,qBAAiB;AAelB,kDAAkD;AAClD,MAAM,MAAM,qBAAqB,GAAG;IAClC,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG,qBAAqB,GAAG;IAC3D,mDAAmD;IACnD,cAAc,EAAE,iBAAiB,CAAC;IAClC,qEAAqE;IACrE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;CAC9C,CAAC;AAoDF;;;;;;;;;GASG;AACH,qBAAa,eAAe;;IAC1B,MAAM,CAAC,QAAQ,CAAC,cAAc,qBAAmB;IAEjD,OAAO,IAAI,MAAM;gBAsBL,OAAO,EAAE,sBAAsB;IAU3C;;;;;;;;;;;;;;OAcG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA0DjC;IA2FD;;;;;;;OAOG;IACG,KAAK,CACT,OAAO,EAAE,WAAW,EACpB,cAAc,CAAC,EAAE,MAAM,6BAA6B,GACnD,OAAO,CAAC,YAAY,CAAC;IA4CxB;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAiExE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}

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_currency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
+var _PriceDataSource_instances, _PriceDataSource_getSelectedCurrency, _PriceDataSource_pollInterval, _PriceDataSource_apiClient, _PriceDataSource_activeSubscriptions, _PriceDataSource_fetchSpotPrices, _PriceDataSource_getAssetIdsFromBalanceState;
 import { ApiPlatformClient } from "@metamask/core-backend";
 import { parseCaipAssetType } from "@metamask/utils";
 import { projectLogger, createModuleLogger } from "../logger.mjs";
@@ -76,13 +76,13 @@ export class PriceDataSource {
     }
     constructor(options) {
         _PriceDataSource_instances.add(this);
-        _PriceDataSource_currency.set(this, void 0);
+        _PriceDataSource_getSelectedCurrency.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_currency, options.currency ?? 'usd', "f");
+        __classPrivateFieldSet(this, _PriceDataSource_getSelectedCurrency, options.getSelectedCurrency, "f");
         __classPrivateFieldSet(this, _PriceDataSource_pollInterval, options.pollInterval ?? DEFAULT_POLL_INTERVAL, "f");
         __classPrivateFieldSet(this, _PriceDataSource_apiClient, options.queryApiClient, "f");
     }
@@ -107,23 +107,26 @@ export class PriceDataSource {
     get assetsMiddleware() {
         return forDataTypes(['price'], async (ctx, next) => {
             // Extract response from context
-            const { response } = ctx;
+            const { response, request } = ctx;
             // Only fetch prices for detected assets (assets without metadata)
             // The subscription handles fetching prices for all existing assets
-            if (!response.detectedAssets) {
+            if (!response.detectedAssets && !request.assetsForPriceUpdate?.length) {
                 return next(ctx);
             }
-            const detectedAssetIds = new Set();
-            for (const detectedIds of Object.values(response.detectedAssets)) {
-                for (const assetId of detectedIds) {
-                    detectedAssetIds.add(assetId);
+            const assetIds = new Set();
+            for (const detectedAccountAssets of Object.values(response.detectedAssets ?? {})) {
+                for (const assetId of detectedAccountAssets) {
+                    assetIds.add(assetId);
                 }
             }
-            if (detectedAssetIds.size === 0) {
+            for (const assetId of request.assetsForPriceUpdate ?? []) {
+                assetIds.add(assetId);
+            }
+            if (assetIds.size === 0) {
                 return next(ctx);
             }
             // Filter to only priceable assets
-            const priceableAssetIds = [...detectedAssetIds].filter(isPriceableAsset);
+            const priceableAssetIds = [...assetIds].filter(isPriceableAsset);
             if (priceableAssetIds.length === 0) {
                 return next(ctx);
             }
@@ -222,7 +225,10 @@ export class PriceDataSource {
                 // Only report if we got prices
                 if (fetchResponse.assetsPrice &&
                     Object.keys(fetchResponse.assetsPrice).length > 0) {
-                    await subscription.onAssetsUpdate(fetchResponse);
+                    await subscription.onAssetsUpdate({
+                        ...fetchResponse,
+                        updateMode: 'merge',
+                    });
                 }
             }
             catch (error) {
@@ -267,7 +273,7 @@ export class PriceDataSource {
         __classPrivateFieldGet(this, _PriceDataSource_activeSubscriptions, "f").clear();
     }
 }
-_PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_fetchSpotPrices = 
+_PriceDataSource_getSelectedCurrency = new WeakMap(), _PriceDataSource_pollInterval = new WeakMap(), _PriceDataSource_apiClient = new WeakMap(), _PriceDataSource_activeSubscriptions = new WeakMap(), _PriceDataSource_instances = new WeakSet(), _PriceDataSource_fetchSpotPrices = 
 // ============================================================================
 // HELPERS
 // ============================================================================
@@ -279,7 +285,7 @@ _PriceDataSource_currency = new WeakMap(), _PriceDataSource_pollInterval = new W
  */
 async function _PriceDataSource_fetchSpotPrices(assetIds) {
     return __classPrivateFieldGet(this, _PriceDataSource_apiClient, "f").prices.fetchV3SpotPrices(assetIds, {
-        currency: __classPrivateFieldGet(this, _PriceDataSource_currency, "f"),
+        currency: __classPrivateFieldGet(this, _PriceDataSource_getSelectedCurrency, "f").call(this),
         includeMarketData: true,
     });
 }, _PriceDataSource_getAssetIdsFromBalanceState = function _PriceDataSource_getAssetIdsFromBalanceState(request, getAssetsState) {