@metamask/assets-controller 0.1.0

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

@@ -0,0 +1,95 @@
+import type { BackendWebSocketServiceActions, BackendWebSocketServiceEvents } from "@metamask/core-backend";
+import type { Messenger } from "@metamask/messenger";
+import { AbstractDataSource } from "./AbstractDataSource.mjs";
+import type { DataSourceState, SubscriptionRequest } from "./AbstractDataSource.mjs";
+import type { ChainId, DataResponse } 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 BackendWebsocketDataSourceState = DataSourceState;
+export type BackendWebsocketDataSourceOptions = {
+    messenger: BackendWebsocketDataSourceMessenger;
+    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);
+    /**
+     * Update active chains when AccountsApiDataSource reports new supported chains.
+     *
+     * @param chains - Array of supported chain IDs.
+     */
+    updateSupportedChains(chains: ChainId[]): void;
+    subscribe(subscriptionRequest: SubscriptionRequest): Promise<void>;
+    destroy(): void;
+}
+/**
+ * Creates a BackendWebsocketDataSource instance.
+ *
+ * @param options - Configuration options for the data source.
+ * @returns A new BackendWebsocketDataSource instance.
+ */
+export declare function createBackendWebsocketDataSource(options: BackendWebsocketDataSourceOptions): BackendWebsocketDataSource;
+export {};
+//# sourceMappingURL=BackendWebsocketDataSource.d.mts.map

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

@@ -0,0 +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"}

package/dist/data-sources/BackendWebsocketDataSource.mjs

