@metamask-previews/assets-controller 14.0.2-preview-3866c0ff1 → 14.0.2-preview-e3275932a

package/dist/data-sources/AccountActivityDataSource.js

@@ -1,272 +0,0 @@
-import { isCaipChainId } from '@metamask/utils';
-import { BigNumber as BigNumberJS } from 'bignumber.js';
-import { projectLogger, createModuleLogger } from '../logger.js';
-import { safeNormalizeAssetId } from '../utils/index.js';
-import { AbstractDataSource } from './AbstractDataSource.js';
-// ============================================================================
-// CONSTANTS
-// ============================================================================
-const CONTROLLER_NAME = 'AccountActivityDataSource';
-const log = createModuleLogger(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 = safeNormalizeAssetId(asset.type);
-        if (asset.decimals === undefined) {
-            continue;
-        }
-        const rawBalanceStr = postBalance.amount.startsWith('0x')
-            ? BigInt(postBalance.amount).toString()
-            : postBalance.amount;
-        const humanReadableAmount = new BigNumberJS(rawBalanceStr)
-            .dividedBy(new BigNumberJS(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.
- */
-export class AccountActivityDataSource extends AbstractDataSource {
-    #messenger;
-    #onActiveChainsUpdated;
-    #getAssetType;
-    #onAssetsUpdate;
-    #onBalanceUpdatedBound;
-    constructor(options) {
-        super(CONTROLLER_NAME, {
-            ...defaultState,
-            ...options.state,
-        });
-        this.#messenger = options.messenger;
-        this.#onActiveChainsUpdated = options.onActiveChainsUpdated;
-        this.#getAssetType = options.getAssetType;
-        this.#onAssetsUpdate = options.onAssetsUpdate;
-        this.#onBalanceUpdatedBound = this.#onBalanceUpdated.bind(this);
-        this.#subscribeToEvents();
-    }
-    // ============================================================================
-    // EVENT SUBSCRIPTIONS
-    // ============================================================================
-    #subscribeToEvents() {
-        this.#messenger.subscribe('AccountActivityService:balanceUpdated', this.#onBalanceUpdatedBound);
-        this.#messenger.subscribe('AccountActivityService:statusChanged', this.#onAccountActivityStatusChanged);
-    }
-    // ============================================================================
-    // 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.
-    }
-    // ============================================================================
-    // BALANCE UPDATES
-    // ============================================================================
-    #onBalanceUpdated({ address, chain, updates, }) {
-        try {
-            if (!address || !chain || !updates || updates.length === 0) {
-                return;
-            }
-            const account = this.#findAccountForAddress(address);
-            if (!account) {
-                return;
-            }
-            const chainId = chain;
-            const response = processAccountActivityBalanceUpdates(updates, account.id, (assetId) => this.#getAssetType(assetId));
-            if (!response.assetsBalance) {
-                return;
-            }
-            const request = {
-                accountsWithSupportedChains: [{ account, supportedChains: [chainId] }],
-                chainIds: [chainId],
-                dataTypes: ['balance', 'metadata'],
-            };
-            Promise.resolve(this.#onAssetsUpdate(response, request)).catch((error) => {
-                log('Failed to report balance update', { error });
-            });
-        }
-        catch (error) {
-            log('Error handling balance update', error);
-        }
-    }
-    /**
-     * Find the wallet account matching the given activity address. EVM addresses
-     * are matched case-insensitively; other namespaces are matched exactly.
-     *
-     * The candidate set is the accounts of the selected account group, resolved
-     * from the wallet at event time (rather than from a stored subscription), so
-     * AADS does not need to be re-subscribed whenever the account set changes.
-     *
-     * @param address - The account address from the activity message.
-     * @returns The matching account, or null.
-     */
-    #findAccountForAddress(address) {
-        const isEvm = address.startsWith('0x');
-        return (this.#getWalletAccounts().find((candidate) => isEvm
-            ? candidate.address.toLowerCase() === address.toLowerCase()
-            : candidate.address === address) ?? null);
-    }
-    /**
-     * Resolve the accounts of the selected account group from the wallet,
-     * mirroring `AssetsController.#getSelectedAccounts`.
-     *
-     * @returns The accounts of the selected account group.
-     */
-    #getWalletAccounts() {
-        const accounts = this.#messenger.call('AccountTreeController:getAccountsFromSelectedAccountGroup');
-        if (accounts.length > 0) {
-            return accounts;
-        }
-        const selectedAccount = this.#messenger.call('AccountsController:getSelectedAccount');
-        return selectedAccount ? [selectedAccount] : [];
-    }
-    // ============================================================================
-    // 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.
-     */
-    #onAccountActivityStatusChanged = ({ 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) => 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) => this.#onActiveChainsUpdated(this.getName(), updatedChains, previous));
-        }
-        catch (error) {
-            log('Error handling status change', error);
-        }
-    };
-    // ============================================================================
-    // CLEANUP
-    // ============================================================================
-    destroy() {
-        try {
-            this.#messenger.unsubscribe('AccountActivityService:balanceUpdated', this.#onBalanceUpdatedBound);
-        }
-        catch (error) {
-            log('Failed to unsubscribe from balanceUpdated', { error });
-        }
-        try {
-            this.#messenger.unsubscribe('AccountActivityService:statusChanged', this.#onAccountActivityStatusChanged);
-        }
-        catch (error) {
-            log('Failed to unsubscribe from statusChanged', { error });
-        }
-        super.destroy();
-    }
-}
-// ============================================================================
-// FACTORY FUNCTION
-// ============================================================================
-/**
- * Creates an AccountActivityDataSource instance.
- *
- * @param options - Configuration options for the data source.
- * @returns A new AccountActivityDataSource instance.
- */
-export function createAccountActivityDataSource(options) {
-    return new AccountActivityDataSource(options);
-}
-//# sourceMappingURL=AccountActivityDataSource.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"AccountActivityDataSource.js","sourceRoot":"","sources":["../../src/data-sources/AccountActivityDataSource.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,SAAS,IAAI,WAAW,EAAE,MAAM,cAAc,CAAC;AAGxD,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AASjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAG7D,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,eAAe,GAAG,2BAA2B,CAAC;AAEpD,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,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,oBAAoB,CAAC,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,WAAW,CAAC,aAAa,CAAC;aACvD,SAAS,CAAC,IAAI,WAAW,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,MAAM,OAAO,yBAA0B,SAAQ,kBAG9C;IACU,UAAU,CAA4B;IAEtC,sBAAsB,CAIrB;IAED,aAAa,CAEU;IAEvB,eAAe,CAGE;IAEjB,sBAAsB,CAIpB;IAEX,YAAY,OAAyC;QACnD,KAAK,CAAC,eAAe,EAAE;YACrB,GAAG,YAAY;YACf,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;QACpC,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;QAC5D,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,YAAY,CAAC;QAC1C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,cAAc,CAAC;QAC9C,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED,+EAA+E;IAC/E,sBAAsB;IACtB,+EAA+E;IAE/E,kBAAkB;QAChB,IAAI,CAAC,UAAU,CAAC,SAAS,CACvB,uCAAuC,EACvC,IAAI,CAAC,sBAAsB,CAC5B,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,SAAS,CACvB,sCAAsC,EACtC,IAAI,CAAC,+BAA+B,CACrC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,0BAA0B;IAC1B,+EAA+E;IAE/E;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS;QACb,wCAAwC;IAC1C,CAAC;IAED,+EAA+E;IAC/E,kBAAkB;IAClB,+EAA+E;IAE/E,iBAAiB,CAAC,EAChB,OAAO,EACP,KAAK,EACL,OAAO,GAKR;QACC,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3D,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YACrD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,KAAgB,CAAC;YAEjC,MAAM,QAAQ,GAAG,oCAAoC,CACnD,OAAO,EACP,OAAO,CAAC,EAAE,EACV,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CACzC,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC5B,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAgB;gBAC3B,2BAA2B,EAAE,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtE,QAAQ,EAAE,CAAC,OAAO,CAAC;gBACnB,SAAS,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;aACnC,CAAC;YAEF,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,KAAK,CAC5D,CAAC,KAAK,EAAE,EAAE;gBACR,GAAG,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACpD,CAAC,CACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACH,sBAAsB,CAAC,OAAe;QACpC,MAAM,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAEvC,OAAO,CACL,IAAI,CAAC,kBAAkB,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAC3C,KAAK;YACH,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE;YAC3D,CAAC,CAAC,SAAS,CAAC,OAAO,KAAK,OAAO,CAClC,IAAI,IAAI,CACV,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,kBAAkB;QAChB,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CACnC,2DAA2D,CAC5D,CAAC;QACF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAC1C,uCAAuC,CACxC,CAAC;QACF,OAAO,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAClD,CAAC;IAED,+EAA+E;IAC/E,iBAAiB;IACjB,+EAA+E;IAE/E;;;;;;;;;;OAUG;IACM,+BAA+B,GAAG,CAAC,EAC1C,QAAQ,EACR,MAAM,GAKP,EAAQ,EAAE;QACT,IAAI,CAAC;YACH,qEAAqE;YACrE,yEAAyE;YACzE,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;YAEzE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,OAAO;YACT,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBACpB,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;oBAClC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACpB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;oBAClC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,EAAE,EAAE,CAC1D,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC,CAAC;IAEF,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,WAAW,CACzB,uCAAuC,EACvC,IAAI,CAAC,sBAAsB,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,IAAI,CAAC,UAAU,CAAC,WAAW,CACzB,sCAAsC,EACtC,IAAI,CAAC,+BAA+B,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;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,+BAA+B,CAC7C,OAAyC;IAEzC,OAAO,IAAI,yBAAyB,CAAC,OAAO,CAAC,CAAC;AAChD,CAAC","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 { BigNumber as 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) => isCaipChainId(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/AccountsApiDataSource.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"AccountsApiDataSource.d.ts","sourceRoot":"","sources":["../../src/data-sources/AccountsApiDataSource.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,KAAK,EACV,yCAAyC,EACzC,2CAA2C,EAC5C,MAAM,0CAA0C,CAAC;AAOlD,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AAExE,OAAO,KAAK,EACV,OAAO,EAGP,WAAW,EACX,YAAY,EACZ,UAAU,EACV,6BAA6B,EAC9B,MAAM,aAAa,CAAC;AAMrB,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAO7D,QAAA,MAAM,eAAe,0BAA0B,CAAC;AAehD,MAAM,MAAM,mCAAmC,GAC7C,yCAAyC,CAAC;AAI5C,MAAM,MAAM,kCAAkC,GAC5C,2CAA2C,CAAC;AAM9C,MAAM,MAAM,0BAA0B,GAAG,eAAe,CAAC;AAUzD,wDAAwD;AACxD,MAAM,MAAM,2BAA2B,GAAG;IACxC,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;;;OAIG;IACH,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;IACtC;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,2BAA2B,GAAG;IACvE;;;OAGG;IACH,SAAS,EAAE,yBAAyB,CAAC;IACrC,mDAAmD;IACnD,cAAc,EAAE,iBAAiB,CAAC;IAClC,qGAAqG;IACrG,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;IACV,KAAK,CAAC,EAAE,OAAO,CAAC,0BAA0B,CAAC,CAAC;CAC7C,CAAC;AAgCF;;;;;;;GAOG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,YAAY,EACtB,WAAW,EAAE,6BAA6B,GACzC,YAAY,CAqCd;AAMD;;;;;GAKG;AACH,qBAAa,qBAAsB,SAAQ,kBAAkB,CAC3D,OAAO,eAAe,EACtB,0BAA0B,CAC3B;;IA0BC,YAAY,OAAO,EAAE,4BAA4B,EAgChD;IA8FD;;;;;;OAMG;IACH,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAEnC;IA0BK,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC,CAoFvD;IAmOD;;;;;;;;;OASG;IACH,IAAI,gBAAgB,IAAI,UAAU,CAuEjC;IAMK,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CAoGvE;IAMD,OAAO,IAAI,IAAI,CAMd;CACF"}