@metamask/assets-controller 0.2.0 → 1.0.0

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

@@ -1,80 +1,33 @@
 import type { BackendWebSocketServiceActions, BackendWebSocketServiceEvents } from "@metamask/core-backend";
-import type { Messenger } from "@metamask/messenger";
+import type { ApiPlatformClient } from "@metamask/core-backend";
 import { AbstractDataSource } from "./AbstractDataSource.mjs";
 import type { DataSourceState, SubscriptionRequest } from "./AbstractDataSource.mjs";
-import type { ChainId, DataResponse } from "../types.mjs";
+import type { AssetsControllerMessenger } from "../AssetsController.mjs";
+import type { ChainId } from "../types.mjs";
 declare const CONTROLLER_NAME = "BackendWebsocketDataSource";
-export type BackendWebsocketDataSourceGetActiveChainsAction = {
-    type: 'BackendWebsocketDataSource:getActiveChains';
-    handler: () => Promise<ChainId[]>;
-};
-export type BackendWebsocketDataSourceSubscribeAction = {
-    type: 'BackendWebsocketDataSource:subscribe';
-    handler: (request: SubscriptionRequest) => Promise<void>;
-};
-export type BackendWebsocketDataSourceUnsubscribeAction = {
-    type: 'BackendWebsocketDataSource:unsubscribe';
-    handler: (subscriptionId: string) => Promise<void>;
-};
-export type BackendWebsocketDataSourceActions = BackendWebsocketDataSourceGetActiveChainsAction | BackendWebsocketDataSourceSubscribeAction | BackendWebsocketDataSourceUnsubscribeAction;
-export type BackendWebsocketDataSourceActiveChainsChangedEvent = {
-    type: 'BackendWebsocketDataSource:activeChainsUpdated';
-    payload: [ChainId[]];
-};
-export type BackendWebsocketDataSourceAssetsUpdatedEvent = {
-    type: 'BackendWebsocketDataSource:assetsUpdated';
-    payload: [DataResponse, string | undefined];
-};
-export type BackendWebsocketDataSourceEvents = BackendWebsocketDataSourceActiveChainsChangedEvent | BackendWebsocketDataSourceAssetsUpdatedEvent;
-type AssetsControllerActiveChainsUpdateAction = {
-    type: 'AssetsController:activeChainsUpdate';
-    handler: (dataSourceId: string, activeChains: ChainId[]) => void;
-};
-type AssetsControllerAssetsUpdateAction = {
-    type: 'AssetsController:assetsUpdate';
-    handler: (response: DataResponse, sourceId: string) => Promise<void>;
-};
-export type BackendWebsocketDataSourceAllowedActions = BackendWebSocketServiceActions | AssetsControllerActiveChainsUpdateAction | AssetsControllerAssetsUpdateAction;
-export type AccountsApiDataSourceActiveChainsChangedEvent = {
-    type: 'AccountsApiDataSource:activeChainsUpdated';
-    payload: [ChainId[]];
-};
-export type BackendWebsocketDataSourceAllowedEvents = BackendWebSocketServiceEvents | AccountsApiDataSourceActiveChainsChangedEvent;
-export type BackendWebsocketDataSourceMessenger = Messenger<typeof CONTROLLER_NAME, BackendWebsocketDataSourceActions | BackendWebsocketDataSourceAllowedActions, BackendWebsocketDataSourceEvents | BackendWebsocketDataSourceAllowedEvents>;
+export type BackendWebsocketDataSourceAllowedActions = BackendWebSocketServiceActions;
+export type BackendWebsocketDataSourceAllowedEvents = BackendWebSocketServiceEvents;
 export type BackendWebsocketDataSourceState = DataSourceState;
 export type BackendWebsocketDataSourceOptions = {
-    messenger: BackendWebsocketDataSourceMessenger;
+    /** The AssetsController messenger (shared by all data sources). */
+    messenger: AssetsControllerMessenger;
+    /** ApiPlatformClient for fetching supported networks at init (same as AccountsApiDataSource). */
+    queryApiClient: ApiPlatformClient;
+    /** Called when active chains are updated. Pass dataSourceName so the controller knows the source. */
+    onActiveChainsUpdated: (dataSourceName: string, chains: ChainId[], previousChains: ChainId[]) => void;
     state?: Partial<BackendWebsocketDataSourceState>;
 };
