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

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

@@ -1,8 +1,20 @@
-import { KeyringClient } from '@metamask/keyring-snap-client';
-import { HandlerType, SnapCaveatType } from '@metamask/snaps-utils';
-import { parseCaipAssetType } from '@metamask/utils';
-import { projectLogger, createModuleLogger } from '../logger.js';
-import { AbstractDataSource } from './AbstractDataSource.js';
+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_onActiveChainsUpdated, _SnapDataSource_onAssetsUpdate, _SnapDataSource_handleSnapBalancesUpdatedBound, _SnapDataSource_handlePermissionStateChangeBound, _SnapDataSource_keyringClientCache, _SnapDataSource_subscribeToEvents, _SnapDataSource_handleSnapBalancesUpdated, _SnapDataSource_isChainSupportedBySnap, _SnapDataSource_getRunnableSnaps, _SnapDataSource_getSnapPermissions, _SnapDataSource_discoverKeyringSnaps, _SnapDataSource_getKeyringClient;
+import { KeyringClient } from "@metamask/keyring-snap-client";
+import { HandlerType, SnapCaveatType } from "@metamask/snaps-utils";
+import { parseCaipAssetType } from "@metamask/utils";
+import { projectLogger, createModuleLogger } from "../logger.mjs";
+import { AbstractDataSource } from "./AbstractDataSource.mjs";
 const log = createModuleLogger(projectLogger, 'SnapDataSource');
 // ============================================================================
 // CONSTANTS
