@metamask-previews/assets-controller 14.0.2-preview-4f4c98e → 14.0.2-preview-eb263b0ee

package/dist/data-sources/AbstractDataSource.cjs

@@ -0,0 +1,111 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AbstractDataSource = void 0;
+// ============================================================================
+// ABSTRACT DATA SOURCE
+// ============================================================================
+/**
+ * Abstract base class for data sources.
+ *
+ * Data sources communicate with AssetsController via Messenger:
+ * - Register actions that AssetsController can call
+ * - Publish events that AssetsController subscribes to
+ */
+class AbstractDataSource {
+    constructor(name, initialState) {
+        /** Active subscriptions by ID */
+        this.activeSubscriptions = new Map();
+        this.name = name;
+        this.state = initialState;
+    }
+    /**
+     * Get the data source name/ID.
+     *
+     * @returns The name of this data source.
+     */
+    getName() {
+        return this.name;
+    }
+    /**
+     * Get currently active chains (supported AND available).
+     *
+     * @returns Array of currently active chain IDs.
+     */
+    async getActiveChains() {
+        return this.state.activeChains;
+    }
+    /**
+     * Get currently active chains synchronously (no state duplication in controller).
+     *
+     * @returns Array of currently active chain IDs.
+     */
+    getActiveChainsSync() {
+        return this.state.activeChains;
+    }
+    /**
+     * Unsubscribe from updates.
+     *
+     * @param subscriptionId - The ID of the subscription to cancel.
+     */
+    async unsubscribe(subscriptionId) {
+        const subscription = this.activeSubscriptions.get(subscriptionId);
+        if (subscription) {
+            subscription.cleanup();
+            this.activeSubscriptions.delete(subscriptionId);
+        }
+    }
+    /**
+     * Update active chains and notify listeners only if changed.
+     *
+     * @param chains - Array of chain IDs to set as active.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    updateActiveChains(chains, publishEvent) {
+        const previousChains = new Set(this.state.activeChains);
+        const newChains = new Set(chains);
+        // Check if chains have actually changed
+        const hasChanges = previousChains.size !== newChains.size ||
+            chains.some((chain) => !previousChains.has(chain));
+        // Always update state
+        this.state.activeChains = chains;
+        // Only publish event if there are actual changes
+        if (hasChanges) {
+            publishEvent(chains);
+        }
+    }
+    /**
+     * Add a chain to active chains.
+     *
+     * @param chainId - The chain ID to add.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    addActiveChain(chainId, publishEvent) {
+        if (!this.state.activeChains.includes(chainId)) {
+            this.state.activeChains = [...this.state.activeChains, chainId];
+            publishEvent(this.state.activeChains);
+        }
+    }
+    /**
+     * Remove a chain from active chains.
+     *
+     * @param chainId - The chain ID to remove.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    removeActiveChain(chainId, publishEvent) {
+        if (this.state.activeChains.includes(chainId)) {
+            this.state.activeChains = this.state.activeChains.filter((chain) => chain !== chainId);
+            publishEvent(this.state.activeChains);
+        }
+    }
+    /**
+     * Destroy the data source and clean up resources.
+     */
+    destroy() {
+        for (const subscription of this.activeSubscriptions.values()) {
+            subscription.cleanup();
+        }
+        this.activeSubscriptions.clear();
+    }
+}
+exports.AbstractDataSource = AbstractDataSource;
+//# sourceMappingURL=AbstractDataSource.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AbstractDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/AbstractDataSource.ts"],"names":[],"mappings":";;;AAgEA,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;GAMG;AACH,MAAsB,kBAAkB;IAYtC,YAAY,IAAU,EAAE,YAAmB;QAJ3C,iCAAiC;QACd,wBAAmB,GACpC,IAAI,GAAG,EAAE,CAAC;QAGV,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACjC,CAAC;IAOD;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAClE,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,kBAAkB,CAC1B,MAAiB,EACjB,YAAyC;QAEzC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAElC,wCAAwC;QACxC,MAAM,UAAU,GACd,cAAc,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI;YACtC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAErD,sBAAsB;QACtB,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;QAEjC,iDAAiD;QACjD,IAAI,UAAU,EAAE,CAAC;YACf,YAAY,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,cAAc,CACtB,OAAgB,EAChB,YAAyC;QAEzC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAChE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,iBAAiB,CACzB,OAAgB,EAChB,YAAyC;QAEzC,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CACtD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,OAAO,CAC7B,CAAC;YACF,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7D,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF;AApID,gDAoIC","sourcesContent":["import type {\n  ChainId,\n  DataRequest,\n  DataResponse,\n  AssetsControllerStateInternal,\n} from '../types.js';\n\n// ============================================================================\n// DATA SOURCE BASE TYPES\n// ============================================================================\n\n/**\n * Subscription request from AssetsController.\n */\nexport type SubscriptionRequest = {\n  request: DataRequest;\n  subscriptionId: string;\n  isUpdate: boolean;\n  /** Called when this data source has new asset data. Passed by the controller when subscribing. Optional request describes the update scope (e.g. dataTypes: ['balance'] to skip enrichment). */\n  onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n  /**\n   * Optional state access (e.g. for price/token data sources that need assetsBalance).\n   * Provided by the controller when subscribing.\n   */\n  getAssetsState?: () => AssetsControllerStateInternal;\n  /**\n   * When true, skip the one-shot fetch that normally runs when a subscription\n   * is created. Used after the controller has already force-fetched balances\n   * for the same accounts/chains so we do not immediately hit the API again.\n   */\n  skipInitialFetch?: boolean;\n};\n\n/**\n * Active subscription entry stored by data sources.\n * Data sources can extend this with additional fields as needed.\n */\nexport type ActiveSubscription = {\n  /** Cleanup function to unsubscribe */\n  cleanup: () => void;\n  /** Currently subscribed chains */\n  chains: ChainId[];\n  /** Original request (for polling data sources to handle account changes) */\n  request?: DataRequest;\n  /** Account addresses (for WebSocket data sources to detect account changes) */\n  addresses?: string[];\n  /** Callback to report asset updates (from SubscriptionRequest) */\n  onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n};\n\n/**\n * Base state for all data sources.\n */\nexport type DataSourceState = {\n  /** Currently active chains (supported AND available) */\n  activeChains: ChainId[];\n};\n\n// ============================================================================\n// ABSTRACT DATA SOURCE\n// ============================================================================\n\n/**\n * Abstract base class for data sources.\n *\n * Data sources communicate with AssetsController via Messenger:\n * - Register actions that AssetsController can call\n * - Publish events that AssetsController subscribes to\n */\nexport abstract class AbstractDataSource<\n  Name extends string,\n  State extends DataSourceState = DataSourceState,\n> {\n  protected readonly name: Name;\n\n  protected state: State;\n\n  /** Active subscriptions by ID */\n  protected readonly activeSubscriptions: Map<string, ActiveSubscription> =\n    new Map();\n\n  constructor(name: Name, initialState: State) {\n    this.name = name;\n    this.state = initialState;\n  }\n\n  /**\n   * Get the data source name/ID.\n   *\n   * @returns The name of this data source.\n   */\n  getName(): Name {\n    return this.name;\n  }\n\n  /**\n   * Get currently active chains (supported AND available).\n   *\n   * @returns Array of currently active chain IDs.\n   */\n  async getActiveChains(): Promise<ChainId[]> {\n    return this.state.activeChains;\n  }\n\n  /**\n   * Get currently active chains synchronously (no state duplication in controller).\n   *\n   * @returns Array of currently active chain IDs.\n   */\n  getActiveChainsSync(): ChainId[] {\n    return this.state.activeChains;\n  }\n\n  /**\n   * Subscribe to updates for the given request.\n   */\n  abstract subscribe(request: SubscriptionRequest): Promise<void>;\n\n  /**\n   * Unsubscribe from updates.\n   *\n   * @param subscriptionId - The ID of the subscription to cancel.\n   */\n  async unsubscribe(subscriptionId: string): Promise<void> {\n    const subscription = this.activeSubscriptions.get(subscriptionId);\n    if (subscription) {\n      subscription.cleanup();\n      this.activeSubscriptions.delete(subscriptionId);\n    }\n  }\n\n  /**\n   * Update active chains and notify listeners only if changed.\n   *\n   * @param chains - Array of chain IDs to set as active.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected updateActiveChains(\n    chains: ChainId[],\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    const previousChains = new Set(this.state.activeChains);\n    const newChains = new Set(chains);\n\n    // Check if chains have actually changed\n    const hasChanges =\n      previousChains.size !== newChains.size ||\n      chains.some((chain) => !previousChains.has(chain));\n\n    // Always update state\n    this.state.activeChains = chains;\n\n    // Only publish event if there are actual changes\n    if (hasChanges) {\n      publishEvent(chains);\n    }\n  }\n\n  /**\n   * Add a chain to active chains.\n   *\n   * @param chainId - The chain ID to add.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected addActiveChain(\n    chainId: ChainId,\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    if (!this.state.activeChains.includes(chainId)) {\n      this.state.activeChains = [...this.state.activeChains, chainId];\n      publishEvent(this.state.activeChains);\n    }\n  }\n\n  /**\n   * Remove a chain from active chains.\n   *\n   * @param chainId - The chain ID to remove.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected removeActiveChain(\n    chainId: ChainId,\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    if (this.state.activeChains.includes(chainId)) {\n      this.state.activeChains = this.state.activeChains.filter(\n        (chain) => chain !== chainId,\n      );\n      publishEvent(this.state.activeChains);\n    }\n  }\n\n  /**\n   * Destroy the data source and clean up resources.\n   */\n  destroy(): void {\n    for (const subscription of this.activeSubscriptions.values()) {\n      subscription.cleanup();\n    }\n    this.activeSubscriptions.clear();\n  }\n}\n"]}