-/**
- * Data source for receiving real-time balance updates via WebSocket.
- *
- * This data source connects directly to BackendWebSocketService to receive
- * push notifications for account balance changes. Unlike AccountsApiDataSource
- * which polls for data, this provides instant updates.
- *
- * Uses Messenger pattern for all interactions:
- * - Calls BackendWebSocketService methods via messenger actions
- * - Exposes its own actions for AssetsController to call
- * - Publishes events for AssetsController to subscribe to
- *
- * Actions exposed:
- * - BackendWebsocketDataSource:getActiveChains
- * - BackendWebsocketDataSource:subscribe
- * - BackendWebsocketDataSource:unsubscribe
- *
- * Events published:
- * - BackendWebsocketDataSource:activeChainsUpdated
- * - BackendWebsocketDataSource:assetsUpdated
- *
- * Actions called (from BackendWebSocketService):
- * - BackendWebSocketService:subscribe
- * - BackendWebSocketService:getConnectionInfo
- * - BackendWebSocketService:findSubscriptionsByChannelPrefix
- */
 export declare class BackendWebsocketDataSource extends AbstractDataSource<typeof CONTROLLER_NAME, BackendWebsocketDataSourceState> {
     #private;
     constructor(options: BackendWebsocketDataSourceOptions);
+    /**
+     * Sync active chains from AccountsApiDataSource.
+     * Called by AssetsController.handleActiveChainsUpdate when the callback is
+     * invoked for BackendWebsocketDataSource (no messenger call; controller already updated).
+     *
+     * @param chains - Updated active chain IDs from AccountsApiDataSource.
+     */
+    setActiveChainsFromAccountsApi(chains: ChainId[]): void;
     /**
      * Update active chains when AccountsApiDataSource reports new supported chains.
      *

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

@@ -1 +1 @@
-{"version":3,"file":"BackendWebsocketDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/BackendWebsocketDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,8BAA8B,EAC9B,6BAA6B,EAM9B,+BAA+B;AAChC,OAAO,KAAK,EAAE,SAAS,EAAE,4BAA4B;AAErD,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAE9B,OAAO,KAAK,EACV,OAAO,EAKP,YAAY,EACb,qBAAiB;AAMlB,QAAA,MAAM,eAAe,+BAA+B,CAAC;AAUrD,MAAM,MAAM,+CAA+C,GAAG;IAC5D,IAAI,EAAE,4CAA4C,CAAC;IACnD,OAAO,EAAE,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,yCAAyC,GAAG;IACtD,IAAI,EAAE,sCAAsC,CAAC;IAC7C,OAAO,EAAE,CAAC,OAAO,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,2CAA2C,GAAG;IACxD,IAAI,EAAE,wCAAwC,CAAC;IAC/C,OAAO,EAAE,CAAC,cAAc,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACpD,CAAC;AAEF,MAAM,MAAM,iCAAiC,GACzC,+CAA+C,GAC/C,yCAAyC,GACzC,2CAA2C,CAAC;AAGhD,MAAM,MAAM,kDAAkD,GAAG;IAC/D,IAAI,EAAE,gDAAgD,CAAC;IACvD,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,4CAA4C,GAAG;IACzD,IAAI,EAAE,0CAA0C,CAAC;IACjD,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,gCAAgC,GACxC,kDAAkD,GAClD,4CAA4C,CAAC;AAGjD,KAAK,wCAAwC,GAAG;IAC9C,IAAI,EAAE,qCAAqC,CAAC;IAC5C,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CAClE,CAAC;AAEF,KAAK,kCAAkC,GAAG;IACxC,IAAI,EAAE,+BAA+B,CAAC;IACtC,OAAO,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACtE,CAAC;AAGF,MAAM,MAAM,wCAAwC,GAChD,8BAA8B,GAC9B,wCAAwC,GACxC,kCAAkC,CAAC;AAGvC,MAAM,MAAM,6CAA6C,GAAG;IAC1D,IAAI,EAAE,2CAA2C,CAAC;IAClD,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;CACtB,CAAC;AAGF,MAAM,MAAM,uCAAuC,GAC/C,6BAA6B,GAC7B,6CAA6C,CAAC;AAElD,MAAM,MAAM,mCAAmC,GAAG,SAAS,CACzD,OAAO,eAAe,EACtB,iCAAiC,GAAG,wCAAwC,EAC5E,gCAAgC,GAAG,uCAAuC,CAC3E,CAAC;AAMF,MAAM,MAAM,+BAA+B,GAAG,eAAe,CAAC;AAU9D,MAAM,MAAM,iCAAiC,GAAG;IAC9C,SAAS,EAAE,mCAAmC,CAAC;IAC/C,KAAK,CAAC,EAAE,OAAO,CAAC,+BAA+B,CAAC,CAAC;CAClD,CAAC;AAuDF;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,qBAAa,0BAA2B,SAAQ,kBAAkB,CAChE,OAAO,eAAe,EACtB,+BAA+B,CAChC;;gBAYa,OAAO,EAAE,iCAAiC;IAuItD;;;;OAIG;IACH,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI;IAcxC,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAyNxE,OAAO,IAAI,IAAI;CAmBhB;AAMD;;;;;GAKG;AACH,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,iCAAiC,GACzC,0BAA0B,CAE5B"}
+{"version":3,"file":"BackendWebsocketDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/BackendWebsocketDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,8BAA8B,EAC9B,6BAA6B,EAM9B,+BAA+B;AAChC,OAAO,KAAK,EAAE,iBAAiB,EAAE,+BAA+B;AAEhE,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EACV,OAAO,EAKR,qBAAiB;AAMlB,QAAA,MAAM,eAAe,+BAA+B,CAAC;AAUrD,MAAM,MAAM,wCAAwC,GAClD,8BAA8B,CAAC;AAGjC,MAAM,MAAM,uCAAuC,GACjD,6BAA6B,CAAC;AAMhC,MAAM,MAAM,+BAA+B,GAAG,eAAe,CAAC;AAU9D,MAAM,MAAM,iCAAiC,GAAG;IAC9C,mEAAmE;IACnE,SAAS,EAAE,yBAAyB,CAAC;IACrC,iGAAiG;IACjG,cAAc,EAAE,iBAAiB,CAAC;IAClC,qGAAqG;IACrG,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;IACV,KAAK,CAAC,EAAE,OAAO,CAAC,+BAA+B,CAAC,CAAC;CAClD,CAAC;AAoIF,qBAAa,0BAA2B,SAAQ,kBAAkB,CAChE,OAAO,eAAe,EACtB,+BAA+B,CAChC;;gBAuBa,OAAO,EAAE,iCAAiC;IAqFtD;;;;;;OAMG;IACH,8BAA8B,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI;IA8DvD;;;;OAIG;IACH,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI;IAWxC,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwOxE,OAAO,IAAI,IAAI;CAwBhB;AAMD;;;;;GAKG;AACH,wBAAgB,gCAAgC,CAC9C,OAAO,EAAE,iCAAiC,GACzC,0BAA0B,CAE5B"}

package/dist/data-sources/BackendWebsocketDataSource.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 _BackendWebsocketDataSource_instances, _BackendWebsocketDataSource_messenger, _BackendWebsocketDataSource_wsSubscriptions, _BackendWebsocketDataSource_pendingSubscriptions, _BackendWebsocketDataSource_subscriptionRequests, _BackendWebsocketDataSource_registerActionHandlers, _BackendWebsocketDataSource_subscribeToEvents, _BackendWebsocketDataSource_handleDisconnect, _BackendWebsocketDataSource_processPendingSubscriptions, _BackendWebsocketDataSource_handleNotification, _BackendWebsocketDataSource_processBalanceUpdates;
+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 { AbstractDataSource } from "./AbstractDataSource.mjs";
 import { projectLogger, createModuleLogger } from "../logger.mjs";
 // ============================================================================
@@ -35,31 +35,74 @@ function extractNamespace(chainId) {
     const [namespace] = chainId.split(':');
     return namespace;
 }
+/** Namespaces we always subscribe to for account activity (EVM + Solana). */
+const ACCOUNT_ACTIVITY_NAMESPACES = ['eip155', 'solana'];
 /**
- * Get unique namespaces from chain IDs.
+ * Get unique namespaces for account-activity subscriptions.
+ * Always includes eip155 and solana so we subscribe to both EVM and Solana account activity,
+ * plus any additional namespaces from the requested chain IDs.
  *
- * @param chainIds - Array of CAIP-2 chain IDs.
- * @returns Array of unique namespaces.
+ * @param chainIds - Array of CAIP-2 chain IDs (from the subscription request).
+ * @returns Array of unique namespaces (at least eip155 and solana).
  */
-function getUniqueNamespaces(chainIds) {
-    const namespaces = new Set();
+function getNamespacesForAccountActivity(chainIds) {
+    const namespaces = new Set(ACCOUNT_ACTIVITY_NAMESPACES);
     for (const chainId of chainIds) {
         namespaces.add(extractNamespace(chainId));
     }
     return Array.from(namespaces);
 }
+/**
+ * Returns the address to use for account-activity subscription in the given namespace.
+ * EIP-155 accounts use hex (0x...) address; Solana accounts use base58.
+ * Returns null if this account type does not have an address in that namespace.
+ *
+ * @param account - Internal account (type + address).
+ * @param account.type - Account type (e.g. "eip155:eoa", "solana:data-account").
+ * @param account.address - Account address (hex for eip155, base58 for solana).
+ * @param namespace - The chain namespace (e.g., "eip155", "solana").
+ * @returns The address for that namespace, or null if the account does not support the namespace.
+ */
+function getAddressForAccountActivity(account, namespace) {
+    if (namespace === 'eip155') {
+        return account.type.startsWith('eip155') ? account.address : null;
+    }
+    if (namespace === 'solana') {
+        return account.type.startsWith('solana') ? account.address : null;
+    }
+    // Other namespaces (e.g. from chainIds): use address if account type matches namespace
+    const typePrefix = `${namespace}:`;
+    return account.type.startsWith(typePrefix) ? account.address : null;
+}
 /**
  * Build WebSocket channel name for account activity using CAIP-10 wildcard format.
  * Uses 0 as the chain reference to subscribe to all chains in the namespace.
- * Format: account-activity.v1.eip155:0:0x1234... (all EVM chains)
- * Format: account-activity.v1.solana:0:ABC123... (all Solana chains)
+ * EIP-155 addresses are lowercased (hex); Solana addresses are left as-is (base58).
  *
  * @param namespace - The chain namespace (e.g., "eip155", "solana").
- * @param address - The account address.
+ * @param address - The account address (hex for eip155, base58 for solana).
  * @returns The WebSocket channel name.
  */
 function buildAccountActivityChannel(namespace, address) {
-    return `${CHANNEL_TYPE}.${namespace}:0:${address.toLowerCase()}`;
+    const formatted = namespace === 'eip155' ? address.toLowerCase() : address;
+    return `${CHANNEL_TYPE}.${namespace}:0:${formatted}`;
+}
+/**
+ * Normalize API chain identifier to CAIP-2 ChainId.
+ * Passes through strings already in namespace:reference form (e.g. eip155:1, solana:5eykt...).
+ * Converts bare decimals to eip155:decimal.
+ *
+ * @param chainIdOrDecimal - Chain ID string (CAIP-2 or decimal) or decimal number.
+ * @returns CAIP-2 ChainId.
+ */
+function toChainId(chainIdOrDecimal) {
+    if (typeof chainIdOrDecimal === 'string') {
+        if (chainIdOrDecimal.includes(':')) {
+            return chainIdOrDecimal;
+        }
+        return `eip155:${chainIdOrDecimal}`;
+    }
+    return `eip155:${chainIdOrDecimal}`;
 }
 // Note: AccountActivityMessage and BalanceUpdate types are imported from @metamask/core-backend
 // ============================================================================
@@ -91,6 +134,7 @@ function buildAccountActivityChannel(namespace, address) {
  * - BackendWebSocketService:getConnectionInfo
  * - BackendWebSocketService:findSubscriptionsByChannelPrefix
  */
+const DEFAULT_CHAINS_REFRESH_INTERVAL_MS = 20 * 60 * 1000; // 20 minutes
 export class BackendWebsocketDataSource extends AbstractDataSource {
     constructor(options) {
         super(CONTROLLER_NAME, {
@@ -99,6 +143,10 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
         });
         _BackendWebsocketDataSource_instances.add(this);
         _BackendWebsocketDataSource_messenger.set(this, void 0);
+        _BackendWebsocketDataSource_apiClient.set(this, void 0);
+        _BackendWebsocketDataSource_onActiveChainsUpdated.set(this, void 0);
+        /** Chains refresh timer */
+        _BackendWebsocketDataSource_chainsRefreshTimer.set(this, null);
         /** WebSocket subscriptions by our internal subscription ID */
         _BackendWebsocketDataSource_wsSubscriptions.set(this, new Map());
         /** Pending subscription requests to process when WebSocket connects */
@@ -106,8 +154,20 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
         /** Store original subscription requests for reconnection */
         _BackendWebsocketDataSource_subscriptionRequests.set(this, new Map());
         __classPrivateFieldSet(this, _BackendWebsocketDataSource_messenger, options.messenger, "f");
-        __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_registerActionHandlers).call(this);
+        __classPrivateFieldSet(this, _BackendWebsocketDataSource_apiClient, options.queryApiClient, "f");
+        __classPrivateFieldSet(this, _BackendWebsocketDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
         __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_subscribeToEvents).call(this);
+        __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_initializeActiveChains).call(this).catch(console.error);
+    }
+    /**
+     * Sync active chains from AccountsApiDataSource.
+     * Called by AssetsController.handleActiveChainsUpdate when the callback is
+     * invoked for BackendWebsocketDataSource (no messenger call; controller already updated).
+     *
+     * @param chains - Updated active chain IDs from AccountsApiDataSource.
+     */
+    setActiveChainsFromAccountsApi(chains) {
+        this.updateActiveChains(chains, () => undefined);
     }
     // ============================================================================
     // ACTIVE CHAINS
@@ -118,7 +178,8 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
      * @param chains - Array of supported chain IDs.
      */
     updateSupportedChains(chains) {
-        this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', CONTROLLER_NAME, updatedChains));
+        const previous = [...this.state.activeChains];
+        this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous));
     }
     // ============================================================================
     // SUBSCRIBE