@@ -70,202 +82,35 @@ const defaultSnapState = {
  * ```
  */
 export class SnapDataSource extends AbstractDataSource {
-    #messenger;
-    #onActiveChainsUpdated;
-    #onAssetsUpdate;
-    /** Bound handler for snap keyring balance updates, stored for cleanup */
-    #handleSnapBalancesUpdatedBound;
-    #handlePermissionStateChangeBound;
-    /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */
-    #keyringClientCache = new Map();
     constructor(options) {
         super(SNAP_DATA_SOURCE_NAME, {
             ...defaultSnapState,
             ...options.state,
         });
-        this.#messenger = options.messenger;
-        this.#onActiveChainsUpdated = options.onActiveChainsUpdated;
-        this.#onAssetsUpdate = options.onAssetsUpdate;
+        _SnapDataSource_instances.add(this);
+        _SnapDataSource_messenger.set(this, void 0);
+        _SnapDataSource_onActiveChainsUpdated.set(this, void 0);
+        _SnapDataSource_onAssetsUpdate.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");
+        __classPrivateFieldSet(this, _SnapDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
+        __classPrivateFieldSet(this, _SnapDataSource_onAssetsUpdate, options.onAssetsUpdate, "f");
         // Bind handlers for cleanup in destroy()
-        this.#handleSnapBalancesUpdatedBound = this.#handleSnapBalancesUpdated.bind(this);
-        this.#handlePermissionStateChangeBound =
-            this.#discoverKeyringSnaps.bind(this);
-        this.#subscribeToEvents();
+        __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_subscribeToEvents).call(this);
         // Discover keyring-capable snaps and populate activeChains dynamically
-        this.#discoverKeyringSnaps();
-    }
-    /**
-     * Subscribe to all events needed by SnapDataSource.
-     * Groups snap keyring events and permission change events.
-     */
-    #subscribeToEvents() {
-        // Subscribe to snap keyring events and permission changes (not in AssetsControllerEvents).
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const messenger = this.#messenger;
-        messenger.subscribe('AccountsController:accountBalancesUpdated', this.#handleSnapBalancesUpdatedBound);
-        messenger.subscribe('PermissionController:stateChange', this.#handlePermissionStateChangeBound);
-        // Rediscover keyring snaps when any snap gets installed
-        messenger.subscribe('SnapController:snapInstalled', () => {
-            this.#discoverKeyringSnaps();
-        });
-    }
-    /**
-     * Handle snap balance updates from the keyring.
-     * Transforms the payload and publishes to AssetsController.
-     *
-     * @param payload - The balance update payload from AccountsController.
-     */
-    #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)) {
-                let chainId;
-                try {
-                    chainId = extractChainFromAssetId(assetId);
-                }
-                catch (error) {
-                    log('Skipping snap balance for malformed asset ID', {
-                        assetId,
-                        error,
-                    });
-                    continue;
-                }
-                if (this.#isChainSupportedBySnap(chainId)) {
-                    accountAssets ??= {};
-                    accountAssets[assetId] = {
-                        amount: balance.amount,
-                        ...(balance.metadata ? { metadata: balance.metadata } : {}),
-                    };
-                }
-            }
-            if (accountAssets) {
-                assetsBalance ??= {};
-                assetsBalance[accountId] = accountAssets;
-            }
-        }
-        // Only report if we have snap-related updates
-        if (assetsBalance) {
-            const response = { assetsBalance, updateMode: 'merge' };
-            Promise.resolve(this.#onAssetsUpdate(response)).catch(console.error);
-        }
-    }
-    /**
-     * Check if a chain ID is supported by any discovered snap.
-     *
-     * @param chainId - The CAIP-2 chain ID to check.
-     * @returns True if we have a snap that supports this chain.
-     */
-    #isChainSupportedBySnap(chainId) {
-        return this.state.activeChains.includes(chainId);
-    }
-    // ============================================================================
-    // SNAP DISCOVERY (Dynamic via PermissionController)
-    // ============================================================================
-    /**
-     * Get all runnable snaps from SnapController.
-     * Runnable snaps are enabled and not blocked.
-     *
-     * @returns Array of runnable snaps.
-     */
-    #getRunnableSnaps() {
-        try {
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            return this.#messenger.call('SnapController:getRunnableSnaps');
-        }
-        catch (error) {
-            log('Failed to get runnable snaps', error);
-            return [];
-        }
-    }
-    /**
-     * Get permissions for a snap from PermissionController.
-     *
-     * @param snapId - The snap ID to get permissions for.
-     * @returns The snap's permissions, or undefined if none.
-     */
-    #getSnapPermissions(snapId) {
-        try {
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            return this.#messenger.call('PermissionController:getPermissions', snapId);
-        }
-        catch (error) {
-            log('Failed to get permissions for snap', { snapId, error });
-            return undefined;
-        }
-    }
-    /**
-     * Discover all snaps with keyring capabilities and their supported chains.
-     * Uses PermissionController to find snaps with endowment:keyring permission.
-     * Updates chainToSnap mapping and activeChains.
-     *
-     * Called on initialization and whenever PermissionController state changes
-     * (e.g., new snaps installed, permissions granted/revoked).
-     *
-     * @remarks
-     * **Known limitation:** If discovery fails (e.g., SnapController not ready),
-     * the data source continues with empty chainToSnap. This means no snap
-     * chains will be supported until a re-discovery is triggered by a permission
-     * change. Callers should be aware that initialization may complete with no
-     * active chains.
-     */
-    #discoverKeyringSnaps() {
-        try {
-            const runnableSnaps = this.#getRunnableSnaps();
-            const chainToSnap = {};
-            const supportedChains = [];
-            for (const snap of runnableSnaps) {
-                const permissions = this.#getSnapPermissions(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 {
-                const previous = [...this.state.activeChains];
-                this.updateActiveChains(supportedChains, (updatedChains) => {
-                    this.#onActiveChainsUpdated(this.getName(), updatedChains, previous);
-                });
-            }
-            catch {
-                // AssetsController not ready yet - expected during initialization
-            }
-        }
-        catch (error) {
-            log('Keyring snap discovery failed', { error });
-            this.state.chainToSnap = {};
-            try {
-                const previous = [...this.state.activeChains];
-                this.updateActiveChains([], (updatedChains) => {
-                    this.#onActiveChainsUpdated(this.getName(), updatedChains, previous);
-                });
-            }
-            catch {
-                // AssetsController not ready yet - expected during initialization
-            }
-        }
+        __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?.chainIds?.length) {
@@ -293,7 +138,7 @@ export class SnapDataSource extends AbstractDataSource {
             }
             const accountId = account.id;
             try {
-                const client = this.#getKeyringClient(snapId);
+                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
@@ -305,7 +150,7 @@ export class SnapDataSource extends AbstractDataSource {
                 // Transform keyring response to DataResponse format
                 if (balances && typeof balances === 'object' && results.assetsBalance) {
                     for (const [assetId, balance] of Object.entries(balances)) {
-                        results.assetsBalance[accountId] ??= {};
+                        (_a = results.assetsBalance)[accountId] ?? (_a[accountId] = {});
                         const accountBalances = results.assetsBalance[accountId];
                         if (accountBalances) {
                             accountBalances[assetId] = {
@@ -338,6 +183,7 @@ export class SnapDataSource extends AbstractDataSource {
      */
     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));
@@ -354,7 +200,7 @@ export class SnapDataSource extends AbstractDataSource {
                 });
                 // Merge response into context
                 if (response.assetsBalance) {
-                    context.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],
@@ -408,7 +254,7 @@ export class SnapDataSource extends AbstractDataSource {
             return;
         }
         // Filter to chains we have a snap for
-        const supportedChains = request.chainIds.filter((chainId) => this.#isChainSupportedBySnap(chainId));
+        const supportedChains = request.chainIds.filter((chainId) => __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_isChainSupportedBySnap).call(this, chainId));
         if (supportedChains.length === 0) {
             return;
         }
@@ -418,7 +264,7 @@ export class SnapDataSource extends AbstractDataSource {
                 chainIds: supportedChains,
             });
             if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {
-                await this.#onAssetsUpdate(fetchResponse);
+                await __classPrivateFieldGet(this, _SnapDataSource_onAssetsUpdate, "f").call(this, fetchResponse);
             }
         }
         catch (error) {
@@ -426,59 +272,165 @@ export class SnapDataSource extends AbstractDataSource {
         }
     }
     // ============================================================================
-    // KEYRING CLIENT
-    // ============================================================================
-    /**
-     * Gets a `KeyringClient` for a Snap.
-     * Caches clients per snap ID to avoid re-instantiation across multiple calls.
-     *
-     * @param snapId - ID of the Snap to get the client for.
-     * @returns A `KeyringClient` for the Snap.
-     */
-    #getKeyringClient(snapId) {
-        const cachedClient = this.#keyringClientCache.get(snapId);
-        if (cachedClient) {
-            return cachedClient;
-        }
-        const client = new KeyringClient({
-            send: async (request) => await this.#messenger.call('SnapController:handleRequest', {
-                snapId: snapId,
-                origin: 'metamask',
-                handler: HandlerType.OnKeyringRequest,
-                request,
-            }),
-        });
-        this.#keyringClientCache.set(snapId, client);
-        return client;
-    }
-    // ============================================================================
     // CLEANUP
     // ============================================================================
     destroy() {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const messenger = this.#messenger;
+        const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
         // Unsubscribe from snap keyring events
         try {
-            messenger.unsubscribe('AccountsController:accountBalancesUpdated', this.#handleSnapBalancesUpdatedBound);
+            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', this.#handlePermissionStateChangeBound);
+            messenger.unsubscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
         }
         catch (error) {
             log('Failed to unsubscribe from permission changes', { error });
         }
         // Clear keyring client cache
-        this.#keyringClientCache.clear();
+        __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").clear();
     }
 }
+_SnapDataSource_messenger = new WeakMap(), _SnapDataSource_onActiveChainsUpdated = new WeakMap(), _SnapDataSource_onAssetsUpdate = 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 and permission changes (not in AssetsControllerEvents).
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
+    messenger.subscribe('AccountsController:accountBalancesUpdated', __classPrivateFieldGet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, "f"));
+    messenger.subscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
+    // Rediscover keyring snaps when any snap gets installed
+    messenger.subscribe('SnapController:snapInstalled', () => {
+        __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_discoverKeyringSnaps).call(this);
+    });
+}, _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)) {
+            let chainId;
+            try {
+                chainId = extractChainFromAssetId(assetId);
+            }
+            catch (error) {
+                log('Skipping snap balance for malformed asset ID', {
+                    assetId,
+                    error,
+                });
+                continue;
+            }
+            if (__classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_isChainSupportedBySnap).call(this, chainId)) {
+                accountAssets ?? (accountAssets = {});
+                accountAssets[assetId] = {
+                    amount: balance.amount,
+                    ...(balance.metadata ? { metadata: balance.metadata } : {}),
+                };
+            }
+        }
+        if (accountAssets) {
+            assetsBalance ?? (assetsBalance = {});
+            assetsBalance[accountId] = accountAssets;
+        }
+    }
+    // Only report if we have snap-related updates
+    if (assetsBalance) {
+        const response = { assetsBalance, updateMode: 'merge' };
+        Promise.resolve(__classPrivateFieldGet(this, _SnapDataSource_onAssetsUpdate, "f").call(this, response)).catch(console.error);
+    }
+}, _SnapDataSource_isChainSupportedBySnap = function _SnapDataSource_isChainSupportedBySnap(chainId) {
+    return this.state.activeChains.includes(chainId);
+}, _SnapDataSource_getRunnableSnaps = function _SnapDataSource_getRunnableSnaps() {
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        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 {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        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 {
+            const previous = [...this.state.activeChains];
+            this.updateActiveChains(supportedChains, (updatedChains) => {
+                __classPrivateFieldGet(this, _SnapDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous);
+            });
+        }
+        catch {
+            // AssetsController not ready yet - expected during initialization
+        }
+    }
+    catch (error) {
+        log('Keyring snap discovery failed', { error });
+        this.state.chainToSnap = {};
+        try {
+            const previous = [...this.state.activeChains];
+            this.updateActiveChains([], (updatedChains) => {
+                __classPrivateFieldGet(this, _SnapDataSource_onActiveChainsUpdated, "f").call(this, this.getName(), updatedChains, previous);
+            });
+        }
+        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.js.map
+//# 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;AACpE,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAIrD,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAqB;AASjE,OAAO,EAAE,kBAAkB,EAAE,iCAAgC;AAkC7D,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;;;;;;;GAOG;AACH,MAAM,UAAU,uBAAuB,CAAC,OAAe;IACrD,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAwB,CAAC,CAAC;IAC5D,OAAO,MAAM,CAAC,OAAO,CAAC;AACxB,CAAC;AAkBD,MAAM,gBAAgB,GAAwB;IAC5C,YAAY,EAAE,EAAE;IAChB,WAAW,EAAE,EAAE;CAChB,CAAC;AA8CF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,cAAe,SAAQ,kBAGnC;IAqBC,YAAY,OAA8B;QACxC,KAAK,CAAC,qBAAqB,EAAE;YAC3B,GAAG,gBAAgB;YACnB,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QAxBI,4CAAsC;QAEtC,wDAIC;QAED,iDAAkE;QAE3E,yEAAyE;QAChE,iEAEC;QAED,mEAA8C;QAEvD,6EAA6E;QACpE,6CAAkD,IAAI,GAAG,EAAE,EAAC;QAQnE,uBAAA,IAAI,6BAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,yCAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,kCAAmB,OAAO,CAAC,cAAc,MAAA,CAAC;QAE9C,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,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1B,uEAAuE;QACvE,uBAAA,IAAI,uEAAsB,MAA1B,IAAI,CAAwB,CAAC;IAC/B,CAAC;IAmMD,+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,EAAE,MAAM,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,2BAA2B,EAAE,MAAM,EAAE,CAAC;YAClD,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;QACpE,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,aAAa,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE;YACd,UAAU,EAAE,OAAO;SACpB,CAAC;QAEF,kEAAkE;QAClE,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YAC9D,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;gCACtB,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;6BAC5D,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,UAAU,EAAE,CAAC;oBACxB,OAAO,CAAC,QAAQ,CAAC,UAAU,GAAG;wBAC5B,GAAG,OAAO,CAAC,QAAQ,CAAC,UAAU;wBAC9B,GAAG,QAAQ,CAAC,UAAU;qBACvB,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,GAAG,mBAAmB,CAAC;QAExD,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,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,sCAAgB,MAApB,IAAI,EAAiB,aAAa,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,sBAAsB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAqCD,+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,6BAA6B;QAC7B,uBAAA,IAAI,0CAAoB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF;;IA1cG,2FAA2F;IAC3F,8DAA8D;IAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;IACzC,SAAS,CAAC,SAAS,CACjB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;IACF,SAAS,CAAC,SAAS,CACjB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;IACF,wDAAwD;IACxD,SAAS,CAAC,SAAS,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACvD,uBAAA,IAAI,uEAAsB,MAA1B,IAAI,CAAwB,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,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,IAAI,OAAgB,CAAC;YACrB,IAAI,CAAC;gBACH,OAAO,GAAG,uBAAuB,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,8CAA8C,EAAE;oBAClD,OAAO;oBACP,KAAK;iBACN,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YACD,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;oBACtB,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5D,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,UAAU,EAAE,OAAO,EAAE,CAAC;QACtE,OAAO,CAAC,OAAO,CAAC,uBAAA,IAAI,sCAAgB,MAApB,IAAI,EAAiB,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvE,CAAC;AACH,CAAC,2FAQuB,OAAgB;IACtC,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;IAaC,IAAI,CAAC;QACH,8DAA8D;QAC9D,OAAQ,uBAAA,IAAI,iCAAmB,CAAC,IAAI,CAClC,iCAAiC,CACxB,CAAC;IACd,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,8DAA8D;QAC9D,OAAQ,uBAAA,IAAI,iCAAmB,CAAC,IAAI,CAClC,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,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC,aAAa,EAAE,EAAE;gBACzD,uBAAA,IAAI,6CAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;YACvE,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,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE;gBAC5C,uBAAA,IAAI,6CAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;AACH,CAAC,+EAyNiB,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,MACE,uBAAA,IAAI,iCAGL,CAAC,IAAI,CAAC,8BAA8B,EAAE;YACrC,MAAM,EAAE,MAAgB;YACxB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,WAAW,CAAC,gBAAgB;YACrC,OAAO;SACR,CAAC;KACL,CAAC,CAAC;IAEH,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;AAmCH,+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 {\n  Caveat,\n  GetPermissions,\n  PermissionConstraint,\n  PermissionControllerStateChange,\n  SubjectPermissions,\n} from '@metamask/permission-controller';\nimport type {\n  SnapControllerGetRunnableSnapsAction,\n  SnapControllerHandleRequestAction,\n  SnapControllerSnapInstalledEvent,\n} from '@metamask/snaps-controllers';\nimport type { Snap, SnapId } from '@metamask/snaps-sdk';\nimport { HandlerType, SnapCaveatType } from '@metamask/snaps-utils';\nimport { parseCaipAssetType } from '@metamask/utils';\nimport type { Json, JsonRpcRequest } from '@metamask/utils';\n\nimport type { AssetsControllerMessenger } from '../AssetsController.js';\nimport { projectLogger, createModuleLogger } from '../logger.js';\nimport type {\n  AssetBalance,\n  ChainId,\n  Caip19AssetId,\n  DataRequest,\n  DataResponse,\n  Middleware,\n} from '../types.js';\nimport { AbstractDataSource } from './AbstractDataSource.js';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource.js';\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        metadata?: Record<string, Json>;\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 * Extracts the CAIP-2 chain ID from a CAIP-19 asset ID.\n * e.g., \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp/slip44:501\" -> \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\"\n * Uses @metamask/utils parseCaipAssetType for CAIP parsing.\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 parsed = parseCaipAssetType(assetId as CaipAssetType);\n  return parsed.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\n/**\n * Allowed events that SnapDataSource can subscribe to.\n */\nexport type SnapDataSourceAllowedEvents =\n  | AccountsControllerAccountBalancesUpdatedEvent\n  | PermissionControllerStateChange\n  | SnapControllerSnapInstalledEvent;\n\nexport type SnapDataSourceAllowedActions =\n  | SnapControllerGetRunnableSnapsAction\n  | SnapControllerHandleRequestAction\n  | GetPermissions;\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type SnapDataSourceOptions = {\n  /** The AssetsController messenger (shared by all data sources). */\n  messenger: AssetsControllerMessenger;\n  /** Called when this data source's active chains change. Pass dataSourceName so the controller knows the source. */\n  onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n  /**\n   * Called directly with snap-sourced balance updates (from\n   * AccountsController:accountBalancesUpdated), independent of whether the\n   * controller currently has an active subscription for the chain.\n   */\n  onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\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 *   onActiveChainsUpdated: (chains) => { /* ... *\\/ },\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: AssetsControllerMessenger;\n\n  readonly #onActiveChainsUpdated: (\n    dataSourceName: string,\n    chains: ChainId[],\n    previousChains: ChainId[],\n  ) => void;\n\n  readonly #onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\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    this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n    this.#onAssetsUpdate = options.onAssetsUpdate;\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.#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 and permission changes (not in AssetsControllerEvents).\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n    messenger.subscribe(\n      'AccountsController:accountBalancesUpdated',\n      this.#handleSnapBalancesUpdatedBound,\n    );\n    messenger.subscribe(\n      'PermissionController:stateChange',\n      this.#handlePermissionStateChangeBound,\n    );\n    // Rediscover keyring snaps when any snap gets installed\n    messenger.subscribe('SnapController:snapInstalled', () => {\n      this.#discoverKeyringSnaps();\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        let chainId: ChainId;\n        try {\n          chainId = extractChainFromAssetId(assetId);\n        } catch (error) {\n          log('Skipping snap balance for malformed asset ID', {\n            assetId,\n            error,\n          });\n          continue;\n        }\n        if (this.#isChainSupportedBySnap(chainId)) {\n          accountAssets ??= {};\n          accountAssets[assetId as Caip19AssetId] = {\n            amount: balance.amount,\n            ...(balance.metadata ? { metadata: balance.metadata } : {}),\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, updateMode: 'merge' };\n      Promise.resolve(this.#onAssetsUpdate(response)).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  // ============================================================================\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      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return (this.#messenger as any).call(\n        'SnapController:getRunnableSnaps',\n      ) as Snap[];\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      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return (this.#messenger as any).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        const previous = [...this.state.activeChains];\n        this.updateActiveChains(supportedChains, (updatedChains) => {\n          this.#onActiveChainsUpdated(this.getName(), updatedChains, previous);\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        const previous = [...this.state.activeChains];\n        this.updateActiveChains([], (updatedChains) => {\n          this.#onActiveChainsUpdated(this.getName(), updatedChains, previous);\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?.chainIds?.length) {\n      return {};\n    }\n    if (!request?.accountsWithSupportedChains?.length) {\n      return { assetsBalance: {}, assetsInfo: {}, updateMode: 'merge' };\n    }\n\n    const results: DataResponse = {\n      assetsBalance: {},\n      assetsInfo: {},\n      updateMode: 'merge',\n    };\n\n    // Fetch balances for each account using its snap ID from metadata\n    for (const { account } of request.accountsWithSupportedChains) {\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                ...(balance.metadata ? { metadata: balance.metadata } : {}),\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.assetsInfo) {\n          context.response.assetsInfo = {\n            ...context.response.assetsInfo,\n            ...response.assetsInfo,\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 } = 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    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.#onAssetsUpdate(fetchResponse);\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 (\n          this.#messenger as unknown as {\n            call: (action: string, ...args: unknown[]) => Promise<Json> | Json;\n          }\n        ).call('SnapController:handleRequest', {\n          snapId: snapId as SnapId,\n          origin: 'metamask',\n          handler: HandlerType.OnKeyringRequest,\n          request,\n        }),\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    // 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"]}