package/dist/data-sources/{AbstractDataSource.d.ts → AbstractDataSource.d.cts}

@@ -1,4 +1,4 @@
-import type { ChainId, DataRequest, DataResponse, AssetsControllerStateInternal } from '../types.js';
+import type { ChainId, DataRequest, DataResponse, AssetsControllerStateInternal } from "../types.cjs";
 /**
  * Subscription request from AssetsController.
  */
@@ -110,4 +110,4 @@ export declare abstract class AbstractDataSource<Name extends string, State exte
      */
     destroy(): void;
 }
-//# sourceMappingURL=AbstractDataSource.d.ts.map
+//# sourceMappingURL=AbstractDataSource.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AbstractDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/AbstractDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,YAAY,EACZ,6BAA6B,EAC9B,qBAAoB;AAMrB;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,WAAW,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,gMAAgM;IAChM,cAAc,EAAE,CACd,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE,WAAW,KAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,6BAA6B,CAAC;IACrD;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,sCAAsC;IACtC,OAAO,EAAE,MAAM,IAAI,CAAC;IACpB,kCAAkC;IAClC,MAAM,EAAE,OAAO,EAAE,CAAC;IAClB,4EAA4E;IAC5E,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,+EAA+E;IAC/E,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,kEAAkE;IAClE,cAAc,EAAE,CACd,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE,WAAW,KAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,wDAAwD;IACxD,YAAY,EAAE,OAAO,EAAE,CAAC;CACzB,CAAC;AAMF;;;;;;GAMG;AACH,8BAAsB,kBAAkB,CACtC,IAAI,SAAS,MAAM,EACnB,KAAK,SAAS,eAAe,GAAG,eAAe;IAE/C,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;IAE9B,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC;IAEvB,iCAAiC;IACjC,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAC3D;gBAEA,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK;IAK3C;;;;OAIG;IACH,OAAO,IAAI,IAAI;IAIf;;;;OAIG;IACG,eAAe,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAI3C;;;;OAIG;IACH,mBAAmB,IAAI,OAAO,EAAE;IAIhC;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAE/D;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;;;;OAKG;IACH,SAAS,CAAC,kBAAkB,CAC1B,MAAM,EAAE,OAAO,EAAE,EACjB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IAkBP;;;;;OAKG;IACH,SAAS,CAAC,cAAc,CACtB,OAAO,EAAE,OAAO,EAChB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IAOP;;;;;OAKG;IACH,SAAS,CAAC,iBAAiB,CACzB,OAAO,EAAE,OAAO,EAChB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IASP;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}

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