@@ -127,8 +188,8 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
         const { request, subscriptionId, isUpdate } = subscriptionRequest;
         // Filter to active chains only
         const chainsToSubscribe = request.chainIds.filter((chainId) => this.state.activeChains.includes(chainId));
-        const addresses = request.accounts.map((a) => a.address);
-        if (chainsToSubscribe.length === 0) {
+        const addresses = request.accountsWithSupportedChains.map((a) => a.account.address);
+        if (addresses.length === 0) {
             return;
         }
         // Check WebSocket connection status
@@ -165,13 +226,16 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
         }
         // Clean up existing subscription if any
         await this.unsubscribe(subscriptionId);
-        // Extract unique namespaces from chains (e.g., eip155, solana)
-        const namespaces = getUniqueNamespaces(chainsToSubscribe);
-        // Build channel names using CAIP-10 wildcard format
+        // Always subscribe to eip155 and solana account activity, plus any namespaces from requested chains
+        const namespaces = getNamespacesForAccountActivity(chainsToSubscribe);
+        // Build channel names: use namespace-appropriate address per account (eip155 = hex, solana = base58)
         const channels = [];
         for (const namespace of namespaces) {
-            for (const address of addresses) {
-                channels.push(buildAccountActivityChannel(namespace, address));
+            for (const { account } of request.accountsWithSupportedChains) {
+                const address = getAddressForAccountActivity(account, namespace);
+                if (address) {
+                    channels.push(buildAccountActivityChannel(namespace, address));
+                }
             }
         }
         try {
@@ -180,7 +244,7 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
                 channels,
                 channelType: CHANNEL_TYPE,
                 callback: (notification) => {
-                    __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_handleNotification).call(this, notification, request);
+                    __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_handleNotification).call(this, notification, subscriptionId);
                 },
             });
             // Store WebSocket subscription