@@ -0,0 +1,383 @@
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    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;
+import { AbstractDataSource } from "./AbstractDataSource.mjs";
+import { projectLogger, createModuleLogger } from "../logger.mjs";
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+const CONTROLLER_NAME = 'BackendWebsocketDataSource';
+const CHANNEL_TYPE = 'account-activity.v1';
+const log = createModuleLogger(projectLogger, CONTROLLER_NAME);
+const defaultState = {
+    activeChains: [],
+};
+// ============================================================================
+// HELPER FUNCTIONS
+// ============================================================================
+/**
+ * Extract namespace from a CAIP-2 chain ID.
+ * E.g., "eip155:1" -> "eip155", "solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp" -> "solana"
+ *
+ * @param chainId - The CAIP-2 chain ID to extract namespace from.
+ * @returns The namespace portion of the chain ID.
+ */
+function extractNamespace(chainId) {
+    const [namespace] = chainId.split(':');
+    return namespace;
+}
+/**
+ * Get unique namespaces from chain IDs.
+ *
+ * @param chainIds - Array of CAIP-2 chain IDs.
+ * @returns Array of unique namespaces.
+ */
+function getUniqueNamespaces(chainIds) {
+    const namespaces = new Set();
+    for (const chainId of chainIds) {
+        namespaces.add(extractNamespace(chainId));
+    }
+    return Array.from(namespaces);
+}
+/**
+ * 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)
+ *
+ * @param namespace - The chain namespace (e.g., "eip155", "solana").
+ * @param address - The account address.
+ * @returns The WebSocket channel name.
+ */
+function buildAccountActivityChannel(namespace, address) {
+    return `${CHANNEL_TYPE}.${namespace}:0:${address.toLowerCase()}`;
+}
+// Note: AccountActivityMessage and BalanceUpdate types are imported from @metamask/core-backend
+// ============================================================================
+// BACKEND WEBSOCKET DATA SOURCE
+// ============================================================================
+/**
+ * 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 class BackendWebsocketDataSource extends AbstractDataSource {
+    constructor(options) {
+        super(CONTROLLER_NAME, {
+            ...defaultState,
+            ...options.state,
+        });
+        _BackendWebsocketDataSource_instances.add(this);
+        _BackendWebsocketDataSource_messenger.set(this, void 0);
+        /** WebSocket subscriptions by our internal subscription ID */
+        _BackendWebsocketDataSource_wsSubscriptions.set(this, new Map());
+        /** Pending subscription requests to process when WebSocket connects */
+        _BackendWebsocketDataSource_pendingSubscriptions.set(this, new Map());
+        /** 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);
+        __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_subscribeToEvents).call(this);
+    }
+    // ============================================================================
+    // ACTIVE CHAINS
+    // ============================================================================
+    /**
+     * Update active chains when AccountsApiDataSource reports new supported chains.
+     *
+     * @param chains - Array of supported chain IDs.
+     */
+    updateSupportedChains(chains) {
+        this.updateActiveChains(chains, (updatedChains) => __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', CONTROLLER_NAME, updatedChains));
+    }
+    // ============================================================================
+    // SUBSCRIBE
+    // ============================================================================
+    async subscribe(subscriptionRequest) {
+        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) {
+            return;
+        }
+        // Check WebSocket connection status
+        try {
+            const connectionInfo = __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").call('BackendWebSocketService:getConnectionInfo');
+            if (connectionInfo.state !== 'connected') {
+                // Store the subscription request to process when WebSocket connects
+                __classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").set(subscriptionId, subscriptionRequest);
+                return;
+            }
+        }
+        catch {
+            // Store anyway - will be processed when we can connect
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").set(subscriptionId, subscriptionRequest);
+            return;
+        }
+        // Remove from pending if it was there (we're processing it now)
+        __classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").delete(subscriptionId);
+        // Handle subscription update
+        if (isUpdate) {
+            const existing = this.activeSubscriptions.get(subscriptionId);
+            if (existing) {
+                // Check if accounts changed - if so, we need to re-subscribe to different channels
+                const existingAddresses = existing.addresses ?? [];
+                const addressesChanged = addresses.length !== existingAddresses.length ||
+                    addresses.some((addr) => !existingAddresses.includes(addr));
+                if (!addressesChanged) {
+                    // Only chains changed - just update chains and return
+                    existing.chains = chainsToSubscribe;
+                    return;
+                }
+                // Accounts changed - fall through to re-subscribe with new channels
+            }
+        }
+        // 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
+        const channels = [];
+        for (const namespace of namespaces) {
+            for (const address of addresses) {
+                channels.push(buildAccountActivityChannel(namespace, address));
+            }
+        }
+        try {
+            // Create WebSocket subscription
+            const wsSubscription = await __classPrivateFieldGet(this, _BackendWebsocketDataSource_messenger, "f").call('BackendWebSocketService:subscribe', {
+                channels,
+                channelType: CHANNEL_TYPE,
+                callback: (notification) => {
+                    __classPrivateFieldGet(this, _BackendWebsocketDataSource_instances, "m", _BackendWebsocketDataSource_handleNotification).call(this, notification, request);
+                },
+            });
+            // Store WebSocket subscription
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").set(subscriptionId, wsSubscription);
+            // Store in abstract class tracking
+            this.activeSubscriptions.set(subscriptionId, {
+                cleanup: () => {
+                    const wsSub = __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").get(subscriptionId);
+                    if (wsSub) {
+                        wsSub.unsubscribe().catch((unsubErr) => {
+                            log('Error unsubscribing', { subscriptionId, error: unsubErr });
+                        });
+                        __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").delete(subscriptionId);
+                    }
+                    // Also clean up the stored request
+                    __classPrivateFieldGet(this, _BackendWebsocketDataSource_subscriptionRequests, "f").delete(subscriptionId);
+                },
+                chains: chainsToSubscribe,
+                addresses,
+            });
+            // Store original request for reconnection
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_subscriptionRequests, "f").set(subscriptionId, subscriptionRequest);
+        }
+        catch (error) {
+            log('WebSocket subscription FAILED', {
+                subscriptionId,
+                error,
+                chains: chainsToSubscribe,
+            });
+        }
+    }
+    // ============================================================================
+    // CLEANUP
+    // ============================================================================
+    destroy() {
+        // Clean up WebSocket subscriptions
+        // Convert to array first to avoid modifying map during iteration
+        const subscriptions = [...__classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").values()];
+        for (const wsSub of subscriptions) {
+            try {
+                // Fire and forget - don't await in destroy
+                wsSub.unsubscribe().catch(() => {
+                    // Ignore errors during cleanup
+                });
+            }
+            catch {
+                // Ignore errors during cleanup
+            }
+        }
+        __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").clear();
+        // Clean up base class subscriptions
+        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_subscribeToEvents = function _BackendWebsocketDataSource_subscribeToEvents() {
+    // Listen for WebSocket connection state changes
+    __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,
+        wsSubscriptionCount: __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").size,
+    });
+    // Move active subscriptions to pending for re-subscription
+    for (const [subscriptionId] of this.activeSubscriptions) {
+        const originalRequest = __classPrivateFieldGet(this, _BackendWebsocketDataSource_subscriptionRequests, "f").get(subscriptionId);
+        if (originalRequest) {
+            // Mark as update since it was previously active
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").set(subscriptionId, {
+                ...originalRequest,
+                isUpdate: false, // Treat as new subscription since server cleared it
+            });
+        }
+    }
+    // Clear WebSocket subscriptions (server-side already cleared)
+    __classPrivateFieldGet(this, _BackendWebsocketDataSource_wsSubscriptions, "f").clear();
+    // Clear active subscriptions (they're no longer valid)
+    this.activeSubscriptions.clear();
+}, _BackendWebsocketDataSource_processPendingSubscriptions = 
+/**
+ * Process any pending subscriptions that were queued while WebSocket was disconnected.
+ */
+async function _BackendWebsocketDataSource_processPendingSubscriptions() {
+    if (__classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").size === 0) {
+        return;
+    }
+    // Process all pending subscriptions
+    const pendingEntries = Array.from(__classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").entries());
+    for (const [subscriptionId, request] of pendingEntries) {
+        try {
+            // Remove from pending before processing to avoid infinite loop
+            __classPrivateFieldGet(this, _BackendWebsocketDataSource_pendingSubscriptions, "f").delete(subscriptionId);
+            await this.subscribe(request);
+        }
+        catch (error) {
+            log('Failed to process pending subscription', {
+                subscriptionId,
+                error,
+            });
+        }
+    }
+}, _BackendWebsocketDataSource_handleNotification = function _BackendWebsocketDataSource_handleNotification(notification, request) {
+    try {
+        const activityMessage = notification.data;
+        const { address, tx, updates } = activityMessage;
+        if (!address || !tx || !updates) {
+            return;
+        }
+        // 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());
+        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);
+        }
+    }
+    catch (error) {
+        log('Error handling notification', error);
+    }
+}, _BackendWebsocketDataSource_processBalanceUpdates = function _BackendWebsocketDataSource_processBalanceUpdates(updates, _chainId, accountId) {
+    const assetsBalance = {
+        [accountId]: {},
+    };
+    const assetsMetadata = {};
+    for (const update of updates) {
+        const { asset, postBalance } = update;
+        if (!asset || !postBalance) {
+            continue;
+        }
+        // Asset type is in CAIP format: "eip155:1/erc20:0x..." or "eip155:1/slip44:60"
+        // We can use it directly as the asset ID
+        const assetId = asset.type;
+        // Determine token type from asset type string
+        const isNative = asset.type.includes('/slip44:');
+        const tokenType = isNative ? 'native' : 'erc20';
+        // Parse balance amount (already in hex format like "0xc350")
+        const balanceAmount = postBalance.amount.startsWith('0x')
+            ? BigInt(postBalance.amount).toString()
+            : postBalance.amount;
+        assetsBalance[accountId][assetId] = {
+            amount: balanceAmount,
+        };
+        assetsMetadata[assetId] = {
+            type: tokenType,
+            symbol: asset.unit,
+            name: asset.unit, // Use unit as name (actual name may not be in the message)
+            decimals: asset.decimals,
+        };
+    }
+    const response = {};
+    if (Object.keys(assetsBalance[accountId]).length > 0) {
+        response.assetsBalance = assetsBalance;
+        response.assetsMetadata = assetsMetadata;
+    }
+    return response;
+};
+// ============================================================================
+// FACTORY FUNCTION
+// ============================================================================
+/**
+ * Creates a BackendWebsocketDataSource instance.
+ *
+ * @param options - Configuration options for the data source.
+ * @returns A new BackendWebsocketDataSource instance.
+ */
+export function createBackendWebsocketDataSource(options) {
+    return new BackendWebsocketDataSource(options);
+}
+//# sourceMappingURL=BackendWebsocketDataSource.mjs.map

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

@@ -0,0 +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"]}