@@ -0,0 +1,113 @@
+import type { ChainId, DataRequest, DataResponse, AssetsControllerStateInternal } from "../types.mjs";
+/**
+ * Subscription request from AssetsController.
+ */
+export type SubscriptionRequest = {
+    request: DataRequest;
+    subscriptionId: string;
+    isUpdate: boolean;
+    /** Called when this data source has new asset data. Passed by the controller when subscribing. Optional request describes the update scope (e.g. dataTypes: ['balance'] to skip enrichment). */
+    onAssetsUpdate: (response: DataResponse, request?: DataRequest) => void | Promise<void>;
+    /**
+     * Optional state access (e.g. for price/token data sources that need assetsBalance).
+     * Provided by the controller when subscribing.
+     */
+    getAssetsState?: () => AssetsControllerStateInternal;
+    /**
+     * When true, skip the one-shot fetch that normally runs when a subscription
+     * is created. Used after the controller has already force-fetched balances
+     * for the same accounts/chains so we do not immediately hit the API again.
+     */
+    skipInitialFetch?: boolean;
+};
+/**
+ * Active subscription entry stored by data sources.
+ * Data sources can extend this with additional fields as needed.
+ */
+export type ActiveSubscription = {
+    /** Cleanup function to unsubscribe */
+    cleanup: () => void;
+    /** Currently subscribed chains */
+    chains: ChainId[];
+    /** Original request (for polling data sources to handle account changes) */
+    request?: DataRequest;
+    /** Account addresses (for WebSocket data sources to detect account changes) */
+    addresses?: string[];
+    /** Callback to report asset updates (from SubscriptionRequest) */
+    onAssetsUpdate: (response: DataResponse, request?: DataRequest) => void | Promise<void>;
+};
+/**
+ * Base state for all data sources.
+ */
+export type DataSourceState = {
+    /** Currently active chains (supported AND available) */
+    activeChains: ChainId[];
+};
+/**
+ * Abstract base class for data sources.
+ *
+ * Data sources communicate with AssetsController via Messenger:
+ * - Register actions that AssetsController can call
+ * - Publish events that AssetsController subscribes to
+ */
+export declare abstract class AbstractDataSource<Name extends string, State extends DataSourceState = DataSourceState> {
+    protected readonly name: Name;
+    protected state: State;
+    /** Active subscriptions by ID */
+    protected readonly activeSubscriptions: Map<string, ActiveSubscription>;
+    constructor(name: Name, initialState: State);
+    /**
+     * Get the data source name/ID.
+     *
+     * @returns The name of this data source.
+     */
+    getName(): Name;
+    /**
+     * Get currently active chains (supported AND available).
+     *
+     * @returns Array of currently active chain IDs.
+     */
+    getActiveChains(): Promise<ChainId[]>;
+    /**
+     * Get currently active chains synchronously (no state duplication in controller).
+     *
+     * @returns Array of currently active chain IDs.
+     */
+    getActiveChainsSync(): ChainId[];
+    /**
+     * Subscribe to updates for the given request.
+     */
+    abstract subscribe(request: SubscriptionRequest): Promise<void>;
+    /**
+     * Unsubscribe from updates.
+     *
+     * @param subscriptionId - The ID of the subscription to cancel.
+     */
+    unsubscribe(subscriptionId: string): Promise<void>;
+    /**
+     * Update active chains and notify listeners only if changed.
+     *
+     * @param chains - Array of chain IDs to set as active.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    protected updateActiveChains(chains: ChainId[], publishEvent: (chains: ChainId[]) => void): void;
+    /**
+     * Add a chain to active chains.
+     *
+     * @param chainId - The chain ID to add.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    protected addActiveChain(chainId: ChainId, publishEvent: (chains: ChainId[]) => void): void;
+    /**
+     * Remove a chain from active chains.
+     *
+     * @param chainId - The chain ID to remove.
+     * @param publishEvent - Callback to publish chain changes to listeners.
+     */
+    protected removeActiveChain(chainId: ChainId, publishEvent: (chains: ChainId[]) => void): void;
+    /**
+     * Destroy the data source and clean up resources.
+     */
+    destroy(): void;
+}
+//# sourceMappingURL=AbstractDataSource.d.mts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AbstractDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/AbstractDataSource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,YAAY,EACZ,6BAA6B,EAC9B,qBAAoB;AAMrB;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,WAAW,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,gMAAgM;IAChM,cAAc,EAAE,CACd,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE,WAAW,KAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,6BAA6B,CAAC;IACrD;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,sCAAsC;IACtC,OAAO,EAAE,MAAM,IAAI,CAAC;IACpB,kCAAkC;IAClC,MAAM,EAAE,OAAO,EAAE,CAAC;IAClB,4EAA4E;IAC5E,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,+EAA+E;IAC/E,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,kEAAkE;IAClE,cAAc,EAAE,CACd,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE,WAAW,KAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,wDAAwD;IACxD,YAAY,EAAE,OAAO,EAAE,CAAC;CACzB,CAAC;AAMF;;;;;;GAMG;AACH,8BAAsB,kBAAkB,CACtC,IAAI,SAAS,MAAM,EACnB,KAAK,SAAS,eAAe,GAAG,eAAe;IAE/C,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;IAE9B,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC;IAEvB,iCAAiC;IACjC,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAC3D;gBAEA,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK;IAK3C;;;;OAIG;IACH,OAAO,IAAI,IAAI;IAIf;;;;OAIG;IACG,eAAe,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAI3C;;;;OAIG;IACH,mBAAmB,IAAI,OAAO,EAAE;IAIhC;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAE/D;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxD;;;;;OAKG;IACH,SAAS,CAAC,kBAAkB,CAC1B,MAAM,EAAE,OAAO,EAAE,EACjB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IAkBP;;;;;OAKG;IACH,SAAS,CAAC,cAAc,CACtB,OAAO,EAAE,OAAO,EAChB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IAOP;;;;;OAKG;IACH,SAAS,CAAC,iBAAiB,CACzB,OAAO,EAAE,OAAO,EAChB,YAAY,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,GACxC,IAAI;IASP;;OAEG;IACH,OAAO,IAAI,IAAI;CAMhB"}

package/dist/data-sources/{AbstractDataSource.js → AbstractDataSource.mjs}

@@ -104,4 +104,4 @@ export class AbstractDataSource {
         this.activeSubscriptions.clear();
     }
 }
-//# sourceMappingURL=AbstractDataSource.js.map
+//# sourceMappingURL=AbstractDataSource.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AbstractDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/AbstractDataSource.ts"],"names":[],"mappings":"AAgEA,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;GAMG;AACH,MAAM,OAAgB,kBAAkB;IAYtC,YAAY,IAAU,EAAE,YAAmB;QAJ3C,iCAAiC;QACd,wBAAmB,GACpC,IAAI,GAAG,EAAE,CAAC;QAGV,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACjC,CAAC;IAOD;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAClE,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,kBAAkB,CAC1B,MAAiB,EACjB,YAAyC;QAEzC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAElC,wCAAwC;QACxC,MAAM,UAAU,GACd,cAAc,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI;YACtC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAErD,sBAAsB;QACtB,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;QAEjC,iDAAiD;QACjD,IAAI,UAAU,EAAE,CAAC;YACf,YAAY,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,cAAc,CACtB,OAAgB,EAChB,YAAyC;QAEzC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAChE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACO,iBAAiB,CACzB,OAAgB,EAChB,YAAyC;QAEzC,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CACtD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,OAAO,CAC7B,CAAC;YACF,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7D,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF","sourcesContent":["import type {\n  ChainId,\n  DataRequest,\n  DataResponse,\n  AssetsControllerStateInternal,\n} from '../types.js';\n\n// ============================================================================\n// DATA SOURCE BASE TYPES\n// ============================================================================\n\n/**\n * Subscription request from AssetsController.\n */\nexport type SubscriptionRequest = {\n  request: DataRequest;\n  subscriptionId: string;\n  isUpdate: boolean;\n  /** Called when this data source has new asset data. Passed by the controller when subscribing. Optional request describes the update scope (e.g. dataTypes: ['balance'] to skip enrichment). */\n  onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n  /**\n   * Optional state access (e.g. for price/token data sources that need assetsBalance).\n   * Provided by the controller when subscribing.\n   */\n  getAssetsState?: () => AssetsControllerStateInternal;\n  /**\n   * When true, skip the one-shot fetch that normally runs when a subscription\n   * is created. Used after the controller has already force-fetched balances\n   * for the same accounts/chains so we do not immediately hit the API again.\n   */\n  skipInitialFetch?: boolean;\n};\n\n/**\n * Active subscription entry stored by data sources.\n * Data sources can extend this with additional fields as needed.\n */\nexport type ActiveSubscription = {\n  /** Cleanup function to unsubscribe */\n  cleanup: () => void;\n  /** Currently subscribed chains */\n  chains: ChainId[];\n  /** Original request (for polling data sources to handle account changes) */\n  request?: DataRequest;\n  /** Account addresses (for WebSocket data sources to detect account changes) */\n  addresses?: string[];\n  /** Callback to report asset updates (from SubscriptionRequest) */\n  onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n};\n\n/**\n * Base state for all data sources.\n */\nexport type DataSourceState = {\n  /** Currently active chains (supported AND available) */\n  activeChains: ChainId[];\n};\n\n// ============================================================================\n// ABSTRACT DATA SOURCE\n// ============================================================================\n\n/**\n * Abstract base class for data sources.\n *\n * Data sources communicate with AssetsController via Messenger:\n * - Register actions that AssetsController can call\n * - Publish events that AssetsController subscribes to\n */\nexport abstract class AbstractDataSource<\n  Name extends string,\n  State extends DataSourceState = DataSourceState,\n> {\n  protected readonly name: Name;\n\n  protected state: State;\n\n  /** Active subscriptions by ID */\n  protected readonly activeSubscriptions: Map<string, ActiveSubscription> =\n    new Map();\n\n  constructor(name: Name, initialState: State) {\n    this.name = name;\n    this.state = initialState;\n  }\n\n  /**\n   * Get the data source name/ID.\n   *\n   * @returns The name of this data source.\n   */\n  getName(): Name {\n    return this.name;\n  }\n\n  /**\n   * Get currently active chains (supported AND available).\n   *\n   * @returns Array of currently active chain IDs.\n   */\n  async getActiveChains(): Promise<ChainId[]> {\n    return this.state.activeChains;\n  }\n\n  /**\n   * Get currently active chains synchronously (no state duplication in controller).\n   *\n   * @returns Array of currently active chain IDs.\n   */\n  getActiveChainsSync(): ChainId[] {\n    return this.state.activeChains;\n  }\n\n  /**\n   * Subscribe to updates for the given request.\n   */\n  abstract subscribe(request: SubscriptionRequest): Promise<void>;\n\n  /**\n   * Unsubscribe from updates.\n   *\n   * @param subscriptionId - The ID of the subscription to cancel.\n   */\n  async unsubscribe(subscriptionId: string): Promise<void> {\n    const subscription = this.activeSubscriptions.get(subscriptionId);\n    if (subscription) {\n      subscription.cleanup();\n      this.activeSubscriptions.delete(subscriptionId);\n    }\n  }\n\n  /**\n   * Update active chains and notify listeners only if changed.\n   *\n   * @param chains - Array of chain IDs to set as active.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected updateActiveChains(\n    chains: ChainId[],\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    const previousChains = new Set(this.state.activeChains);\n    const newChains = new Set(chains);\n\n    // Check if chains have actually changed\n    const hasChanges =\n      previousChains.size !== newChains.size ||\n      chains.some((chain) => !previousChains.has(chain));\n\n    // Always update state\n    this.state.activeChains = chains;\n\n    // Only publish event if there are actual changes\n    if (hasChanges) {\n      publishEvent(chains);\n    }\n  }\n\n  /**\n   * Add a chain to active chains.\n   *\n   * @param chainId - The chain ID to add.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected addActiveChain(\n    chainId: ChainId,\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    if (!this.state.activeChains.includes(chainId)) {\n      this.state.activeChains = [...this.state.activeChains, chainId];\n      publishEvent(this.state.activeChains);\n    }\n  }\n\n  /**\n   * Remove a chain from active chains.\n   *\n   * @param chainId - The chain ID to remove.\n   * @param publishEvent - Callback to publish chain changes to listeners.\n   */\n  protected removeActiveChain(\n    chainId: ChainId,\n    publishEvent: (chains: ChainId[]) => void,\n  ): void {\n    if (this.state.activeChains.includes(chainId)) {\n      this.state.activeChains = this.state.activeChains.filter(\n        (chain) => chain !== chainId,\n      );\n      publishEvent(this.state.activeChains);\n    }\n  }\n\n  /**\n   * Destroy the data source and clean up resources.\n   */\n  destroy(): void {\n    for (const subscription of this.activeSubscriptions.values()) {\n      subscription.cleanup();\n    }\n    this.activeSubscriptions.clear();\n  }\n}\n"]}

