@metamask/assets-controllers 74.3.2 → 74.3.3

package/CHANGELOG.md

@@ -7,6 +7,13 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [74.3.3]
+
+### Changed
+
+- Enhance `TokenBalancesController` with internal dynamic polling per chain support, enabling configurable polling intervals for different networks with automatic interval grouping for improved performance (transparent to existing API) ([#6357](https://github.com/MetaMask/core/pull/6357))
+- Bump `@metamask/base-controller` from `^8.2.0` to `^8.3.0` ([#6465](https://github.com/MetaMask/core/pull/6465))
+
 ## [74.3.2]
 
 ### Changed
@@ -1954,7 +1961,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Use Ethers for AssetsContractController ([#845](https://github.com/MetaMask/core/pull/845))
 
-[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.3.2...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.3.3...HEAD
+[74.3.3]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.3.2...@metamask/assets-controllers@74.3.3
 [74.3.2]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.3.1...@metamask/assets-controllers@74.3.2
 [74.3.1]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.3.0...@metamask/assets-controllers@74.3.1
 [74.3.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.2.0...@metamask/assets-controllers@74.3.0

package/dist/TokenBalancesController.cjs

@@ -10,7 +10,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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 _TokenBalancesController_instances, _TokenBalancesController_queryAllAccounts, _TokenBalancesController_balanceFetchers, _TokenBalancesController_allTokens, _TokenBalancesController_detectedTokens, _TokenBalancesController_chainIdsWithTokens, _TokenBalancesController_getProvider, _TokenBalancesController_getNetworkClient, _TokenBalancesController_onTokensChanged, _TokenBalancesController_onNetworkChanged, _TokenBalancesController_onAccountRemoved;
+var _TokenBalancesController_instances, _TokenBalancesController_queryAllAccounts, _TokenBalancesController_balanceFetchers, _TokenBalancesController_allTokens, _TokenBalancesController_detectedTokens, _TokenBalancesController_defaultInterval, _TokenBalancesController_chainPollingConfig, _TokenBalancesController_intervalPollingTimers, _TokenBalancesController_isControllerPollingActive, _TokenBalancesController_requestedChainIds, _TokenBalancesController_chainIdsWithTokens, _TokenBalancesController_getProvider, _TokenBalancesController_getNetworkClient, _TokenBalancesController_startIntervalGroupPolling, _TokenBalancesController_startPollingForInterval, _TokenBalancesController_setPollingTimer, _TokenBalancesController_onTokensChanged, _TokenBalancesController_onNetworkChanged, _TokenBalancesController_onAccountRemoved;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TokenBalancesController = void 0;
 const providers_1 = require("@ethersproject/providers");
@@ -37,7 +37,7 @@ const checksum = (addr) => (0, controller_utils_1.toChecksumHexAddress)(addr);
 // ────────────────────────────────────────────────────────────────────────────
 // region: Main controller
 class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPollingController)() {
-    constructor({ messenger, interval = DEFAULT_INTERVAL_MS, state = {}, queryMultipleAccounts = true, useAccountsAPI = false, allowExternalServices = () => true, }) {
+    constructor({ messenger, interval = DEFAULT_INTERVAL_MS, chainPollingIntervals = {}, state = {}, queryMultipleAccounts = true, useAccountsAPI = false, allowExternalServices = () => true, }) {
         super({
             name: CONTROLLER,
             messenger,
@@ -49,6 +49,16 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
         _TokenBalancesController_balanceFetchers.set(this, void 0);
         _TokenBalancesController_allTokens.set(this, {});
         _TokenBalancesController_detectedTokens.set(this, {});
+        /** Default polling interval for chains without specific configuration */
+        _TokenBalancesController_defaultInterval.set(this, void 0);
+        /** Per-chain polling configuration */
+        _TokenBalancesController_chainPollingConfig.set(this, void 0);
+        /** Active polling timers grouped by interval */
+        _TokenBalancesController_intervalPollingTimers.set(this, new Map());
+        /** Track if controller-level polling is active */
+        _TokenBalancesController_isControllerPollingActive.set(this, false);
+        /** Store original chainIds from startPolling to preserve intent */
+        _TokenBalancesController_requestedChainIds.set(this, []);
         _TokenBalancesController_getProvider.set(this, (chainId) => {
             const { networkConfigurationsByChainId } = this.messagingSystem.call('NetworkController:getState');
             const cfg = networkConfigurationsByChainId[chainId];
@@ -176,6 +186,8 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
             });
         });
         __classPrivateFieldSet(this, _TokenBalancesController_queryAllAccounts, queryMultipleAccounts, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_defaultInterval, interval, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_chainPollingConfig, { ...chainPollingIntervals }, "f");
         // Strategy order: API first, then RPC fallback
         __classPrivateFieldSet(this, _TokenBalancesController_balanceFetchers, [
             ...(useAccountsAPI && allowExternalServices()
@@ -198,10 +210,61 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
         });
         this.messagingSystem.subscribe('NetworkController:stateChange', __classPrivateFieldGet(this, _TokenBalancesController_onNetworkChanged, "f"));
         this.messagingSystem.subscribe('KeyringController:accountRemoved', __classPrivateFieldGet(this, _TokenBalancesController_onAccountRemoved, "f"));
+        // Register action handlers for polling interval control
+        this.messagingSystem.registerActionHandler(`TokenBalancesController:updateChainPollingConfigs`, this.updateChainPollingConfigs.bind(this));
+        this.messagingSystem.registerActionHandler(`TokenBalancesController:getChainPollingConfig`, this.getChainPollingConfig.bind(this));
+    }
+    /**
+     * Override to support per-chain polling intervals by grouping chains by interval
+     *
+     * @param options0 - The polling options
+     * @param options0.chainIds - Chain IDs to start polling for
+     */
+    _startPolling({ chainIds }) {
+        // Store the original chainIds to preserve intent across config updates
+        __classPrivateFieldSet(this, _TokenBalancesController_requestedChainIds, [...chainIds], "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, true, "f");
+        __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startIntervalGroupPolling).call(this, chainIds, true);
+    }
+    /**
+     * Override to handle our custom polling approach
+     */
+    _stopPollingByPollingTokenSetId() {
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, false, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_requestedChainIds, [], "f"); // Clear original intent when stopping
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+    }
+    /**
+     * Get polling configuration for a chain (includes default fallback)
+     *
+     * @param chainId - The chain ID to get config for
+     * @returns The polling configuration for the chain
+     */
+    getChainPollingConfig(chainId) {
+        return (__classPrivateFieldGet(this, _TokenBalancesController_chainPollingConfig, "f")[chainId] ?? {
+            interval: __classPrivateFieldGet(this, _TokenBalancesController_defaultInterval, "f"),
+        });
     }
     async _executePoll({ chainIds }) {
+        // This won't be called with our custom implementation, but keep for compatibility
         await this.updateBalances({ chainIds });
     }
+    /**
+     * Update multiple chain polling configurations at once
+     *
+     * @param configs - Object mapping chain IDs to polling configurations
+     * @param options - Optional configuration for the update behavior
+     * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)
+     */
+    updateChainPollingConfigs(configs, options = { immediateUpdate: true }) {
+        Object.assign(__classPrivateFieldGet(this, _TokenBalancesController_chainPollingConfig, "f"), configs);
+        // If polling is currently active, restart with new interval groupings
+        if (__classPrivateFieldGet(this, _TokenBalancesController_isControllerPollingActive, "f")) {
+            // Restart polling with immediate fetch by default, unless explicitly disabled
+            __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startIntervalGroupPolling).call(this, __classPrivateFieldGet(this, _TokenBalancesController_requestedChainIds, "f"), options.immediateUpdate);
+        }
+    }
     async updateBalances({ chainIds } = {}) {
         const targetChains = chainIds ?? __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_chainIdsWithTokens).call(this);
         if (!targetChains.length) {
@@ -312,15 +375,75 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
     resetState() {
         this.update(() => ({ tokenBalances: {} }));
     }
+    /**
+     * Clean up all timers and resources when controller is destroyed
+     */
+    destroy() {
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, false, "f");
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+        // Unregister action handlers
+        this.messagingSystem.unregisterActionHandler(`TokenBalancesController:updateChainPollingConfigs`);
+        this.messagingSystem.unregisterActionHandler(`TokenBalancesController:getChainPollingConfig`);
+        super.destroy();
+    }
 }
 exports.TokenBalancesController = TokenBalancesController;
-_TokenBalancesController_queryAllAccounts = new WeakMap(), _TokenBalancesController_balanceFetchers = new WeakMap(), _TokenBalancesController_allTokens = new WeakMap(), _TokenBalancesController_detectedTokens = new WeakMap(), _TokenBalancesController_getProvider = new WeakMap(), _TokenBalancesController_getNetworkClient = new WeakMap(), _TokenBalancesController_onTokensChanged = new WeakMap(), _TokenBalancesController_onNetworkChanged = new WeakMap(), _TokenBalancesController_onAccountRemoved = new WeakMap(), _TokenBalancesController_instances = new WeakSet(), _TokenBalancesController_chainIdsWithTokens = function _TokenBalancesController_chainIdsWithTokens() {
+_TokenBalancesController_queryAllAccounts = new WeakMap(), _TokenBalancesController_balanceFetchers = new WeakMap(), _TokenBalancesController_allTokens = new WeakMap(), _TokenBalancesController_detectedTokens = new WeakMap(), _TokenBalancesController_defaultInterval = new WeakMap(), _TokenBalancesController_chainPollingConfig = new WeakMap(), _TokenBalancesController_intervalPollingTimers = new WeakMap(), _TokenBalancesController_isControllerPollingActive = new WeakMap(), _TokenBalancesController_requestedChainIds = new WeakMap(), _TokenBalancesController_getProvider = new WeakMap(), _TokenBalancesController_getNetworkClient = new WeakMap(), _TokenBalancesController_onTokensChanged = new WeakMap(), _TokenBalancesController_onNetworkChanged = new WeakMap(), _TokenBalancesController_onAccountRemoved = new WeakMap(), _TokenBalancesController_instances = new WeakSet(), _TokenBalancesController_chainIdsWithTokens = function _TokenBalancesController_chainIdsWithTokens() {
     return [
         ...new Set([
             ...Object.keys(__classPrivateFieldGet(this, _TokenBalancesController_allTokens, "f")),
             ...Object.keys(__classPrivateFieldGet(this, _TokenBalancesController_detectedTokens, "f")),
         ]),
     ];
+}, _TokenBalancesController_startIntervalGroupPolling = function _TokenBalancesController_startIntervalGroupPolling(chainIds, immediate = true) {
+    // Stop any existing interval timers
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+    // Group chains by their polling intervals
+    const intervalGroups = new Map();
+    for (const chainId of chainIds) {
+        const config = this.getChainPollingConfig(chainId);
+        const existing = intervalGroups.get(config.interval) || [];
+        existing.push(chainId);
+        intervalGroups.set(config.interval, existing);
+    }
+    // Start separate polling loop for each interval group
+    for (const [interval, chainIdsGroup] of intervalGroups) {
+        __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startPollingForInterval).call(this, interval, chainIdsGroup, immediate);
+    }
+}, _TokenBalancesController_startPollingForInterval = function _TokenBalancesController_startPollingForInterval(interval, chainIds, immediate = true) {
+    const pollFunction = async () => {
+        if (!__classPrivateFieldGet(this, _TokenBalancesController_isControllerPollingActive, "f")) {
+            return;
+        }
+        try {
+            await this._executePoll({ chainIds });
+        }
+        catch (error) {
+            console.warn(`Polling failed for chains ${chainIds.join(', ')} with interval ${interval}:`, error);
+        }
+    };
+    // Poll immediately first if requested
+    if (immediate) {
+        pollFunction().catch((error) => {
+            console.warn(`Immediate polling failed for chains ${chainIds.join(', ')}:`, error);
+        });
+    }
+    // Then start regular interval polling
+    __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_setPollingTimer).call(this, interval, chainIds, pollFunction);
+}, _TokenBalancesController_setPollingTimer = function _TokenBalancesController_setPollingTimer(interval, chainIds, pollFunction) {
+    // Clear any existing timer for this interval first
+    const existingTimer = __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").get(interval);
+    if (existingTimer) {
+        clearInterval(existingTimer);
+    }
+    const timer = setInterval(() => {
+        pollFunction().catch((error) => {
+            console.warn(`Interval polling failed for chains ${chainIds.join(', ')}:`, error);
+        });
+    }, interval);
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").set(interval, timer);
 };
 exports.default = TokenBalancesController;
 //# sourceMappingURL=TokenBalancesController.cjs.map

package/dist/TokenBalancesController.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"TokenBalancesController.cjs","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,wDAAwD;AAUxD,iEAKoC;AAQpC,qEAA+E;AAM/E,2CAAoD;AACpD,iCAAgC;AAChC,mCAAiC;AAMjC,6EAAiF;AACjF,gGAI4D;AAC5D,+EAAsE;AAUtE,MAAM,UAAU,GAAG,yBAAkC,CAAC;AACtD,MAAM,mBAAmB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEjD,MAAM,QAAQ,GAAG;IACf,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;CACnD,CAAC;AAqEF,YAAY;AAEZ,+EAA+E;AAC/E,2BAA2B;AAC3B,MAAM,KAAK,GAAG,CAAI,IAAO,EAAE,EAAkB,EAAK,EAAE,CAAC,IAAA,eAAO,EAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAEvE,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAmB,EAAE,CACjD,IAAA,uCAAoB,EAAC,IAAI,CAAoB,CAAC;AAChD,YAAY;AAEZ,+EAA+E;AAC/E,0BAA0B;AAC1B,MAAa,uBAAwB,SAAQ,IAAA,oDAA+B,GAM3E;IASC,YAAY,EACV,SAAS,EACT,QAAQ,GAAG,mBAAmB,EAC9B,KAAK,GAAG,EAAE,EACV,qBAAqB,GAAG,IAAI,EAC5B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GACH;QAC/B,KAAK,CAAC;YACJ,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,QAAQ;YACR,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,GAAG,KAAK,EAAE;SACvC,CAAC,CAAC;;QArBI,4DAA2B;QAE3B,2DAAmC;QAE5C,6CAAiD,EAAE,EAAC;QAEpD,kDAA8D,EAAE,EAAC;QAkExD,+CAAe,CAAC,OAAmB,EAAgB,EAAE;YAC5D,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACtC,wCAAwC,EACxC,eAAe,CAChB,CAAC;YACF,OAAO,IAAI,wBAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,EAAC;QAEO,oDAAoB,CAAC,OAAmB,EAAE,EAAE;YACnD,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9B,wCAAwC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC,EAAC;QAkKO,mDAAmB,KAAK,EAAE,KAA4B,EAAE,EAAE;YACjE,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAc,CAAC;YACjD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,kBAAkB,CAAC,GAAG,CAAC,OAAqB,CAAC,CAAC;iBAC/C;aACF;YAED,iFAAiF;YACjF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC;aACxC,CAAC,CAAC;YAEH,2FAA2F;YAC3F,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvE,MAAM,EAAE,GAAG,OAAqB,CAAC;gBAEjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;wBAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;wBACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEtD,mDAAmD;gBACnD,MAAM,cAAc,GAClB,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;oBAClD,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;gBAElE,2EAA2E;gBAC3E,OAAO,cAAc,IAAI,CAAC,CAAC,eAAe,IAAI,YAAY,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE;gBACjC,kDAAkD;gBAClD,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;gBAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;gBAC/C,OAAO;aACR;YAED,2DAA2D;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE;oBACtC,MAAM,EAAE,GAAG,OAAqB,CAAC;oBACjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;4BACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAEtD,IACE,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;wBAClD,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,EAC/D;wBACA,IAAI,YAAY,EAAE;4BAChB,yDAAyD;4BACzD,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;yBAClB;6BAAM,IAAI,eAAe,EAAE;4BAC1B,0EAA0E;4BAC1E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;gCAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;gCAC9C,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oCACrC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;oCACrC,UAAU,GAAG,IAAI,CAAC;iCACnB;6BACF;yBACF;qBACF;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;YAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;YAE/C,wGAAwG;YACxG,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE;gBACjC,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACzD,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,KAAmB,EAAE,EAAE;YACnD,sEAAsE;YACtE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAClD,CAAC;YAEF,gDAAgD;YAChD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;iBACnC;aACF;YAED,kCAAkC;YAClC,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAC7D,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC3C,CAAC;YAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9B,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAChB,2DAA2D;oBAC3D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;wBAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;wBAC9C,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE;4BAC5C,MAAM,UAAU,GAAG,cAA4B,CAAC;4BAChD,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE;gCAC7C,OAAO,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;6BAChD;yBACF;qBACF;gBACH,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,IAAY,EAAE,EAAE;YAC5C,IAAI,CAAC,IAAA,yBAAiB,EAAC,IAAI,CAAC,IAAI,CAAC,IAAA,oCAAiB,EAAC,IAAI,CAAC,EAAE;gBACxD,OAAO;aACR;YACD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,OAAO,CAAC,CAAC,aAAa,CAAC,IAAuB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,EAAC;QAxXA,uBAAA,IAAI,6CAAqB,qBAAqB,MAAA,CAAC;QAE/C,+CAA+C;QAC/C,uBAAA,IAAI,4CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,+CAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,4CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,uCAAiB,CAAC,uBAAA,IAAI,4CAAa,EAAE,uBAAA,IAAI,iDAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;gBACtE,SAAS,EAAE,uBAAA,IAAI,0CAAW;gBAC1B,iBAAiB,EAAE,uBAAA,IAAI,+CAAgB;aACxC,CAAC,CAAC;SACJ,MAAA,CAAC;QAEF,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,sCAAsC;QACtC,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAChE,2BAA2B,CAC5B,CAAC;QACF,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,2CAAmB,iBAAiB,MAAA,CAAC;QAEzC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,8BAA8B,EAC9B,CAAC,WAAkC,EAAE,EAAE;YACrC,uBAAA,IAAI,gDAAiB,MAArB,IAAI,EAAkB,WAAW,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjD,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,kCAAkC,EAClC,uBAAA,IAAI,iDAAkB,CACvB,CAAC;IACJ,CAAC;IAoCD,KAAK,CAAC,YAAY,CAAC,EAAE,QAAQ,EAA8B;QACzD,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAE,QAAQ,KAAkC,EAAE;QACjE,MAAM,YAAY,GAAG,QAAQ,IAAI,uBAAA,IAAI,uFAAoB,MAAxB,IAAI,CAAsB,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,OAAO;SACR;QAED,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACrD,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QAEF,MAAM,UAAU,GAAuB,EAAE,CAAC;QAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;QAExC,oEAAoE;QACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,gDAAiB,EAAE;YAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;gBAC3B,SAAS;aACV;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;oBACnC,QAAQ,EAAE,eAAe;oBACzB,gBAAgB,EAAE,uBAAA,IAAI,iDAAkB;oBACxC,eAAe,EAAE,QAA2B;oBAC5C,WAAW;iBACZ,CAAC,CAAC;gBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;oBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;oBAC7B,iDAAiD;oBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;iBACH;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;gBACF,sCAAsC;aACvC;YAED,iDAAiD;YACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;gBAChC,MAAM;aACP;SACF;QAED,sCAAsC;QACtC,MAAM,iBAAiB,GAAG,uBAAA,IAAI,iDAAkB;YAC9C,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAA0B,CAAC;YACtD,CAAC,CAAC,CAAC,QAA2B,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YAC7B,mEAAmE;YACnE,+CAA+C;YAC/C,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE;gBAClC,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE;oBACvC,mCAAmC;oBACnC,MAAM,WAAW,GAAG,uBAAA,IAAI,0CAAW,CAAC,OAAO,CAAC,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE;wBAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACzC,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;oBAED,2CAA2C;oBAC3C,MAAM,mBAAmB,GAAG,uBAAA,IAAI,+CAAgB,CAAC,OAAO,CAAC,CAAC;oBAC1D,IAAI,mBAAmB,EAAE,CAAC,OAAO,CAAC,EAAE;wBAClC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACjD,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;iBACF;aACF;YAED,2DAA2D;YAC3D,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;;gBACjE,IAAI,OAAO,IAAI,KAAK,KAAK,SAAS,EAAE;oBAClC,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClE,IAAA,wBAAK,EAAC,KAAK,CAAC,CAAC;iBAChB;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAA,gBAAO,EAAC,IAAI,EAAE,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAExB,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,YAAY,CAC7C,CAAC;YAEF,kFAAkF;YAClF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACtD,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAA,0BAAO,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBACxD,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,cAAc,CACf,CAAC;aACH;YAED,+CAA+C;YAC/C,MAAM,wBAAwB,GAAG,MAAM,CAAC,MAAM,CAC5C,8DAAmC,CACpC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YAEpC,uFAAuF;YACvF,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7C,OAAO,CACL,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,KAAK,KAAK,YAAY;oBACxB,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CACzD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC5D,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAA,wBAAK,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBAC5D,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,oBAAoB,CACrB,CAAC;aACH;SACF;IACH,CAAC;IAED,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;CAiJF;AAvZD,0DAuZC;;IAhVG,OAAO;QACL,GAAG,IAAI,GAAG,CAAC;YACT,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC;YAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC;SACrC,CAAC;KACa,CAAC;AACpB,CAAC;AA4UH,kBAAe,uBAAuB,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  BNToHex,\n  isValidHexAddress,\n  toChecksumHexAddress,\n  toHex,\n} from '@metamask/controller-utils';\nimport type { KeyringControllerAccountRemovedEvent } from '@metamask/keyring-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n  NetworkState,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type {\n  PreferencesControllerGetStateAction,\n  PreferencesControllerStateChangeEvent,\n} from '@metamask/preferences-controller';\nimport type { Hex } from '@metamask/utils';\nimport { isStrictHexString } from '@metamask/utils';\nimport { produce } from 'immer';\nimport { isEqual } from 'lodash';\n\nimport type {\n  AccountTrackerUpdateNativeBalancesAction,\n  AccountTrackerUpdateStakedBalancesAction,\n} from './AccountTrackerController';\nimport { STAKING_CONTRACT_ADDRESS_BY_CHAINID } from './AssetsContractController';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\nimport { RpcBalanceFetcher } from './rpc-service/rpc-balance-fetcher';\nimport type {\n  TokensControllerGetStateAction,\n  TokensControllerState,\n  TokensControllerStateChangeEvent,\n} from './TokensController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst CONTROLLER = 'TokenBalancesController' as const;\nconst DEFAULT_INTERVAL_MS = 180_000; // 3 minutes\n\nconst metadata = {\n  tokenBalances: { persist: true, anonymous: false },\n};\n\n// account → chain → token → balance\nexport type TokenBalances = Record<\n  ChecksumAddress,\n  Record<ChainIdHex, Record<ChecksumAddress, Hex>>\n>;\n\nexport type TokenBalancesControllerState = {\n  tokenBalances: TokenBalances;\n};\n\nexport type TokenBalancesControllerGetStateAction = ControllerGetStateAction<\n  typeof CONTROLLER,\n  TokenBalancesControllerState\n>;\n\nexport type TokenBalancesControllerActions =\n  TokenBalancesControllerGetStateAction;\n\nexport type TokenBalancesControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;\n\nexport type NativeBalanceEvent = {\n  type: `${typeof CONTROLLER}:updatedNativeBalance`;\n  payload: unknown[];\n};\n\nexport type TokenBalancesControllerEvents =\n  | TokenBalancesControllerStateChangeEvent\n  | NativeBalanceEvent;\n\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | TokensControllerGetStateAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | AccountsControllerListAccountsAction\n  | AccountTrackerUpdateNativeBalancesAction\n  | AccountTrackerUpdateStakedBalancesAction;\n\nexport type AllowedEvents =\n  | TokensControllerStateChangeEvent\n  | PreferencesControllerStateChangeEvent\n  | NetworkControllerStateChangeEvent\n  | KeyringControllerAccountRemovedEvent;\n\nexport type TokenBalancesControllerMessenger = RestrictedMessenger<\n  typeof CONTROLLER,\n  TokenBalancesControllerActions | AllowedActions,\n  TokenBalancesControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\nexport type TokenBalancesControllerOptions = {\n  messenger: TokenBalancesControllerMessenger;\n  interval?: number;\n  state?: Partial<TokenBalancesControllerState>;\n  /** When `true`, balances for *all* known accounts are queried. */\n  queryMultipleAccounts?: boolean;\n  /** Enable Accounts‑API strategy (if supported chain). */\n  useAccountsAPI?: boolean;\n  /** Disable external HTTP calls (privacy / offline mode). */\n  allowExternalServices?: () => boolean;\n  /** Custom logger. */\n  log?: (...args: unknown[]) => void;\n};\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Helper utilities\nconst draft = <T>(base: T, fn: (d: T) => void): T => produce(base, fn);\n\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\nconst checksum = (addr: string): ChecksumAddress =>\n  toChecksumHexAddress(addr) as ChecksumAddress;\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Main controller\nexport class TokenBalancesController extends StaticIntervalPollingController<{\n  chainIds: ChainIdHex[];\n}>()<\n  typeof CONTROLLER,\n  TokenBalancesControllerState,\n  TokenBalancesControllerMessenger\n> {\n  readonly #queryAllAccounts: boolean;\n\n  readonly #balanceFetchers: BalanceFetcher[];\n\n  #allTokens: TokensControllerState['allTokens'] = {};\n\n  #detectedTokens: TokensControllerState['allDetectedTokens'] = {};\n\n  constructor({\n    messenger,\n    interval = DEFAULT_INTERVAL_MS,\n    state = {},\n    queryMultipleAccounts = true,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: TokenBalancesControllerOptions) {\n    super({\n      name: CONTROLLER,\n      messenger,\n      metadata,\n      state: { tokenBalances: {}, ...state },\n    });\n\n    this.#queryAllAccounts = queryMultipleAccounts;\n\n    // Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new RpcBalanceFetcher(this.#getProvider, this.#getNetworkClient, () => ({\n        allTokens: this.#allTokens,\n        allDetectedTokens: this.#detectedTokens,\n      })),\n    ];\n\n    this.setIntervalLength(interval);\n\n    // initial token state & subscriptions\n    const { allTokens, allDetectedTokens } = this.messagingSystem.call(\n      'TokensController:getState',\n    );\n    this.#allTokens = allTokens;\n    this.#detectedTokens = allDetectedTokens;\n\n    this.messagingSystem.subscribe(\n      'TokensController:stateChange',\n      (tokensState: TokensControllerState) => {\n        this.#onTokensChanged(tokensState).catch((error) => {\n          console.warn('Error handling token state change:', error);\n        });\n      },\n    );\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      this.#onNetworkChanged,\n    );\n    this.messagingSystem.subscribe(\n      'KeyringController:accountRemoved',\n      this.#onAccountRemoved,\n    );\n  }\n\n  #chainIdsWithTokens(): ChainIdHex[] {\n    return [\n      ...new Set([\n        ...Object.keys(this.#allTokens),\n        ...Object.keys(this.#detectedTokens),\n      ]),\n    ] as ChainIdHex[];\n  }\n\n  readonly #getProvider = (chainId: ChainIdHex): Web3Provider => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    const client = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n    return new Web3Provider(client.provider);\n  };\n\n  readonly #getNetworkClient = (chainId: ChainIdHex) => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    return this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n  };\n\n  async _executePoll({ chainIds }: { chainIds: ChainIdHex[] }) {\n    await this.updateBalances({ chainIds });\n  }\n\n  async updateBalances({ chainIds }: { chainIds?: ChainIdHex[] } = {}) {\n    const targetChains = chainIds ?? this.#chainIdsWithTokens();\n    if (!targetChains.length) {\n      return;\n    }\n\n    const { address: selected } = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const aggregated: ProcessedBalance[] = [];\n    let remainingChains = [...targetChains];\n\n    // Try each fetcher in order, removing successfully processed chains\n    for (const fetcher of this.#balanceFetchers) {\n      const supportedChains = remainingChains.filter((c) =>\n        fetcher.supports(c),\n      );\n      if (!supportedChains.length) {\n        continue;\n      }\n\n      try {\n        const balances = await fetcher.fetch({\n          chainIds: supportedChains,\n          queryAllAccounts: this.#queryAllAccounts,\n          selectedAccount: selected as ChecksumAddress,\n          allAccounts,\n        });\n\n        if (balances && balances.length > 0) {\n          aggregated.push(...balances);\n          // Remove chains that were successfully processed\n          const processedChains = new Set(balances.map((b) => b.chainId));\n          remainingChains = remainingChains.filter(\n            (chain) => !processedChains.has(chain),\n          );\n        }\n      } catch (error) {\n        console.warn(\n          `Balance fetcher failed for chains ${supportedChains.join(', ')}: ${String(error)}`,\n        );\n        // Continue to next fetcher (fallback)\n      }\n\n      // If all chains have been processed, break early\n      if (remainingChains.length === 0) {\n        break;\n      }\n    }\n\n    // Determine which accounts to process\n    const accountsToProcess = this.#queryAllAccounts\n      ? allAccounts.map((a) => a.address as ChecksumAddress)\n      : [selected as ChecksumAddress];\n\n    const prev = this.state;\n    const next = draft(prev, (d) => {\n      // First, initialize all tokens from allTokens state with balance 0\n      // for the accounts and chains we're processing\n      for (const chainId of targetChains) {\n        for (const account of accountsToProcess) {\n          // Initialize tokens from allTokens\n          const chainTokens = this.#allTokens[chainId];\n          if (chainTokens?.[account]) {\n            Object.values(chainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n\n          // Initialize tokens from allDetectedTokens\n          const detectedChainTokens = this.#detectedTokens[chainId];\n          if (detectedChainTokens?.[account]) {\n            Object.values(detectedChainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n        }\n      }\n\n      // Then update with actual fetched balances where available\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          ((d.tokenBalances[account] ??= {})[chainId] ??= {})[checksum(token)] =\n            toHex(value);\n        }\n      });\n    });\n\n    if (!isEqual(prev, next)) {\n      this.update(() => next);\n\n      const nativeBalances = aggregated.filter(\n        (r) => r.success && r.token === ZERO_ADDRESS,\n      );\n\n      // Update native token balances in a single batch operation for better performance\n      if (nativeBalances.length > 0) {\n        const balanceUpdates = nativeBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          balance: balance.value ? BNToHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateNativeBalances',\n          balanceUpdates,\n        );\n      }\n\n      // Get staking contract addresses for filtering\n      const stakingContractAddresses = Object.values(\n        STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n      ).map((addr) => addr.toLowerCase());\n\n      // Filter and update staked balances in a single batch operation for better performance\n      const stakedBalances = aggregated.filter((r) => {\n        return (\n          r.success &&\n          r.token !== ZERO_ADDRESS &&\n          stakingContractAddresses.includes(r.token.toLowerCase())\n        );\n      });\n\n      if (stakedBalances.length > 0) {\n        const stakedBalanceUpdates = stakedBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          stakedBalance: balance.value ? toHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateStakedBalances',\n          stakedBalanceUpdates,\n        );\n      }\n    }\n  }\n\n  resetState() {\n    this.update(() => ({ tokenBalances: {} }));\n  }\n\n  readonly #onTokensChanged = async (state: TokensControllerState) => {\n    const changed: ChainIdHex[] = [];\n    let hasChanges = false;\n\n    // Get chains that have existing balances\n    const chainsWithBalances = new Set<ChainIdHex>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const chainId of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        chainsWithBalances.add(chainId as ChainIdHex);\n      }\n    }\n\n    // Only process chains that are explicitly mentioned in the incoming state change\n    const incomingChainIds = new Set([\n      ...Object.keys(state.allTokens),\n      ...Object.keys(state.allDetectedTokens),\n    ]);\n\n    // Only proceed if there are actual changes to chains that have balances or are being added\n    const relevantChainIds = Array.from(incomingChainIds).filter((chainId) => {\n      const id = chainId as ChainIdHex;\n\n      const hasTokensNow =\n        (state.allTokens[id] && Object.keys(state.allTokens[id]).length > 0) ||\n        (state.allDetectedTokens[id] &&\n          Object.keys(state.allDetectedTokens[id]).length > 0);\n      const hadTokensBefore =\n        (this.#allTokens[id] && Object.keys(this.#allTokens[id]).length > 0) ||\n        (this.#detectedTokens[id] &&\n          Object.keys(this.#detectedTokens[id]).length > 0);\n\n      // Check if there's an actual change in token state\n      const hasTokenChange =\n        !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n        !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id]);\n\n      // Process chains that have actual changes OR are new chains getting tokens\n      return hasTokenChange || (!hadTokensBefore && hasTokensNow);\n    });\n\n    if (relevantChainIds.length === 0) {\n      // No relevant changes, just update internal state\n      this.#allTokens = state.allTokens;\n      this.#detectedTokens = state.allDetectedTokens;\n      return;\n    }\n\n    // Handle both cleanup and updates in a single state update\n    this.update((s) => {\n      for (const chainId of relevantChainIds) {\n        const id = chainId as ChainIdHex;\n        const hasTokensNow =\n          (state.allTokens[id] &&\n            Object.keys(state.allTokens[id]).length > 0) ||\n          (state.allDetectedTokens[id] &&\n            Object.keys(state.allDetectedTokens[id]).length > 0);\n        const hadTokensBefore =\n          (this.#allTokens[id] &&\n            Object.keys(this.#allTokens[id]).length > 0) ||\n          (this.#detectedTokens[id] &&\n            Object.keys(this.#detectedTokens[id]).length > 0);\n\n        if (\n          !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n          !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id])\n        ) {\n          if (hasTokensNow) {\n            // Chain still has tokens - mark for async balance update\n            changed.push(id);\n          } else if (hadTokensBefore) {\n            // Chain had tokens before but doesn't now - clean up balances immediately\n            for (const address of Object.keys(s.tokenBalances)) {\n              const addressKey = address as ChecksumAddress;\n              if (s.tokenBalances[addressKey]?.[id]) {\n                s.tokenBalances[addressKey][id] = {};\n                hasChanges = true;\n              }\n            }\n          }\n        }\n      }\n    });\n\n    this.#allTokens = state.allTokens;\n    this.#detectedTokens = state.allDetectedTokens;\n\n    // Only update balances for chains that still have tokens (and only if we haven't already updated state)\n    if (changed.length && !hasChanges) {\n      this.updateBalances({ chainIds: changed }).catch((error) => {\n        console.warn('Error updating balances after token change:', error);\n      });\n    }\n  };\n\n  readonly #onNetworkChanged = (state: NetworkState) => {\n    // Check if any networks were removed by comparing with previous state\n    const currentNetworks = new Set(\n      Object.keys(state.networkConfigurationsByChainId),\n    );\n\n    // Get all networks that currently have balances\n    const networksWithBalances = new Set<string>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const network of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        networksWithBalances.add(network);\n      }\n    }\n\n    // Find networks that were removed\n    const removedNetworks = Array.from(networksWithBalances).filter(\n      (network) => !currentNetworks.has(network),\n    );\n\n    if (removedNetworks.length > 0) {\n      this.update((s) => {\n        // Remove balances for all accounts on the deleted networks\n        for (const address of Object.keys(s.tokenBalances)) {\n          const addressKey = address as ChecksumAddress;\n          for (const removedNetwork of removedNetworks) {\n            const networkKey = removedNetwork as ChainIdHex;\n            if (s.tokenBalances[addressKey]?.[networkKey]) {\n              delete s.tokenBalances[addressKey][networkKey];\n            }\n          }\n        }\n      });\n    }\n  };\n\n  readonly #onAccountRemoved = (addr: string) => {\n    if (!isStrictHexString(addr) || !isValidHexAddress(addr)) {\n      return;\n    }\n    this.update((s) => {\n      delete s.tokenBalances[addr as ChecksumAddress];\n    });\n  };\n}\n\nexport default TokenBalancesController;\n"]}
+{"version":3,"file":"TokenBalancesController.cjs","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,wDAAwD;AAUxD,iEAKoC;AAQpC,qEAA+E;AAM/E,2CAAoD;AACpD,iCAAgC;AAChC,mCAAiC;AAMjC,6EAAiF;AACjF,gGAI4D;AAC5D,+EAAsE;AAUtE,MAAM,UAAU,GAAG,yBAAkC,CAAC;AACtD,MAAM,mBAAmB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEjD,MAAM,QAAQ,GAAG;IACf,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;CACnD,CAAC;AA8FF,YAAY;AAEZ,+EAA+E;AAC/E,2BAA2B;AAC3B,MAAM,KAAK,GAAG,CAAI,IAAO,EAAE,EAAkB,EAAK,EAAE,CAAC,IAAA,eAAO,EAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAEvE,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAmB,EAAE,CACjD,IAAA,uCAAoB,EAAC,IAAI,CAAoB,CAAC;AAChD,YAAY;AAEZ,+EAA+E;AAC/E,0BAA0B;AAC1B,MAAa,uBAAwB,SAAQ,IAAA,oDAA+B,GAM3E;IAwBC,YAAY,EACV,SAAS,EACT,QAAQ,GAAG,mBAAmB,EAC9B,qBAAqB,GAAG,EAAE,EAC1B,KAAK,GAAG,EAAE,EACV,qBAAqB,GAAG,IAAI,EAC5B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GACH;QAC/B,KAAK,CAAC;YACJ,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,QAAQ;YACR,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,GAAG,KAAK,EAAE;SACvC,CAAC,CAAC;;QArCI,4DAA2B;QAE3B,2DAAmC;QAE5C,6CAAiD,EAAE,EAAC;QAEpD,kDAA8D,EAAE,EAAC;QAEjE,yEAAyE;QAChE,2DAAyB;QAElC,sCAAsC;QAC7B,8DAA4D;QAErE,gDAAgD;QACvC,yDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,kDAAkD;QAClD,6DAA6B,KAAK,EAAC;QAEnC,mEAAmE;QACnE,qDAAmC,EAAE,EAAC;QAgF7B,+CAAe,CAAC,OAAmB,EAAgB,EAAE;YAC5D,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACtC,wCAAwC,EACxC,eAAe,CAChB,CAAC;YACF,OAAO,IAAI,wBAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,EAAC;QAEO,oDAAoB,CAAC,OAAmB,EAAE,EAAE;YACnD,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9B,wCAAwC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC,EAAC;QA+TO,mDAAmB,KAAK,EAAE,KAA4B,EAAE,EAAE;YACjE,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAc,CAAC;YACjD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,kBAAkB,CAAC,GAAG,CAAC,OAAqB,CAAC,CAAC;iBAC/C;aACF;YAED,iFAAiF;YACjF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC;aACxC,CAAC,CAAC;YAEH,2FAA2F;YAC3F,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvE,MAAM,EAAE,GAAG,OAAqB,CAAC;gBAEjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;wBAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;wBACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEtD,mDAAmD;gBACnD,MAAM,cAAc,GAClB,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;oBAClD,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;gBAElE,2EAA2E;gBAC3E,OAAO,cAAc,IAAI,CAAC,CAAC,eAAe,IAAI,YAAY,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE;gBACjC,kDAAkD;gBAClD,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;gBAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;gBAC/C,OAAO;aACR;YAED,2DAA2D;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE;oBACtC,MAAM,EAAE,GAAG,OAAqB,CAAC;oBACjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;4BACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAEtD,IACE,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;wBAClD,CAAC,IAAA,gBAAO,EAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,EAC/D;wBACA,IAAI,YAAY,EAAE;4BAChB,yDAAyD;4BACzD,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;yBAClB;6BAAM,IAAI,eAAe,EAAE;4BAC1B,0EAA0E;4BAC1E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;gCAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;gCAC9C,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oCACrC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;oCACrC,UAAU,GAAG,IAAI,CAAC;iCACnB;6BACF;yBACF;qBACF;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;YAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;YAE/C,wGAAwG;YACxG,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE;gBACjC,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACzD,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,KAAmB,EAAE,EAAE;YACnD,sEAAsE;YACtE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAClD,CAAC;YAEF,gDAAgD;YAChD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;iBACnC;aACF;YAED,kCAAkC;YAClC,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAC7D,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC3C,CAAC;YAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9B,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAChB,2DAA2D;oBAC3D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;wBAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;wBAC9C,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE;4BAC5C,MAAM,UAAU,GAAG,cAA4B,CAAC;4BAChD,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE;gCAC7C,OAAO,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;6BAChD;yBACF;qBACF;gBACH,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,IAAY,EAAE,EAAE;YAC5C,IAAI,CAAC,IAAA,yBAAiB,EAAC,IAAI,CAAC,IAAI,CAAC,IAAA,oCAAiB,EAAC,IAAI,CAAC,EAAE;gBACxD,OAAO;aACR;YACD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,OAAO,CAAC,CAAC,aAAa,CAAC,IAAuB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,EAAC;QAliBA,uBAAA,IAAI,6CAAqB,qBAAqB,MAAA,CAAC;QAC/C,uBAAA,IAAI,4CAAoB,QAAQ,MAAA,CAAC;QACjC,uBAAA,IAAI,+CAAuB,EAAE,GAAG,qBAAqB,EAAE,MAAA,CAAC;QAExD,+CAA+C;QAC/C,uBAAA,IAAI,4CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,+CAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,4CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,uCAAiB,CAAC,uBAAA,IAAI,4CAAa,EAAE,uBAAA,IAAI,iDAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;gBACtE,SAAS,EAAE,uBAAA,IAAI,0CAAW;gBAC1B,iBAAiB,EAAE,uBAAA,IAAI,+CAAgB;aACxC,CAAC,CAAC;SACJ,MAAA,CAAC;QAEF,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,sCAAsC;QACtC,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAChE,2BAA2B,CAC5B,CAAC;QACF,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,2CAAmB,iBAAiB,MAAA,CAAC;QAEzC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,8BAA8B,EAC9B,CAAC,WAAkC,EAAE,EAAE;YACrC,uBAAA,IAAI,gDAAiB,MAArB,IAAI,EAAkB,WAAW,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjD,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,kCAAkC,EAClC,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QAEF,wDAAwD;QACxD,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,mDAAmD,EACnD,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,+CAA+C,EAC/C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;IACJ,CAAC;IAoCD;;;;;OAKG;IACM,aAAa,CAAC,EAAE,QAAQ,EAA8B;QAC7D,uEAAuE;QACvE,uBAAA,IAAI,8CAAsB,CAAC,GAAG,QAAQ,CAAC,MAAA,CAAC;QACxC,uBAAA,IAAI,sDAA8B,IAAI,MAAA,CAAC;QACvC,uBAAA,IAAI,8FAA2B,MAA/B,IAAI,EAA4B,QAAQ,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IAkGD;;OAEG;IACM,+BAA+B;QACtC,uBAAA,IAAI,sDAA8B,KAAK,MAAA,CAAC;QACxC,uBAAA,IAAI,8CAAsB,EAAE,MAAA,CAAC,CAAC,sCAAsC;QACpE,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;IACtC,CAAC;IAED;;;;;OAKG;IACH,qBAAqB,CAAC,OAAmB;QACvC,OAAO,CACL,uBAAA,IAAI,mDAAoB,CAAC,OAAO,CAAC,IAAI;YACnC,QAAQ,EAAE,uBAAA,IAAI,gDAAiB;SAChC,CACF,CAAC;IACJ,CAAC;IAEQ,KAAK,CAAC,YAAY,CAAC,EAAE,QAAQ,EAA8B;QAClE,kFAAkF;QAClF,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;;OAMG;IACH,yBAAyB,CACvB,OAA+C,EAC/C,UAA4C,EAAE,eAAe,EAAE,IAAI,EAAE;QAErE,MAAM,CAAC,MAAM,CAAC,uBAAA,IAAI,mDAAoB,EAAE,OAAO,CAAC,CAAC;QAEjD,sEAAsE;QACtE,IAAI,uBAAA,IAAI,0DAA2B,EAAE;YACnC,8EAA8E;YAC9E,uBAAA,IAAI,8FAA2B,MAA/B,IAAI,EACF,uBAAA,IAAI,kDAAmB,EACvB,OAAO,CAAC,eAAe,CACxB,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAE,QAAQ,KAAkC,EAAE;QACjE,MAAM,YAAY,GAAG,QAAQ,IAAI,uBAAA,IAAI,uFAAoB,MAAxB,IAAI,CAAsB,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,OAAO;SACR;QAED,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACrD,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QAEF,MAAM,UAAU,GAAuB,EAAE,CAAC;QAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;QAExC,oEAAoE;QACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,gDAAiB,EAAE;YAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;gBAC3B,SAAS;aACV;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;oBACnC,QAAQ,EAAE,eAAe;oBACzB,gBAAgB,EAAE,uBAAA,IAAI,iDAAkB;oBACxC,eAAe,EAAE,QAA2B;oBAC5C,WAAW;iBACZ,CAAC,CAAC;gBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;oBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;oBAC7B,iDAAiD;oBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;iBACH;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;gBACF,sCAAsC;aACvC;YAED,iDAAiD;YACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;gBAChC,MAAM;aACP;SACF;QAED,sCAAsC;QACtC,MAAM,iBAAiB,GAAG,uBAAA,IAAI,iDAAkB;YAC9C,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAA0B,CAAC;YACtD,CAAC,CAAC,CAAC,QAA2B,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YAC7B,mEAAmE;YACnE,+CAA+C;YAC/C,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE;gBAClC,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE;oBACvC,mCAAmC;oBACnC,MAAM,WAAW,GAAG,uBAAA,IAAI,0CAAW,CAAC,OAAO,CAAC,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE;wBAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACzC,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;oBAED,2CAA2C;oBAC3C,MAAM,mBAAmB,GAAG,uBAAA,IAAI,+CAAgB,CAAC,OAAO,CAAC,CAAC;oBAC1D,IAAI,mBAAmB,EAAE,CAAC,OAAO,CAAC,EAAE;wBAClC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACjD,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;iBACF;aACF;YAED,2DAA2D;YAC3D,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;;gBACjE,IAAI,OAAO,IAAI,KAAK,KAAK,SAAS,EAAE;oBAClC,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClE,IAAA,wBAAK,EAAC,KAAK,CAAC,CAAC;iBAChB;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAA,gBAAO,EAAC,IAAI,EAAE,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAExB,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,YAAY,CAC7C,CAAC;YAEF,kFAAkF;YAClF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACtD,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAA,0BAAO,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBACxD,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,cAAc,CACf,CAAC;aACH;YAED,+CAA+C;YAC/C,MAAM,wBAAwB,GAAG,MAAM,CAAC,MAAM,CAC5C,8DAAmC,CACpC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YAEpC,uFAAuF;YACvF,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7C,OAAO,CACL,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,KAAK,KAAK,YAAY;oBACxB,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CACzD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC5D,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAA,wBAAK,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBAC5D,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,oBAAoB,CACrB,CAAC;aACH;SACF;IACH,CAAC;IAED,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;IAkJD;;OAEG;IACM,OAAO;QACd,uBAAA,IAAI,sDAA8B,KAAK,MAAA,CAAC;QACxC,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;QAEpC,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAC1C,mDAAmD,CACpD,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAC1C,+CAA+C,CAChD,CAAC;QAEF,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;AApmBD,0DAomBC;;IAhgBG,OAAO;QACL,GAAG,IAAI,GAAG,CAAC;YACT,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC;YAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC;SACrC,CAAC;KACa,CAAC;AACpB,CAAC,mHA8C0B,QAAsB,EAAE,SAAS,GAAG,IAAI;IACjE,oCAAoC;IACpC,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;IAEpC,0CAA0C;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEvD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3D,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;KAC/C;IAED,sDAAsD;IACtD,KAAK,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,IAAI,cAAc,EAAE;QACtD,uBAAA,IAAI,4FAAyB,MAA7B,IAAI,EAA0B,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;KACnE;AACH,CAAC,+GAUC,QAAgB,EAChB,QAAsB,EACtB,SAAS,GAAG,IAAI;IAEhB,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC9B,IAAI,CAAC,uBAAA,IAAI,0DAA2B,EAAE;YACpC,OAAO;SACR;QACD,IAAI;YACF,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;SACvC;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,IAAI,CACV,6BAA6B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,QAAQ,GAAG,EAC7E,KAAK,CACN,CAAC;SACH;IACH,CAAC,CAAC;IAEF,sCAAsC;IACtC,IAAI,SAAS,EAAE;QACb,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7B,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAC7D,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;KACJ;IAED,sCAAsC;IACtC,uBAAA,IAAI,oFAAiB,MAArB,IAAI,EAAkB,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;AAC1D,CAAC,+FAUC,QAAgB,EAChB,QAAsB,EACtB,YAAiC;IAEjC,mDAAmD;IACnD,MAAM,aAAa,GAAG,uBAAA,IAAI,sDAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAChE,IAAI,aAAa,EAAE;QACjB,aAAa,CAAC,aAAa,CAAC,CAAC;KAC9B;IAED,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7B,OAAO,CAAC,IAAI,CACV,sCAAsC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAC5D,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,QAAQ,CAAC,CAAC;IACb,uBAAA,IAAI,sDAAuB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;AACnD,CAAC;AAsXH,kBAAe,uBAAuB,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  BNToHex,\n  isValidHexAddress,\n  toChecksumHexAddress,\n  toHex,\n} from '@metamask/controller-utils';\nimport type { KeyringControllerAccountRemovedEvent } from '@metamask/keyring-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n  NetworkState,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type {\n  PreferencesControllerGetStateAction,\n  PreferencesControllerStateChangeEvent,\n} from '@metamask/preferences-controller';\nimport type { Hex } from '@metamask/utils';\nimport { isStrictHexString } from '@metamask/utils';\nimport { produce } from 'immer';\nimport { isEqual } from 'lodash';\n\nimport type {\n  AccountTrackerUpdateNativeBalancesAction,\n  AccountTrackerUpdateStakedBalancesAction,\n} from './AccountTrackerController';\nimport { STAKING_CONTRACT_ADDRESS_BY_CHAINID } from './AssetsContractController';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\nimport { RpcBalanceFetcher } from './rpc-service/rpc-balance-fetcher';\nimport type {\n  TokensControllerGetStateAction,\n  TokensControllerState,\n  TokensControllerStateChangeEvent,\n} from './TokensController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst CONTROLLER = 'TokenBalancesController' as const;\nconst DEFAULT_INTERVAL_MS = 180_000; // 3 minutes\n\nconst metadata = {\n  tokenBalances: { persist: true, anonymous: false },\n};\n\n// account → chain → token → balance\nexport type TokenBalances = Record<\n  ChecksumAddress,\n  Record<ChainIdHex, Record<ChecksumAddress, Hex>>\n>;\n\nexport type TokenBalancesControllerState = {\n  tokenBalances: TokenBalances;\n};\n\nexport type TokenBalancesControllerGetStateAction = ControllerGetStateAction<\n  typeof CONTROLLER,\n  TokenBalancesControllerState\n>;\n\nexport type TokenBalancesControllerUpdateChainPollingConfigsAction = {\n  type: `TokenBalancesController:updateChainPollingConfigs`;\n  handler: TokenBalancesController['updateChainPollingConfigs'];\n};\n\nexport type TokenBalancesControllerGetChainPollingConfigAction = {\n  type: `TokenBalancesController:getChainPollingConfig`;\n  handler: TokenBalancesController['getChainPollingConfig'];\n};\n\nexport type TokenBalancesControllerActions =\n  | TokenBalancesControllerGetStateAction\n  | TokenBalancesControllerUpdateChainPollingConfigsAction\n  | TokenBalancesControllerGetChainPollingConfigAction;\n\nexport type TokenBalancesControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;\n\nexport type NativeBalanceEvent = {\n  type: `${typeof CONTROLLER}:updatedNativeBalance`;\n  payload: unknown[];\n};\n\nexport type TokenBalancesControllerEvents =\n  | TokenBalancesControllerStateChangeEvent\n  | NativeBalanceEvent;\n\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | TokensControllerGetStateAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | AccountsControllerListAccountsAction\n  | AccountTrackerUpdateNativeBalancesAction\n  | AccountTrackerUpdateStakedBalancesAction;\n\nexport type AllowedEvents =\n  | TokensControllerStateChangeEvent\n  | PreferencesControllerStateChangeEvent\n  | NetworkControllerStateChangeEvent\n  | KeyringControllerAccountRemovedEvent;\n\nexport type TokenBalancesControllerMessenger = RestrictedMessenger<\n  typeof CONTROLLER,\n  TokenBalancesControllerActions | AllowedActions,\n  TokenBalancesControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\nexport type ChainPollingConfig = {\n  /** Polling interval in milliseconds for this chain */\n  interval: number;\n};\n\nexport type UpdateChainPollingConfigsOptions = {\n  /** Whether to immediately fetch balances after updating configs (default: true) */\n  immediateUpdate?: boolean;\n};\n\nexport type TokenBalancesControllerOptions = {\n  messenger: TokenBalancesControllerMessenger;\n  /** Default interval for chains not specified in chainPollingIntervals */\n  interval?: number;\n  /** Per-chain polling configuration */\n  chainPollingIntervals?: Record<ChainIdHex, ChainPollingConfig>;\n  state?: Partial<TokenBalancesControllerState>;\n  /** When `true`, balances for *all* known accounts are queried. */\n  queryMultipleAccounts?: boolean;\n  /** Enable Accounts‑API strategy (if supported chain). */\n  useAccountsAPI?: boolean;\n  /** Disable external HTTP calls (privacy / offline mode). */\n  allowExternalServices?: () => boolean;\n  /** Custom logger. */\n  log?: (...args: unknown[]) => void;\n};\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Helper utilities\nconst draft = <T>(base: T, fn: (d: T) => void): T => produce(base, fn);\n\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\nconst checksum = (addr: string): ChecksumAddress =>\n  toChecksumHexAddress(addr) as ChecksumAddress;\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Main controller\nexport class TokenBalancesController extends StaticIntervalPollingController<{\n  chainIds: ChainIdHex[];\n}>()<\n  typeof CONTROLLER,\n  TokenBalancesControllerState,\n  TokenBalancesControllerMessenger\n> {\n  readonly #queryAllAccounts: boolean;\n\n  readonly #balanceFetchers: BalanceFetcher[];\n\n  #allTokens: TokensControllerState['allTokens'] = {};\n\n  #detectedTokens: TokensControllerState['allDetectedTokens'] = {};\n\n  /** Default polling interval for chains without specific configuration */\n  readonly #defaultInterval: number;\n\n  /** Per-chain polling configuration */\n  readonly #chainPollingConfig: Record<ChainIdHex, ChainPollingConfig>;\n\n  /** Active polling timers grouped by interval */\n  readonly #intervalPollingTimers: Map<number, NodeJS.Timeout> = new Map();\n\n  /** Track if controller-level polling is active */\n  #isControllerPollingActive = false;\n\n  /** Store original chainIds from startPolling to preserve intent */\n  #requestedChainIds: ChainIdHex[] = [];\n\n  constructor({\n    messenger,\n    interval = DEFAULT_INTERVAL_MS,\n    chainPollingIntervals = {},\n    state = {},\n    queryMultipleAccounts = true,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: TokenBalancesControllerOptions) {\n    super({\n      name: CONTROLLER,\n      messenger,\n      metadata,\n      state: { tokenBalances: {}, ...state },\n    });\n\n    this.#queryAllAccounts = queryMultipleAccounts;\n    this.#defaultInterval = interval;\n    this.#chainPollingConfig = { ...chainPollingIntervals };\n\n    // Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new RpcBalanceFetcher(this.#getProvider, this.#getNetworkClient, () => ({\n        allTokens: this.#allTokens,\n        allDetectedTokens: this.#detectedTokens,\n      })),\n    ];\n\n    this.setIntervalLength(interval);\n\n    // initial token state & subscriptions\n    const { allTokens, allDetectedTokens } = this.messagingSystem.call(\n      'TokensController:getState',\n    );\n    this.#allTokens = allTokens;\n    this.#detectedTokens = allDetectedTokens;\n\n    this.messagingSystem.subscribe(\n      'TokensController:stateChange',\n      (tokensState: TokensControllerState) => {\n        this.#onTokensChanged(tokensState).catch((error) => {\n          console.warn('Error handling token state change:', error);\n        });\n      },\n    );\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      this.#onNetworkChanged,\n    );\n    this.messagingSystem.subscribe(\n      'KeyringController:accountRemoved',\n      this.#onAccountRemoved,\n    );\n\n    // Register action handlers for polling interval control\n    this.messagingSystem.registerActionHandler(\n      `TokenBalancesController:updateChainPollingConfigs`,\n      this.updateChainPollingConfigs.bind(this),\n    );\n\n    this.messagingSystem.registerActionHandler(\n      `TokenBalancesController:getChainPollingConfig`,\n      this.getChainPollingConfig.bind(this),\n    );\n  }\n\n  #chainIdsWithTokens(): ChainIdHex[] {\n    return [\n      ...new Set([\n        ...Object.keys(this.#allTokens),\n        ...Object.keys(this.#detectedTokens),\n      ]),\n    ] as ChainIdHex[];\n  }\n\n  readonly #getProvider = (chainId: ChainIdHex): Web3Provider => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    const client = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n    return new Web3Provider(client.provider);\n  };\n\n  readonly #getNetworkClient = (chainId: ChainIdHex) => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    return this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n  };\n\n  /**\n   * Override to support per-chain polling intervals by grouping chains by interval\n   *\n   * @param options0 - The polling options\n   * @param options0.chainIds - Chain IDs to start polling for\n   */\n  override _startPolling({ chainIds }: { chainIds: ChainIdHex[] }) {\n    // Store the original chainIds to preserve intent across config updates\n    this.#requestedChainIds = [...chainIds];\n    this.#isControllerPollingActive = true;\n    this.#startIntervalGroupPolling(chainIds, true);\n  }\n\n  /**\n   * Start or restart interval-based polling for multiple chains\n   *\n   * @param chainIds - Chain IDs to start polling for\n   * @param immediate - Whether to poll immediately before starting timers (default: true)\n   */\n  #startIntervalGroupPolling(chainIds: ChainIdHex[], immediate = true) {\n    // Stop any existing interval timers\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n\n    // Group chains by their polling intervals\n    const intervalGroups = new Map<number, ChainIdHex[]>();\n\n    for (const chainId of chainIds) {\n      const config = this.getChainPollingConfig(chainId);\n      const existing = intervalGroups.get(config.interval) || [];\n      existing.push(chainId);\n      intervalGroups.set(config.interval, existing);\n    }\n\n    // Start separate polling loop for each interval group\n    for (const [interval, chainIdsGroup] of intervalGroups) {\n      this.#startPollingForInterval(interval, chainIdsGroup, immediate);\n    }\n  }\n\n  /**\n   * Start polling loop for chains that share the same interval\n   *\n   * @param interval - The polling interval in milliseconds\n   * @param chainIds - Chain IDs that share this interval\n   * @param immediate - Whether to poll immediately before starting the timer (default: true)\n   */\n  #startPollingForInterval(\n    interval: number,\n    chainIds: ChainIdHex[],\n    immediate = true,\n  ) {\n    const pollFunction = async () => {\n      if (!this.#isControllerPollingActive) {\n        return;\n      }\n      try {\n        await this._executePoll({ chainIds });\n      } catch (error) {\n        console.warn(\n          `Polling failed for chains ${chainIds.join(', ')} with interval ${interval}:`,\n          error,\n        );\n      }\n    };\n\n    // Poll immediately first if requested\n    if (immediate) {\n      pollFunction().catch((error) => {\n        console.warn(\n          `Immediate polling failed for chains ${chainIds.join(', ')}:`,\n          error,\n        );\n      });\n    }\n\n    // Then start regular interval polling\n    this.#setPollingTimer(interval, chainIds, pollFunction);\n  }\n\n  /**\n   * Helper method to set up polling timer\n   *\n   * @param interval - The polling interval in milliseconds\n   * @param chainIds - Chain IDs for this interval\n   * @param pollFunction - The function to call on each poll\n   */\n  #setPollingTimer(\n    interval: number,\n    chainIds: ChainIdHex[],\n    pollFunction: () => Promise<void>,\n  ) {\n    // Clear any existing timer for this interval first\n    const existingTimer = this.#intervalPollingTimers.get(interval);\n    if (existingTimer) {\n      clearInterval(existingTimer);\n    }\n\n    const timer = setInterval(() => {\n      pollFunction().catch((error) => {\n        console.warn(\n          `Interval polling failed for chains ${chainIds.join(', ')}:`,\n          error,\n        );\n      });\n    }, interval);\n    this.#intervalPollingTimers.set(interval, timer);\n  }\n\n  /**\n   * Override to handle our custom polling approach\n   */\n  override _stopPollingByPollingTokenSetId() {\n    this.#isControllerPollingActive = false;\n    this.#requestedChainIds = []; // Clear original intent when stopping\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n  }\n\n  /**\n   * Get polling configuration for a chain (includes default fallback)\n   *\n   * @param chainId - The chain ID to get config for\n   * @returns The polling configuration for the chain\n   */\n  getChainPollingConfig(chainId: ChainIdHex): ChainPollingConfig {\n    return (\n      this.#chainPollingConfig[chainId] ?? {\n        interval: this.#defaultInterval,\n      }\n    );\n  }\n\n  override async _executePoll({ chainIds }: { chainIds: ChainIdHex[] }) {\n    // This won't be called with our custom implementation, but keep for compatibility\n    await this.updateBalances({ chainIds });\n  }\n\n  /**\n   * Update multiple chain polling configurations at once\n   *\n   * @param configs - Object mapping chain IDs to polling configurations\n   * @param options - Optional configuration for the update behavior\n   * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)\n   */\n  updateChainPollingConfigs(\n    configs: Record<ChainIdHex, ChainPollingConfig>,\n    options: UpdateChainPollingConfigsOptions = { immediateUpdate: true },\n  ): void {\n    Object.assign(this.#chainPollingConfig, configs);\n\n    // If polling is currently active, restart with new interval groupings\n    if (this.#isControllerPollingActive) {\n      // Restart polling with immediate fetch by default, unless explicitly disabled\n      this.#startIntervalGroupPolling(\n        this.#requestedChainIds,\n        options.immediateUpdate,\n      );\n    }\n  }\n\n  async updateBalances({ chainIds }: { chainIds?: ChainIdHex[] } = {}) {\n    const targetChains = chainIds ?? this.#chainIdsWithTokens();\n    if (!targetChains.length) {\n      return;\n    }\n\n    const { address: selected } = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const aggregated: ProcessedBalance[] = [];\n    let remainingChains = [...targetChains];\n\n    // Try each fetcher in order, removing successfully processed chains\n    for (const fetcher of this.#balanceFetchers) {\n      const supportedChains = remainingChains.filter((c) =>\n        fetcher.supports(c),\n      );\n      if (!supportedChains.length) {\n        continue;\n      }\n\n      try {\n        const balances = await fetcher.fetch({\n          chainIds: supportedChains,\n          queryAllAccounts: this.#queryAllAccounts,\n          selectedAccount: selected as ChecksumAddress,\n          allAccounts,\n        });\n\n        if (balances && balances.length > 0) {\n          aggregated.push(...balances);\n          // Remove chains that were successfully processed\n          const processedChains = new Set(balances.map((b) => b.chainId));\n          remainingChains = remainingChains.filter(\n            (chain) => !processedChains.has(chain),\n          );\n        }\n      } catch (error) {\n        console.warn(\n          `Balance fetcher failed for chains ${supportedChains.join(', ')}: ${String(error)}`,\n        );\n        // Continue to next fetcher (fallback)\n      }\n\n      // If all chains have been processed, break early\n      if (remainingChains.length === 0) {\n        break;\n      }\n    }\n\n    // Determine which accounts to process\n    const accountsToProcess = this.#queryAllAccounts\n      ? allAccounts.map((a) => a.address as ChecksumAddress)\n      : [selected as ChecksumAddress];\n\n    const prev = this.state;\n    const next = draft(prev, (d) => {\n      // First, initialize all tokens from allTokens state with balance 0\n      // for the accounts and chains we're processing\n      for (const chainId of targetChains) {\n        for (const account of accountsToProcess) {\n          // Initialize tokens from allTokens\n          const chainTokens = this.#allTokens[chainId];\n          if (chainTokens?.[account]) {\n            Object.values(chainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n\n          // Initialize tokens from allDetectedTokens\n          const detectedChainTokens = this.#detectedTokens[chainId];\n          if (detectedChainTokens?.[account]) {\n            Object.values(detectedChainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n        }\n      }\n\n      // Then update with actual fetched balances where available\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          ((d.tokenBalances[account] ??= {})[chainId] ??= {})[checksum(token)] =\n            toHex(value);\n        }\n      });\n    });\n\n    if (!isEqual(prev, next)) {\n      this.update(() => next);\n\n      const nativeBalances = aggregated.filter(\n        (r) => r.success && r.token === ZERO_ADDRESS,\n      );\n\n      // Update native token balances in a single batch operation for better performance\n      if (nativeBalances.length > 0) {\n        const balanceUpdates = nativeBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          balance: balance.value ? BNToHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateNativeBalances',\n          balanceUpdates,\n        );\n      }\n\n      // Get staking contract addresses for filtering\n      const stakingContractAddresses = Object.values(\n        STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n      ).map((addr) => addr.toLowerCase());\n\n      // Filter and update staked balances in a single batch operation for better performance\n      const stakedBalances = aggregated.filter((r) => {\n        return (\n          r.success &&\n          r.token !== ZERO_ADDRESS &&\n          stakingContractAddresses.includes(r.token.toLowerCase())\n        );\n      });\n\n      if (stakedBalances.length > 0) {\n        const stakedBalanceUpdates = stakedBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          stakedBalance: balance.value ? toHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateStakedBalances',\n          stakedBalanceUpdates,\n        );\n      }\n    }\n  }\n\n  resetState() {\n    this.update(() => ({ tokenBalances: {} }));\n  }\n\n  readonly #onTokensChanged = async (state: TokensControllerState) => {\n    const changed: ChainIdHex[] = [];\n    let hasChanges = false;\n\n    // Get chains that have existing balances\n    const chainsWithBalances = new Set<ChainIdHex>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const chainId of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        chainsWithBalances.add(chainId as ChainIdHex);\n      }\n    }\n\n    // Only process chains that are explicitly mentioned in the incoming state change\n    const incomingChainIds = new Set([\n      ...Object.keys(state.allTokens),\n      ...Object.keys(state.allDetectedTokens),\n    ]);\n\n    // Only proceed if there are actual changes to chains that have balances or are being added\n    const relevantChainIds = Array.from(incomingChainIds).filter((chainId) => {\n      const id = chainId as ChainIdHex;\n\n      const hasTokensNow =\n        (state.allTokens[id] && Object.keys(state.allTokens[id]).length > 0) ||\n        (state.allDetectedTokens[id] &&\n          Object.keys(state.allDetectedTokens[id]).length > 0);\n      const hadTokensBefore =\n        (this.#allTokens[id] && Object.keys(this.#allTokens[id]).length > 0) ||\n        (this.#detectedTokens[id] &&\n          Object.keys(this.#detectedTokens[id]).length > 0);\n\n      // Check if there's an actual change in token state\n      const hasTokenChange =\n        !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n        !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id]);\n\n      // Process chains that have actual changes OR are new chains getting tokens\n      return hasTokenChange || (!hadTokensBefore && hasTokensNow);\n    });\n\n    if (relevantChainIds.length === 0) {\n      // No relevant changes, just update internal state\n      this.#allTokens = state.allTokens;\n      this.#detectedTokens = state.allDetectedTokens;\n      return;\n    }\n\n    // Handle both cleanup and updates in a single state update\n    this.update((s) => {\n      for (const chainId of relevantChainIds) {\n        const id = chainId as ChainIdHex;\n        const hasTokensNow =\n          (state.allTokens[id] &&\n            Object.keys(state.allTokens[id]).length > 0) ||\n          (state.allDetectedTokens[id] &&\n            Object.keys(state.allDetectedTokens[id]).length > 0);\n        const hadTokensBefore =\n          (this.#allTokens[id] &&\n            Object.keys(this.#allTokens[id]).length > 0) ||\n          (this.#detectedTokens[id] &&\n            Object.keys(this.#detectedTokens[id]).length > 0);\n\n        if (\n          !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n          !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id])\n        ) {\n          if (hasTokensNow) {\n            // Chain still has tokens - mark for async balance update\n            changed.push(id);\n          } else if (hadTokensBefore) {\n            // Chain had tokens before but doesn't now - clean up balances immediately\n            for (const address of Object.keys(s.tokenBalances)) {\n              const addressKey = address as ChecksumAddress;\n              if (s.tokenBalances[addressKey]?.[id]) {\n                s.tokenBalances[addressKey][id] = {};\n                hasChanges = true;\n              }\n            }\n          }\n        }\n      }\n    });\n\n    this.#allTokens = state.allTokens;\n    this.#detectedTokens = state.allDetectedTokens;\n\n    // Only update balances for chains that still have tokens (and only if we haven't already updated state)\n    if (changed.length && !hasChanges) {\n      this.updateBalances({ chainIds: changed }).catch((error) => {\n        console.warn('Error updating balances after token change:', error);\n      });\n    }\n  };\n\n  readonly #onNetworkChanged = (state: NetworkState) => {\n    // Check if any networks were removed by comparing with previous state\n    const currentNetworks = new Set(\n      Object.keys(state.networkConfigurationsByChainId),\n    );\n\n    // Get all networks that currently have balances\n    const networksWithBalances = new Set<string>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const network of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        networksWithBalances.add(network);\n      }\n    }\n\n    // Find networks that were removed\n    const removedNetworks = Array.from(networksWithBalances).filter(\n      (network) => !currentNetworks.has(network),\n    );\n\n    if (removedNetworks.length > 0) {\n      this.update((s) => {\n        // Remove balances for all accounts on the deleted networks\n        for (const address of Object.keys(s.tokenBalances)) {\n          const addressKey = address as ChecksumAddress;\n          for (const removedNetwork of removedNetworks) {\n            const networkKey = removedNetwork as ChainIdHex;\n            if (s.tokenBalances[addressKey]?.[networkKey]) {\n              delete s.tokenBalances[addressKey][networkKey];\n            }\n          }\n        }\n      });\n    }\n  };\n\n  readonly #onAccountRemoved = (addr: string) => {\n    if (!isStrictHexString(addr) || !isValidHexAddress(addr)) {\n      return;\n    }\n    this.update((s) => {\n      delete s.tokenBalances[addr as ChecksumAddress];\n    });\n  };\n\n  /**\n   * Clean up all timers and resources when controller is destroyed\n   */\n  override destroy(): void {\n    this.#isControllerPollingActive = false;\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n\n    // Unregister action handlers\n    this.messagingSystem.unregisterActionHandler(\n      `TokenBalancesController:updateChainPollingConfigs`,\n    );\n    this.messagingSystem.unregisterActionHandler(\n      `TokenBalancesController:getChainPollingConfig`,\n    );\n\n    super.destroy();\n  }\n}\n\nexport default TokenBalancesController;\n"]}

package/dist/TokenBalancesController.d.cts

@@ -14,7 +14,15 @@ export type TokenBalancesControllerState = {
     tokenBalances: TokenBalances;
 };
 export type TokenBalancesControllerGetStateAction = ControllerGetStateAction<typeof CONTROLLER, TokenBalancesControllerState>;
-export type TokenBalancesControllerActions = TokenBalancesControllerGetStateAction;
+export type TokenBalancesControllerUpdateChainPollingConfigsAction = {
+    type: `TokenBalancesController:updateChainPollingConfigs`;
+    handler: TokenBalancesController['updateChainPollingConfigs'];
+};
+export type TokenBalancesControllerGetChainPollingConfigAction = {
+    type: `TokenBalancesController:getChainPollingConfig`;
+    handler: TokenBalancesController['getChainPollingConfig'];
+};
+export type TokenBalancesControllerActions = TokenBalancesControllerGetStateAction | TokenBalancesControllerUpdateChainPollingConfigsAction | TokenBalancesControllerGetChainPollingConfigAction;
 export type TokenBalancesControllerStateChangeEvent = ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;
 export type NativeBalanceEvent = {
     type: `${typeof CONTROLLER}:updatedNativeBalance`;
@@ -24,9 +32,20 @@ export type TokenBalancesControllerEvents = TokenBalancesControllerStateChangeEv
 export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 export type AllowedEvents = TokensControllerStateChangeEvent | PreferencesControllerStateChangeEvent | NetworkControllerStateChangeEvent | KeyringControllerAccountRemovedEvent;
 export type TokenBalancesControllerMessenger = RestrictedMessenger<typeof CONTROLLER, TokenBalancesControllerActions | AllowedActions, TokenBalancesControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
+export type ChainPollingConfig = {
+    /** Polling interval in milliseconds for this chain */
+    interval: number;
+};
+export type UpdateChainPollingConfigsOptions = {
+    /** Whether to immediately fetch balances after updating configs (default: true) */
+    immediateUpdate?: boolean;
+};
 export type TokenBalancesControllerOptions = {
     messenger: TokenBalancesControllerMessenger;
+    /** Default interval for chains not specified in chainPollingIntervals */
     interval?: number;
+    /** Per-chain polling configuration */
+    chainPollingIntervals?: Record<ChainIdHex, ChainPollingConfig>;
     state?: Partial<TokenBalancesControllerState>;
     /** When `true`, balances for *all* known accounts are queried. */
     queryMultipleAccounts?: boolean;
@@ -66,14 +85,46 @@ declare const TokenBalancesController_base: (abstract new (...args: any[]) => {
 }) & typeof import("@metamask/base-controller").BaseController;
 export declare class TokenBalancesController extends TokenBalancesController_base<typeof CONTROLLER, TokenBalancesControllerState, TokenBalancesControllerMessenger> {
     #private;
-    constructor({ messenger, interval, state, queryMultipleAccounts, useAccountsAPI, allowExternalServices, }: TokenBalancesControllerOptions);
+    constructor({ messenger, interval, chainPollingIntervals, state, queryMultipleAccounts, useAccountsAPI, allowExternalServices, }: TokenBalancesControllerOptions);
+    /**
+     * Override to support per-chain polling intervals by grouping chains by interval
+     *
+     * @param options0 - The polling options
+     * @param options0.chainIds - Chain IDs to start polling for
+     */
+    _startPolling({ chainIds }: {
+        chainIds: ChainIdHex[];
+    }): void;
+    /**
+     * Override to handle our custom polling approach
+     */
+    _stopPollingByPollingTokenSetId(): void;
+    /**
+     * Get polling configuration for a chain (includes default fallback)
+     *
+     * @param chainId - The chain ID to get config for
+     * @returns The polling configuration for the chain
+     */
+    getChainPollingConfig(chainId: ChainIdHex): ChainPollingConfig;
     _executePoll({ chainIds }: {
         chainIds: ChainIdHex[];
     }): Promise<void>;
+    /**
+     * Update multiple chain polling configurations at once
+     *
+     * @param configs - Object mapping chain IDs to polling configurations
+     * @param options - Optional configuration for the update behavior
+     * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)
+     */
+    updateChainPollingConfigs(configs: Record<ChainIdHex, ChainPollingConfig>, options?: UpdateChainPollingConfigsOptions): void;
     updateBalances({ chainIds }?: {
         chainIds?: ChainIdHex[];
     }): Promise<void>;
     resetState(): void;
+    /**
+     * Clean up all timers and resources when controller is destroyed
+     */
+    destroy(): void;
 }
 export default TokenBalancesController;
 //# sourceMappingURL=TokenBalancesController.d.cts.map

package/dist/TokenBalancesController.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"TokenBalancesController.d.cts","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EAAE,oCAAoC,EAAE,qCAAqC;AACzF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAElC,qCAAqC;AAEtC,OAAO,KAAK,EACV,mCAAmC,EACnC,qCAAqC,EACtC,yCAAyC;AAC1C,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACzC,uCAAmC;AAQpC,OAAO,KAAK,EACV,8BAA8B,EAE9B,gCAAgC,EACjC,+BAA2B;AAE5B,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AAElC,QAAA,MAAM,UAAU,2BAAqC,CAAC;AAQtD,MAAM,MAAM,aAAa,GAAG,MAAM,CAChC,eAAe,EACf,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CACjD,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,aAAa,EAAE,aAAa,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,UAAU,EACjB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CAAC,OAAO,UAAU,EAAE,4BAA4B,CAAC,CAAC;AAE9E,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,GAAG,OAAO,UAAU,uBAAuB,CAAC;IAClD,OAAO,EAAE,OAAO,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACrC,uCAAuC,GACvC,kBAAkB,CAAC;AAEvB,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,8BAA8B,GAC9B,mCAAmC,GACnC,0CAA0C,GAC1C,oCAAoC,GACpC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C,MAAM,MAAM,aAAa,GACrB,gCAAgC,GAChC,qCAAqC,GACrC,iCAAiC,GACjC,oCAAoC,CAAC;AAEzC,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,UAAU,EACjB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,SAAS,EAAE,gCAAgC,CAAC;IAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAC9C,kEAAkE;IAClE,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,yDAAyD;IACzD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,4DAA4D;IAC5D,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;IACtC,qBAAqB;IACrB,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACpC,CAAC;;;;;;;kBAiBU,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;kBAAZ,UAAU,EAAE;;;AADxB,qBAAa,uBAAwB,SAAQ,6BAG3C,OAAO,UAAU,EACjB,4BAA4B,EAC5B,gCAAgC,CACjC;;gBASa,EACV,SAAS,EACT,QAA8B,EAC9B,KAAU,EACV,qBAA4B,EAC5B,cAAsB,EACtB,qBAAkC,GACnC,EAAE,8BAA8B;IAkF3B,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAIrD,cAAc,CAAC,EAAE,QAAQ,EAAE,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAA;KAAO;IAwJnE,UAAU;CAmJX;AAED,eAAe,uBAAuB,CAAC"}
+{"version":3,"file":"TokenBalancesController.d.cts","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EAAE,oCAAoC,EAAE,qCAAqC;AACzF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAElC,qCAAqC;AAEtC,OAAO,KAAK,EACV,mCAAmC,EACnC,qCAAqC,EACtC,yCAAyC;AAC1C,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACzC,uCAAmC;AAQpC,OAAO,KAAK,EACV,8BAA8B,EAE9B,gCAAgC,EACjC,+BAA2B;AAE5B,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AAElC,QAAA,MAAM,UAAU,2BAAqC,CAAC;AAQtD,MAAM,MAAM,aAAa,GAAG,MAAM,CAChC,eAAe,EACf,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CACjD,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,aAAa,EAAE,aAAa,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,UAAU,EACjB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,sDAAsD,GAAG;IACnE,IAAI,EAAE,mDAAmD,CAAC;IAC1D,OAAO,EAAE,uBAAuB,CAAC,2BAA2B,CAAC,CAAC;CAC/D,CAAC;AAEF,MAAM,MAAM,kDAAkD,GAAG;IAC/D,IAAI,EAAE,+CAA+C,CAAC;IACtD,OAAO,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,CAAC;CAC3D,CAAC;AAEF,MAAM,MAAM,8BAA8B,GACtC,qCAAqC,GACrC,sDAAsD,GACtD,kDAAkD,CAAC;AAEvD,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CAAC,OAAO,UAAU,EAAE,4BAA4B,CAAC,CAAC;AAE9E,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,GAAG,OAAO,UAAU,uBAAuB,CAAC;IAClD,OAAO,EAAE,OAAO,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACrC,uCAAuC,GACvC,kBAAkB,CAAC;AAEvB,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,8BAA8B,GAC9B,mCAAmC,GACnC,0CAA0C,GAC1C,oCAAoC,GACpC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C,MAAM,MAAM,aAAa,GACrB,gCAAgC,GAChC,qCAAqC,GACrC,iCAAiC,GACjC,oCAAoC,CAAC;AAEzC,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,UAAU,EACjB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,sDAAsD;IACtD,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,gCAAgC,GAAG;IAC7C,mFAAmF;IACnF,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,SAAS,EAAE,gCAAgC,CAAC;IAC5C,yEAAyE;IACzE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,sCAAsC;IACtC,qBAAqB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;IAC/D,KAAK,CAAC,EAAE,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAC9C,kEAAkE;IAClE,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,yDAAyD;IACzD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,4DAA4D;IAC5D,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;IACtC,qBAAqB;IACrB,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACpC,CAAC;;;;;;;kBAiBU,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;kBAAZ,UAAU,EAAE;;;AADxB,qBAAa,uBAAwB,SAAQ,6BAG3C,OAAO,UAAU,EACjB,4BAA4B,EAC5B,gCAAgC,CACjC;;gBAwBa,EACV,SAAS,EACT,QAA8B,EAC9B,qBAA0B,EAC1B,KAAU,EACV,qBAA4B,EAC5B,cAAsB,EACtB,qBAAkC,GACnC,EAAE,8BAA8B;IA+FjC;;;;;OAKG;IACM,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAuG/D;;OAEG;IACM,+BAA+B;IAOxC;;;;;OAKG;IACH,qBAAqB,CAAC,OAAO,EAAE,UAAU,GAAG,kBAAkB;IAQ/C,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAKpE;;;;;;OAMG;IACH,yBAAyB,CACvB,OAAO,EAAE,MAAM,CAAC,UAAU,EAAE,kBAAkB,CAAC,EAC/C,OAAO,GAAE,gCAA4D,GACpE,IAAI;IAaD,cAAc,CAAC,EAAE,QAAQ,EAAE,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAA;KAAO;IAwJnE,UAAU;IAoJV;;OAEG;IACM,OAAO,IAAI,IAAI;CAezB;AAED,eAAe,uBAAuB,CAAC"}

package/dist/TokenBalancesController.d.mts

@@ -14,7 +14,15 @@ export type TokenBalancesControllerState = {
     tokenBalances: TokenBalances;
 };
 export type TokenBalancesControllerGetStateAction = ControllerGetStateAction<typeof CONTROLLER, TokenBalancesControllerState>;
-export type TokenBalancesControllerActions = TokenBalancesControllerGetStateAction;
+export type TokenBalancesControllerUpdateChainPollingConfigsAction = {
+    type: `TokenBalancesController:updateChainPollingConfigs`;
+    handler: TokenBalancesController['updateChainPollingConfigs'];
+};
+export type TokenBalancesControllerGetChainPollingConfigAction = {
+    type: `TokenBalancesController:getChainPollingConfig`;
+    handler: TokenBalancesController['getChainPollingConfig'];
+};
+export type TokenBalancesControllerActions = TokenBalancesControllerGetStateAction | TokenBalancesControllerUpdateChainPollingConfigsAction | TokenBalancesControllerGetChainPollingConfigAction;
 export type TokenBalancesControllerStateChangeEvent = ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;
 export type NativeBalanceEvent = {
     type: `${typeof CONTROLLER}:updatedNativeBalance`;
@@ -24,9 +32,20 @@ export type TokenBalancesControllerEvents = TokenBalancesControllerStateChangeEv
 export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 export type AllowedEvents = TokensControllerStateChangeEvent | PreferencesControllerStateChangeEvent | NetworkControllerStateChangeEvent | KeyringControllerAccountRemovedEvent;
 export type TokenBalancesControllerMessenger = RestrictedMessenger<typeof CONTROLLER, TokenBalancesControllerActions | AllowedActions, TokenBalancesControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
+export type ChainPollingConfig = {
+    /** Polling interval in milliseconds for this chain */
+    interval: number;
+};
+export type UpdateChainPollingConfigsOptions = {
+    /** Whether to immediately fetch balances after updating configs (default: true) */
+    immediateUpdate?: boolean;
+};
 export type TokenBalancesControllerOptions = {
     messenger: TokenBalancesControllerMessenger;
+    /** Default interval for chains not specified in chainPollingIntervals */
     interval?: number;
+    /** Per-chain polling configuration */
+    chainPollingIntervals?: Record<ChainIdHex, ChainPollingConfig>;
     state?: Partial<TokenBalancesControllerState>;
     /** When `true`, balances for *all* known accounts are queried. */
     queryMultipleAccounts?: boolean;
@@ -66,14 +85,46 @@ declare const TokenBalancesController_base: (abstract new (...args: any[]) => {
 }) & typeof import("@metamask/base-controller").BaseController;
 export declare class TokenBalancesController extends TokenBalancesController_base<typeof CONTROLLER, TokenBalancesControllerState, TokenBalancesControllerMessenger> {
     #private;
-    constructor({ messenger, interval, state, queryMultipleAccounts, useAccountsAPI, allowExternalServices, }: TokenBalancesControllerOptions);
+    constructor({ messenger, interval, chainPollingIntervals, state, queryMultipleAccounts, useAccountsAPI, allowExternalServices, }: TokenBalancesControllerOptions);
+    /**
+     * Override to support per-chain polling intervals by grouping chains by interval
+     *
+     * @param options0 - The polling options
+     * @param options0.chainIds - Chain IDs to start polling for
+     */
+    _startPolling({ chainIds }: {
+        chainIds: ChainIdHex[];
+    }): void;
+    /**
+     * Override to handle our custom polling approach
+     */
+    _stopPollingByPollingTokenSetId(): void;
+    /**
+     * Get polling configuration for a chain (includes default fallback)
+     *
+     * @param chainId - The chain ID to get config for
+     * @returns The polling configuration for the chain
+     */
+    getChainPollingConfig(chainId: ChainIdHex): ChainPollingConfig;
     _executePoll({ chainIds }: {
         chainIds: ChainIdHex[];
     }): Promise<void>;
+    /**
+     * Update multiple chain polling configurations at once
+     *
+     * @param configs - Object mapping chain IDs to polling configurations
+     * @param options - Optional configuration for the update behavior
+     * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)
+     */
+    updateChainPollingConfigs(configs: Record<ChainIdHex, ChainPollingConfig>, options?: UpdateChainPollingConfigsOptions): void;
     updateBalances({ chainIds }?: {
         chainIds?: ChainIdHex[];
     }): Promise<void>;
     resetState(): void;
+    /**
+     * Clean up all timers and resources when controller is destroyed
+     */
+    destroy(): void;
 }
 export default TokenBalancesController;
 //# sourceMappingURL=TokenBalancesController.d.mts.map

package/dist/TokenBalancesController.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"TokenBalancesController.d.mts","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EAAE,oCAAoC,EAAE,qCAAqC;AACzF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAElC,qCAAqC;AAEtC,OAAO,KAAK,EACV,mCAAmC,EACnC,qCAAqC,EACtC,yCAAyC;AAC1C,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACzC,uCAAmC;AAQpC,OAAO,KAAK,EACV,8BAA8B,EAE9B,gCAAgC,EACjC,+BAA2B;AAE5B,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AAElC,QAAA,MAAM,UAAU,2BAAqC,CAAC;AAQtD,MAAM,MAAM,aAAa,GAAG,MAAM,CAChC,eAAe,EACf,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CACjD,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,aAAa,EAAE,aAAa,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,UAAU,EACjB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CAAC,OAAO,UAAU,EAAE,4BAA4B,CAAC,CAAC;AAE9E,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,GAAG,OAAO,UAAU,uBAAuB,CAAC;IAClD,OAAO,EAAE,OAAO,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACrC,uCAAuC,GACvC,kBAAkB,CAAC;AAEvB,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,8BAA8B,GAC9B,mCAAmC,GACnC,0CAA0C,GAC1C,oCAAoC,GACpC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C,MAAM,MAAM,aAAa,GACrB,gCAAgC,GAChC,qCAAqC,GACrC,iCAAiC,GACjC,oCAAoC,CAAC;AAEzC,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,UAAU,EACjB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,SAAS,EAAE,gCAAgC,CAAC;IAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAC9C,kEAAkE;IAClE,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,yDAAyD;IACzD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,4DAA4D;IAC5D,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;IACtC,qBAAqB;IACrB,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACpC,CAAC;;;;;;;kBAiBU,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;kBAAZ,UAAU,EAAE;;;AADxB,qBAAa,uBAAwB,SAAQ,6BAG3C,OAAO,UAAU,EACjB,4BAA4B,EAC5B,gCAAgC,CACjC;;gBASa,EACV,SAAS,EACT,QAA8B,EAC9B,KAAU,EACV,qBAA4B,EAC5B,cAAsB,EACtB,qBAAkC,GACnC,EAAE,8BAA8B;IAkF3B,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAIrD,cAAc,CAAC,EAAE,QAAQ,EAAE,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAA;KAAO;IAwJnE,UAAU;CAmJX;AAED,eAAe,uBAAuB,CAAC"}
+{"version":3,"file":"TokenBalancesController.d.mts","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EAAE,oCAAoC,EAAE,qCAAqC;AACzF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAElC,qCAAqC;AAEtC,OAAO,KAAK,EACV,mCAAmC,EACnC,qCAAqC,EACtC,yCAAyC;AAC1C,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACzC,uCAAmC;AAQpC,OAAO,KAAK,EACV,8BAA8B,EAE9B,gCAAgC,EACjC,+BAA2B;AAE5B,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AAElC,QAAA,MAAM,UAAU,2BAAqC,CAAC;AAQtD,MAAM,MAAM,aAAa,GAAG,MAAM,CAChC,eAAe,EACf,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CACjD,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,aAAa,EAAE,aAAa,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,UAAU,EACjB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,sDAAsD,GAAG;IACnE,IAAI,EAAE,mDAAmD,CAAC;IAC1D,OAAO,EAAE,uBAAuB,CAAC,2BAA2B,CAAC,CAAC;CAC/D,CAAC;AAEF,MAAM,MAAM,kDAAkD,GAAG;IAC/D,IAAI,EAAE,+CAA+C,CAAC;IACtD,OAAO,EAAE,uBAAuB,CAAC,uBAAuB,CAAC,CAAC;CAC3D,CAAC;AAEF,MAAM,MAAM,8BAA8B,GACtC,qCAAqC,GACrC,sDAAsD,GACtD,kDAAkD,CAAC;AAEvD,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CAAC,OAAO,UAAU,EAAE,4BAA4B,CAAC,CAAC;AAE9E,MAAM,MAAM,kBAAkB,GAAG;IAC/B,IAAI,EAAE,GAAG,OAAO,UAAU,uBAAuB,CAAC;IAClD,OAAO,EAAE,OAAO,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACrC,uCAAuC,GACvC,kBAAkB,CAAC;AAEvB,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,8BAA8B,GAC9B,mCAAmC,GACnC,0CAA0C,GAC1C,oCAAoC,GACpC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C,MAAM,MAAM,aAAa,GACrB,gCAAgC,GAChC,qCAAqC,GACrC,iCAAiC,GACjC,oCAAoC,CAAC;AAEzC,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,UAAU,EACjB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,sDAAsD;IACtD,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,gCAAgC,GAAG;IAC7C,mFAAmF;IACnF,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,SAAS,EAAE,gCAAgC,CAAC;IAC5C,yEAAyE;IACzE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,sCAAsC;IACtC,qBAAqB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;IAC/D,KAAK,CAAC,EAAE,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAC9C,kEAAkE;IAClE,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,yDAAyD;IACzD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,4DAA4D;IAC5D,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;IACtC,qBAAqB;IACrB,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACpC,CAAC;;;;;;;kBAiBU,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;kBAAZ,UAAU,EAAE;;;;;kBAAZ,UAAU,EAAE;;kBAAZ,UAAU,EAAE;;;AADxB,qBAAa,uBAAwB,SAAQ,6BAG3C,OAAO,UAAU,EACjB,4BAA4B,EAC5B,gCAAgC,CACjC;;gBAwBa,EACV,SAAS,EACT,QAA8B,EAC9B,qBAA0B,EAC1B,KAAU,EACV,qBAA4B,EAC5B,cAAsB,EACtB,qBAAkC,GACnC,EAAE,8BAA8B;IA+FjC;;;;;OAKG;IACM,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAuG/D;;OAEG;IACM,+BAA+B;IAOxC;;;;;OAKG;IACH,qBAAqB,CAAC,OAAO,EAAE,UAAU,GAAG,kBAAkB;IAQ/C,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE;IAKpE;;;;;;OAMG;IACH,yBAAyB,CACvB,OAAO,EAAE,MAAM,CAAC,UAAU,EAAE,kBAAkB,CAAC,EAC/C,OAAO,GAAE,gCAA4D,GACpE,IAAI;IAaD,cAAc,CAAC,EAAE,QAAQ,EAAE,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAA;KAAO;IAwJnE,UAAU;IAoJV;;OAEG;IACM,OAAO,IAAI,IAAI;CAezB;AAED,eAAe,uBAAuB,CAAC"}

package/dist/TokenBalancesController.mjs

@@ -9,7 +9,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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 _TokenBalancesController_instances, _TokenBalancesController_queryAllAccounts, _TokenBalancesController_balanceFetchers, _TokenBalancesController_allTokens, _TokenBalancesController_detectedTokens, _TokenBalancesController_chainIdsWithTokens, _TokenBalancesController_getProvider, _TokenBalancesController_getNetworkClient, _TokenBalancesController_onTokensChanged, _TokenBalancesController_onNetworkChanged, _TokenBalancesController_onAccountRemoved;
+var _TokenBalancesController_instances, _TokenBalancesController_queryAllAccounts, _TokenBalancesController_balanceFetchers, _TokenBalancesController_allTokens, _TokenBalancesController_detectedTokens, _TokenBalancesController_defaultInterval, _TokenBalancesController_chainPollingConfig, _TokenBalancesController_intervalPollingTimers, _TokenBalancesController_isControllerPollingActive, _TokenBalancesController_requestedChainIds, _TokenBalancesController_chainIdsWithTokens, _TokenBalancesController_getProvider, _TokenBalancesController_getNetworkClient, _TokenBalancesController_startIntervalGroupPolling, _TokenBalancesController_startPollingForInterval, _TokenBalancesController_setPollingTimer, _TokenBalancesController_onTokensChanged, _TokenBalancesController_onNetworkChanged, _TokenBalancesController_onAccountRemoved;
 import { Web3Provider } from "@ethersproject/providers";
 import { BNToHex, isValidHexAddress, toChecksumHexAddress, toHex } from "@metamask/controller-utils";
 import { StaticIntervalPollingController } from "@metamask/polling-controller";
@@ -35,7 +35,7 @@ const checksum = (addr) => toChecksumHexAddress(addr);
 // ────────────────────────────────────────────────────────────────────────────
 // region: Main controller
 export class TokenBalancesController extends StaticIntervalPollingController() {
-    constructor({ messenger, interval = DEFAULT_INTERVAL_MS, state = {}, queryMultipleAccounts = true, useAccountsAPI = false, allowExternalServices = () => true, }) {
+    constructor({ messenger, interval = DEFAULT_INTERVAL_MS, chainPollingIntervals = {}, state = {}, queryMultipleAccounts = true, useAccountsAPI = false, allowExternalServices = () => true, }) {
         super({
             name: CONTROLLER,
             messenger,
@@ -47,6 +47,16 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
         _TokenBalancesController_balanceFetchers.set(this, void 0);
         _TokenBalancesController_allTokens.set(this, {});
         _TokenBalancesController_detectedTokens.set(this, {});
+        /** Default polling interval for chains without specific configuration */
+        _TokenBalancesController_defaultInterval.set(this, void 0);
+        /** Per-chain polling configuration */
+        _TokenBalancesController_chainPollingConfig.set(this, void 0);
+        /** Active polling timers grouped by interval */
+        _TokenBalancesController_intervalPollingTimers.set(this, new Map());
+        /** Track if controller-level polling is active */
+        _TokenBalancesController_isControllerPollingActive.set(this, false);
+        /** Store original chainIds from startPolling to preserve intent */
+        _TokenBalancesController_requestedChainIds.set(this, []);
         _TokenBalancesController_getProvider.set(this, (chainId) => {
             const { networkConfigurationsByChainId } = this.messagingSystem.call('NetworkController:getState');
             const cfg = networkConfigurationsByChainId[chainId];
@@ -174,6 +184,8 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
             });
         });
         __classPrivateFieldSet(this, _TokenBalancesController_queryAllAccounts, queryMultipleAccounts, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_defaultInterval, interval, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_chainPollingConfig, { ...chainPollingIntervals }, "f");
         // Strategy order: API first, then RPC fallback
         __classPrivateFieldSet(this, _TokenBalancesController_balanceFetchers, [
             ...(useAccountsAPI && allowExternalServices()
@@ -196,10 +208,61 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
         });
         this.messagingSystem.subscribe('NetworkController:stateChange', __classPrivateFieldGet(this, _TokenBalancesController_onNetworkChanged, "f"));
         this.messagingSystem.subscribe('KeyringController:accountRemoved', __classPrivateFieldGet(this, _TokenBalancesController_onAccountRemoved, "f"));
+        // Register action handlers for polling interval control
+        this.messagingSystem.registerActionHandler(`TokenBalancesController:updateChainPollingConfigs`, this.updateChainPollingConfigs.bind(this));
+        this.messagingSystem.registerActionHandler(`TokenBalancesController:getChainPollingConfig`, this.getChainPollingConfig.bind(this));
+    }
+    /**
+     * Override to support per-chain polling intervals by grouping chains by interval
+     *
+     * @param options0 - The polling options
+     * @param options0.chainIds - Chain IDs to start polling for
+     */
+    _startPolling({ chainIds }) {
+        // Store the original chainIds to preserve intent across config updates
+        __classPrivateFieldSet(this, _TokenBalancesController_requestedChainIds, [...chainIds], "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, true, "f");
+        __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startIntervalGroupPolling).call(this, chainIds, true);
+    }
+    /**
+     * Override to handle our custom polling approach
+     */
+    _stopPollingByPollingTokenSetId() {
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, false, "f");
+        __classPrivateFieldSet(this, _TokenBalancesController_requestedChainIds, [], "f"); // Clear original intent when stopping
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+    }
+    /**
+     * Get polling configuration for a chain (includes default fallback)
+     *
+     * @param chainId - The chain ID to get config for
+     * @returns The polling configuration for the chain
+     */
+    getChainPollingConfig(chainId) {
+        return (__classPrivateFieldGet(this, _TokenBalancesController_chainPollingConfig, "f")[chainId] ?? {
+            interval: __classPrivateFieldGet(this, _TokenBalancesController_defaultInterval, "f"),
+        });
     }
     async _executePoll({ chainIds }) {
+        // This won't be called with our custom implementation, but keep for compatibility
         await this.updateBalances({ chainIds });
     }
+    /**
+     * Update multiple chain polling configurations at once
+     *
+     * @param configs - Object mapping chain IDs to polling configurations
+     * @param options - Optional configuration for the update behavior
+     * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)
+     */
+    updateChainPollingConfigs(configs, options = { immediateUpdate: true }) {
+        Object.assign(__classPrivateFieldGet(this, _TokenBalancesController_chainPollingConfig, "f"), configs);
+        // If polling is currently active, restart with new interval groupings
+        if (__classPrivateFieldGet(this, _TokenBalancesController_isControllerPollingActive, "f")) {
+            // Restart polling with immediate fetch by default, unless explicitly disabled
+            __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startIntervalGroupPolling).call(this, __classPrivateFieldGet(this, _TokenBalancesController_requestedChainIds, "f"), options.immediateUpdate);
+        }
+    }
     async updateBalances({ chainIds } = {}) {
         const targetChains = chainIds ?? __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_chainIdsWithTokens).call(this);
         if (!targetChains.length) {
@@ -310,14 +373,74 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
     resetState() {
         this.update(() => ({ tokenBalances: {} }));
     }
+    /**
+     * Clean up all timers and resources when controller is destroyed
+     */
+    destroy() {
+        __classPrivateFieldSet(this, _TokenBalancesController_isControllerPollingActive, false, "f");
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+        __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+        // Unregister action handlers
+        this.messagingSystem.unregisterActionHandler(`TokenBalancesController:updateChainPollingConfigs`);
+        this.messagingSystem.unregisterActionHandler(`TokenBalancesController:getChainPollingConfig`);
+        super.destroy();
+    }
 }
-_TokenBalancesController_queryAllAccounts = new WeakMap(), _TokenBalancesController_balanceFetchers = new WeakMap(), _TokenBalancesController_allTokens = new WeakMap(), _TokenBalancesController_detectedTokens = new WeakMap(), _TokenBalancesController_getProvider = new WeakMap(), _TokenBalancesController_getNetworkClient = new WeakMap(), _TokenBalancesController_onTokensChanged = new WeakMap(), _TokenBalancesController_onNetworkChanged = new WeakMap(), _TokenBalancesController_onAccountRemoved = new WeakMap(), _TokenBalancesController_instances = new WeakSet(), _TokenBalancesController_chainIdsWithTokens = function _TokenBalancesController_chainIdsWithTokens() {
+_TokenBalancesController_queryAllAccounts = new WeakMap(), _TokenBalancesController_balanceFetchers = new WeakMap(), _TokenBalancesController_allTokens = new WeakMap(), _TokenBalancesController_detectedTokens = new WeakMap(), _TokenBalancesController_defaultInterval = new WeakMap(), _TokenBalancesController_chainPollingConfig = new WeakMap(), _TokenBalancesController_intervalPollingTimers = new WeakMap(), _TokenBalancesController_isControllerPollingActive = new WeakMap(), _TokenBalancesController_requestedChainIds = new WeakMap(), _TokenBalancesController_getProvider = new WeakMap(), _TokenBalancesController_getNetworkClient = new WeakMap(), _TokenBalancesController_onTokensChanged = new WeakMap(), _TokenBalancesController_onNetworkChanged = new WeakMap(), _TokenBalancesController_onAccountRemoved = new WeakMap(), _TokenBalancesController_instances = new WeakSet(), _TokenBalancesController_chainIdsWithTokens = function _TokenBalancesController_chainIdsWithTokens() {
     return [
         ...new Set([
             ...Object.keys(__classPrivateFieldGet(this, _TokenBalancesController_allTokens, "f")),
             ...Object.keys(__classPrivateFieldGet(this, _TokenBalancesController_detectedTokens, "f")),
         ]),
     ];
+}, _TokenBalancesController_startIntervalGroupPolling = function _TokenBalancesController_startIntervalGroupPolling(chainIds, immediate = true) {
+    // Stop any existing interval timers
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").forEach((timer) => clearInterval(timer));
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").clear();
+    // Group chains by their polling intervals
+    const intervalGroups = new Map();
+    for (const chainId of chainIds) {
+        const config = this.getChainPollingConfig(chainId);
+        const existing = intervalGroups.get(config.interval) || [];
+        existing.push(chainId);
+        intervalGroups.set(config.interval, existing);
+    }
+    // Start separate polling loop for each interval group
+    for (const [interval, chainIdsGroup] of intervalGroups) {
+        __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_startPollingForInterval).call(this, interval, chainIdsGroup, immediate);
+    }
+}, _TokenBalancesController_startPollingForInterval = function _TokenBalancesController_startPollingForInterval(interval, chainIds, immediate = true) {
+    const pollFunction = async () => {
+        if (!__classPrivateFieldGet(this, _TokenBalancesController_isControllerPollingActive, "f")) {
+            return;
+        }
+        try {
+            await this._executePoll({ chainIds });
+        }
+        catch (error) {
+            console.warn(`Polling failed for chains ${chainIds.join(', ')} with interval ${interval}:`, error);
+        }
+    };
+    // Poll immediately first if requested
+    if (immediate) {
+        pollFunction().catch((error) => {
+            console.warn(`Immediate polling failed for chains ${chainIds.join(', ')}:`, error);
+        });
+    }
+    // Then start regular interval polling
+    __classPrivateFieldGet(this, _TokenBalancesController_instances, "m", _TokenBalancesController_setPollingTimer).call(this, interval, chainIds, pollFunction);
+}, _TokenBalancesController_setPollingTimer = function _TokenBalancesController_setPollingTimer(interval, chainIds, pollFunction) {
+    // Clear any existing timer for this interval first
+    const existingTimer = __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").get(interval);
+    if (existingTimer) {
+        clearInterval(existingTimer);
+    }
+    const timer = setInterval(() => {
+        pollFunction().catch((error) => {
+            console.warn(`Interval polling failed for chains ${chainIds.join(', ')}:`, error);
+        });
+    }, interval);
+    __classPrivateFieldGet(this, _TokenBalancesController_intervalPollingTimers, "f").set(interval, timer);
 };
 export default TokenBalancesController;
 //# sourceMappingURL=TokenBalancesController.mjs.map

package/dist/TokenBalancesController.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"TokenBalancesController.mjs","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAUxD,OAAO,EACL,OAAO,EACP,iBAAiB,EACjB,oBAAoB,EACpB,KAAK,EACN,mCAAmC;AAQpC,OAAO,EAAE,+BAA+B,EAAE,qCAAqC;AAM/E,OAAO,EAAE,iBAAiB,EAAE,wBAAwB;AACpD,OAAO,EAAE,OAAO,EAAE,cAAc;;;AAOhC,OAAO,EAAE,mCAAmC,EAAE,uCAAmC;AACjF,OAAO,EACL,yBAAyB,EAG1B,+DAA2D;AAC5D,OAAO,EAAE,iBAAiB,EAAE,8CAA0C;AAUtE,MAAM,UAAU,GAAG,yBAAkC,CAAC;AACtD,MAAM,mBAAmB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEjD,MAAM,QAAQ,GAAG;IACf,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;CACnD,CAAC;AAqEF,YAAY;AAEZ,+EAA+E;AAC/E,2BAA2B;AAC3B,MAAM,KAAK,GAAG,CAAI,IAAO,EAAE,EAAkB,EAAK,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAEvE,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAmB,EAAE,CACjD,oBAAoB,CAAC,IAAI,CAAoB,CAAC;AAChD,YAAY;AAEZ,+EAA+E;AAC/E,0BAA0B;AAC1B,MAAM,OAAO,uBAAwB,SAAQ,+BAA+B,EAM3E;IASC,YAAY,EACV,SAAS,EACT,QAAQ,GAAG,mBAAmB,EAC9B,KAAK,GAAG,EAAE,EACV,qBAAqB,GAAG,IAAI,EAC5B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GACH;QAC/B,KAAK,CAAC;YACJ,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,QAAQ;YACR,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,GAAG,KAAK,EAAE;SACvC,CAAC,CAAC;;QArBI,4DAA2B;QAE3B,2DAAmC;QAE5C,6CAAiD,EAAE,EAAC;QAEpD,kDAA8D,EAAE,EAAC;QAkExD,+CAAe,CAAC,OAAmB,EAAgB,EAAE;YAC5D,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACtC,wCAAwC,EACxC,eAAe,CAChB,CAAC;YACF,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,EAAC;QAEO,oDAAoB,CAAC,OAAmB,EAAE,EAAE;YACnD,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9B,wCAAwC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC,EAAC;QAkKO,mDAAmB,KAAK,EAAE,KAA4B,EAAE,EAAE;YACjE,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAc,CAAC;YACjD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,kBAAkB,CAAC,GAAG,CAAC,OAAqB,CAAC,CAAC;iBAC/C;aACF;YAED,iFAAiF;YACjF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC;aACxC,CAAC,CAAC;YAEH,2FAA2F;YAC3F,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvE,MAAM,EAAE,GAAG,OAAqB,CAAC;gBAEjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;wBAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;wBACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEtD,mDAAmD;gBACnD,MAAM,cAAc,GAClB,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;oBAClD,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;gBAElE,2EAA2E;gBAC3E,OAAO,cAAc,IAAI,CAAC,CAAC,eAAe,IAAI,YAAY,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE;gBACjC,kDAAkD;gBAClD,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;gBAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;gBAC/C,OAAO;aACR;YAED,2DAA2D;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE;oBACtC,MAAM,EAAE,GAAG,OAAqB,CAAC;oBACjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;4BACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAEtD,IACE,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;wBAClD,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,EAC/D;wBACA,IAAI,YAAY,EAAE;4BAChB,yDAAyD;4BACzD,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;yBAClB;6BAAM,IAAI,eAAe,EAAE;4BAC1B,0EAA0E;4BAC1E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;gCAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;gCAC9C,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oCACrC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;oCACrC,UAAU,GAAG,IAAI,CAAC;iCACnB;6BACF;yBACF;qBACF;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;YAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;YAE/C,wGAAwG;YACxG,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE;gBACjC,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACzD,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,KAAmB,EAAE,EAAE;YACnD,sEAAsE;YACtE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAClD,CAAC;YAEF,gDAAgD;YAChD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;iBACnC;aACF;YAED,kCAAkC;YAClC,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAC7D,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC3C,CAAC;YAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9B,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAChB,2DAA2D;oBAC3D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;wBAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;wBAC9C,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE;4BAC5C,MAAM,UAAU,GAAG,cAA4B,CAAC;4BAChD,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE;gCAC7C,OAAO,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;6BAChD;yBACF;qBACF;gBACH,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,IAAY,EAAE,EAAE;YAC5C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;gBACxD,OAAO;aACR;YACD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,OAAO,CAAC,CAAC,aAAa,CAAC,IAAuB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,EAAC;QAxXA,uBAAA,IAAI,6CAAqB,qBAAqB,MAAA,CAAC;QAE/C,+CAA+C;QAC/C,uBAAA,IAAI,4CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,yBAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,4CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,iBAAiB,CAAC,uBAAA,IAAI,4CAAa,EAAE,uBAAA,IAAI,iDAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;gBACtE,SAAS,EAAE,uBAAA,IAAI,0CAAW;gBAC1B,iBAAiB,EAAE,uBAAA,IAAI,+CAAgB;aACxC,CAAC,CAAC;SACJ,MAAA,CAAC;QAEF,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,sCAAsC;QACtC,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAChE,2BAA2B,CAC5B,CAAC;QACF,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,2CAAmB,iBAAiB,MAAA,CAAC;QAEzC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,8BAA8B,EAC9B,CAAC,WAAkC,EAAE,EAAE;YACrC,uBAAA,IAAI,gDAAiB,MAArB,IAAI,EAAkB,WAAW,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjD,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,kCAAkC,EAClC,uBAAA,IAAI,iDAAkB,CACvB,CAAC;IACJ,CAAC;IAoCD,KAAK,CAAC,YAAY,CAAC,EAAE,QAAQ,EAA8B;QACzD,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAE,QAAQ,KAAkC,EAAE;QACjE,MAAM,YAAY,GAAG,QAAQ,IAAI,uBAAA,IAAI,uFAAoB,MAAxB,IAAI,CAAsB,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,OAAO;SACR;QAED,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACrD,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QAEF,MAAM,UAAU,GAAuB,EAAE,CAAC;QAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;QAExC,oEAAoE;QACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,gDAAiB,EAAE;YAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;gBAC3B,SAAS;aACV;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;oBACnC,QAAQ,EAAE,eAAe;oBACzB,gBAAgB,EAAE,uBAAA,IAAI,iDAAkB;oBACxC,eAAe,EAAE,QAA2B;oBAC5C,WAAW;iBACZ,CAAC,CAAC;gBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;oBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;oBAC7B,iDAAiD;oBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;iBACH;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;gBACF,sCAAsC;aACvC;YAED,iDAAiD;YACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;gBAChC,MAAM;aACP;SACF;QAED,sCAAsC;QACtC,MAAM,iBAAiB,GAAG,uBAAA,IAAI,iDAAkB;YAC9C,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAA0B,CAAC;YACtD,CAAC,CAAC,CAAC,QAA2B,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YAC7B,mEAAmE;YACnE,+CAA+C;YAC/C,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE;gBAClC,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE;oBACvC,mCAAmC;oBACnC,MAAM,WAAW,GAAG,uBAAA,IAAI,0CAAW,CAAC,OAAO,CAAC,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE;wBAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACzC,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;oBAED,2CAA2C;oBAC3C,MAAM,mBAAmB,GAAG,uBAAA,IAAI,+CAAgB,CAAC,OAAO,CAAC,CAAC;oBAC1D,IAAI,mBAAmB,EAAE,CAAC,OAAO,CAAC,EAAE;wBAClC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACjD,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;iBACF;aACF;YAED,2DAA2D;YAC3D,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;;gBACjE,IAAI,OAAO,IAAI,KAAK,KAAK,SAAS,EAAE;oBAClC,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClE,KAAK,CAAC,KAAK,CAAC,CAAC;iBAChB;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAExB,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,YAAY,CAC7C,CAAC;YAEF,kFAAkF;YAClF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACtD,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBACxD,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,cAAc,CACf,CAAC;aACH;YAED,+CAA+C;YAC/C,MAAM,wBAAwB,GAAG,MAAM,CAAC,MAAM,CAC5C,mCAAmC,CACpC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YAEpC,uFAAuF;YACvF,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7C,OAAO,CACL,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,KAAK,KAAK,YAAY;oBACxB,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CACzD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC5D,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBAC5D,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,oBAAoB,CACrB,CAAC;aACH;SACF;IACH,CAAC;IAED,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;CAiJF;;IAhVG,OAAO;QACL,GAAG,IAAI,GAAG,CAAC;YACT,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC;YAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC;SACrC,CAAC;KACa,CAAC;AACpB,CAAC;AA4UH,eAAe,uBAAuB,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  BNToHex,\n  isValidHexAddress,\n  toChecksumHexAddress,\n  toHex,\n} from '@metamask/controller-utils';\nimport type { KeyringControllerAccountRemovedEvent } from '@metamask/keyring-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n  NetworkState,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type {\n  PreferencesControllerGetStateAction,\n  PreferencesControllerStateChangeEvent,\n} from '@metamask/preferences-controller';\nimport type { Hex } from '@metamask/utils';\nimport { isStrictHexString } from '@metamask/utils';\nimport { produce } from 'immer';\nimport { isEqual } from 'lodash';\n\nimport type {\n  AccountTrackerUpdateNativeBalancesAction,\n  AccountTrackerUpdateStakedBalancesAction,\n} from './AccountTrackerController';\nimport { STAKING_CONTRACT_ADDRESS_BY_CHAINID } from './AssetsContractController';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\nimport { RpcBalanceFetcher } from './rpc-service/rpc-balance-fetcher';\nimport type {\n  TokensControllerGetStateAction,\n  TokensControllerState,\n  TokensControllerStateChangeEvent,\n} from './TokensController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst CONTROLLER = 'TokenBalancesController' as const;\nconst DEFAULT_INTERVAL_MS = 180_000; // 3 minutes\n\nconst metadata = {\n  tokenBalances: { persist: true, anonymous: false },\n};\n\n// account → chain → token → balance\nexport type TokenBalances = Record<\n  ChecksumAddress,\n  Record<ChainIdHex, Record<ChecksumAddress, Hex>>\n>;\n\nexport type TokenBalancesControllerState = {\n  tokenBalances: TokenBalances;\n};\n\nexport type TokenBalancesControllerGetStateAction = ControllerGetStateAction<\n  typeof CONTROLLER,\n  TokenBalancesControllerState\n>;\n\nexport type TokenBalancesControllerActions =\n  TokenBalancesControllerGetStateAction;\n\nexport type TokenBalancesControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;\n\nexport type NativeBalanceEvent = {\n  type: `${typeof CONTROLLER}:updatedNativeBalance`;\n  payload: unknown[];\n};\n\nexport type TokenBalancesControllerEvents =\n  | TokenBalancesControllerStateChangeEvent\n  | NativeBalanceEvent;\n\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | TokensControllerGetStateAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | AccountsControllerListAccountsAction\n  | AccountTrackerUpdateNativeBalancesAction\n  | AccountTrackerUpdateStakedBalancesAction;\n\nexport type AllowedEvents =\n  | TokensControllerStateChangeEvent\n  | PreferencesControllerStateChangeEvent\n  | NetworkControllerStateChangeEvent\n  | KeyringControllerAccountRemovedEvent;\n\nexport type TokenBalancesControllerMessenger = RestrictedMessenger<\n  typeof CONTROLLER,\n  TokenBalancesControllerActions | AllowedActions,\n  TokenBalancesControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\nexport type TokenBalancesControllerOptions = {\n  messenger: TokenBalancesControllerMessenger;\n  interval?: number;\n  state?: Partial<TokenBalancesControllerState>;\n  /** When `true`, balances for *all* known accounts are queried. */\n  queryMultipleAccounts?: boolean;\n  /** Enable Accounts‑API strategy (if supported chain). */\n  useAccountsAPI?: boolean;\n  /** Disable external HTTP calls (privacy / offline mode). */\n  allowExternalServices?: () => boolean;\n  /** Custom logger. */\n  log?: (...args: unknown[]) => void;\n};\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Helper utilities\nconst draft = <T>(base: T, fn: (d: T) => void): T => produce(base, fn);\n\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\nconst checksum = (addr: string): ChecksumAddress =>\n  toChecksumHexAddress(addr) as ChecksumAddress;\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Main controller\nexport class TokenBalancesController extends StaticIntervalPollingController<{\n  chainIds: ChainIdHex[];\n}>()<\n  typeof CONTROLLER,\n  TokenBalancesControllerState,\n  TokenBalancesControllerMessenger\n> {\n  readonly #queryAllAccounts: boolean;\n\n  readonly #balanceFetchers: BalanceFetcher[];\n\n  #allTokens: TokensControllerState['allTokens'] = {};\n\n  #detectedTokens: TokensControllerState['allDetectedTokens'] = {};\n\n  constructor({\n    messenger,\n    interval = DEFAULT_INTERVAL_MS,\n    state = {},\n    queryMultipleAccounts = true,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: TokenBalancesControllerOptions) {\n    super({\n      name: CONTROLLER,\n      messenger,\n      metadata,\n      state: { tokenBalances: {}, ...state },\n    });\n\n    this.#queryAllAccounts = queryMultipleAccounts;\n\n    // Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new RpcBalanceFetcher(this.#getProvider, this.#getNetworkClient, () => ({\n        allTokens: this.#allTokens,\n        allDetectedTokens: this.#detectedTokens,\n      })),\n    ];\n\n    this.setIntervalLength(interval);\n\n    // initial token state & subscriptions\n    const { allTokens, allDetectedTokens } = this.messagingSystem.call(\n      'TokensController:getState',\n    );\n    this.#allTokens = allTokens;\n    this.#detectedTokens = allDetectedTokens;\n\n    this.messagingSystem.subscribe(\n      'TokensController:stateChange',\n      (tokensState: TokensControllerState) => {\n        this.#onTokensChanged(tokensState).catch((error) => {\n          console.warn('Error handling token state change:', error);\n        });\n      },\n    );\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      this.#onNetworkChanged,\n    );\n    this.messagingSystem.subscribe(\n      'KeyringController:accountRemoved',\n      this.#onAccountRemoved,\n    );\n  }\n\n  #chainIdsWithTokens(): ChainIdHex[] {\n    return [\n      ...new Set([\n        ...Object.keys(this.#allTokens),\n        ...Object.keys(this.#detectedTokens),\n      ]),\n    ] as ChainIdHex[];\n  }\n\n  readonly #getProvider = (chainId: ChainIdHex): Web3Provider => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    const client = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n    return new Web3Provider(client.provider);\n  };\n\n  readonly #getNetworkClient = (chainId: ChainIdHex) => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    return this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n  };\n\n  async _executePoll({ chainIds }: { chainIds: ChainIdHex[] }) {\n    await this.updateBalances({ chainIds });\n  }\n\n  async updateBalances({ chainIds }: { chainIds?: ChainIdHex[] } = {}) {\n    const targetChains = chainIds ?? this.#chainIdsWithTokens();\n    if (!targetChains.length) {\n      return;\n    }\n\n    const { address: selected } = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const aggregated: ProcessedBalance[] = [];\n    let remainingChains = [...targetChains];\n\n    // Try each fetcher in order, removing successfully processed chains\n    for (const fetcher of this.#balanceFetchers) {\n      const supportedChains = remainingChains.filter((c) =>\n        fetcher.supports(c),\n      );\n      if (!supportedChains.length) {\n        continue;\n      }\n\n      try {\n        const balances = await fetcher.fetch({\n          chainIds: supportedChains,\n          queryAllAccounts: this.#queryAllAccounts,\n          selectedAccount: selected as ChecksumAddress,\n          allAccounts,\n        });\n\n        if (balances && balances.length > 0) {\n          aggregated.push(...balances);\n          // Remove chains that were successfully processed\n          const processedChains = new Set(balances.map((b) => b.chainId));\n          remainingChains = remainingChains.filter(\n            (chain) => !processedChains.has(chain),\n          );\n        }\n      } catch (error) {\n        console.warn(\n          `Balance fetcher failed for chains ${supportedChains.join(', ')}: ${String(error)}`,\n        );\n        // Continue to next fetcher (fallback)\n      }\n\n      // If all chains have been processed, break early\n      if (remainingChains.length === 0) {\n        break;\n      }\n    }\n\n    // Determine which accounts to process\n    const accountsToProcess = this.#queryAllAccounts\n      ? allAccounts.map((a) => a.address as ChecksumAddress)\n      : [selected as ChecksumAddress];\n\n    const prev = this.state;\n    const next = draft(prev, (d) => {\n      // First, initialize all tokens from allTokens state with balance 0\n      // for the accounts and chains we're processing\n      for (const chainId of targetChains) {\n        for (const account of accountsToProcess) {\n          // Initialize tokens from allTokens\n          const chainTokens = this.#allTokens[chainId];\n          if (chainTokens?.[account]) {\n            Object.values(chainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n\n          // Initialize tokens from allDetectedTokens\n          const detectedChainTokens = this.#detectedTokens[chainId];\n          if (detectedChainTokens?.[account]) {\n            Object.values(detectedChainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n        }\n      }\n\n      // Then update with actual fetched balances where available\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          ((d.tokenBalances[account] ??= {})[chainId] ??= {})[checksum(token)] =\n            toHex(value);\n        }\n      });\n    });\n\n    if (!isEqual(prev, next)) {\n      this.update(() => next);\n\n      const nativeBalances = aggregated.filter(\n        (r) => r.success && r.token === ZERO_ADDRESS,\n      );\n\n      // Update native token balances in a single batch operation for better performance\n      if (nativeBalances.length > 0) {\n        const balanceUpdates = nativeBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          balance: balance.value ? BNToHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateNativeBalances',\n          balanceUpdates,\n        );\n      }\n\n      // Get staking contract addresses for filtering\n      const stakingContractAddresses = Object.values(\n        STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n      ).map((addr) => addr.toLowerCase());\n\n      // Filter and update staked balances in a single batch operation for better performance\n      const stakedBalances = aggregated.filter((r) => {\n        return (\n          r.success &&\n          r.token !== ZERO_ADDRESS &&\n          stakingContractAddresses.includes(r.token.toLowerCase())\n        );\n      });\n\n      if (stakedBalances.length > 0) {\n        const stakedBalanceUpdates = stakedBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          stakedBalance: balance.value ? toHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateStakedBalances',\n          stakedBalanceUpdates,\n        );\n      }\n    }\n  }\n\n  resetState() {\n    this.update(() => ({ tokenBalances: {} }));\n  }\n\n  readonly #onTokensChanged = async (state: TokensControllerState) => {\n    const changed: ChainIdHex[] = [];\n    let hasChanges = false;\n\n    // Get chains that have existing balances\n    const chainsWithBalances = new Set<ChainIdHex>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const chainId of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        chainsWithBalances.add(chainId as ChainIdHex);\n      }\n    }\n\n    // Only process chains that are explicitly mentioned in the incoming state change\n    const incomingChainIds = new Set([\n      ...Object.keys(state.allTokens),\n      ...Object.keys(state.allDetectedTokens),\n    ]);\n\n    // Only proceed if there are actual changes to chains that have balances or are being added\n    const relevantChainIds = Array.from(incomingChainIds).filter((chainId) => {\n      const id = chainId as ChainIdHex;\n\n      const hasTokensNow =\n        (state.allTokens[id] && Object.keys(state.allTokens[id]).length > 0) ||\n        (state.allDetectedTokens[id] &&\n          Object.keys(state.allDetectedTokens[id]).length > 0);\n      const hadTokensBefore =\n        (this.#allTokens[id] && Object.keys(this.#allTokens[id]).length > 0) ||\n        (this.#detectedTokens[id] &&\n          Object.keys(this.#detectedTokens[id]).length > 0);\n\n      // Check if there's an actual change in token state\n      const hasTokenChange =\n        !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n        !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id]);\n\n      // Process chains that have actual changes OR are new chains getting tokens\n      return hasTokenChange || (!hadTokensBefore && hasTokensNow);\n    });\n\n    if (relevantChainIds.length === 0) {\n      // No relevant changes, just update internal state\n      this.#allTokens = state.allTokens;\n      this.#detectedTokens = state.allDetectedTokens;\n      return;\n    }\n\n    // Handle both cleanup and updates in a single state update\n    this.update((s) => {\n      for (const chainId of relevantChainIds) {\n        const id = chainId as ChainIdHex;\n        const hasTokensNow =\n          (state.allTokens[id] &&\n            Object.keys(state.allTokens[id]).length > 0) ||\n          (state.allDetectedTokens[id] &&\n            Object.keys(state.allDetectedTokens[id]).length > 0);\n        const hadTokensBefore =\n          (this.#allTokens[id] &&\n            Object.keys(this.#allTokens[id]).length > 0) ||\n          (this.#detectedTokens[id] &&\n            Object.keys(this.#detectedTokens[id]).length > 0);\n\n        if (\n          !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n          !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id])\n        ) {\n          if (hasTokensNow) {\n            // Chain still has tokens - mark for async balance update\n            changed.push(id);\n          } else if (hadTokensBefore) {\n            // Chain had tokens before but doesn't now - clean up balances immediately\n            for (const address of Object.keys(s.tokenBalances)) {\n              const addressKey = address as ChecksumAddress;\n              if (s.tokenBalances[addressKey]?.[id]) {\n                s.tokenBalances[addressKey][id] = {};\n                hasChanges = true;\n              }\n            }\n          }\n        }\n      }\n    });\n\n    this.#allTokens = state.allTokens;\n    this.#detectedTokens = state.allDetectedTokens;\n\n    // Only update balances for chains that still have tokens (and only if we haven't already updated state)\n    if (changed.length && !hasChanges) {\n      this.updateBalances({ chainIds: changed }).catch((error) => {\n        console.warn('Error updating balances after token change:', error);\n      });\n    }\n  };\n\n  readonly #onNetworkChanged = (state: NetworkState) => {\n    // Check if any networks were removed by comparing with previous state\n    const currentNetworks = new Set(\n      Object.keys(state.networkConfigurationsByChainId),\n    );\n\n    // Get all networks that currently have balances\n    const networksWithBalances = new Set<string>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const network of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        networksWithBalances.add(network);\n      }\n    }\n\n    // Find networks that were removed\n    const removedNetworks = Array.from(networksWithBalances).filter(\n      (network) => !currentNetworks.has(network),\n    );\n\n    if (removedNetworks.length > 0) {\n      this.update((s) => {\n        // Remove balances for all accounts on the deleted networks\n        for (const address of Object.keys(s.tokenBalances)) {\n          const addressKey = address as ChecksumAddress;\n          for (const removedNetwork of removedNetworks) {\n            const networkKey = removedNetwork as ChainIdHex;\n            if (s.tokenBalances[addressKey]?.[networkKey]) {\n              delete s.tokenBalances[addressKey][networkKey];\n            }\n          }\n        }\n      });\n    }\n  };\n\n  readonly #onAccountRemoved = (addr: string) => {\n    if (!isStrictHexString(addr) || !isValidHexAddress(addr)) {\n      return;\n    }\n    this.update((s) => {\n      delete s.tokenBalances[addr as ChecksumAddress];\n    });\n  };\n}\n\nexport default TokenBalancesController;\n"]}
+{"version":3,"file":"TokenBalancesController.mjs","sourceRoot":"","sources":["../src/TokenBalancesController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAUxD,OAAO,EACL,OAAO,EACP,iBAAiB,EACjB,oBAAoB,EACpB,KAAK,EACN,mCAAmC;AAQpC,OAAO,EAAE,+BAA+B,EAAE,qCAAqC;AAM/E,OAAO,EAAE,iBAAiB,EAAE,wBAAwB;AACpD,OAAO,EAAE,OAAO,EAAE,cAAc;;;AAOhC,OAAO,EAAE,mCAAmC,EAAE,uCAAmC;AACjF,OAAO,EACL,yBAAyB,EAG1B,+DAA2D;AAC5D,OAAO,EAAE,iBAAiB,EAAE,8CAA0C;AAUtE,MAAM,UAAU,GAAG,yBAAkC,CAAC;AACtD,MAAM,mBAAmB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEjD,MAAM,QAAQ,GAAG;IACf,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;CACnD,CAAC;AA8FF,YAAY;AAEZ,+EAA+E;AAC/E,2BAA2B;AAC3B,MAAM,KAAK,GAAG,CAAI,IAAO,EAAE,EAAkB,EAAK,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAEvE,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAmB,EAAE,CACjD,oBAAoB,CAAC,IAAI,CAAoB,CAAC;AAChD,YAAY;AAEZ,+EAA+E;AAC/E,0BAA0B;AAC1B,MAAM,OAAO,uBAAwB,SAAQ,+BAA+B,EAM3E;IAwBC,YAAY,EACV,SAAS,EACT,QAAQ,GAAG,mBAAmB,EAC9B,qBAAqB,GAAG,EAAE,EAC1B,KAAK,GAAG,EAAE,EACV,qBAAqB,GAAG,IAAI,EAC5B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GACH;QAC/B,KAAK,CAAC;YACJ,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,QAAQ;YACR,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,GAAG,KAAK,EAAE;SACvC,CAAC,CAAC;;QArCI,4DAA2B;QAE3B,2DAAmC;QAE5C,6CAAiD,EAAE,EAAC;QAEpD,kDAA8D,EAAE,EAAC;QAEjE,yEAAyE;QAChE,2DAAyB;QAElC,sCAAsC;QAC7B,8DAA4D;QAErE,gDAAgD;QACvC,yDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,kDAAkD;QAClD,6DAA6B,KAAK,EAAC;QAEnC,mEAAmE;QACnE,qDAAmC,EAAE,EAAC;QAgF7B,+CAAe,CAAC,OAAmB,EAAgB,EAAE;YAC5D,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACtC,wCAAwC,EACxC,eAAe,CAChB,CAAC;YACF,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,EAAC;QAEO,oDAAoB,CAAC,OAAmB,EAAE,EAAE;YACnD,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;YACF,MAAM,GAAG,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC1E,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9B,wCAAwC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC,EAAC;QA+TO,mDAAmB,KAAK,EAAE,KAA4B,EAAE,EAAE;YACjE,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAc,CAAC;YACjD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,kBAAkB,CAAC,GAAG,CAAC,OAAqB,CAAC,CAAC;iBAC/C;aACF;YAED,iFAAiF;YACjF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;gBAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC;aACxC,CAAC,CAAC;YAEH,2FAA2F;YAC3F,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvE,MAAM,EAAE,GAAG,OAAqB,CAAC;gBAEjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;wBAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;oBACpE,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;wBACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEtD,mDAAmD;gBACnD,MAAM,cAAc,GAClB,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;oBAClD,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;gBAElE,2EAA2E;gBAC3E,OAAO,cAAc,IAAI,CAAC,CAAC,eAAe,IAAI,YAAY,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE;gBACjC,kDAAkD;gBAClD,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;gBAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;gBAC/C,OAAO;aACR;YAED,2DAA2D;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE;oBACtC,MAAM,EAAE,GAAG,OAAqB,CAAC;oBACjC,MAAM,YAAY,GAChB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BAC1B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACzD,MAAM,eAAe,GACnB,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC9C,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC;4BACvB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAEtD,IACE,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,0CAAW,CAAC,EAAE,CAAC,CAAC;wBAClD,CAAC,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,uBAAA,IAAI,+CAAgB,CAAC,EAAE,CAAC,CAAC,EAC/D;wBACA,IAAI,YAAY,EAAE;4BAChB,yDAAyD;4BACzD,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;yBAClB;6BAAM,IAAI,eAAe,EAAE;4BAC1B,0EAA0E;4BAC1E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;gCAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;gCAC9C,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oCACrC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;oCACrC,UAAU,GAAG,IAAI,CAAC;iCACnB;6BACF;yBACF;qBACF;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,uBAAA,IAAI,sCAAc,KAAK,CAAC,SAAS,MAAA,CAAC;YAClC,uBAAA,IAAI,2CAAmB,KAAK,CAAC,iBAAiB,MAAA,CAAC;YAE/C,wGAAwG;YACxG,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,EAAE;gBACjC,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACzD,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,KAAmB,EAAE,EAAE;YACnD,sEAAsE;YACtE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAClD,CAAC;YAEF,gDAAgD;YAChD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE;gBAC3D,MAAM,UAAU,GAAG,OAA0B,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAC3C,EAAE;oBACD,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;iBACnC;aACF;YAED,kCAAkC;YAClC,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAC7D,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC3C,CAAC;YAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9B,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAChB,2DAA2D;oBAC3D,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE;wBAClD,MAAM,UAAU,GAAG,OAA0B,CAAC;wBAC9C,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE;4BAC5C,MAAM,UAAU,GAAG,cAA4B,CAAC;4BAChD,IAAI,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE;gCAC7C,OAAO,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;6BAChD;yBACF;qBACF;gBACH,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,EAAC;QAEO,oDAAoB,CAAC,IAAY,EAAE,EAAE;YAC5C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;gBACxD,OAAO;aACR;YACD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,OAAO,CAAC,CAAC,aAAa,CAAC,IAAuB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,EAAC;QAliBA,uBAAA,IAAI,6CAAqB,qBAAqB,MAAA,CAAC;QAC/C,uBAAA,IAAI,4CAAoB,QAAQ,MAAA,CAAC;QACjC,uBAAA,IAAI,+CAAuB,EAAE,GAAG,qBAAqB,EAAE,MAAA,CAAC;QAExD,+CAA+C;QAC/C,uBAAA,IAAI,4CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,yBAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,4CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,iBAAiB,CAAC,uBAAA,IAAI,4CAAa,EAAE,uBAAA,IAAI,iDAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;gBACtE,SAAS,EAAE,uBAAA,IAAI,0CAAW;gBAC1B,iBAAiB,EAAE,uBAAA,IAAI,+CAAgB;aACxC,CAAC,CAAC;SACJ,MAAA,CAAC;QAEF,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,sCAAsC;QACtC,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAChE,2BAA2B,CAC5B,CAAC;QACF,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,2CAAmB,iBAAiB,MAAA,CAAC;QAEzC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,8BAA8B,EAC9B,CAAC,WAAkC,EAAE,EAAE;YACrC,uBAAA,IAAI,gDAAiB,MAArB,IAAI,EAAkB,WAAW,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjD,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,kCAAkC,EAClC,uBAAA,IAAI,iDAAkB,CACvB,CAAC;QAEF,wDAAwD;QACxD,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,mDAAmD,EACnD,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1C,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,+CAA+C,EAC/C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;IACJ,CAAC;IAoCD;;;;;OAKG;IACM,aAAa,CAAC,EAAE,QAAQ,EAA8B;QAC7D,uEAAuE;QACvE,uBAAA,IAAI,8CAAsB,CAAC,GAAG,QAAQ,CAAC,MAAA,CAAC;QACxC,uBAAA,IAAI,sDAA8B,IAAI,MAAA,CAAC;QACvC,uBAAA,IAAI,8FAA2B,MAA/B,IAAI,EAA4B,QAAQ,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IAkGD;;OAEG;IACM,+BAA+B;QACtC,uBAAA,IAAI,sDAA8B,KAAK,MAAA,CAAC;QACxC,uBAAA,IAAI,8CAAsB,EAAE,MAAA,CAAC,CAAC,sCAAsC;QACpE,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;IACtC,CAAC;IAED;;;;;OAKG;IACH,qBAAqB,CAAC,OAAmB;QACvC,OAAO,CACL,uBAAA,IAAI,mDAAoB,CAAC,OAAO,CAAC,IAAI;YACnC,QAAQ,EAAE,uBAAA,IAAI,gDAAiB;SAChC,CACF,CAAC;IACJ,CAAC;IAEQ,KAAK,CAAC,YAAY,CAAC,EAAE,QAAQ,EAA8B;QAClE,kFAAkF;QAClF,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;;OAMG;IACH,yBAAyB,CACvB,OAA+C,EAC/C,UAA4C,EAAE,eAAe,EAAE,IAAI,EAAE;QAErE,MAAM,CAAC,MAAM,CAAC,uBAAA,IAAI,mDAAoB,EAAE,OAAO,CAAC,CAAC;QAEjD,sEAAsE;QACtE,IAAI,uBAAA,IAAI,0DAA2B,EAAE;YACnC,8EAA8E;YAC9E,uBAAA,IAAI,8FAA2B,MAA/B,IAAI,EACF,uBAAA,IAAI,kDAAmB,EACvB,OAAO,CAAC,eAAe,CACxB,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAE,QAAQ,KAAkC,EAAE;QACjE,MAAM,YAAY,GAAG,QAAQ,IAAI,uBAAA,IAAI,uFAAoB,MAAxB,IAAI,CAAsB,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,OAAO;SACR;QAED,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACrD,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QAEF,MAAM,UAAU,GAAuB,EAAE,CAAC;QAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;QAExC,oEAAoE;QACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,gDAAiB,EAAE;YAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;gBAC3B,SAAS;aACV;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;oBACnC,QAAQ,EAAE,eAAe;oBACzB,gBAAgB,EAAE,uBAAA,IAAI,iDAAkB;oBACxC,eAAe,EAAE,QAA2B;oBAC5C,WAAW;iBACZ,CAAC,CAAC;gBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;oBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;oBAC7B,iDAAiD;oBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;iBACH;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;gBACF,sCAAsC;aACvC;YAED,iDAAiD;YACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;gBAChC,MAAM;aACP;SACF;QAED,sCAAsC;QACtC,MAAM,iBAAiB,GAAG,uBAAA,IAAI,iDAAkB;YAC9C,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAA0B,CAAC;YACtD,CAAC,CAAC,CAAC,QAA2B,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YAC7B,mEAAmE;YACnE,+CAA+C;YAC/C,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE;gBAClC,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE;oBACvC,mCAAmC;oBACnC,MAAM,WAAW,GAAG,uBAAA,IAAI,0CAAW,CAAC,OAAO,CAAC,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE;wBAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACzC,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;oBAED,2CAA2C;oBAC3C,MAAM,mBAAmB,GAAG,uBAAA,IAAI,+CAAgB,CAAC,OAAO,CAAC,CAAC;oBAC1D,IAAI,mBAAmB,EAAE,CAAC,OAAO,CAAC,EAAE;wBAClC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CACjD,CAAC,KAA0B,EAAE,EAAE;;4BAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC7C,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CACjD,YAAY,CACb,GAAG,KAAK,CAAC;wBACZ,CAAC,CACF,CAAC;qBACH;iBACF;aACF;YAED,2DAA2D;YAC3D,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;;gBACjE,IAAI,OAAO,IAAI,KAAK,KAAK,SAAS,EAAE;oBAClC,OAAC,OAAC,CAAC,CAAC,aAAa,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,EAAC,OAAO,SAAP,OAAO,IAAM,EAAE,EAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClE,KAAK,CAAC,KAAK,CAAC,CAAC;iBAChB;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAExB,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,YAAY,CAC7C,CAAC;YAEF,kFAAkF;YAClF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACtD,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBACxD,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,cAAc,CACf,CAAC;aACH;YAED,+CAA+C;YAC/C,MAAM,wBAAwB,GAAG,MAAM,CAAC,MAAM,CAC5C,mCAAmC,CACpC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YAEpC,uFAAuF;YACvF,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7C,OAAO,CACL,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,KAAK,KAAK,YAAY;oBACxB,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CACzD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC5D,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;iBAC5D,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,+CAA+C,EAC/C,oBAAoB,CACrB,CAAC;aACH;SACF;IACH,CAAC;IAED,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC7C,CAAC;IAkJD;;OAEG;IACM,OAAO;QACd,uBAAA,IAAI,sDAA8B,KAAK,MAAA,CAAC;QACxC,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;QAEpC,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAC1C,mDAAmD,CACpD,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAC1C,+CAA+C,CAChD,CAAC;QAEF,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;;IAhgBG,OAAO;QACL,GAAG,IAAI,GAAG,CAAC;YACT,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,0CAAW,CAAC;YAC/B,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,+CAAgB,CAAC;SACrC,CAAC;KACa,CAAC;AACpB,CAAC,mHA8C0B,QAAsB,EAAE,SAAS,GAAG,IAAI;IACjE,oCAAoC;IACpC,uBAAA,IAAI,sDAAuB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACrE,uBAAA,IAAI,sDAAuB,CAAC,KAAK,EAAE,CAAC;IAEpC,0CAA0C;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEvD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3D,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;KAC/C;IAED,sDAAsD;IACtD,KAAK,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,IAAI,cAAc,EAAE;QACtD,uBAAA,IAAI,4FAAyB,MAA7B,IAAI,EAA0B,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;KACnE;AACH,CAAC,+GAUC,QAAgB,EAChB,QAAsB,EACtB,SAAS,GAAG,IAAI;IAEhB,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC9B,IAAI,CAAC,uBAAA,IAAI,0DAA2B,EAAE;YACpC,OAAO;SACR;QACD,IAAI;YACF,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;SACvC;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,IAAI,CACV,6BAA6B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,QAAQ,GAAG,EAC7E,KAAK,CACN,CAAC;SACH;IACH,CAAC,CAAC;IAEF,sCAAsC;IACtC,IAAI,SAAS,EAAE;QACb,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7B,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAC7D,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;KACJ;IAED,sCAAsC;IACtC,uBAAA,IAAI,oFAAiB,MAArB,IAAI,EAAkB,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;AAC1D,CAAC,+FAUC,QAAgB,EAChB,QAAsB,EACtB,YAAiC;IAEjC,mDAAmD;IACnD,MAAM,aAAa,GAAG,uBAAA,IAAI,sDAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAChE,IAAI,aAAa,EAAE;QACjB,aAAa,CAAC,aAAa,CAAC,CAAC;KAC9B;IAED,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7B,OAAO,CAAC,IAAI,CACV,sCAAsC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAC5D,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,QAAQ,CAAC,CAAC;IACb,uBAAA,IAAI,sDAAuB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;AACnD,CAAC;AAsXH,eAAe,uBAAuB,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  BNToHex,\n  isValidHexAddress,\n  toChecksumHexAddress,\n  toHex,\n} from '@metamask/controller-utils';\nimport type { KeyringControllerAccountRemovedEvent } from '@metamask/keyring-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n  NetworkState,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type {\n  PreferencesControllerGetStateAction,\n  PreferencesControllerStateChangeEvent,\n} from '@metamask/preferences-controller';\nimport type { Hex } from '@metamask/utils';\nimport { isStrictHexString } from '@metamask/utils';\nimport { produce } from 'immer';\nimport { isEqual } from 'lodash';\n\nimport type {\n  AccountTrackerUpdateNativeBalancesAction,\n  AccountTrackerUpdateStakedBalancesAction,\n} from './AccountTrackerController';\nimport { STAKING_CONTRACT_ADDRESS_BY_CHAINID } from './AssetsContractController';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\nimport { RpcBalanceFetcher } from './rpc-service/rpc-balance-fetcher';\nimport type {\n  TokensControllerGetStateAction,\n  TokensControllerState,\n  TokensControllerStateChangeEvent,\n} from './TokensController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst CONTROLLER = 'TokenBalancesController' as const;\nconst DEFAULT_INTERVAL_MS = 180_000; // 3 minutes\n\nconst metadata = {\n  tokenBalances: { persist: true, anonymous: false },\n};\n\n// account → chain → token → balance\nexport type TokenBalances = Record<\n  ChecksumAddress,\n  Record<ChainIdHex, Record<ChecksumAddress, Hex>>\n>;\n\nexport type TokenBalancesControllerState = {\n  tokenBalances: TokenBalances;\n};\n\nexport type TokenBalancesControllerGetStateAction = ControllerGetStateAction<\n  typeof CONTROLLER,\n  TokenBalancesControllerState\n>;\n\nexport type TokenBalancesControllerUpdateChainPollingConfigsAction = {\n  type: `TokenBalancesController:updateChainPollingConfigs`;\n  handler: TokenBalancesController['updateChainPollingConfigs'];\n};\n\nexport type TokenBalancesControllerGetChainPollingConfigAction = {\n  type: `TokenBalancesController:getChainPollingConfig`;\n  handler: TokenBalancesController['getChainPollingConfig'];\n};\n\nexport type TokenBalancesControllerActions =\n  | TokenBalancesControllerGetStateAction\n  | TokenBalancesControllerUpdateChainPollingConfigsAction\n  | TokenBalancesControllerGetChainPollingConfigAction;\n\nexport type TokenBalancesControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof CONTROLLER, TokenBalancesControllerState>;\n\nexport type NativeBalanceEvent = {\n  type: `${typeof CONTROLLER}:updatedNativeBalance`;\n  payload: unknown[];\n};\n\nexport type TokenBalancesControllerEvents =\n  | TokenBalancesControllerStateChangeEvent\n  | NativeBalanceEvent;\n\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | TokensControllerGetStateAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | AccountsControllerListAccountsAction\n  | AccountTrackerUpdateNativeBalancesAction\n  | AccountTrackerUpdateStakedBalancesAction;\n\nexport type AllowedEvents =\n  | TokensControllerStateChangeEvent\n  | PreferencesControllerStateChangeEvent\n  | NetworkControllerStateChangeEvent\n  | KeyringControllerAccountRemovedEvent;\n\nexport type TokenBalancesControllerMessenger = RestrictedMessenger<\n  typeof CONTROLLER,\n  TokenBalancesControllerActions | AllowedActions,\n  TokenBalancesControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\nexport type ChainPollingConfig = {\n  /** Polling interval in milliseconds for this chain */\n  interval: number;\n};\n\nexport type UpdateChainPollingConfigsOptions = {\n  /** Whether to immediately fetch balances after updating configs (default: true) */\n  immediateUpdate?: boolean;\n};\n\nexport type TokenBalancesControllerOptions = {\n  messenger: TokenBalancesControllerMessenger;\n  /** Default interval for chains not specified in chainPollingIntervals */\n  interval?: number;\n  /** Per-chain polling configuration */\n  chainPollingIntervals?: Record<ChainIdHex, ChainPollingConfig>;\n  state?: Partial<TokenBalancesControllerState>;\n  /** When `true`, balances for *all* known accounts are queried. */\n  queryMultipleAccounts?: boolean;\n  /** Enable Accounts‑API strategy (if supported chain). */\n  useAccountsAPI?: boolean;\n  /** Disable external HTTP calls (privacy / offline mode). */\n  allowExternalServices?: () => boolean;\n  /** Custom logger. */\n  log?: (...args: unknown[]) => void;\n};\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Helper utilities\nconst draft = <T>(base: T, fn: (d: T) => void): T => produce(base, fn);\n\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\nconst checksum = (addr: string): ChecksumAddress =>\n  toChecksumHexAddress(addr) as ChecksumAddress;\n// endregion\n\n// ────────────────────────────────────────────────────────────────────────────\n// region: Main controller\nexport class TokenBalancesController extends StaticIntervalPollingController<{\n  chainIds: ChainIdHex[];\n}>()<\n  typeof CONTROLLER,\n  TokenBalancesControllerState,\n  TokenBalancesControllerMessenger\n> {\n  readonly #queryAllAccounts: boolean;\n\n  readonly #balanceFetchers: BalanceFetcher[];\n\n  #allTokens: TokensControllerState['allTokens'] = {};\n\n  #detectedTokens: TokensControllerState['allDetectedTokens'] = {};\n\n  /** Default polling interval for chains without specific configuration */\n  readonly #defaultInterval: number;\n\n  /** Per-chain polling configuration */\n  readonly #chainPollingConfig: Record<ChainIdHex, ChainPollingConfig>;\n\n  /** Active polling timers grouped by interval */\n  readonly #intervalPollingTimers: Map<number, NodeJS.Timeout> = new Map();\n\n  /** Track if controller-level polling is active */\n  #isControllerPollingActive = false;\n\n  /** Store original chainIds from startPolling to preserve intent */\n  #requestedChainIds: ChainIdHex[] = [];\n\n  constructor({\n    messenger,\n    interval = DEFAULT_INTERVAL_MS,\n    chainPollingIntervals = {},\n    state = {},\n    queryMultipleAccounts = true,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: TokenBalancesControllerOptions) {\n    super({\n      name: CONTROLLER,\n      messenger,\n      metadata,\n      state: { tokenBalances: {}, ...state },\n    });\n\n    this.#queryAllAccounts = queryMultipleAccounts;\n    this.#defaultInterval = interval;\n    this.#chainPollingConfig = { ...chainPollingIntervals };\n\n    // Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new RpcBalanceFetcher(this.#getProvider, this.#getNetworkClient, () => ({\n        allTokens: this.#allTokens,\n        allDetectedTokens: this.#detectedTokens,\n      })),\n    ];\n\n    this.setIntervalLength(interval);\n\n    // initial token state & subscriptions\n    const { allTokens, allDetectedTokens } = this.messagingSystem.call(\n      'TokensController:getState',\n    );\n    this.#allTokens = allTokens;\n    this.#detectedTokens = allDetectedTokens;\n\n    this.messagingSystem.subscribe(\n      'TokensController:stateChange',\n      (tokensState: TokensControllerState) => {\n        this.#onTokensChanged(tokensState).catch((error) => {\n          console.warn('Error handling token state change:', error);\n        });\n      },\n    );\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      this.#onNetworkChanged,\n    );\n    this.messagingSystem.subscribe(\n      'KeyringController:accountRemoved',\n      this.#onAccountRemoved,\n    );\n\n    // Register action handlers for polling interval control\n    this.messagingSystem.registerActionHandler(\n      `TokenBalancesController:updateChainPollingConfigs`,\n      this.updateChainPollingConfigs.bind(this),\n    );\n\n    this.messagingSystem.registerActionHandler(\n      `TokenBalancesController:getChainPollingConfig`,\n      this.getChainPollingConfig.bind(this),\n    );\n  }\n\n  #chainIdsWithTokens(): ChainIdHex[] {\n    return [\n      ...new Set([\n        ...Object.keys(this.#allTokens),\n        ...Object.keys(this.#detectedTokens),\n      ]),\n    ] as ChainIdHex[];\n  }\n\n  readonly #getProvider = (chainId: ChainIdHex): Web3Provider => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    const client = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n    return new Web3Provider(client.provider);\n  };\n\n  readonly #getNetworkClient = (chainId: ChainIdHex) => {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const cfg = networkConfigurationsByChainId[chainId];\n    const { networkClientId } = cfg.rpcEndpoints[cfg.defaultRpcEndpointIndex];\n    return this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      networkClientId,\n    );\n  };\n\n  /**\n   * Override to support per-chain polling intervals by grouping chains by interval\n   *\n   * @param options0 - The polling options\n   * @param options0.chainIds - Chain IDs to start polling for\n   */\n  override _startPolling({ chainIds }: { chainIds: ChainIdHex[] }) {\n    // Store the original chainIds to preserve intent across config updates\n    this.#requestedChainIds = [...chainIds];\n    this.#isControllerPollingActive = true;\n    this.#startIntervalGroupPolling(chainIds, true);\n  }\n\n  /**\n   * Start or restart interval-based polling for multiple chains\n   *\n   * @param chainIds - Chain IDs to start polling for\n   * @param immediate - Whether to poll immediately before starting timers (default: true)\n   */\n  #startIntervalGroupPolling(chainIds: ChainIdHex[], immediate = true) {\n    // Stop any existing interval timers\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n\n    // Group chains by their polling intervals\n    const intervalGroups = new Map<number, ChainIdHex[]>();\n\n    for (const chainId of chainIds) {\n      const config = this.getChainPollingConfig(chainId);\n      const existing = intervalGroups.get(config.interval) || [];\n      existing.push(chainId);\n      intervalGroups.set(config.interval, existing);\n    }\n\n    // Start separate polling loop for each interval group\n    for (const [interval, chainIdsGroup] of intervalGroups) {\n      this.#startPollingForInterval(interval, chainIdsGroup, immediate);\n    }\n  }\n\n  /**\n   * Start polling loop for chains that share the same interval\n   *\n   * @param interval - The polling interval in milliseconds\n   * @param chainIds - Chain IDs that share this interval\n   * @param immediate - Whether to poll immediately before starting the timer (default: true)\n   */\n  #startPollingForInterval(\n    interval: number,\n    chainIds: ChainIdHex[],\n    immediate = true,\n  ) {\n    const pollFunction = async () => {\n      if (!this.#isControllerPollingActive) {\n        return;\n      }\n      try {\n        await this._executePoll({ chainIds });\n      } catch (error) {\n        console.warn(\n          `Polling failed for chains ${chainIds.join(', ')} with interval ${interval}:`,\n          error,\n        );\n      }\n    };\n\n    // Poll immediately first if requested\n    if (immediate) {\n      pollFunction().catch((error) => {\n        console.warn(\n          `Immediate polling failed for chains ${chainIds.join(', ')}:`,\n          error,\n        );\n      });\n    }\n\n    // Then start regular interval polling\n    this.#setPollingTimer(interval, chainIds, pollFunction);\n  }\n\n  /**\n   * Helper method to set up polling timer\n   *\n   * @param interval - The polling interval in milliseconds\n   * @param chainIds - Chain IDs for this interval\n   * @param pollFunction - The function to call on each poll\n   */\n  #setPollingTimer(\n    interval: number,\n    chainIds: ChainIdHex[],\n    pollFunction: () => Promise<void>,\n  ) {\n    // Clear any existing timer for this interval first\n    const existingTimer = this.#intervalPollingTimers.get(interval);\n    if (existingTimer) {\n      clearInterval(existingTimer);\n    }\n\n    const timer = setInterval(() => {\n      pollFunction().catch((error) => {\n        console.warn(\n          `Interval polling failed for chains ${chainIds.join(', ')}:`,\n          error,\n        );\n      });\n    }, interval);\n    this.#intervalPollingTimers.set(interval, timer);\n  }\n\n  /**\n   * Override to handle our custom polling approach\n   */\n  override _stopPollingByPollingTokenSetId() {\n    this.#isControllerPollingActive = false;\n    this.#requestedChainIds = []; // Clear original intent when stopping\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n  }\n\n  /**\n   * Get polling configuration for a chain (includes default fallback)\n   *\n   * @param chainId - The chain ID to get config for\n   * @returns The polling configuration for the chain\n   */\n  getChainPollingConfig(chainId: ChainIdHex): ChainPollingConfig {\n    return (\n      this.#chainPollingConfig[chainId] ?? {\n        interval: this.#defaultInterval,\n      }\n    );\n  }\n\n  override async _executePoll({ chainIds }: { chainIds: ChainIdHex[] }) {\n    // This won't be called with our custom implementation, but keep for compatibility\n    await this.updateBalances({ chainIds });\n  }\n\n  /**\n   * Update multiple chain polling configurations at once\n   *\n   * @param configs - Object mapping chain IDs to polling configurations\n   * @param options - Optional configuration for the update behavior\n   * @param options.immediateUpdate - Whether to immediately fetch balances after updating configs (default: true)\n   */\n  updateChainPollingConfigs(\n    configs: Record<ChainIdHex, ChainPollingConfig>,\n    options: UpdateChainPollingConfigsOptions = { immediateUpdate: true },\n  ): void {\n    Object.assign(this.#chainPollingConfig, configs);\n\n    // If polling is currently active, restart with new interval groupings\n    if (this.#isControllerPollingActive) {\n      // Restart polling with immediate fetch by default, unless explicitly disabled\n      this.#startIntervalGroupPolling(\n        this.#requestedChainIds,\n        options.immediateUpdate,\n      );\n    }\n  }\n\n  async updateBalances({ chainIds }: { chainIds?: ChainIdHex[] } = {}) {\n    const targetChains = chainIds ?? this.#chainIdsWithTokens();\n    if (!targetChains.length) {\n      return;\n    }\n\n    const { address: selected } = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const aggregated: ProcessedBalance[] = [];\n    let remainingChains = [...targetChains];\n\n    // Try each fetcher in order, removing successfully processed chains\n    for (const fetcher of this.#balanceFetchers) {\n      const supportedChains = remainingChains.filter((c) =>\n        fetcher.supports(c),\n      );\n      if (!supportedChains.length) {\n        continue;\n      }\n\n      try {\n        const balances = await fetcher.fetch({\n          chainIds: supportedChains,\n          queryAllAccounts: this.#queryAllAccounts,\n          selectedAccount: selected as ChecksumAddress,\n          allAccounts,\n        });\n\n        if (balances && balances.length > 0) {\n          aggregated.push(...balances);\n          // Remove chains that were successfully processed\n          const processedChains = new Set(balances.map((b) => b.chainId));\n          remainingChains = remainingChains.filter(\n            (chain) => !processedChains.has(chain),\n          );\n        }\n      } catch (error) {\n        console.warn(\n          `Balance fetcher failed for chains ${supportedChains.join(', ')}: ${String(error)}`,\n        );\n        // Continue to next fetcher (fallback)\n      }\n\n      // If all chains have been processed, break early\n      if (remainingChains.length === 0) {\n        break;\n      }\n    }\n\n    // Determine which accounts to process\n    const accountsToProcess = this.#queryAllAccounts\n      ? allAccounts.map((a) => a.address as ChecksumAddress)\n      : [selected as ChecksumAddress];\n\n    const prev = this.state;\n    const next = draft(prev, (d) => {\n      // First, initialize all tokens from allTokens state with balance 0\n      // for the accounts and chains we're processing\n      for (const chainId of targetChains) {\n        for (const account of accountsToProcess) {\n          // Initialize tokens from allTokens\n          const chainTokens = this.#allTokens[chainId];\n          if (chainTokens?.[account]) {\n            Object.values(chainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n\n          // Initialize tokens from allDetectedTokens\n          const detectedChainTokens = this.#detectedTokens[chainId];\n          if (detectedChainTokens?.[account]) {\n            Object.values(detectedChainTokens[account]).forEach(\n              (token: { address: string }) => {\n                const tokenAddress = checksum(token.address);\n                ((d.tokenBalances[account] ??= {})[chainId] ??= {})[\n                  tokenAddress\n                ] = '0x0';\n              },\n            );\n          }\n        }\n      }\n\n      // Then update with actual fetched balances where available\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          ((d.tokenBalances[account] ??= {})[chainId] ??= {})[checksum(token)] =\n            toHex(value);\n        }\n      });\n    });\n\n    if (!isEqual(prev, next)) {\n      this.update(() => next);\n\n      const nativeBalances = aggregated.filter(\n        (r) => r.success && r.token === ZERO_ADDRESS,\n      );\n\n      // Update native token balances in a single batch operation for better performance\n      if (nativeBalances.length > 0) {\n        const balanceUpdates = nativeBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          balance: balance.value ? BNToHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateNativeBalances',\n          balanceUpdates,\n        );\n      }\n\n      // Get staking contract addresses for filtering\n      const stakingContractAddresses = Object.values(\n        STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n      ).map((addr) => addr.toLowerCase());\n\n      // Filter and update staked balances in a single batch operation for better performance\n      const stakedBalances = aggregated.filter((r) => {\n        return (\n          r.success &&\n          r.token !== ZERO_ADDRESS &&\n          stakingContractAddresses.includes(r.token.toLowerCase())\n        );\n      });\n\n      if (stakedBalances.length > 0) {\n        const stakedBalanceUpdates = stakedBalances.map((balance) => ({\n          address: balance.account,\n          chainId: balance.chainId,\n          stakedBalance: balance.value ? toHex(balance.value) : '0x0',\n        }));\n\n        this.messagingSystem.call(\n          'AccountTrackerController:updateStakedBalances',\n          stakedBalanceUpdates,\n        );\n      }\n    }\n  }\n\n  resetState() {\n    this.update(() => ({ tokenBalances: {} }));\n  }\n\n  readonly #onTokensChanged = async (state: TokensControllerState) => {\n    const changed: ChainIdHex[] = [];\n    let hasChanges = false;\n\n    // Get chains that have existing balances\n    const chainsWithBalances = new Set<ChainIdHex>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const chainId of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        chainsWithBalances.add(chainId as ChainIdHex);\n      }\n    }\n\n    // Only process chains that are explicitly mentioned in the incoming state change\n    const incomingChainIds = new Set([\n      ...Object.keys(state.allTokens),\n      ...Object.keys(state.allDetectedTokens),\n    ]);\n\n    // Only proceed if there are actual changes to chains that have balances or are being added\n    const relevantChainIds = Array.from(incomingChainIds).filter((chainId) => {\n      const id = chainId as ChainIdHex;\n\n      const hasTokensNow =\n        (state.allTokens[id] && Object.keys(state.allTokens[id]).length > 0) ||\n        (state.allDetectedTokens[id] &&\n          Object.keys(state.allDetectedTokens[id]).length > 0);\n      const hadTokensBefore =\n        (this.#allTokens[id] && Object.keys(this.#allTokens[id]).length > 0) ||\n        (this.#detectedTokens[id] &&\n          Object.keys(this.#detectedTokens[id]).length > 0);\n\n      // Check if there's an actual change in token state\n      const hasTokenChange =\n        !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n        !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id]);\n\n      // Process chains that have actual changes OR are new chains getting tokens\n      return hasTokenChange || (!hadTokensBefore && hasTokensNow);\n    });\n\n    if (relevantChainIds.length === 0) {\n      // No relevant changes, just update internal state\n      this.#allTokens = state.allTokens;\n      this.#detectedTokens = state.allDetectedTokens;\n      return;\n    }\n\n    // Handle both cleanup and updates in a single state update\n    this.update((s) => {\n      for (const chainId of relevantChainIds) {\n        const id = chainId as ChainIdHex;\n        const hasTokensNow =\n          (state.allTokens[id] &&\n            Object.keys(state.allTokens[id]).length > 0) ||\n          (state.allDetectedTokens[id] &&\n            Object.keys(state.allDetectedTokens[id]).length > 0);\n        const hadTokensBefore =\n          (this.#allTokens[id] &&\n            Object.keys(this.#allTokens[id]).length > 0) ||\n          (this.#detectedTokens[id] &&\n            Object.keys(this.#detectedTokens[id]).length > 0);\n\n        if (\n          !isEqual(state.allTokens[id], this.#allTokens[id]) ||\n          !isEqual(state.allDetectedTokens[id], this.#detectedTokens[id])\n        ) {\n          if (hasTokensNow) {\n            // Chain still has tokens - mark for async balance update\n            changed.push(id);\n          } else if (hadTokensBefore) {\n            // Chain had tokens before but doesn't now - clean up balances immediately\n            for (const address of Object.keys(s.tokenBalances)) {\n              const addressKey = address as ChecksumAddress;\n              if (s.tokenBalances[addressKey]?.[id]) {\n                s.tokenBalances[addressKey][id] = {};\n                hasChanges = true;\n              }\n            }\n          }\n        }\n      }\n    });\n\n    this.#allTokens = state.allTokens;\n    this.#detectedTokens = state.allDetectedTokens;\n\n    // Only update balances for chains that still have tokens (and only if we haven't already updated state)\n    if (changed.length && !hasChanges) {\n      this.updateBalances({ chainIds: changed }).catch((error) => {\n        console.warn('Error updating balances after token change:', error);\n      });\n    }\n  };\n\n  readonly #onNetworkChanged = (state: NetworkState) => {\n    // Check if any networks were removed by comparing with previous state\n    const currentNetworks = new Set(\n      Object.keys(state.networkConfigurationsByChainId),\n    );\n\n    // Get all networks that currently have balances\n    const networksWithBalances = new Set<string>();\n    for (const address of Object.keys(this.state.tokenBalances)) {\n      const addressKey = address as ChecksumAddress;\n      for (const network of Object.keys(\n        this.state.tokenBalances[addressKey] || {},\n      )) {\n        networksWithBalances.add(network);\n      }\n    }\n\n    // Find networks that were removed\n    const removedNetworks = Array.from(networksWithBalances).filter(\n      (network) => !currentNetworks.has(network),\n    );\n\n    if (removedNetworks.length > 0) {\n      this.update((s) => {\n        // Remove balances for all accounts on the deleted networks\n        for (const address of Object.keys(s.tokenBalances)) {\n          const addressKey = address as ChecksumAddress;\n          for (const removedNetwork of removedNetworks) {\n            const networkKey = removedNetwork as ChainIdHex;\n            if (s.tokenBalances[addressKey]?.[networkKey]) {\n              delete s.tokenBalances[addressKey][networkKey];\n            }\n          }\n        }\n      });\n    }\n  };\n\n  readonly #onAccountRemoved = (addr: string) => {\n    if (!isStrictHexString(addr) || !isValidHexAddress(addr)) {\n      return;\n    }\n    this.update((s) => {\n      delete s.tokenBalances[addr as ChecksumAddress];\n    });\n  };\n\n  /**\n   * Clean up all timers and resources when controller is destroyed\n   */\n  override destroy(): void {\n    this.#isControllerPollingActive = false;\n    this.#intervalPollingTimers.forEach((timer) => clearInterval(timer));\n    this.#intervalPollingTimers.clear();\n\n    // Unregister action handlers\n    this.messagingSystem.unregisterActionHandler(\n      `TokenBalancesController:updateChainPollingConfigs`,\n    );\n    this.messagingSystem.unregisterActionHandler(\n      `TokenBalancesController:getChainPollingConfig`,\n    );\n\n    super.destroy();\n  }\n}\n\nexport default TokenBalancesController;\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metamask/assets-controllers",
-  "version": "74.3.2",
+  "version": "74.3.3",
   "description": "Controllers which manage interactions involving ERC-20, ERC-721, and ERC-1155 tokens (including NFTs)",
   "keywords": [
     "MetaMask",
@@ -54,7 +54,7 @@
     "@ethersproject/contracts": "^5.7.0",
     "@ethersproject/providers": "^5.7.0",
     "@metamask/abi-utils": "^2.0.3",
-    "@metamask/base-controller": "^8.2.0",
+    "@metamask/base-controller": "^8.3.0",
     "@metamask/contract-metadata": "^2.4.0",
     "@metamask/controller-utils": "^11.12.0",
     "@metamask/eth-query": "^4.0.0",
@@ -80,7 +80,7 @@
   "devDependencies": {
     "@babel/runtime": "^7.23.9",
     "@metamask/account-api": "^0.9.0",
-    "@metamask/account-tree-controller": "^0.12.0",
+    "@metamask/account-tree-controller": "^0.12.1",
     "@metamask/accounts-controller": "^33.0.0",
     "@metamask/approval-controller": "^7.1.3",
     "@metamask/auto-changelog": "^3.4.4",
@@ -95,7 +95,7 @@
     "@metamask/preferences-controller": "^19.0.0",
     "@metamask/providers": "^22.1.0",
     "@metamask/snaps-controllers": "^14.0.1",
-    "@metamask/transaction-controller": "^60.1.0",
+    "@metamask/transaction-controller": "^60.2.0",
     "@types/jest": "^27.4.1",
     "@types/lodash": "^4.14.191",
     "@types/node": "^16.18.54",