@metamask/assets-controller 0.1.0

package/dist/data-sources/SnapDataSource.mjs

@@ -0,0 +1,456 @@
+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 _SnapDataSource_instances, _SnapDataSource_messenger, _SnapDataSource_handleSnapBalancesUpdatedBound, _SnapDataSource_handlePermissionStateChangeBound, _SnapDataSource_keyringClientCache, _SnapDataSource_subscribeToEvents, _SnapDataSource_handleSnapBalancesUpdated, _SnapDataSource_isChainSupportedBySnap, _SnapDataSource_registerActionHandlers, _SnapDataSource_getRunnableSnaps, _SnapDataSource_getSnapPermissions, _SnapDataSource_discoverKeyringSnaps, _SnapDataSource_getKeyringClient;
+import { KeyringClient } from "@metamask/keyring-snap-client";
+import { HandlerType, SnapCaveatType } from "@metamask/snaps-utils";
+import { AbstractDataSource } from "./AbstractDataSource.mjs";
+import { projectLogger, createModuleLogger } from "../logger.mjs";
+const log = createModuleLogger(projectLogger, 'SnapDataSource');
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+export const SNAP_DATA_SOURCE_NAME = 'SnapDataSource';
+/** The permission name for snap keyring endowment */
+export const KEYRING_PERMISSION = 'endowment:keyring';
+/** The permission name for snap assets endowment (contains chainIds) */
+export const ASSETS_PERMISSION = 'endowment:assets';
+// ============================================================================
+// PERMISSION UTILITIES
+// ============================================================================
+/**
+ * Getter function to get the chainIds caveat from a permission.
+ *
+ * This does basic validation of the caveat, but does not validate the type or
+ * value of the namespaces object itself, as this is handled by the
+ * `PermissionsController` when the permission is requested.
+ *
+ * @param permission - The permission to get the `chainIds` caveat from.
+ * @returns An array of `chainIds` that the snap supports, or null if none.
+ */
+export function getChainIdsCaveat(permission) {
+    if (!permission?.caveats) {
+        return null;
+    }
+    const caveat = permission.caveats.find((permCaveat) => permCaveat.type === SnapCaveatType.ChainIds);
+    return caveat ? caveat.value : null;
+}
+/**
+ * Extract chain ID from a CAIP-19 asset ID.
+ * e.g., "solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp/slip44:501" -> "solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp"
+ *
+ * @param assetId - The CAIP-19 asset ID to extract chain from.
+ * @returns The CAIP-2 chain ID portion of the asset ID.
+ */
+export function extractChainFromAssetId(assetId) {
+    const parts = assetId.split('/');
+    return parts[0];
+}
+const defaultSnapState = {
+    activeChains: [],
+    chainToSnap: {},
+};
+// ============================================================================
+// SNAP DATA SOURCE
+// ============================================================================
+/**
+ * Unified Snap data source that routes requests to the appropriate wallet snap
+ * based on the chain ID prefix.
+ *
+ * @example
+ * ```typescript
+ * const snapDataSource = new SnapDataSource({
+ *   messenger,
+ * });
+ *
+ * // Fetch will automatically route to the correct snap
+ * await snapDataSource.fetch({
+ *   chainIds: ['solana:mainnet', 'bip122:000000000019d6689c085ae165831e93'],
+ *   accountIds: ['account1'],
+ * });
+ * ```
+ */
+export class SnapDataSource extends AbstractDataSource {
+    constructor(options) {
+        super(SNAP_DATA_SOURCE_NAME, {
+            ...defaultSnapState,
+            ...options.state,
+        });
+        _SnapDataSource_instances.add(this);
+        _SnapDataSource_messenger.set(this, void 0);
+        /** Bound handler for snap keyring balance updates, stored for cleanup */
+        _SnapDataSource_handleSnapBalancesUpdatedBound.set(this, void 0);
+        _SnapDataSource_handlePermissionStateChangeBound.set(this, void 0);
+        /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */
+        _SnapDataSource_keyringClientCache.set(this, new Map());
+        __classPrivateFieldSet(this, _SnapDataSource_messenger, options.messenger, "f");
+        // Bind handlers for cleanup in destroy()
+        __classPrivateFieldSet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_handleSnapBalancesUpdated).bind(this), "f");
+        __classPrivateFieldSet(this, _SnapDataSource_handlePermissionStateChangeBound, __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_discoverKeyringSnaps).bind(this), "f");
+        __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_registerActionHandlers).call(this);
+        __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_subscribeToEvents).call(this);
+        // Discover keyring-capable snaps and populate activeChains dynamically
+        __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_discoverKeyringSnaps).call(this);
+    }
+    // ============================================================================
+    // FETCH
+    // ============================================================================
+    async fetch(request) {
+        var _a;
+        // Guard against undefined request
+        // Note: chainIds filtering is done by middleware/subscribe before calling fetch
+        if (!request?.accounts || !request?.chainIds?.length) {
+            return {};
+        }
+        const results = {
+            assetsBalance: {},
+            assetsMetadata: {},
+        };
+        // Fetch balances for each account using its snap ID from metadata
+        for (const account of request.accounts) {
+            // Skip accounts without snap metadata (non-snap accounts)
+            const snapId = account.metadata.snap?.id;
+            if (!snapId) {
+                continue;
+            }
+            // Skip accounts whose snap doesn't support any of the requested chains
+            const snapSupportsRequestedChains = request.chainIds.some((chainId) => this.state.chainToSnap[chainId] === snapId);
+            if (!snapSupportsRequestedChains) {
+                continue;
+            }
+            const accountId = account.id;
+            try {
+                const client = __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_getKeyringClient).call(this, snapId);
+                // Step 1: Get the list of assets for this account
+                const accountAssets = await client.listAccountAssets(accountId);
+                // If no assets, skip to next account
+                if (!accountAssets || accountAssets.length === 0) {
+                    continue;
+                }
+                // Step 2: Get balances for those specific assets
+                const balances = await client.getAccountBalances(accountId, accountAssets);
+                // Transform keyring response to DataResponse format
+                if (balances && typeof balances === 'object' && results.assetsBalance) {
+                    for (const [assetId, balance] of Object.entries(balances)) {
+                        (_a = results.assetsBalance)[accountId] ?? (_a[accountId] = {});
+                        const accountBalances = results.assetsBalance[accountId];
+                        if (accountBalances) {
+                            accountBalances[assetId] = {
+                                amount: balance.amount,
+                            };
+                        }
+                    }
+                }
+            }
+            catch {
+                // Expected when account doesn't belong to this snap
+            }
+        }
+        return results;
+    }
+    // ============================================================================
+    // MIDDLEWARE
+    // ============================================================================
+    /**
+     * Get the middleware for fetching balances via Snaps.
+     * This middleware:
+     * - Supports multiple accounts in a single request
+     * - Filters request to only chains this data source supports
+     * - Fetches balances for those chains for all accounts
+     * - Merges response into context
+     * - Removes handled chains from request for next middleware
+     *
+     * @returns The middleware function for the assets pipeline.
+     */
+    get assetsMiddleware() {
+        return async (context, next) => {
+            var _a;
+            const { request } = context;
+            // Filter to chains this data source supports
+            const supportedChains = request.chainIds.filter((chainId) => this.state.activeChains.includes(chainId));
+            // If no supported chains, skip and pass to next middleware
+            if (supportedChains.length === 0) {
+                return next(context);
+            }
+            let successfullyHandledChains = [];
+            try {
+                // Fetch for supported chains
+                const response = await this.fetch({
+                    ...request,
+                    chainIds: supportedChains,
+                });
+                // Merge response into context
+                if (response.assetsBalance) {
+                    (_a = context.response).assetsBalance ?? (_a.assetsBalance = {});
+                    for (const [accountId, accountBalances] of Object.entries(response.assetsBalance)) {
+                        context.response.assetsBalance[accountId] = {
+                            ...context.response.assetsBalance[accountId],
+                            ...accountBalances,
+                        };
+                    }
+                }
+                if (response.assetsMetadata) {
+                    context.response.assetsMetadata = {
+                        ...context.response.assetsMetadata,
+                        ...response.assetsMetadata,
+                    };
+                }
+                if (response.assetsPrice) {
+                    context.response.assetsPrice = {
+                        ...context.response.assetsPrice,
+                        ...response.assetsPrice,
+                    };
+                }
+                // Determine successfully handled chains (exclude errors)
+                const failedChains = new Set(Object.keys(response.errors ?? {}));
+                successfullyHandledChains = supportedChains.filter((chainId) => !failedChains.has(chainId));
+            }
+            catch (error) {
+                log('Middleware fetch failed', { error });
+                successfullyHandledChains = [];
+            }
+            // Prepare context for next middleware
+            let nextContext = context;
+            if (successfullyHandledChains.length > 0) {
+                const remainingChains = request.chainIds.filter((chainId) => !successfullyHandledChains.includes(chainId));
+                nextContext = {
+                    ...context,
+                    request: {
+                        ...request,
+                        chainIds: remainingChains,
+                    },
+                };
+            }
+            // Call next middleware
+            return next(nextContext);
+        };
+    }
+    // ============================================================================
+    // SUBSCRIBE - Routes to appropriate snap(s)
+    // ============================================================================
+    async subscribe(subscriptionRequest) {
+        const { request, subscriptionId, isUpdate } = subscriptionRequest;
+        // Guard against undefined request or chainIds
+        if (!request?.chainIds) {
+            return;
+        }
+        // Filter to chains we have a snap for
+        const supportedChains = request.chainIds.filter((chainId) => __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_isChainSupportedBySnap).call(this, chainId));
+        if (supportedChains.length === 0) {
+            return;
+        }
+        if (isUpdate) {
+            const existing = this.activeSubscriptions.get(subscriptionId);
+            if (existing) {
+                existing.chains = supportedChains;
+                // Do a fetch to get latest data on subscription update
+                this.fetch({
+                    ...request,
+                    chainIds: supportedChains,
+                })
+                    .then(async (fetchResponse) => {
+                    if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {
+                        await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:assetsUpdate', fetchResponse, SNAP_DATA_SOURCE_NAME);
+                    }
+                    return fetchResponse;
+                })
+                    .catch((error) => {
+                    log('Subscription update fetch failed', { subscriptionId, error });
+                });
+                return;
+            }
+        }
+        await this.unsubscribe(subscriptionId);
+        // Snaps provide real-time updates via AccountsController:accountBalancesUpdated
+        // We only need to track the subscription and do an initial fetch
+        // No polling needed - updates come through #handleSnapBalancesUpdated
+        this.activeSubscriptions.set(subscriptionId, {
+            cleanup: () => {
+                // No timer to clear - we use event-based updates
+            },
+            chains: supportedChains,
+        });
+        // Initial fetch to get current balances
+        try {
+            const fetchResponse = await this.fetch({
+                ...request,
+                chainIds: supportedChains,
+            });
+            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {
+                await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:assetsUpdate', fetchResponse, SNAP_DATA_SOURCE_NAME);
+            }
+        }
+        catch (error) {
+            log('Initial fetch failed', { subscriptionId, error });
+        }
+    }
+    // ============================================================================
+    // CLEANUP
+    // ============================================================================
+    destroy() {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
+        // Unsubscribe from snap keyring events
+        try {
+            messenger.unsubscribe('AccountsController:accountBalancesUpdated', __classPrivateFieldGet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, "f"));
+        }
+        catch (error) {
+            log('Failed to unsubscribe from snap keyring events', { error });
+        }
+        // Unsubscribe from permission changes
+        try {
+            messenger.unsubscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
+        }
+        catch (error) {
+            log('Failed to unsubscribe from permission changes', { error });
+        }
+        // Clean up active subscriptions
+        for (const [subscriptionId] of this.activeSubscriptions) {
+            this.unsubscribe(subscriptionId).catch(() => {
+                // Ignore cleanup errors
+            });
+        }
+        // Clear keyring client cache
+        __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").clear();
+    }
+}
+_SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpdatedBound = new WeakMap(), _SnapDataSource_handlePermissionStateChangeBound = new WeakMap(), _SnapDataSource_keyringClientCache = new WeakMap(), _SnapDataSource_instances = new WeakSet(), _SnapDataSource_subscribeToEvents = function _SnapDataSource_subscribeToEvents() {
+    // Subscribe to snap keyring events for real-time balance updates.
+    // The snaps emit AccountBalancesUpdated events when balances change,
+    // which are re-published by AccountsController.
+    __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").subscribe('AccountsController:accountBalancesUpdated', __classPrivateFieldGet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, "f"));
+    // Subscribe to permission changes to detect new keyring snaps at runtime.
+    // Re-runs snap discovery when permissions change.
+    __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").subscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
+}, _SnapDataSource_handleSnapBalancesUpdated = function _SnapDataSource_handleSnapBalancesUpdated(payload) {
+    // Transform the snap keyring payload to DataResponse format
+    let assetsBalance;
+    for (const [accountId, assets] of Object.entries(payload.balances)) {
+        let accountAssets;
+        for (const [assetId, balance] of Object.entries(assets)) {
+            const chainId = extractChainFromAssetId(assetId);
+            if (__classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_isChainSupportedBySnap).call(this, chainId)) {
+                accountAssets ?? (accountAssets = {});
+                accountAssets[assetId] = {
+                    amount: balance.amount,
+                };
+            }
+        }
+        if (accountAssets) {
+            assetsBalance ?? (assetsBalance = {});
+            assetsBalance[accountId] = accountAssets;
+        }
+    }
+    // Only report if we have snap-related updates
+    if (assetsBalance) {
+        const response = { assetsBalance };
+        __classPrivateFieldGet(this, _SnapDataSource_messenger, "f")
+            .call('AssetsController:assetsUpdate', response, SNAP_DATA_SOURCE_NAME)
+            .catch(console.error);
+    }
+}, _SnapDataSource_isChainSupportedBySnap = function _SnapDataSource_isChainSupportedBySnap(chainId) {
+    return this.state.activeChains.includes(chainId);
+}, _SnapDataSource_registerActionHandlers = function _SnapDataSource_registerActionHandlers() {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
+    messenger.registerActionHandler('SnapDataSource:getAssetsMiddleware', () => this.assetsMiddleware);
+    messenger.registerActionHandler('SnapDataSource:getActiveChains', async () => this.getActiveChains());
+    messenger.registerActionHandler('SnapDataSource:fetch', async (request) => this.fetch(request));
+    messenger.registerActionHandler('SnapDataSource:subscribe', async (request) => this.subscribe(request));
+    messenger.registerActionHandler('SnapDataSource:unsubscribe', async (subscriptionId) => this.unsubscribe(subscriptionId));
+}, _SnapDataSource_getRunnableSnaps = function _SnapDataSource_getRunnableSnaps() {
+    try {
+        return __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('SnapController:getRunnableSnaps');
+    }
+    catch (error) {
+        log('Failed to get runnable snaps', error);
+        return [];
+    }
+}, _SnapDataSource_getSnapPermissions = function _SnapDataSource_getSnapPermissions(snapId) {
+    try {
+        return __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('PermissionController:getPermissions', snapId);
+    }
+    catch (error) {
+        log('Failed to get permissions for snap', { snapId, error });
+        return undefined;
+    }
+}, _SnapDataSource_discoverKeyringSnaps = function _SnapDataSource_discoverKeyringSnaps() {
+    try {
+        const runnableSnaps = __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_getRunnableSnaps).call(this);
+        const chainToSnap = {};
+        const supportedChains = [];
+        for (const snap of runnableSnaps) {
+            const permissions = __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_getSnapPermissions).call(this, snap.id);
+            // Must have endowment:keyring permission to be a keyring snap
+            if (!permissions?.[KEYRING_PERMISSION]) {
+                continue;
+            }
+            // Get chainIds caveat from the assets permission (not keyring permission)
+            // The chainIds are stored in endowment:assets
+            const assetsPermission = permissions[ASSETS_PERMISSION];
+            const chainIds = getChainIdsCaveat(assetsPermission);
+            // Map each chain to this snap (first snap wins if multiple support same chain)
+            if (chainIds) {
+                for (const chainId of chainIds) {
+                    if (!(chainId in chainToSnap)) {
+                        chainToSnap[chainId] = snap.id;
+                        supportedChains.push(chainId);
+                    }
+                }
+            }
+        }
+        // Update chainToSnap mapping
+        this.state.chainToSnap = chainToSnap;
+        // Notify if chains changed
+        try {
+            this.updateActiveChains(supportedChains, (updatedChains) => {
+                __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', SNAP_DATA_SOURCE_NAME, updatedChains);
+            });
+        }
+        catch {
+            // AssetsController not ready yet - expected during initialization
+        }
+    }
+    catch (error) {
+        log('Keyring snap discovery failed', { error });
+        this.state.chainToSnap = {};
+        try {
+            this.updateActiveChains([], (updatedChains) => {
+                __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', SNAP_DATA_SOURCE_NAME, updatedChains);
+            });
+        }
+        catch {
+            // AssetsController not ready yet - expected during initialization
+        }
+    }
+}, _SnapDataSource_getKeyringClient = function _SnapDataSource_getKeyringClient(snapId) {
+    const cachedClient = __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").get(snapId);
+    if (cachedClient) {
+        return cachedClient;
+    }
+    const client = new KeyringClient({
+        send: async (request) => (await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('SnapController:handleRequest', {
+            snapId: snapId,
+            origin: 'metamask',
+            handler: HandlerType.OnKeyringRequest,
+            request,
+        })),
+    });
+    __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").set(snapId, client);
+    return client;
+};
+// ============================================================================
+// FACTORY
+// ============================================================================
+export function createSnapDataSource(options) {
+    return new SnapDataSource(options);
+}
+//# sourceMappingURL=SnapDataSource.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"SnapDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,OAAO,EAAE,aAAa,EAAE,sCAAsC;AAc9D,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,8BAA8B;AAGpE,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAK1D,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAqC9D,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;AAEhE,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,MAAM,CAAC,MAAM,qBAAqB,GAAG,gBAAgB,CAAC;AAEtD,qDAAqD;AACrD,MAAM,CAAC,MAAM,kBAAkB,GAAG,mBAAmB,CAAC;AAEtD,wEAAwE;AACxE,MAAM,CAAC,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;AAEpD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;GASG;AACH,MAAM,UAAU,iBAAiB,CAC/B,UAAiC;IAEjC,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,CACpC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ,CACpB,CAAC;IAE1C,OAAO,MAAM,CAAC,CAAC,CAAE,MAAM,CAAC,KAAmB,CAAC,CAAC,CAAC,IAAI,CAAC;AACrD,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CAAC,OAAe;IACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,KAAK,CAAC,CAAC,CAAY,CAAC;AAC7B,CAAC;AAkBD,MAAM,gBAAgB,GAAwB;IAC5C,YAAY,EAAE,EAAE;IAChB,WAAW,EAAE,EAAE;CAChB,CAAC;AAkGF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,OAAO,cAAe,SAAQ,kBAGnC;IAaC,YAAY,OAA8B;QACxC,KAAK,CAAC,qBAAqB,EAAE;YAC3B,GAAG,gBAAgB;YACnB,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QAhBI,4CAAoC;QAE7C,yEAAyE;QAChE,iEAEC;QAED,mEAA8C;QAEvD,6EAA6E;QACpE,6CAAkD,IAAI,GAAG,EAAE,EAAC;QAQnE,uBAAA,IAAI,6BAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QAEpC,yCAAyC;QACzC,uBAAA,IAAI,kDAAmC,uBAAA,IAAI,4EAA2B,CAAC,IAAI,CACzE,IAAI,CACoD,MAAA,CAAC;QAC3D,uBAAA,IAAI,oDACF,uBAAA,IAAI,uEAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAA,CAAC;QAExC,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,CAA0B,CAAC;QAC/B,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1B,uEAAuE;QACvE,uBAAA,IAAI,uEAAsB,MAA1B,IAAI,CAAwB,CAAC;IAC/B,CAAC;IA0ND,+EAA+E;IAC/E,QAAQ;IACR,+EAA+E;IAE/E,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,kCAAkC;QAClC,gFAAgF;QAChF,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;YACrD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,kEAAkE;QAClE,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,0DAA0D;YAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YAED,uEAAuE;YACvE,MAAM,2BAA2B,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CACvD,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,MAAM,CACxD,CAAC;YACF,IAAI,CAAC,2BAA2B,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,MAAM,CAAC,CAAC;gBAE9C,kDAAkD;gBAClD,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAEhE,qCAAqC;gBACrC,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACjD,SAAS;gBACX,CAAC;gBAED,iDAAiD;gBACjD,MAAM,QAAQ,GACZ,MAAM,MAAM,CAAC,kBAAkB,CAC7B,SAAS,EACT,aAAgC,CACjC,CAAC;gBAEJ,oDAAoD;gBACpD,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;oBACtE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC1D,MAAA,OAAO,CAAC,aAAa,EAAC,SAAS,SAAT,SAAS,IAAM,EAAE,EAAC;wBACxC,MAAM,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;wBACzD,IAAI,eAAe,EAAE,CAAC;4BACnB,eAA2C,CAAC,OAAO,CAAC,GAAG;gCACtD,MAAM,EAAE,OAAO,CAAC,MAAM;6BACvB,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,oDAAoD;YACtD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,6CAA6C;YAC7C,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;YAEF,2DAA2D;YAC3D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,yBAAyB,GAAc,EAAE,CAAC;YAE9C,IAAI,CAAC;gBACH,6BAA6B;gBAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;oBAChC,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC,CAAC;gBAEH,8BAA8B;gBAC9B,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAC3B,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;wBACF,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;4BAC1C,GAAG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC;4BAC5C,GAAG,eAAe;yBACnB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;oBAC5B,OAAO,CAAC,QAAQ,CAAC,cAAc,GAAG;wBAChC,GAAG,OAAO,CAAC,QAAQ,CAAC,cAAc;wBAClC,GAAG,QAAQ,CAAC,cAAc;qBAC3B,CAAC;gBACJ,CAAC;gBACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACzB,OAAO,CAAC,QAAQ,CAAC,WAAW,GAAG;wBAC7B,GAAG,OAAO,CAAC,QAAQ,CAAC,WAAW;wBAC/B,GAAG,QAAQ,CAAC,WAAW;qBACxB,CAAC;gBACJ,CAAC;gBAED,yDAAyD;gBACzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC;gBACjE,yBAAyB,GAAG,eAAe,CAAC,MAAM,CAChD,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CACxC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC1C,yBAAyB,GAAG,EAAE,CAAC;YACjC,CAAC;YAED,sCAAsC;YACtC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,yBAAyB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAC7C,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,yBAAyB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1D,CAAC;gBACF,WAAW,GAAG;oBACZ,GAAG,OAAO;oBACV,OAAO,EAAE;wBACP,GAAG,OAAO;wBACV,QAAQ,EAAE,eAAe;qBAC1B;iBACF,CAAC;YACJ,CAAC;YAED,uBAAuB;YACvB,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3B,CAAC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,4CAA4C;IAC5C,+EAA+E;IAE/E,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,8CAA8C;QAC9C,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,sCAAsC;QACtC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,CACtC,CAAC;QAEF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9D,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,MAAM,GAAG,eAAe,CAAC;gBAClC,uDAAuD;gBACvD,IAAI,CAAC,KAAK,CAAC;oBACT,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC;qBACC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,EAAE;oBAC5B,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9D,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACxB,+BAA+B,EAC/B,aAAa,EACb,qBAAqB,CACtB,CAAC;oBACJ,CAAC;oBACD,OAAO,aAAa,CAAC;gBACvB,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;gBACL,OAAO;YACT,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,gFAAgF;QAChF,iEAAiE;QACjE,sEAAsE;QAEtE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,iDAAiD;YACnD,CAAC;YACD,MAAM,EAAE,eAAe;SACxB,CAAC,CAAC;QAEH,wCAAwC;QACxC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;gBACrC,GAAG,OAAO;gBACV,QAAQ,EAAE,eAAe;aAC1B,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9D,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACxB,+BAA+B,EAC/B,aAAa,EACb,qBAAqB,CACtB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,sBAAsB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAiCD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,8DAA8D;QAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;QAEzC,uCAAuC;QACvC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,sCAAsC;QACtC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+CAA+C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,gCAAgC;QAChC,KAAK,MAAM,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;gBAC1C,wBAAwB;YAC1B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,uBAAA,IAAI,0CAAoB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF;;IA3gBG,kEAAkE;IAClE,qEAAqE;IACrE,gDAAgD;IAChD,uBAAA,IAAI,iCAAW,CAAC,SAAS,CACvB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;IAEF,0EAA0E;IAC1E,kDAAkD;IAClD,uBAAA,IAAI,iCAAW,CAAC,SAAS,CACvB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;AACJ,CAAC,iGASC,OAA2C;IAE3C,4DAA4D;IAC5D,IAAI,aAAqE,CAAC;IAE1E,KAAK,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnE,IAAI,aAA8D,CAAC;QAEnE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,CAAC,CAAC;YACjD,IAAI,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,EAAE,CAAC;gBAC1C,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;gBACrB,aAAa,CAAC,OAAwB,CAAC,GAAG;oBACxC,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;YACrB,aAAa,CAAC,SAAS,CAAC,GAAG,aAAa,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,QAAQ,GAAiB,EAAE,aAAa,EAAE,CAAC;QACjD,uBAAA,IAAI,iCAAW;aACZ,IAAI,CAAC,+BAA+B,EAAE,QAAQ,EAAE,qBAAqB,CAAC;aACtE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC,2FAQuB,OAAgB;IACtC,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;IAGC,8DAA8D;IAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;IAEzC,SAAS,CAAC,qBAAqB,CAC7B,oCAAoC,EACpC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAC5B,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,gCAAgC,EAChC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CACnC,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,sBAAsB,EACtB,KAAK,EAAE,OAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CACpD,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,0BAA0B,EAC1B,KAAK,EAAE,OAA4B,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAChE,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,4BAA4B,EAC5B,KAAK,EAAE,cAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CACnE,CAAC;AACJ,CAAC;IAaC,IAAI,CAAC;QACH,OAAO,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IACjE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC3C,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC,mFASC,MAAc;IAEd,IAAI,CAAC;QACH,OAAO,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACzB,qCAAqC,EACrC,MAAM,CACqC,CAAC;IAChD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;IAkBC,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC/C,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,MAAM,eAAe,GAAc,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,uBAAA,IAAI,qEAAoB,MAAxB,IAAI,EAAqB,IAAI,CAAC,EAAE,CAAC,CAAC;YACtD,8DAA8D;YAC9D,IAAI,CAAC,WAAW,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBACvC,SAAS;YACX,CAAC;YAED,0EAA0E;YAC1E,8CAA8C;YAC9C,MAAM,gBAAgB,GAAG,WAAW,CAAC,iBAAiB,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAErD,+EAA+E;YAC/E,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC/B,IAAI,CAAC,CAAC,OAAO,IAAI,WAAW,CAAC,EAAE,CAAC;wBAC9B,WAAW,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;wBAC/B,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,WAAW,CAAC;QAErC,2BAA2B;QAC3B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC,aAAa,EAAE,EAAE;gBACzD,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,qBAAqB,EACrB,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE;gBAC5C,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,qBAAqB,EACrB,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;AACH,CAAC,+EAgQiB,MAAc;IAC9B,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;QAC/B,IAAI,EAAE,KAAK,EAAE,OAAuB,EAAiB,EAAE,CACrD,CAAC,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC1D,MAAM,EAAE,MAAgB;YACxB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,WAAW,CAAC,gBAAgB;YACrC,OAAO;SACR,CAAC,CAAS;KACd,CAAC,CAAC;IAEH,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;AA0CH,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,MAAM,UAAU,oBAAoB,CAClC,OAA8B;IAE9B,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC","sourcesContent":["import type { Balance, CaipAssetType } from '@metamask/keyring-api';\nimport { KeyringClient } from '@metamask/keyring-snap-client';\nimport type { Messenger } from '@metamask/messenger';\nimport type {\n  Caveat,\n  GetPermissions,\n  PermissionConstraint,\n  PermissionControllerStateChange,\n  SubjectPermissions,\n} from '@metamask/permission-controller';\nimport type {\n  GetRunnableSnaps,\n  HandleSnapRequest,\n} from '@metamask/snaps-controllers';\nimport type { Snap, SnapId } from '@metamask/snaps-sdk';\nimport { HandlerType, SnapCaveatType } from '@metamask/snaps-utils';\nimport type { Json, JsonRpcRequest } from '@metamask/utils';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  AssetBalance,\n  ChainId,\n  Caip19AssetId,\n  DataRequest,\n  DataResponse,\n  Middleware,\n} from '../types';\n\n// ============================================================================\n// SNAP KEYRING EVENT TYPES\n// ============================================================================\n\n/**\n * Payload for AccountsController:accountBalancesUpdated event.\n * Re-published from SnapKeyring:accountBalancesUpdated.\n */\nexport type AccountBalancesUpdatedEventPayload = {\n  balances: {\n    [accountId: string]: {\n      [assetId: string]: {\n        amount: string;\n        unit: string;\n      };\n    };\n  };\n};\n\n/**\n * Event from AccountsController when snap balances are updated.\n */\nexport type AccountsControllerAccountBalancesUpdatedEvent = {\n  type: 'AccountsController:accountBalancesUpdated';\n  payload: [AccountBalancesUpdatedEventPayload];\n};\n\nconst log = createModuleLogger(projectLogger, 'SnapDataSource');\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nexport const SNAP_DATA_SOURCE_NAME = 'SnapDataSource';\n\n/** The permission name for snap keyring endowment */\nexport const KEYRING_PERMISSION = 'endowment:keyring';\n\n/** The permission name for snap assets endowment (contains chainIds) */\nexport const ASSETS_PERMISSION = 'endowment:assets';\n\n// ============================================================================\n// PERMISSION UTILITIES\n// ============================================================================\n\n/**\n * Getter function to get the chainIds caveat from a permission.\n *\n * This does basic validation of the caveat, but does not validate the type or\n * value of the namespaces object itself, as this is handled by the\n * `PermissionsController` when the permission is requested.\n *\n * @param permission - The permission to get the `chainIds` caveat from.\n * @returns An array of `chainIds` that the snap supports, or null if none.\n */\nexport function getChainIdsCaveat(\n  permission?: PermissionConstraint,\n): ChainId[] | null {\n  if (!permission?.caveats) {\n    return null;\n  }\n\n  const caveat = permission.caveats.find(\n    (permCaveat) => permCaveat.type === SnapCaveatType.ChainIds,\n  ) as Caveat<string, string[]> | undefined;\n\n  return caveat ? (caveat.value as ChainId[]) : null;\n}\n\n/**\n * Extract chain ID from a CAIP-19 asset ID.\n * e.g., \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp/slip44:501\" -> \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\"\n *\n * @param assetId - The CAIP-19 asset ID to extract chain from.\n * @returns The CAIP-2 chain ID portion of the asset ID.\n */\nexport function extractChainFromAssetId(assetId: string): ChainId {\n  const parts = assetId.split('/');\n  return parts[0] as ChainId;\n}\n\n// ============================================================================\n// STATE\n// ============================================================================\n\n/**\n * State for the SnapDataSource.\n * Uses dynamic snap discovery - chains are populated from PermissionController.\n */\nexport type SnapDataSourceState = {\n  /**\n   * Mapping of chain IDs to snap IDs that support them.\n   * Used to filter which accounts to process for a given chain request.\n   */\n  chainToSnap: Record<ChainId, string>;\n} & DataSourceState;\n\nconst defaultSnapState: SnapDataSourceState = {\n  activeChains: [],\n  chainToSnap: {},\n};\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\nexport type SnapDataSourceGetAssetsMiddlewareAction = {\n  type: 'SnapDataSource:getAssetsMiddleware';\n  handler: () => Middleware;\n};\n\nexport type SnapDataSourceGetActiveChainsAction = {\n  type: 'SnapDataSource:getActiveChains';\n  handler: () => Promise<ChainId[]>;\n};\n\nexport type SnapDataSourceFetchAction = {\n  type: 'SnapDataSource:fetch';\n  handler: (request: DataRequest) => Promise<DataResponse>;\n};\n\nexport type SnapDataSourceSubscribeAction = {\n  type: 'SnapDataSource:subscribe';\n  handler: (request: SubscriptionRequest) => Promise<void>;\n};\n\nexport type SnapDataSourceUnsubscribeAction = {\n  type: 'SnapDataSource:unsubscribe';\n  handler: (subscriptionId: string) => Promise<void>;\n};\n\nexport type SnapDataSourceActions =\n  | SnapDataSourceGetAssetsMiddlewareAction\n  | SnapDataSourceGetActiveChainsAction\n  | SnapDataSourceFetchAction\n  | SnapDataSourceSubscribeAction\n  | SnapDataSourceUnsubscribeAction;\n\nexport type SnapDataSourceActiveChainsChangedEvent = {\n  type: 'SnapDataSource:activeChainsUpdated';\n  payload: [ChainId[]];\n};\n\nexport type SnapDataSourceAssetsUpdatedEvent = {\n  type: 'SnapDataSource:assetsUpdated';\n  payload: [DataResponse, string | undefined];\n};\n\nexport type SnapDataSourceEvents =\n  | SnapDataSourceActiveChainsChangedEvent\n  | SnapDataSourceAssetsUpdatedEvent;\n\n/**\n * Allowed events that SnapDataSource can subscribe to.\n */\nexport type SnapDataSourceAllowedEvents =\n  | AccountsControllerAccountBalancesUpdatedEvent\n  | PermissionControllerStateChange;\n\n// Actions to report to AssetsController\ntype AssetsControllerActiveChainsUpdateAction = {\n  type: 'AssetsController:activeChainsUpdate';\n  handler: (dataSourceId: string, activeChains: ChainId[]) => void;\n};\n\ntype AssetsControllerAssetsUpdateAction = {\n  type: 'AssetsController:assetsUpdate';\n  handler: (response: DataResponse, sourceId: string) => Promise<void>;\n};\n\nexport type SnapDataSourceAllowedActions =\n  | AssetsControllerActiveChainsUpdateAction\n  | AssetsControllerAssetsUpdateAction\n  | GetRunnableSnaps\n  | HandleSnapRequest\n  | GetPermissions;\n\nexport type SnapDataSourceMessenger = Messenger<\n  typeof SNAP_DATA_SOURCE_NAME,\n  SnapDataSourceActions | SnapDataSourceAllowedActions,\n  SnapDataSourceEvents | SnapDataSourceAllowedEvents\n>;\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type SnapDataSourceOptions = {\n  /** Messenger for this data source */\n  messenger: SnapDataSourceMessenger;\n  /** Configured networks to support (defaults to all snap networks) */\n  configuredNetworks?: ChainId[];\n  /** Default polling interval in ms for subscriptions */\n  pollInterval?: number;\n  /** Initial state */\n  state?: Partial<SnapDataSourceState>;\n};\n\n// ============================================================================\n// SNAP DATA SOURCE\n// ============================================================================\n\n/**\n * Unified Snap data source that routes requests to the appropriate wallet snap\n * based on the chain ID prefix.\n *\n * @example\n * ```typescript\n * const snapDataSource = new SnapDataSource({\n *   messenger,\n * });\n *\n * // Fetch will automatically route to the correct snap\n * await snapDataSource.fetch({\n *   chainIds: ['solana:mainnet', 'bip122:000000000019d6689c085ae165831e93'],\n *   accountIds: ['account1'],\n * });\n * ```\n */\nexport class SnapDataSource extends AbstractDataSource<\n  typeof SNAP_DATA_SOURCE_NAME,\n  SnapDataSourceState\n> {\n  readonly #messenger: SnapDataSourceMessenger;\n\n  /** Bound handler for snap keyring balance updates, stored for cleanup */\n  readonly #handleSnapBalancesUpdatedBound: (\n    payload: AccountBalancesUpdatedEventPayload,\n  ) => void;\n\n  readonly #handlePermissionStateChangeBound: () => void;\n\n  /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */\n  readonly #keyringClientCache: Map<string, KeyringClient> = new Map();\n\n  constructor(options: SnapDataSourceOptions) {\n    super(SNAP_DATA_SOURCE_NAME, {\n      ...defaultSnapState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n\n    // Bind handlers for cleanup in destroy()\n    this.#handleSnapBalancesUpdatedBound = this.#handleSnapBalancesUpdated.bind(\n      this,\n    ) as (payload: AccountBalancesUpdatedEventPayload) => void;\n    this.#handlePermissionStateChangeBound =\n      this.#discoverKeyringSnaps.bind(this);\n\n    this.#registerActionHandlers();\n    this.#subscribeToEvents();\n\n    // Discover keyring-capable snaps and populate activeChains dynamically\n    this.#discoverKeyringSnaps();\n  }\n\n  /**\n   * Subscribe to all events needed by SnapDataSource.\n   * Groups snap keyring events and permission change events.\n   */\n  #subscribeToEvents(): void {\n    // Subscribe to snap keyring events for real-time balance updates.\n    // The snaps emit AccountBalancesUpdated events when balances change,\n    // which are re-published by AccountsController.\n    this.#messenger.subscribe(\n      'AccountsController:accountBalancesUpdated',\n      this.#handleSnapBalancesUpdatedBound,\n    );\n\n    // Subscribe to permission changes to detect new keyring snaps at runtime.\n    // Re-runs snap discovery when permissions change.\n    this.#messenger.subscribe(\n      'PermissionController:stateChange',\n      this.#handlePermissionStateChangeBound,\n    );\n  }\n\n  /**\n   * Handle snap balance updates from the keyring.\n   * Transforms the payload and publishes to AssetsController.\n   *\n   * @param payload - The balance update payload from AccountsController.\n   */\n  #handleSnapBalancesUpdated(\n    payload: AccountBalancesUpdatedEventPayload,\n  ): void {\n    // Transform the snap keyring payload to DataResponse format\n    let assetsBalance: NonNullable<DataResponse['assetsBalance']> | undefined;\n\n    for (const [accountId, assets] of Object.entries(payload.balances)) {\n      let accountAssets: Record<Caip19AssetId, AssetBalance> | undefined;\n\n      for (const [assetId, balance] of Object.entries(assets)) {\n        const chainId = extractChainFromAssetId(assetId);\n        if (this.#isChainSupportedBySnap(chainId)) {\n          accountAssets ??= {};\n          accountAssets[assetId as Caip19AssetId] = {\n            amount: balance.amount,\n          };\n        }\n      }\n\n      if (accountAssets) {\n        assetsBalance ??= {};\n        assetsBalance[accountId] = accountAssets;\n      }\n    }\n\n    // Only report if we have snap-related updates\n    if (assetsBalance) {\n      const response: DataResponse = { assetsBalance };\n      this.#messenger\n        .call('AssetsController:assetsUpdate', response, SNAP_DATA_SOURCE_NAME)\n        .catch(console.error);\n    }\n  }\n\n  /**\n   * Check if a chain ID is supported by any discovered snap.\n   *\n   * @param chainId - The CAIP-2 chain ID to check.\n   * @returns True if we have a snap that supports this chain.\n   */\n  #isChainSupportedBySnap(chainId: ChainId): boolean {\n    return this.state.activeChains.includes(chainId);\n  }\n\n  #registerActionHandlers(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    messenger.registerActionHandler(\n      'SnapDataSource:getAssetsMiddleware',\n      () => this.assetsMiddleware,\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:getActiveChains',\n      async () => this.getActiveChains(),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:fetch',\n      async (request: DataRequest) => this.fetch(request),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:subscribe',\n      async (request: SubscriptionRequest) => this.subscribe(request),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:unsubscribe',\n      async (subscriptionId: string) => this.unsubscribe(subscriptionId),\n    );\n  }\n\n  // ============================================================================\n  // SNAP DISCOVERY (Dynamic via PermissionController)\n  // ============================================================================\n\n  /**\n   * Get all runnable snaps from SnapController.\n   * Runnable snaps are enabled and not blocked.\n   *\n   * @returns Array of runnable snaps.\n   */\n  #getRunnableSnaps(): Snap[] {\n    try {\n      return this.#messenger.call('SnapController:getRunnableSnaps');\n    } catch (error) {\n      log('Failed to get runnable snaps', error);\n      return [];\n    }\n  }\n\n  /**\n   * Get permissions for a snap from PermissionController.\n   *\n   * @param snapId - The snap ID to get permissions for.\n   * @returns The snap's permissions, or undefined if none.\n   */\n  #getSnapPermissions(\n    snapId: string,\n  ): SubjectPermissions<PermissionConstraint> | undefined {\n    try {\n      return this.#messenger.call(\n        'PermissionController:getPermissions',\n        snapId,\n      ) as SubjectPermissions<PermissionConstraint>;\n    } catch (error) {\n      log('Failed to get permissions for snap', { snapId, error });\n      return undefined;\n    }\n  }\n\n  /**\n   * Discover all snaps with keyring capabilities and their supported chains.\n   * Uses PermissionController to find snaps with endowment:keyring permission.\n   * Updates chainToSnap mapping and activeChains.\n   *\n   * Called on initialization and whenever PermissionController state changes\n   * (e.g., new snaps installed, permissions granted/revoked).\n   *\n   * @remarks\n   * **Known limitation:** If discovery fails (e.g., SnapController not ready),\n   * the data source continues with empty chainToSnap. This means no snap\n   * chains will be supported until a re-discovery is triggered by a permission\n   * change. Callers should be aware that initialization may complete with no\n   * active chains.\n   */\n  #discoverKeyringSnaps(): void {\n    try {\n      const runnableSnaps = this.#getRunnableSnaps();\n      const chainToSnap: Record<ChainId, string> = {};\n      const supportedChains: ChainId[] = [];\n\n      for (const snap of runnableSnaps) {\n        const permissions = this.#getSnapPermissions(snap.id);\n        // Must have endowment:keyring permission to be a keyring snap\n        if (!permissions?.[KEYRING_PERMISSION]) {\n          continue;\n        }\n\n        // Get chainIds caveat from the assets permission (not keyring permission)\n        // The chainIds are stored in endowment:assets\n        const assetsPermission = permissions[ASSETS_PERMISSION];\n        const chainIds = getChainIdsCaveat(assetsPermission);\n\n        // Map each chain to this snap (first snap wins if multiple support same chain)\n        if (chainIds) {\n          for (const chainId of chainIds) {\n            if (!(chainId in chainToSnap)) {\n              chainToSnap[chainId] = snap.id;\n              supportedChains.push(chainId);\n            }\n          }\n        }\n      }\n\n      // Update chainToSnap mapping\n      this.state.chainToSnap = chainToSnap;\n\n      // Notify if chains changed\n      try {\n        this.updateActiveChains(supportedChains, (updatedChains) => {\n          this.#messenger.call(\n            'AssetsController:activeChainsUpdate',\n            SNAP_DATA_SOURCE_NAME,\n            updatedChains,\n          );\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    } catch (error) {\n      log('Keyring snap discovery failed', { error });\n      this.state.chainToSnap = {};\n      try {\n        this.updateActiveChains([], (updatedChains) => {\n          this.#messenger.call(\n            'AssetsController:activeChainsUpdate',\n            SNAP_DATA_SOURCE_NAME,\n            updatedChains,\n          );\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    }\n  }\n\n  // ============================================================================\n  // FETCH\n  // ============================================================================\n\n  async fetch(request: DataRequest): Promise<DataResponse> {\n    // Guard against undefined request\n    // Note: chainIds filtering is done by middleware/subscribe before calling fetch\n    if (!request?.accounts || !request?.chainIds?.length) {\n      return {};\n    }\n\n    const results: DataResponse = {\n      assetsBalance: {},\n      assetsMetadata: {},\n    };\n\n    // Fetch balances for each account using its snap ID from metadata\n    for (const account of request.accounts) {\n      // Skip accounts without snap metadata (non-snap accounts)\n      const snapId = account.metadata.snap?.id;\n      if (!snapId) {\n        continue;\n      }\n\n      // Skip accounts whose snap doesn't support any of the requested chains\n      const snapSupportsRequestedChains = request.chainIds.some(\n        (chainId) => this.state.chainToSnap[chainId] === snapId,\n      );\n      if (!snapSupportsRequestedChains) {\n        continue;\n      }\n\n      const accountId = account.id;\n      try {\n        const client = this.#getKeyringClient(snapId);\n\n        // Step 1: Get the list of assets for this account\n        const accountAssets = await client.listAccountAssets(accountId);\n\n        // If no assets, skip to next account\n        if (!accountAssets || accountAssets.length === 0) {\n          continue;\n        }\n\n        // Step 2: Get balances for those specific assets\n        const balances: Record<CaipAssetType, Balance> =\n          await client.getAccountBalances(\n            accountId,\n            accountAssets as CaipAssetType[],\n          );\n\n        // Transform keyring response to DataResponse format\n        if (balances && typeof balances === 'object' && results.assetsBalance) {\n          for (const [assetId, balance] of Object.entries(balances)) {\n            results.assetsBalance[accountId] ??= {};\n            const accountBalances = results.assetsBalance[accountId];\n            if (accountBalances) {\n              (accountBalances as Record<string, unknown>)[assetId] = {\n                amount: balance.amount,\n              };\n            }\n          }\n        }\n      } catch {\n        // Expected when account doesn't belong to this snap\n      }\n    }\n\n    return results;\n  }\n\n  // ============================================================================\n  // MIDDLEWARE\n  // ============================================================================\n\n  /**\n   * Get the middleware for fetching balances via Snaps.\n   * This middleware:\n   * - Supports multiple accounts in a single request\n   * - Filters request to only chains this data source supports\n   * - Fetches balances for those chains for all accounts\n   * - Merges response into context\n   * - Removes handled chains from request for next middleware\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return async (context, next) => {\n      const { request } = context;\n\n      // Filter to chains this data source supports\n      const supportedChains = request.chainIds.filter((chainId) =>\n        this.state.activeChains.includes(chainId),\n      );\n\n      // If no supported chains, skip and pass to next middleware\n      if (supportedChains.length === 0) {\n        return next(context);\n      }\n\n      let successfullyHandledChains: ChainId[] = [];\n\n      try {\n        // Fetch for supported chains\n        const response = await this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        });\n\n        // Merge response into context\n        if (response.assetsBalance) {\n          context.response.assetsBalance ??= {};\n          for (const [accountId, accountBalances] of Object.entries(\n            response.assetsBalance,\n          )) {\n            context.response.assetsBalance[accountId] = {\n              ...context.response.assetsBalance[accountId],\n              ...accountBalances,\n            };\n          }\n        }\n        if (response.assetsMetadata) {\n          context.response.assetsMetadata = {\n            ...context.response.assetsMetadata,\n            ...response.assetsMetadata,\n          };\n        }\n        if (response.assetsPrice) {\n          context.response.assetsPrice = {\n            ...context.response.assetsPrice,\n            ...response.assetsPrice,\n          };\n        }\n\n        // Determine successfully handled chains (exclude errors)\n        const failedChains = new Set(Object.keys(response.errors ?? {}));\n        successfullyHandledChains = supportedChains.filter(\n          (chainId) => !failedChains.has(chainId),\n        );\n      } catch (error) {\n        log('Middleware fetch failed', { error });\n        successfullyHandledChains = [];\n      }\n\n      // Prepare context for next middleware\n      let nextContext = context;\n      if (successfullyHandledChains.length > 0) {\n        const remainingChains = request.chainIds.filter(\n          (chainId) => !successfullyHandledChains.includes(chainId),\n        );\n        nextContext = {\n          ...context,\n          request: {\n            ...request,\n            chainIds: remainingChains,\n          },\n        };\n      }\n\n      // Call next middleware\n      return next(nextContext);\n    };\n  }\n\n  // ============================================================================\n  // SUBSCRIBE - Routes to appropriate snap(s)\n  // ============================================================================\n\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Guard against undefined request or chainIds\n    if (!request?.chainIds) {\n      return;\n    }\n\n    // Filter to chains we have a snap for\n    const supportedChains = request.chainIds.filter((chainId) =>\n      this.#isChainSupportedBySnap(chainId),\n    );\n\n    if (supportedChains.length === 0) {\n      return;\n    }\n\n    if (isUpdate) {\n      const existing = this.activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        existing.chains = supportedChains;\n        // Do a fetch to get latest data on subscription update\n        this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        })\n          .then(async (fetchResponse) => {\n            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {\n              await this.#messenger.call(\n                'AssetsController:assetsUpdate',\n                fetchResponse,\n                SNAP_DATA_SOURCE_NAME,\n              );\n            }\n            return fetchResponse;\n          })\n          .catch((error) => {\n            log('Subscription update fetch failed', { subscriptionId, error });\n          });\n        return;\n      }\n    }\n\n    await this.unsubscribe(subscriptionId);\n\n    // Snaps provide real-time updates via AccountsController:accountBalancesUpdated\n    // We only need to track the subscription and do an initial fetch\n    // No polling needed - updates come through #handleSnapBalancesUpdated\n\n    this.activeSubscriptions.set(subscriptionId, {\n      cleanup: () => {\n        // No timer to clear - we use event-based updates\n      },\n      chains: supportedChains,\n    });\n\n    // Initial fetch to get current balances\n    try {\n      const fetchResponse = await this.fetch({\n        ...request,\n        chainIds: supportedChains,\n      });\n\n      if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {\n        await this.#messenger.call(\n          'AssetsController:assetsUpdate',\n          fetchResponse,\n          SNAP_DATA_SOURCE_NAME,\n        );\n      }\n    } catch (error) {\n      log('Initial fetch failed', { subscriptionId, error });\n    }\n  }\n\n  // ============================================================================\n  // KEYRING CLIENT\n  // ============================================================================\n\n  /**\n   * Gets a `KeyringClient` for a Snap.\n   * Caches clients per snap ID to avoid re-instantiation across multiple calls.\n   *\n   * @param snapId - ID of the Snap to get the client for.\n   * @returns A `KeyringClient` for the Snap.\n   */\n  #getKeyringClient(snapId: string): KeyringClient {\n    const cachedClient = this.#keyringClientCache.get(snapId);\n    if (cachedClient) {\n      return cachedClient;\n    }\n\n    const client = new KeyringClient({\n      send: async (request: JsonRpcRequest): Promise<Json> =>\n        (await this.#messenger.call('SnapController:handleRequest', {\n          snapId: snapId as SnapId,\n          origin: 'metamask',\n          handler: HandlerType.OnKeyringRequest,\n          request,\n        })) as Json,\n    });\n\n    this.#keyringClientCache.set(snapId, client);\n    return client;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    // Unsubscribe from snap keyring events\n    try {\n      messenger.unsubscribe(\n        'AccountsController:accountBalancesUpdated',\n        this.#handleSnapBalancesUpdatedBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from snap keyring events', { error });\n    }\n\n    // Unsubscribe from permission changes\n    try {\n      messenger.unsubscribe(\n        'PermissionController:stateChange',\n        this.#handlePermissionStateChangeBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from permission changes', { error });\n    }\n\n    // Clean up active subscriptions\n    for (const [subscriptionId] of this.activeSubscriptions) {\n      this.unsubscribe(subscriptionId).catch(() => {\n        // Ignore cleanup errors\n      });\n    }\n\n    // Clear keyring client cache\n    this.#keyringClientCache.clear();\n  }\n}\n\n// ============================================================================\n// FACTORY\n// ============================================================================\n\nexport function createSnapDataSource(\n  options: SnapDataSourceOptions,\n): SnapDataSource {\n  return new SnapDataSource(options);\n}\n"]}