package/dist/data-sources/AccountActivityDataSource.cjs

@@ -0,0 +1,267 @@
+"use strict";
+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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _AccountActivityDataSource_instances, _AccountActivityDataSource_messenger, _AccountActivityDataSource_onActiveChainsUpdated, _AccountActivityDataSource_getAssetType, _AccountActivityDataSource_onAssetsUpdate, _AccountActivityDataSource_onBalanceUpdatedBound, _AccountActivityDataSource_subscribeToEvents, _AccountActivityDataSource_onBalanceUpdated, _AccountActivityDataSource_findAccountForAddress, _AccountActivityDataSource_getWalletAccounts, _AccountActivityDataSource_onAccountActivityStatusChanged;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createAccountActivityDataSource = exports.AccountActivityDataSource = void 0;
+const utils_1 = require("@metamask/utils");
+const bignumber_js_1 = __importDefault(require("bignumber.js"));
+const logger_js_1 = require("../logger.cjs");
+const index_js_1 = require("../utils/index.cjs");
+const AbstractDataSource_js_1 = require("./AbstractDataSource.cjs");
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+const CONTROLLER_NAME = 'AccountActivityDataSource';
+const log = (0, logger_js_1.createModuleLogger)(logger_js_1.projectLogger, CONTROLLER_NAME);
+// ============================================================================
+// BALANCE UPDATE PROCESSING
+// ============================================================================
+/**
+ * Convert AccountActivityMessage balance updates into a {@link DataResponse}
+ * for AssetsController.
+ *
+ * @param updates - Balance updates from account-activity websocket payload.
+ * @param accountId - Internal account UUID.
+ * @param getAssetType - Resolver for asset metadata type.
+ * @returns DataResponse with merge mode when balances are present.
+ */
+function processAccountActivityBalanceUpdates(updates, accountId, getAssetType) {
+    const assetsBalance = Object.create(null);
+    assetsBalance[accountId] = Object.create(null);
+    const assetsMetadata = Object.create(null);
+    for (const update of updates) {
+        const { asset, postBalance } = update;
+        if (!asset || !postBalance) {
+            continue;
+        }
+        // The websocket sends lower-case ERC-20 addresses, while state and the
+        // rest of the pipeline key assets by their checksummed ID. Normalizing
+        // here keeps middleware comparisons (detection, custom-asset graduation,
+        // occurrence filtering) from treating an existing holding as brand new.
+        const assetId = (0, index_js_1.safeNormalizeAssetId)(asset.type);
+        if (asset.decimals === undefined) {
+            continue;
+        }
+        const rawBalanceStr = postBalance.amount.startsWith('0x')
+            ? BigInt(postBalance.amount).toString()
+            : postBalance.amount;
+        const humanReadableAmount = new bignumber_js_1.default(rawBalanceStr)
+            .dividedBy(new bignumber_js_1.default(10).pow(asset.decimals))
+            .toFixed();
+        assetsBalance[accountId][assetId] = {
+            amount: humanReadableAmount,
+        };
+        assetsMetadata[assetId] = {
+            type: getAssetType(assetId),
+            symbol: asset.unit,
+            name: asset.unit,
+            decimals: asset.decimals,
+        };
+    }
+    const response = { updateMode: 'merge' };
+    if (Object.keys(assetsBalance[accountId]).length > 0) {
+        response.assetsBalance = assetsBalance;
+        response.assetsInfo = assetsMetadata;
+    }
+    return response;
+}
+const defaultState = {
+    activeChains: [],
+};
+// ============================================================================
+// ACCOUNT ACTIVITY DATA SOURCE
+// ============================================================================
+/**
+ * Data source that consumes real-time updates from `AccountActivityService`.
+ *
+ * `AccountActivityService` owns the WebSocket connection and channel
+ * subscriptions; this data source is a thin consumer of the two high-level
+ * events that it publishes:
+ *
+ * - `AccountActivityService:balanceUpdated` — post-transaction balances for the
+ *   subscribed account(s). The address is resolved against the wallet's
+ *   selected account group, transformed into a {@link DataResponse} (via
+ *   {@link processAccountActivityBalanceUpdates}), and pushed to the controller
+ *   through the injected `onAssetsUpdate` callback.
+ * - `AccountActivityService:statusChanged` — per-chain "up"/"down" notifications.
+ *   A chain reported "up" is claimed as an active chain (WebSocket is providing
+ *   real-time data); a chain reported "down" is released so polling data sources
+ *   can take over. Each change is applied directly via `updateActiveChains`,
+ *   which notifies the controller through `onActiveChainsUpdated`.
+ *
+ * This data source does NOT debounce, jitter, or gate chain updates itself.
+ * Coalescing (collapsing bursts) and jitter (staggering the WS-subscribe herd
+ * across clients) live in `AssetsController`, where the expensive re-subscribe
+ * happens and where updates from all data sources converge. `activeChains` are
+ * never seeded from the accounts API; they only ever reflect live
+ * `statusChanged` events (the service flushes all tracked chains as "down" when
+ * the WebSocket disconnects, releasing them).
+ *
+ * It subscribes to these events in its constructor (the same way
+ * `TokenBalancesController` does). Unlike the polling data sources it does not
+ * take part in the controller's subscribe/unsubscribe handoff: `subscribe` is a
+ * no-op, balance updates are routed by resolving the address against the
+ * wallet, and updates are pushed through the injected `onAssetsUpdate` callback.
+ */
+class AccountActivityDataSource extends AbstractDataSource_js_1.AbstractDataSource {
+    constructor(options) {
+        super(CONTROLLER_NAME, {
+            ...defaultState,
+            ...options.state,
+        });
+        _AccountActivityDataSource_instances.add(this);
+        _AccountActivityDataSource_messenger.set(this, void 0);
+        _AccountActivityDataSource_onActiveChainsUpdated.set(this, void 0);
+        _AccountActivityDataSource_getAssetType.set(this, void 0);
+        _AccountActivityDataSource_onAssetsUpdate.set(this, void 0);
+        _AccountActivityDataSource_onBalanceUpdatedBound.set(this, void 0);
+        // ============================================================================
+        // STATUS CHANGES
+        // ============================================================================
+        /**
+         * Handle a `statusChanged` notification. Chains reported "up" are claimed as
+         * active (WebSocket provides real-time data); chains reported "down" are
+         * released so polling data sources take over. The change is applied directly;
+         * coalescing and jitter are handled by `AssetsController`.
+         *
+         * @param options - The status change notification.
+         * @param options.chainIds - The CAIP-2 chain IDs whose status changed.
+         * @param options.status - Whether the chains are `up` or `down`.
+         * @param options.timestamp - Optional timestamp of the status change.
+         */
+        _AccountActivityDataSource_onAccountActivityStatusChanged.set(this, ({ chainIds, status, }) => {
+            try {
+                // Act on every namespace (eip155, solana, etc.); AssetsController is
+                // multichain. Only skip identifiers that are not valid CAIP-2 chain IDs.
+                const validChains = chainIds.filter((chainId) => (0, utils_1.isCaipChainId)(chainId));
+                if (validChains.length === 0) {
+                    return;
+                }
+                const next = new Set(this.state.activeChains);
+                if (status === 'up') {
+                    for (const chainId of validChains) {
+                        next.add(chainId);
+                    }
+                }
+                else {
+                    for (const chainId of validChains) {
+                        next.delete(chainId);
+                    }
+                }
+                const previous = [...this.state.activeChains];
+                this.updateActiveChains(Array.from(next), (updatedChains) => __classPrivateFieldGet(this, _AccountActivityDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous));
+            }
+            catch (error) {
+                log('Error handling status change', error);
+            }
+        });
+        __classPrivateFieldSet(this, _AccountActivityDataSource_messenger, options.messenger, "f");
+        __classPrivateFieldSet(this, _AccountActivityDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
+        __classPrivateFieldSet(this, _AccountActivityDataSource_getAssetType, options.getAssetType, "f");
+        __classPrivateFieldSet(this, _AccountActivityDataSource_onAssetsUpdate, options.onAssetsUpdate, "f");
+        __classPrivateFieldSet(this, _AccountActivityDataSource_onBalanceUpdatedBound, __classPrivateFieldGet(this, _AccountActivityDataSource_instances, "m", _AccountActivityDataSource_onBalanceUpdated).bind(this), "f");
+        __classPrivateFieldGet(this, _AccountActivityDataSource_instances, "m", _AccountActivityDataSource_subscribeToEvents).call(this);
+    }
+    // ============================================================================
+    // SUBSCRIBE / UNSUBSCRIBE
+    // ============================================================================
+    /**
+     * AADS is event-driven and chain-agnostic: it never participates in the
+     * controller's subscribe/unsubscribe handoff. Incoming `balanceUpdated`
+     * events are routed by resolving the address against the wallet's accounts
+     * (see `#onBalanceUpdated`), and updates are pushed through the injected
+     * `#onAssetsUpdate` callback. This override exists only to satisfy the
+     * abstract contract and is intentionally a no-op.
+     */
+    async subscribe() {
+        // Intentionally empty — see method doc.
+    }
+    // ============================================================================
+    // CLEANUP
+    // ============================================================================
+    destroy() {
+        try {
+            __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").unsubscribe('AccountActivityService:balanceUpdated', __classPrivateFieldGet(this, _AccountActivityDataSource_onBalanceUpdatedBound, "f"));
+        }
+        catch (error) {
+            log('Failed to unsubscribe from balanceUpdated', { error });
+        }
+        try {
+            __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").unsubscribe('AccountActivityService:statusChanged', __classPrivateFieldGet(this, _AccountActivityDataSource_onAccountActivityStatusChanged, "f"));
+        }
+        catch (error) {
+            log('Failed to unsubscribe from statusChanged', { error });
+        }
+        super.destroy();
+    }
+}
+exports.AccountActivityDataSource = AccountActivityDataSource;
+_AccountActivityDataSource_messenger = new WeakMap(), _AccountActivityDataSource_onActiveChainsUpdated = new WeakMap(), _AccountActivityDataSource_getAssetType = new WeakMap(), _AccountActivityDataSource_onAssetsUpdate = new WeakMap(), _AccountActivityDataSource_onBalanceUpdatedBound = new WeakMap(), _AccountActivityDataSource_onAccountActivityStatusChanged = new WeakMap(), _AccountActivityDataSource_instances = new WeakSet(), _AccountActivityDataSource_subscribeToEvents = function _AccountActivityDataSource_subscribeToEvents() {
+    __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").subscribe('AccountActivityService:balanceUpdated', __classPrivateFieldGet(this, _AccountActivityDataSource_onBalanceUpdatedBound, "f"));
+    __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").subscribe('AccountActivityService:statusChanged', __classPrivateFieldGet(this, _AccountActivityDataSource_onAccountActivityStatusChanged, "f"));
+}, _AccountActivityDataSource_onBalanceUpdated = function _AccountActivityDataSource_onBalanceUpdated({ address, chain, updates, }) {
+    try {
+        if (!address || !chain || !updates || updates.length === 0) {
+            return;
+        }
+        const account = __classPrivateFieldGet(this, _AccountActivityDataSource_instances, "m", _AccountActivityDataSource_findAccountForAddress).call(this, address);
+        if (!account) {
+            return;
+        }
+        const chainId = chain;
+        const response = processAccountActivityBalanceUpdates(updates, account.id, (assetId) => __classPrivateFieldGet(this, _AccountActivityDataSource_getAssetType, "f").call(this, assetId));
+        if (!response.assetsBalance) {
+            return;
+        }
+        const request = {
+            accountsWithSupportedChains: [{ account, supportedChains: [chainId] }],
+            chainIds: [chainId],
+            dataTypes: ['balance', 'metadata'],
+        };
+        Promise.resolve(__classPrivateFieldGet(this, _AccountActivityDataSource_onAssetsUpdate, "f").call(this, response, request)).catch((error) => {
+            log('Failed to report balance update', { error });
+        });
+    }
+    catch (error) {
+        log('Error handling balance update', error);
+    }
+}, _AccountActivityDataSource_findAccountForAddress = function _AccountActivityDataSource_findAccountForAddress(address) {
+    const isEvm = address.startsWith('0x');
+    return (__classPrivateFieldGet(this, _AccountActivityDataSource_instances, "m", _AccountActivityDataSource_getWalletAccounts).call(this).find((candidate) => isEvm
+        ? candidate.address.toLowerCase() === address.toLowerCase()
+        : candidate.address === address) ?? null);
+}, _AccountActivityDataSource_getWalletAccounts = function _AccountActivityDataSource_getWalletAccounts() {
+    const accounts = __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").call('AccountTreeController:getAccountsFromSelectedAccountGroup');
+    if (accounts.length > 0) {
+        return accounts;
+    }
+    const selectedAccount = __classPrivateFieldGet(this, _AccountActivityDataSource_messenger, "f").call('AccountsController:getSelectedAccount');
+    return selectedAccount ? [selectedAccount] : [];
+};
+// ============================================================================
+// FACTORY FUNCTION
+// ============================================================================
+/**
+ * Creates an AccountActivityDataSource instance.
+ *
+ * @param options - Configuration options for the data source.
+ * @returns A new AccountActivityDataSource instance.
+ */
+function createAccountActivityDataSource(options) {
+    return new AccountActivityDataSource(options);
+}
+exports.createAccountActivityDataSource = createAccountActivityDataSource;
+//# sourceMappingURL=AccountActivityDataSource.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AccountActivityDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/AccountActivityDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AAMA,2CAAgD;AAChD,gEAAuC;AAGvC,6CAAiE;AASjE,iDAAyD;AACzD,oEAA6D;AAG7D,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,2BAA2B,CAAC;AAEpD,MAAM,GAAG,GAAG,IAAA,8BAAkB,EAAC,yBAAa,EAAE,eAAe,CAAC,CAAC;AAE/D,+EAA+E;AAC/E,4BAA4B;AAC5B,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,SAAS,oCAAoC,CAC3C,OAAwB,EACxB,SAAiB,EACjB,YAAoE;IAEpE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAGvC,CAAC;IACF,aAAa,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAG5C,CAAC;IACF,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAGxC,CAAC;IAEF,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,uEAAuE;QACvE,uEAAuE;QACvE,yEAAyE;QACzE,wEAAwE;QACxE,MAAM,OAAO,GAAG,IAAA,+BAAoB,EAAC,KAAK,CAAC,IAAqB,CAAC,CAAC;QAElE,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACjC,SAAS;QACX,CAAC;QAED,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,MAAM,mBAAmB,GAAG,IAAI,sBAAW,CAAC,aAAa,CAAC;aACvD,SAAS,CAAC,IAAI,sBAAW,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;aAClD,OAAO,EAAE,CAAC;QAEb,aAAa,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG;YAClC,MAAM,EAAE,mBAAmB;SAC5B,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,GAAG;YACxB,IAAI,EAAE,YAAY,CAAC,OAAO,CAAC;YAC3B,MAAM,EAAE,KAAK,CAAC,IAAI;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IACvD,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACvC,QAAQ,CAAC,UAAU,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAiBD,MAAM,YAAY,GAAmC;IACnD,YAAY,EAAE,EAAE;CACjB,CAAC;AA8BF,+EAA+E;AAC/E,+BAA+B;AAC/B,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,MAAa,yBAA0B,SAAQ,0CAG9C;IAwBC,YAAY,OAAyC;QACnD,KAAK,CAAC,eAAe,EAAE;YACrB,GAAG,YAAY;YACf,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QA3BI,uDAAsC;QAEtC,mEAIC;QAED,0DAEuB;QAEvB,4DAGiB;QAEjB,mEAIE;QA8IX,+EAA+E;QAC/E,iBAAiB;QACjB,+EAA+E;QAE/E;;;;;;;;;;WAUG;QACM,oEAAkC,CAAC,EAC1C,QAAQ,EACR,MAAM,GAKP,EAAQ,EAAE;YACT,IAAI,CAAC;gBACH,qEAAqE;gBACrE,yEAAyE;gBACzE,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC9C,IAAA,qBAAa,EAAC,OAAO,CAAC,CACV,CAAC;gBAEf,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7B,OAAO;gBACT,CAAC;gBAED,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC9C,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;oBACpB,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;wBAClC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACpB,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;wBAClC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;gBAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC9C,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,EAAE,EAAE,CAC1D,uBAAA,IAAI,wDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC,EAAC;QA1LA,uBAAA,IAAI,wCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,oDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,2CAAiB,OAAO,CAAC,YAAY,MAAA,CAAC;QAC1C,uBAAA,IAAI,6CAAmB,OAAO,CAAC,cAAc,MAAA,CAAC;QAC9C,uBAAA,IAAI,oDAA0B,uBAAA,IAAI,yFAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAA,CAAC;QAEhE,uBAAA,IAAI,0FAAmB,MAAvB,IAAI,CAAqB,CAAC;IAC5B,CAAC;IAiBD,+EAA+E;IAC/E,0BAA0B;IAC1B,+EAA+E;IAE/E;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS;QACb,wCAAwC;IAC1C,CAAC;IAsJD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,IAAI,CAAC;YACH,uBAAA,IAAI,4CAAW,CAAC,WAAW,CACzB,uCAAuC,EACvC,uBAAA,IAAI,wDAAuB,CAC5B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,2CAA2C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC;YACH,uBAAA,IAAI,4CAAW,CAAC,WAAW,CACzB,sCAAsC,EACtC,uBAAA,IAAI,iEAAgC,CACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,0CAA0C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;AAtPD,8DAsPC;;IAvMG,uBAAA,IAAI,4CAAW,CAAC,SAAS,CACvB,uCAAuC,EACvC,uBAAA,IAAI,wDAAuB,CAC5B,CAAC;IACF,uBAAA,IAAI,4CAAW,CAAC,SAAS,CACvB,sCAAsC,EACtC,uBAAA,IAAI,iEAAgC,CACrC,CAAC;AACJ,CAAC,qGAsBiB,EAChB,OAAO,EACP,KAAK,EACL,OAAO,GAKR;IACC,IAAI,CAAC;QACH,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3D,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,uBAAA,IAAI,8FAAuB,MAA3B,IAAI,EAAwB,OAAO,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,KAAgB,CAAC;QAEjC,MAAM,QAAQ,GAAG,oCAAoC,CACnD,OAAO,EACP,OAAO,CAAC,EAAE,EACV,CAAC,OAAO,EAAE,EAAE,CAAC,uBAAA,IAAI,+CAAc,MAAlB,IAAI,EAAe,OAAO,CAAC,CACzC,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAgB;YAC3B,2BAA2B,EAAE,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;YACtE,QAAQ,EAAE,CAAC,OAAO,CAAC;YACnB,SAAS,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;SACnC,CAAC;QAEF,OAAO,CAAC,OAAO,CAAC,uBAAA,IAAI,iDAAgB,MAApB,IAAI,EAAiB,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,KAAK,CAC5D,CAAC,KAAK,EAAE,EAAE;YACR,GAAG,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC,+GAasB,OAAe;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAEvC,OAAO,CACL,uBAAA,IAAI,0FAAmB,MAAvB,IAAI,CAAqB,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAC3C,KAAK;QACH,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE;QAC3D,CAAC,CAAC,SAAS,CAAC,OAAO,KAAK,OAAO,CAClC,IAAI,IAAI,CACV,CAAC;AACJ,CAAC;IASC,MAAM,QAAQ,GAAG,uBAAA,IAAI,4CAAW,CAAC,IAAI,CACnC,2DAA2D,CAC5D,CAAC;IACF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,eAAe,GAAG,uBAAA,IAAI,4CAAW,CAAC,IAAI,CAC1C,uCAAuC,CACxC,CAAC;IACF,OAAO,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAClD,CAAC;AAmFH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;GAKG;AACH,SAAgB,+BAA+B,CAC7C,OAAyC;IAEzC,OAAO,IAAI,yBAAyB,CAAC,OAAO,CAAC,CAAC;AAChD,CAAC;AAJD,0EAIC","sourcesContent":["import type {\n  AccountActivityServiceBalanceUpdatedEvent,\n  AccountActivityServiceStatusChangedEvent,\n  BalanceUpdate,\n} from '@metamask/core-backend';\nimport type { InternalAccount } from '@metamask/keyring-internal-api';\nimport { isCaipChainId } from '@metamask/utils';\nimport BigNumberJS from 'bignumber.js';\n\nimport type { AssetsControllerMessenger } from '../AssetsController.js';\nimport { projectLogger, createModuleLogger } from '../logger.js';\nimport type {\n  AssetBalance,\n  AssetMetadata,\n  ChainId,\n  Caip19AssetId,\n  DataRequest,\n  DataResponse,\n} from '../types.js';\nimport { safeNormalizeAssetId } from '../utils/index.js';\nimport { AbstractDataSource } from './AbstractDataSource.js';\nimport type { DataSourceState } from './AbstractDataSource.js';\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst CONTROLLER_NAME = 'AccountActivityDataSource';\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n// ============================================================================\n// BALANCE UPDATE PROCESSING\n// ============================================================================\n\n/**\n * Convert AccountActivityMessage balance updates into a {@link DataResponse}\n * for AssetsController.\n *\n * @param updates - Balance updates from account-activity websocket payload.\n * @param accountId - Internal account UUID.\n * @param getAssetType - Resolver for asset metadata type.\n * @returns DataResponse with merge mode when balances are present.\n */\nfunction processAccountActivityBalanceUpdates(\n  updates: BalanceUpdate[],\n  accountId: string,\n  getAssetType: (assetId: Caip19AssetId) => 'native' | 'erc20' | 'spl',\n): DataResponse {\n  const assetsBalance = Object.create(null) as Record<\n    string,\n    Record<Caip19AssetId, AssetBalance>\n  >;\n  assetsBalance[accountId] = Object.create(null) as Record<\n    Caip19AssetId,\n    AssetBalance\n  >;\n  const assetsMetadata = Object.create(null) as Record<\n    Caip19AssetId,\n    AssetMetadata\n  >;\n\n  for (const update of updates) {\n    const { asset, postBalance } = update;\n\n    if (!asset || !postBalance) {\n      continue;\n    }\n\n    // The websocket sends lower-case ERC-20 addresses, while state and the\n    // rest of the pipeline key assets by their checksummed ID. Normalizing\n    // here keeps middleware comparisons (detection, custom-asset graduation,\n    // occurrence filtering) from treating an existing holding as brand new.\n    const assetId = safeNormalizeAssetId(asset.type as Caip19AssetId);\n\n    if (asset.decimals === undefined) {\n      continue;\n    }\n\n    const rawBalanceStr = postBalance.amount.startsWith('0x')\n      ? BigInt(postBalance.amount).toString()\n      : postBalance.amount;\n\n    const humanReadableAmount = new BigNumberJS(rawBalanceStr)\n      .dividedBy(new BigNumberJS(10).pow(asset.decimals))\n      .toFixed();\n\n    assetsBalance[accountId][assetId] = {\n      amount: humanReadableAmount,\n    };\n\n    assetsMetadata[assetId] = {\n      type: getAssetType(assetId),\n      symbol: asset.unit,\n      name: asset.unit,\n      decimals: asset.decimals,\n    };\n  }\n\n  const response: DataResponse = { updateMode: 'merge' };\n  if (Object.keys(assetsBalance[accountId]).length > 0) {\n    response.assetsBalance = assetsBalance;\n    response.assetsInfo = assetsMetadata;\n  }\n\n  return response;\n}\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\n/** Allowed events that AccountActivityDataSource subscribes to. */\nexport type AccountActivityDataSourceAllowedEvents =\n  | AccountActivityServiceBalanceUpdatedEvent\n  | AccountActivityServiceStatusChangedEvent;\n\n// ============================================================================\n// STATE\n// ============================================================================\n\nexport type AccountActivityDataSourceState = DataSourceState;\n\nconst defaultState: AccountActivityDataSourceState = {\n  activeChains: [],\n};\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type AccountActivityDataSourceOptions = {\n  /** The AssetsController messenger (shared by all data sources). */\n  messenger: AssetsControllerMessenger;\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  /** Returns the asset type ('native' | 'erc20' | 'spl') for a given CAIP-19 asset ID. */\n  getAssetType: (assetId: Caip19AssetId) => 'native' | 'erc20' | 'spl';\n  /**\n   * Pushes decoded balance updates to the controller (bound to\n   * `AssetsController.handleAssetsUpdate`). AADS is event-driven and never\n   * takes part in the controller's subscribe handoff, so it receives this\n   * callback directly instead of via a `SubscriptionRequest`.\n   */\n  onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n  state?: Partial<AccountActivityDataSourceState>;\n};\n\n// ============================================================================\n// ACCOUNT ACTIVITY DATA SOURCE\n// ============================================================================\n\n/**\n * Data source that consumes real-time updates from `AccountActivityService`.\n *\n * `AccountActivityService` owns the WebSocket connection and channel\n * subscriptions; this data source is a thin consumer of the two high-level\n * events that it publishes:\n *\n * - `AccountActivityService:balanceUpdated` — post-transaction balances for the\n *   subscribed account(s). The address is resolved against the wallet's\n *   selected account group, transformed into a {@link DataResponse} (via\n *   {@link processAccountActivityBalanceUpdates}), and pushed to the controller\n *   through the injected `onAssetsUpdate` callback.\n * - `AccountActivityService:statusChanged` — per-chain \"up\"/\"down\" notifications.\n *   A chain reported \"up\" is claimed as an active chain (WebSocket is providing\n *   real-time data); a chain reported \"down\" is released so polling data sources\n *   can take over. Each change is applied directly via `updateActiveChains`,\n *   which notifies the controller through `onActiveChainsUpdated`.\n *\n * This data source does NOT debounce, jitter, or gate chain updates itself.\n * Coalescing (collapsing bursts) and jitter (staggering the WS-subscribe herd\n * across clients) live in `AssetsController`, where the expensive re-subscribe\n * happens and where updates from all data sources converge. `activeChains` are\n * never seeded from the accounts API; they only ever reflect live\n * `statusChanged` events (the service flushes all tracked chains as \"down\" when\n * the WebSocket disconnects, releasing them).\n *\n * It subscribes to these events in its constructor (the same way\n * `TokenBalancesController` does). Unlike the polling data sources it does not\n * take part in the controller's subscribe/unsubscribe handoff: `subscribe` is a\n * no-op, balance updates are routed by resolving the address against the\n * wallet, and updates are pushed through the injected `onAssetsUpdate` callback.\n */\nexport class AccountActivityDataSource extends AbstractDataSource<\n  typeof CONTROLLER_NAME,\n  AccountActivityDataSourceState\n> {\n  readonly #messenger: AssetsControllerMessenger;\n\n  readonly #onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n\n  readonly #getAssetType: (\n    assetId: Caip19AssetId,\n  ) => 'native' | 'erc20' | 'spl';\n\n  readonly #onAssetsUpdate: (\n    response: DataResponse,\n    request?: DataRequest,\n  ) => void | Promise<void>;\n\n  readonly #onBalanceUpdatedBound: (event: {\n    address: string;\n    chain: string;\n    updates: BalanceUpdate[];\n  }) => void;\n\n  constructor(options: AccountActivityDataSourceOptions) {\n    super(CONTROLLER_NAME, {\n      ...defaultState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n    this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n    this.#getAssetType = options.getAssetType;\n    this.#onAssetsUpdate = options.onAssetsUpdate;\n    this.#onBalanceUpdatedBound = this.#onBalanceUpdated.bind(this);\n\n    this.#subscribeToEvents();\n  }\n\n  // ============================================================================\n  // EVENT SUBSCRIPTIONS\n  // ============================================================================\n\n  #subscribeToEvents(): void {\n    this.#messenger.subscribe(\n      'AccountActivityService:balanceUpdated',\n      this.#onBalanceUpdatedBound,\n    );\n    this.#messenger.subscribe(\n      'AccountActivityService:statusChanged',\n      this.#onAccountActivityStatusChanged,\n    );\n  }\n\n  // ============================================================================\n  // SUBSCRIBE / UNSUBSCRIBE\n  // ============================================================================\n\n  /**\n   * AADS is event-driven and chain-agnostic: it never participates in the\n   * controller's subscribe/unsubscribe handoff. Incoming `balanceUpdated`\n   * events are routed by resolving the address against the wallet's accounts\n   * (see `#onBalanceUpdated`), and updates are pushed through the injected\n   * `#onAssetsUpdate` callback. This override exists only to satisfy the\n   * abstract contract and is intentionally a no-op.\n   */\n  async subscribe(): Promise<void> {\n    // Intentionally empty — see method doc.\n  }\n\n  // ============================================================================\n  // BALANCE UPDATES\n  // ============================================================================\n\n  #onBalanceUpdated({\n    address,\n    chain,\n    updates,\n  }: {\n    address: string;\n    chain: string;\n    updates: BalanceUpdate[];\n  }): void {\n    try {\n      if (!address || !chain || !updates || updates.length === 0) {\n        return;\n      }\n\n      const account = this.#findAccountForAddress(address);\n      if (!account) {\n        return;\n      }\n\n      const chainId = chain as ChainId;\n\n      const response = processAccountActivityBalanceUpdates(\n        updates,\n        account.id,\n        (assetId) => this.#getAssetType(assetId),\n      );\n\n      if (!response.assetsBalance) {\n        return;\n      }\n\n      const request: DataRequest = {\n        accountsWithSupportedChains: [{ account, supportedChains: [chainId] }],\n        chainIds: [chainId],\n        dataTypes: ['balance', 'metadata'],\n      };\n\n      Promise.resolve(this.#onAssetsUpdate(response, request)).catch(\n        (error) => {\n          log('Failed to report balance update', { error });\n        },\n      );\n    } catch (error) {\n      log('Error handling balance update', error);\n    }\n  }\n\n  /**\n   * Find the wallet account matching the given activity address. EVM addresses\n   * are matched case-insensitively; other namespaces are matched exactly.\n   *\n   * The candidate set is the accounts of the selected account group, resolved\n   * from the wallet at event time (rather than from a stored subscription), so\n   * AADS does not need to be re-subscribed whenever the account set changes.\n   *\n   * @param address - The account address from the activity message.\n   * @returns The matching account, or null.\n   */\n  #findAccountForAddress(address: string): InternalAccount | null {\n    const isEvm = address.startsWith('0x');\n\n    return (\n      this.#getWalletAccounts().find((candidate) =>\n        isEvm\n          ? candidate.address.toLowerCase() === address.toLowerCase()\n          : candidate.address === address,\n      ) ?? null\n    );\n  }\n\n  /**\n   * Resolve the accounts of the selected account group from the wallet,\n   * mirroring `AssetsController.#getSelectedAccounts`.\n   *\n   * @returns The accounts of the selected account group.\n   */\n  #getWalletAccounts(): InternalAccount[] {\n    const accounts = this.#messenger.call(\n      'AccountTreeController:getAccountsFromSelectedAccountGroup',\n    );\n    if (accounts.length > 0) {\n      return accounts;\n    }\n\n    const selectedAccount = this.#messenger.call(\n      'AccountsController:getSelectedAccount',\n    );\n    return selectedAccount ? [selectedAccount] : [];\n  }\n\n  // ============================================================================\n  // STATUS CHANGES\n  // ============================================================================\n\n  /**\n   * Handle a `statusChanged` notification. Chains reported \"up\" are claimed as\n   * active (WebSocket provides real-time data); chains reported \"down\" are\n   * released so polling data sources take over. The change is applied directly;\n   * coalescing and jitter are handled by `AssetsController`.\n   *\n   * @param options - The status change notification.\n   * @param options.chainIds - The CAIP-2 chain IDs whose status changed.\n   * @param options.status - Whether the chains are `up` or `down`.\n   * @param options.timestamp - Optional timestamp of the status change.\n   */\n  readonly #onAccountActivityStatusChanged = ({\n    chainIds,\n    status,\n  }: {\n    chainIds: string[];\n    status: 'up' | 'down';\n    timestamp?: number;\n  }): void => {\n    try {\n      // Act on every namespace (eip155, solana, etc.); AssetsController is\n      // multichain. Only skip identifiers that are not valid CAIP-2 chain IDs.\n      const validChains = chainIds.filter((chainId) =>\n        isCaipChainId(chainId),\n      ) as ChainId[];\n\n      if (validChains.length === 0) {\n        return;\n      }\n\n      const next = new Set(this.state.activeChains);\n      if (status === 'up') {\n        for (const chainId of validChains) {\n          next.add(chainId);\n        }\n      } else {\n        for (const chainId of validChains) {\n          next.delete(chainId);\n        }\n      }\n\n      const previous = [...this.state.activeChains];\n      this.updateActiveChains(Array.from(next), (updatedChains) =>\n        this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n      );\n    } catch (error) {\n      log('Error handling status change', error);\n    }\n  };\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    try {\n      this.#messenger.unsubscribe(\n        'AccountActivityService:balanceUpdated',\n        this.#onBalanceUpdatedBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from balanceUpdated', { error });\n    }\n\n    try {\n      this.#messenger.unsubscribe(\n        'AccountActivityService:statusChanged',\n        this.#onAccountActivityStatusChanged,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from statusChanged', { error });\n    }\n\n    super.destroy();\n  }\n}\n\n// ============================================================================\n// FACTORY FUNCTION\n// ============================================================================\n\n/**\n * Creates an AccountActivityDataSource instance.\n *\n * @param options - Configuration options for the data source.\n * @returns A new AccountActivityDataSource instance.\n */\nexport function createAccountActivityDataSource(\n  options: AccountActivityDataSourceOptions,\n): AccountActivityDataSource {\n  return new AccountActivityDataSource(options);\n}\n"]}

package/dist/data-sources/{AccountActivityDataSource.d.ts → AccountActivityDataSource.d.cts}

@@ -1,8 +1,8 @@
-import type { AccountActivityServiceBalanceUpdatedEvent, AccountActivityServiceStatusChangedEvent } from '@metamask/core-backend';
-import type { AssetsControllerMessenger } from '../AssetsController.js';
-import type { ChainId, Caip19AssetId, DataRequest, DataResponse } from '../types.js';
-import { AbstractDataSource } from './AbstractDataSource.js';
-import type { DataSourceState } from './AbstractDataSource.js';
+import type { AccountActivityServiceBalanceUpdatedEvent, AccountActivityServiceStatusChangedEvent } from "@metamask/core-backend";
+import type { AssetsControllerMessenger } from "../AssetsController.cjs";
+import type { ChainId, Caip19AssetId, DataRequest, DataResponse } from "../types.cjs";
+import { AbstractDataSource } from "./AbstractDataSource.cjs";
+import type { DataSourceState } from "./AbstractDataSource.cjs";
 declare const CONTROLLER_NAME = "AccountActivityDataSource";
 /** Allowed events that AccountActivityDataSource subscribes to. */
 export type AccountActivityDataSourceAllowedEvents = AccountActivityServiceBalanceUpdatedEvent | AccountActivityServiceStatusChangedEvent;
@@ -77,4 +77,4 @@ export declare class AccountActivityDataSource extends AbstractDataSource<typeof
  */
 export declare function createAccountActivityDataSource(options: AccountActivityDataSourceOptions): AccountActivityDataSource;
 export {};
-//# sourceMappingURL=AccountActivityDataSource.d.ts.map
+//# sourceMappingURL=AccountActivityDataSource.d.cts.map