@@ -200,6 +264,7 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
                 },
                 chains: chainsToSubscribe,
                 addresses,
+                onAssetsUpdate: subscriptionRequest.onAssetsUpdate,
             });
             // Store original request for reconnection
             __classPrivateFieldGet(this, _BackendWebsocketDataSource_subscriptionRequests, "f").set(subscriptionId, subscriptionRequest);
@@ -216,6 +281,10 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
     // CLEANUP
     // ============================================================================
     destroy() {
+        if (__classPrivateFieldGet(this, _BackendWebsocketDataSource_chainsRefreshTimer, "f")) {
+            clearInterval(__classPrivateFieldGet(this, _BackendWebsocketDataSource_chainsRefreshTimer, "f"));
+            __classPrivateFieldSet(this, _BackendWebsocketDataSource_chainsRefreshTimer, null, "f");
+        }
         // Clean up WebSocket subscriptions
         // Convert to array first to avoid modifying map during iteration
         const subscriptions = [...__classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").values()];
@@ -235,32 +304,50 @@ export class BackendWebsocketDataSource extends AbstractDataSource {
         super.destroy();
     }
 }
-_BackendWebsocketDataSource_messenger = new WeakMap(), _BackendWebsocketDataSource_wsSubscriptions = new WeakMap(), _BackendWebsocketDataSource_pendingSubscriptions = new WeakMap(), _BackendWebsocketDataSource_subscriptionRequests = new WeakMap(), _BackendWebsocketDataSource_instances = new WeakSet(), _BackendWebsocketDataSource_registerActionHandlers = function _BackendWebsocketDataSource_registerActionHandlers() {
-    const getActiveChainsHandler = async () => this.getActiveChains();
-    const subscribeHandler = async (request) => this.subscribe(request);
-    const unsubscribeHandler = async (subscriptionId) => this.unsubscribe(subscriptionId);
-    __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").registerActionHandler('BackendWebsocketDataSource:getActiveChains', getActiveChainsHandler);
-    __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").registerActionHandler('BackendWebsocketDataSource:subscribe', subscribeHandler);
-    __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").registerActionHandler('BackendWebsocketDataSource:unsubscribe', unsubscribeHandler);
+_BackendWebsocketDataSource_messenger = new WeakMap(), _BackendWebsocketDataSource_apiClient = new WeakMap(), _BackendWebsocketDataSource_onActiveChainsUpdated = new WeakMap(), _BackendWebsocketDataSource_chainsRefreshTimer = new WeakMap(), _BackendWebsocketDataSource_wsSubscriptions = new WeakMap(), _BackendWebsocketDataSource_pendingSubscriptions = new WeakMap(), _BackendWebsocketDataSource_subscriptionRequests = new WeakMap(), _BackendWebsocketDataSource_instances = new WeakSet(), _BackendWebsocketDataSource_initializeActiveChains = 
+// ============================================================================
+// INITIALIZATION
+// ============================================================================
+async function _BackendWebsocketDataSource_initializeActiveChains() {
+    try {
+        const chains = await __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_fetchActiveChains).call(this);
+        const previous = [...this.state.activeChains];
+        this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous));
+        __classPrivateFieldSet(this, _BackendWebsocketDataSource_chainsRefreshTimer, setInterval(() => {
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_refreshActiveChains).call(this).catch(console.error);
+        }, DEFAULT_CHAINS_REFRESH_INTERVAL_MS), "f");
+    }
+    catch (error) {
+        log('Failed to fetch active chains', error);
+    }
+}, _BackendWebsocketDataSource_refreshActiveChains = async function _BackendWebsocketDataSource_refreshActiveChains() {
+    try {
+        const chains = await __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_fetchActiveChains).call(this);
+        const previousChains = new Set(this.state.activeChains);
+        const newChains = new Set(chains);
+        const added = chains.filter((chain) => !previousChains.has(chain));
+        const removed = Array.from(previousChains).filter((chain) => !newChains.has(chain));
+        if (added.length > 0 || removed.length > 0) {
+            const previous = [...this.state.activeChains];
+            this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous));
+        }
+    }
+    catch (error) {
+        log('Failed to refresh active chains', error);
+    }
+}, _BackendWebsocketDataSource_fetchActiveChains = async function _BackendWebsocketDataSource_fetchActiveChains() {
+    const response = await __classPrivateFieldGet(this, _BackendWebsocketDataSource_apiClient, "f").accounts.fetchV2SupportedNetworks();
+    return response.fullSupport.map(toChainId);
 }, _BackendWebsocketDataSource_subscribeToEvents = function _BackendWebsocketDataSource_subscribeToEvents() {
-    // Listen for WebSocket connection state changes
+    // Listen for WebSocket connection state changes (event not in AssetsControllerEvents).
     __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").subscribe('BackendWebSocketService:connectionStateChanged', (connectionInfo) => {
         if (connectionInfo.state === 'connected') {
-            // WebSocket connected - process any pending subscriptions
             __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_processPendingSubscriptions).call(this).catch(console.error);
         }
         else if (connectionInfo.state === 'disconnected') {
-            // When disconnected, all subscriptions are cleared server-side
-            // Move active subscriptions to pending for re-subscription on reconnect
             __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_handleDisconnect).call(this);
         }
     });
-    // Listen for AccountsApiDataSource active chains changes
-    // This keeps BackendWebsocketDataSource in sync with the supported chains
-    // since both use the same backend infrastructure
-    __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").subscribe('AccountsApiDataSource:activeChainsUpdated', (chains) => {
-        this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', CONTROLLER_NAME, updatedChains));
-    });
 }, _BackendWebsocketDataSource_handleDisconnect = function _BackendWebsocketDataSource_handleDisconnect() {
     log('WebSocket disconnected, preserving subscriptions for reconnect', {
         activeSubscriptionCount: this.activeSubscriptions.size,
@@ -304,8 +391,13 @@ async function _BackendWebsocketDataSource_processPendingSubscriptions() {
             });
         }
     }
-}, _BackendWebsocketDataSource_handleNotification = function _BackendWebsocketDataSource_handleNotification(notification, request) {
+}, _BackendWebsocketDataSource_handleNotification = function _BackendWebsocketDataSource_handleNotification(notification, subscriptionId) {
     try {
+        const subscription = this.activeSubscriptions.get(subscriptionId);
+        const request = __classPrivateFieldGet(this, _BackendWebsocketDataSource_subscriptionRequests, "f").get(subscriptionId)?.request;
+        if (!request) {
+            return;
+        }
         const activityMessage = notification.data;
         const { address, tx, updates } = activityMessage;
         if (!address || !tx || !updates) {
@@ -313,19 +405,20 @@ async function _BackendWebsocketDataSource_processPendingSubscriptions() {
         }
         // Extract chain ID from transaction (CAIP-2 format, e.g., "eip155:8453")
         const chainId = tx.chain;
-        // Find matching account in request
-        const account = request.accounts.find((a) => a.address.toLowerCase() === address.toLowerCase());
+        // Find matching account in request (eip155: case-insensitive hex; solana: exact base58)
+        const account = request.accountsWithSupportedChains
+            .map((entry) => entry.account)
+            .find((a) => a.address.startsWith('0x')
+            ? a.address.toLowerCase() === address.toLowerCase()
+            : a.address === address);
         if (!account) {
             return;
         }
         const accountId = account.id;
         // Process all balance updates from the activity message
         const response = __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_processBalanceUpdates).call(this, updates, chainId, accountId);
-        if (Object.keys(response).length > 0) {
-            // Report update to AssetsController
-            __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f")
-                .call('AssetsController:assetsUpdate', response, CONTROLLER_NAME)
-                .catch(console.error);
+        if (Object.keys(response).length > 0 && subscription) {
+            Promise.resolve(subscription.onAssetsUpdate(response)).catch(console.error);
         }
     }
     catch (error) {

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

@@ -1 +1 @@
-{"version":3,"file":"BackendWebsocketDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/BackendWebsocketDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAWA,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAK1D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAU9D,+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;AAkF/D,MAAM,YAAY,GAAoC;IACpD,YAAY,EAAE,EAAE;CACjB,CAAC;AAWF,+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;;;;;GAKG;AACH,SAAS,mBAAmB,CAAC,QAAmB;IAC9C,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;IACrC,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;;;;;;;;;GASG;AACH,SAAS,2BAA2B,CAClC,SAAiB,EACjB,OAAe;IAEf,OAAO,GAAG,YAAY,IAAI,SAAS,MAAM,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;AACnE,CAAC;AAED,gGAAgG;AAEhG,+EAA+E;AAC/E,gCAAgC;AAChC,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,OAAO,0BAA2B,SAAQ,kBAG/C;IAYC,YAAY,OAA0C;QACpD,KAAK,CAAC,eAAe,EAAE;YACrB,GAAG,YAAY;YACf,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QAfI,wDAAgD;QAEzD,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;QAEpC,uBAAA,IAAI,iGAAwB,MAA5B,IAAI,CAA0B,CAAC;QAC/B,uBAAA,IAAI,4FAAmB,MAAvB,IAAI,CAAqB,CAAC;IAC5B,CAAC;IAyHD,+EAA+E;IAC/E,gBAAgB;IAChB,+EAA+E;IAE/E;;;;OAIG;IACH,qBAAqB,CAAC,MAAiB;QACrC,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,6CAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,eAAe,EACf,aAAa,CACd,CACF,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,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAEzD,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,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,+DAA+D;QAC/D,MAAM,UAAU,GAAG,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,oDAAoD;QACpD,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;gBAChC,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;YACjE,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,OAAO,CAAC,CAAC;gBAClD,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,QAAQ,EAAE,EAAE;4BACrC,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;aACV,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;IAyGD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,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;;IArXG,MAAM,sBAAsB,GAC1B,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;IAErC,MAAM,gBAAgB,GACpB,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAE7C,MAAM,kBAAkB,GACtB,KAAK,EAAE,cAAc,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAE7D,uBAAA,IAAI,6CAAW,CAAC,qBAAqB,CACnC,4CAA4C,EAC5C,sBAAsB,CACvB,CAAC;IAEF,uBAAA,IAAI,6CAAW,CAAC,qBAAqB,CACnC,sCAAsC,EACtC,gBAAgB,CACjB,CAAC;IAEF,uBAAA,IAAI,6CAAW,CAAC,qBAAqB,CACnC,wCAAwC,EACxC,kBAAkB,CACnB,CAAC;AACJ,CAAC;IAGC,gDAAgD;IAChD,uBAAA,IAAI,6CAAW,CAAC,SAAS,CACvB,gDAAgD,EAChD,CAAC,cAAc,EAAE,EAAE;QACjB,IAAI,cAAc,CAAC,KAAK,KAAM,WAA8B,EAAE,CAAC;YAC7D,0DAA0D;YAC1D,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,+DAA+D;YAC/D,wEAAwE;YACxE,uBAAA,IAAI,2FAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC3B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,yDAAyD;IACzD,0EAA0E;IAC1E,iDAAiD;IACjD,uBAAA,IAAI,6CAAW,CAAC,SAAS,CACvB,2CAA2C,EAC3C,CAAC,MAAiB,EAAE,EAAE;QACpB,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,CAChD,uBAAA,IAAI,6CAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,eAAe,EACf,aAAa,CACd,CACF,CAAC;IACJ,CAAC,CACF,CAAC;AACJ,CAAC;IAOC,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,2GA4IC,YAAuC,EACvC,OAAoB;IAEpB,IAAI,CAAC;QACH,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,mCAAmC;QACnC,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE,CACzD,CAAC;QACF,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,EAAE,CAAC;YACrC,oCAAoC;YACpC,uBAAA,IAAI,6CAAW;iBACZ,IAAI,CAAC,+BAA+B,EAAE,QAAQ,EAAE,eAAe,CAAC;iBAChE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC1B,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,cAAc,GAAG,cAAc,CAAC;IAC3C,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AA2BH,+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 { Messenger } from '@metamask/messenger';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  ChainId,\n  Caip19AssetId,\n  AssetMetadata,\n  AssetBalance,\n  DataRequest,\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// Action types that BackendWebsocketDataSource exposes\nexport type BackendWebsocketDataSourceGetActiveChainsAction = {\n  type: 'BackendWebsocketDataSource:getActiveChains';\n  handler: () => Promise<ChainId[]>;\n};\n\nexport type BackendWebsocketDataSourceSubscribeAction = {\n  type: 'BackendWebsocketDataSource:subscribe';\n  handler: (request: SubscriptionRequest) => Promise<void>;\n};\n\nexport type BackendWebsocketDataSourceUnsubscribeAction = {\n  type: 'BackendWebsocketDataSource:unsubscribe';\n  handler: (subscriptionId: string) => Promise<void>;\n};\n\nexport type BackendWebsocketDataSourceActions =\n  | BackendWebsocketDataSourceGetActiveChainsAction\n  | BackendWebsocketDataSourceSubscribeAction\n  | BackendWebsocketDataSourceUnsubscribeAction;\n\n// Event types that BackendWebsocketDataSource publishes\nexport type BackendWebsocketDataSourceActiveChainsChangedEvent = {\n  type: 'BackendWebsocketDataSource:activeChainsUpdated';\n  payload: [ChainId[]];\n};\n\nexport type BackendWebsocketDataSourceAssetsUpdatedEvent = {\n  type: 'BackendWebsocketDataSource:assetsUpdated';\n  payload: [DataResponse, string | undefined];\n};\n\nexport type BackendWebsocketDataSourceEvents =\n  | BackendWebsocketDataSourceActiveChainsChangedEvent\n  | BackendWebsocketDataSourceAssetsUpdatedEvent;\n\n// Actions to report to AssetsController\ntype AssetsControllerActiveChainsUpdateAction = {\n  type: 'AssetsController:activeChainsUpdate';\n  handler: (dataSourceId: string, activeChains: ChainId[]) => void;\n};\n\ntype AssetsControllerAssetsUpdateAction = {\n  type: 'AssetsController:assetsUpdate';\n  handler: (response: DataResponse, sourceId: string) => Promise<void>;\n};\n\n// Allowed actions that BackendWebsocketDataSource can call\nexport type BackendWebsocketDataSourceAllowedActions =\n  | BackendWebSocketServiceActions\n  | AssetsControllerActiveChainsUpdateAction\n  | AssetsControllerAssetsUpdateAction;\n\n// Event type from AccountsApiDataSource that we subscribe to\nexport type AccountsApiDataSourceActiveChainsChangedEvent = {\n  type: 'AccountsApiDataSource:activeChainsUpdated';\n  payload: [ChainId[]];\n};\n\n// Allowed events that BackendWebsocketDataSource can subscribe to\nexport type BackendWebsocketDataSourceAllowedEvents =\n  | BackendWebSocketServiceEvents\n  | AccountsApiDataSourceActiveChainsChangedEvent;\n\nexport type BackendWebsocketDataSourceMessenger = Messenger<\n  typeof CONTROLLER_NAME,\n  BackendWebsocketDataSourceActions | BackendWebsocketDataSourceAllowedActions,\n  BackendWebsocketDataSourceEvents | BackendWebsocketDataSourceAllowedEvents\n>;\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  messenger: BackendWebsocketDataSourceMessenger;\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/**\n * Get unique namespaces from chain IDs.\n *\n * @param chainIds - Array of CAIP-2 chain IDs.\n * @returns Array of unique namespaces.\n */\nfunction getUniqueNamespaces(chainIds: ChainId[]): string[] {\n  const namespaces = new Set<string>();\n  for (const chainId of chainIds) {\n    namespaces.add(extractNamespace(chainId));\n  }\n  return Array.from(namespaces);\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 * Format: account-activity.v1.eip155:0:0x1234... (all EVM chains)\n * Format: account-activity.v1.solana:0:ABC123... (all Solana chains)\n *\n * @param namespace - The chain namespace (e.g., \"eip155\", \"solana\").\n * @param address - The account address.\n * @returns The WebSocket channel name.\n */\nfunction buildAccountActivityChannel(\n  namespace: string,\n  address: string,\n): string {\n  return `${CHANNEL_TYPE}.${namespace}:0:${address.toLowerCase()}`;\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 */\nexport class BackendWebsocketDataSource extends AbstractDataSource<\n  typeof CONTROLLER_NAME,\n  BackendWebsocketDataSourceState\n> {\n  readonly #messenger: BackendWebsocketDataSourceMessenger;\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\n    this.#registerActionHandlers();\n    this.#subscribeToEvents();\n  }\n\n  // ============================================================================\n  // INITIALIZATION\n  // ============================================================================\n\n  #registerActionHandlers(): void {\n    const getActiveChainsHandler: BackendWebsocketDataSourceGetActiveChainsAction['handler'] =\n      async () => this.getActiveChains();\n\n    const subscribeHandler: BackendWebsocketDataSourceSubscribeAction['handler'] =\n      async (request) => this.subscribe(request);\n\n    const unsubscribeHandler: BackendWebsocketDataSourceUnsubscribeAction['handler'] =\n      async (subscriptionId) => this.unsubscribe(subscriptionId);\n\n    this.#messenger.registerActionHandler(\n      'BackendWebsocketDataSource:getActiveChains',\n      getActiveChainsHandler,\n    );\n\n    this.#messenger.registerActionHandler(\n      'BackendWebsocketDataSource:subscribe',\n      subscribeHandler,\n    );\n\n    this.#messenger.registerActionHandler(\n      'BackendWebsocketDataSource:unsubscribe',\n      unsubscribeHandler,\n    );\n  }\n\n  #subscribeToEvents(): void {\n    // Listen for WebSocket connection state changes\n    this.#messenger.subscribe(\n      'BackendWebSocketService:connectionStateChanged',\n      (connectionInfo) => {\n        if (connectionInfo.state === ('connected' as WebSocketState)) {\n          // WebSocket connected - process any pending subscriptions\n          this.#processPendingSubscriptions().catch(console.error);\n        } else if (\n          connectionInfo.state === ('disconnected' as WebSocketState)\n        ) {\n          // When disconnected, all subscriptions are cleared server-side\n          // Move active subscriptions to pending for re-subscription on reconnect\n          this.#handleDisconnect();\n        }\n      },\n    );\n\n    // Listen for AccountsApiDataSource active chains changes\n    // This keeps BackendWebsocketDataSource in sync with the supported chains\n    // since both use the same backend infrastructure\n    this.#messenger.subscribe(\n      'AccountsApiDataSource:activeChainsUpdated',\n      (chains: ChainId[]) => {\n        this.updateActiveChains(chains, (updatedChains) =>\n          this.#messenger.call(\n            'AssetsController:activeChainsUpdate',\n            CONTROLLER_NAME,\n            updatedChains,\n          ),\n        );\n      },\n    );\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    this.updateActiveChains(chains, (updatedChains) =>\n      this.#messenger.call(\n        'AssetsController:activeChainsUpdate',\n        CONTROLLER_NAME,\n        updatedChains,\n      ),\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.accounts.map((a) => a.address);\n\n    if (chainsToSubscribe.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    // Extract unique namespaces from chains (e.g., eip155, solana)\n    const namespaces = getUniqueNamespaces(chainsToSubscribe);\n\n    // Build channel names using CAIP-10 wildcard format\n    const channels: string[] = [];\n    for (const namespace of namespaces) {\n      for (const address of addresses) {\n        channels.push(buildAccountActivityChannel(namespace, address));\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, request);\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) => {\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      });\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    request: DataRequest,\n  ): void {\n    try {\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\n      const account = request.accounts.find(\n        (a) => a.address.toLowerCase() === address.toLowerCase(),\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) {\n        // Report update to AssetsController\n        this.#messenger\n          .call('AssetsController:assetsUpdate', response, CONTROLLER_NAME)\n          .catch(console.error);\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.assetsMetadata = assetsMetadata;\n    }\n\n    return response;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\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":";;;;;;;;;;;;AAWA,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;;;;;;;GAOG;AACH,SAAS,SAAS,CAAC,gBAAiC;IAClD,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;QACzC,IAAI,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,OAAO,gBAA2B,CAAC;QACrC,CAAC;QACD,OAAO,UAAU,gBAAgB,EAAa,CAAC;IACjD,CAAC;IACD,OAAO,UAAU,gBAAgB,EAAa,CAAC;AACjD,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,cAAc,GAAG,cAAc,CAAC;IAC3C,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';\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 namespace:reference form (e.g. eip155:1, solana:5eykt...).\n * Converts bare decimals to eip155:decimal.\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 (chainIdOrDecimal.includes(':')) {\n      return chainIdOrDecimal as ChainId;\n    }\n    return `eip155:${chainIdOrDecimal}` as ChainId;\n  }\n  return `eip155:${chainIdOrDecimal}` as ChainId;\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.assetsMetadata = 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"]}