@metamask/assets-controllers 73.1.0 → 73.2.0

package/CHANGELOG.md

@@ -7,6 +7,33 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [73.2.0]
+
+### Added
+
+- Implement balance change calculator and network filtering ([#6285](https://github.com/MetaMask/core/pull/6285))
+  - Add core balance change calculators with period support (1d/7d/30d), network filtering, and group-level computation
+- Add new utility functions for efficient balance fetching using Multicall3 ([#6212](https://github.com/MetaMask/core/pull/6212))
+  - Added `aggregate3` function for direct access to Multicall3's aggregate3 method with individual failure handling
+  - Added `getTokenBalancesForMultipleAddresses` function to efficiently batch ERC20 and native token balance queries for multiple addresses
+  - Supports up to 300 calls per batch with automatic fallback to individual calls on unsupported chains
+  - Returns organized balance data as nested maps for easy consumption by client applications
+
+### Changed
+
+- **BREAKING**: Improved `TokenBalancesController` performance with two-tier balance fetching strategy ([#6232](https://github.com/MetaMask/core/pull/6232))
+  - Implements Accounts API as primary fetching method for supported networks (faster, more efficient)
+  - Falls back to RPC calls using Multicall3's `aggregate3` for unsupported networks or API failures
+  - Significantly reduces RPC calls from N individual requests to batched calls of up to 300 operations
+  - Provides comprehensive network coverage with graceful degradation when services are unavailable
+- Bump `@metamask/base-controller` from `^8.0.1` to `^8.1.0` ([#6284](https://github.com/MetaMask/core/pull/6284))
+- Bump `@metamask/controller-utils` from `^11.11.0` to `^11.12.0` ([#6303](https://github.com/MetaMask/core/pull/6303))
+- Bump `@metamask/transaction-controller` from `^59.1.0` to `^59.2.0` ([#6291](https://github.com/MetaMask/core/pull/6291))
+- Bump `@metamask/account-tree-controller` from `^0.7.0` to `^0.8.0` ([#6273](https://github.com/MetaMask/core/pull/6273))
+- Bump `@metamask/accounts-controller` from `^32.0.1` to `^32.0.2` ([#6273](https://github.com/MetaMask/core/pull/6273))
+- Bump `@metamask/keyring-controller` from `^22.1.0` to `^22.1.1` ([#6273](https://github.com/MetaMask/core/pull/6273))
+- Bump `@metamask/multichain-account-service` from `^0.3.0` to `^0.4.0` ([#6273](https://github.com/MetaMask/core/pull/6273))
+
 ## [73.1.0]
 
 ### Added
@@ -1826,7 +1853,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@73.1.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.2.0...HEAD
+[73.2.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.1.0...@metamask/assets-controllers@73.2.0
 [73.1.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.0.2...@metamask/assets-controllers@73.1.0
 [73.0.2]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.0.1...@metamask/assets-controllers@73.0.2
 [73.0.1]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.0.0...@metamask/assets-controllers@73.0.1

package/dist/AccountTrackerController.cjs

@@ -13,7 +13,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
-var _AccountTrackerController_instances, _AccountTrackerController_refreshMutex, _AccountTrackerController_includeStakedAssets, _AccountTrackerController_getStakedBalanceForChain, _AccountTrackerController_getCorrectNetworkClient, _AccountTrackerController_getNetworkClientIds, _AccountTrackerController_getBalanceFromChain;
+var _AccountTrackerController_instances, _AccountTrackerController_refreshMutex, _AccountTrackerController_includeStakedAssets, _AccountTrackerController_getStakedBalanceForChain, _AccountTrackerController_getCorrectNetworkClient, _AccountTrackerController_getNetworkClientIds, _AccountTrackerController_getBalanceFromChain, _AccountTrackerController_registerMessageHandlers;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AccountTrackerController = void 0;
 const contracts_1 = require("@ethersproject/contracts");
@@ -79,6 +79,7 @@ class AccountTrackerController extends (0, polling_controller_1.StaticIntervalPo
                 this.refresh(__classPrivateFieldGet(this, _AccountTrackerController_instances, "m", _AccountTrackerController_getNetworkClientIds).call(this));
             }
         }, (event) => event.address);
+        __classPrivateFieldGet(this, _AccountTrackerController_instances, "m", _AccountTrackerController_registerMessageHandlers).call(this);
     }
     syncAccounts(newChainIds) {
         const accountsByChainId = (0, lodash_1.cloneDeep)(this.state.accountsByChainId);
@@ -269,6 +270,52 @@ class AccountTrackerController extends (0, polling_controller_1.StaticIntervalPo
             }, {});
         });
     }
+    /**
+     * Updates the balances of multiple native tokens in a single batch operation.
+     * This is more efficient than calling updateNativeToken multiple times as it
+     * triggers only one state update.
+     *
+     * @param balances - Array of balance updates, each containing address, chainId, and balance.
+     */
+    updateNativeBalances(balances) {
+        this.update((state) => {
+            balances.forEach(({ address, chainId, balance }) => {
+                // Ensure the chainId exists in the state
+                if (!state.accountsByChainId[chainId]) {
+                    state.accountsByChainId[chainId] = {};
+                }
+                // Ensure the address exists for this chain
+                if (!state.accountsByChainId[chainId][address]) {
+                    state.accountsByChainId[chainId][address] = { balance: '0x0' };
+                }
+                // Update the balance
+                state.accountsByChainId[chainId][address].balance = balance;
+            });
+        });
+    }
+    /**
+     * Updates the staked balances of multiple accounts in a single batch operation.
+     * This is more efficient than updating staked balances individually as it
+     * triggers only one state update.
+     *
+     * @param stakedBalances - Array of staked balance updates, each containing address, chainId, and stakedBalance.
+     */
+    updateStakedBalances(stakedBalances) {
+        this.update((state) => {
+            stakedBalances.forEach(({ address, chainId, stakedBalance }) => {
+                // Ensure the chainId exists in the state
+                if (!state.accountsByChainId[chainId]) {
+                    state.accountsByChainId[chainId] = {};
+                }
+                // Ensure the address exists for this chain
+                if (!state.accountsByChainId[chainId][address]) {
+                    state.accountsByChainId[chainId][address] = { balance: '0x0' };
+                }
+                // Update the staked balance
+                state.accountsByChainId[chainId][address].stakedBalance = stakedBalance;
+            });
+        });
+    }
 }
 exports.AccountTrackerController = AccountTrackerController;
 _AccountTrackerController_refreshMutex = new WeakMap(), _AccountTrackerController_includeStakedAssets = new WeakMap(), _AccountTrackerController_getStakedBalanceForChain = new WeakMap(), _AccountTrackerController_instances = new WeakSet(), _AccountTrackerController_getCorrectNetworkClient = function _AccountTrackerController_getCorrectNetworkClient(networkClientId) {
@@ -298,6 +345,9 @@ async function _AccountTrackerController_getBalanceFromChain(address, ethQuery)
         (0, utils_1.assert)(ethQuery, 'Provider not set.');
         return await (0, controller_utils_1.query)(ethQuery, 'getBalance', [address]);
     });
+}, _AccountTrackerController_registerMessageHandlers = function _AccountTrackerController_registerMessageHandlers() {
+    this.messagingSystem.registerActionHandler(`${controllerName}:updateNativeBalances`, this.updateNativeBalances.bind(this));
+    this.messagingSystem.registerActionHandler(`${controllerName}:updateStakedBalances`, this.updateStakedBalances.bind(this));
 };
 exports.default = AccountTrackerController;
 //# sourceMappingURL=AccountTrackerController.cjs.map

package/dist/AccountTrackerController.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"AccountTrackerController.cjs","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AACA,wDAAoD;AACpD,wDAAwD;AAYxD,iEAIoC;AACpC,oEAA2C;AAM3C,qEAA+E;AAE/E,2CAAsD;AACtD,6CAAoC;AACpC,mCAA4C;AAC5C,sGAA4E;AAE5E,6EAIoC;AACpC,iDAAgF;AAEhF;;GAEG;AACH,MAAM,cAAc,GAAG,0BAA0B,CAAC;AA2BlD,MAAM,sBAAsB,GAAG;IAC7B,iBAAiB,EAAE;QACjB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAgEF;;GAEG;AACH,MAAa,wBAAyB,SAAQ,IAAA,oDAA+B,GAI5E;IAOC;;;;;;;;;OASG;IACH,YAAY,EACV,QAAQ,GAAG,KAAK,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAAmB,GAAG,KAAK,GAO5B;QACC,MAAM,EAAE,uBAAuB,EAAE,GAAG,SAAS,CAAC,IAAI,CAChD,4BAA4B,CAC7B,CAAC;QACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,SAAS,CAAC,IAAI,CAChB,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;QACF,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,SAAS;YACT,KAAK,EAAE;gBACL,iBAAiB,EAAE;oBACjB,CAAC,OAAO,CAAC,EAAE,EAAE;iBACd;gBACD,GAAG,KAAK;aACT;YACD,QAAQ,EAAE,sBAAsB;SACjC,CAAC,CAAC;;QAhDI,iDAAgB,IAAI,mBAAK,EAAE,EAAC;QAE5B,gEAA8B;QAE9B,qEAAgF;QA6CvF,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAE1D,uBAAA,IAAI,iDAAwB,mBAAmB,MAAA,CAAC;QAEhD,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,6CAA6C,EAC7C,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE;YAC1B,IAAI,UAAU,KAAK,WAAW,EAAE;gBAC9B,0CAA0C;gBAC1C,mEAAmE;gBACnE,IAAI,CAAC,OAAO,CAAC,uBAAA,IAAI,0FAAqB,MAAzB,IAAI,CAAuB,CAAC,CAAC;aAC3C;QACH,CAAC,EACD,CAAC,KAAK,EAAU,EAAE,CAAC,KAAK,CAAC,OAAO,CACjC,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,WAAqB;QACxC,MAAM,iBAAiB,GAAG,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAClE,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,GAC3C,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAExE,+CAA+C;QAC/C,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YACjC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE;gBAClC,iBAAiB,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;gBACnC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC3B,iBAAiB,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC9D,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,CAAC,CAAC;QAEH,oEAAoE;QACpE,4DAA4D;QAC5D,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAC7B,IAAI,CAAC,eAAe;aACjB,IAAI,CAAC,iCAAiC,CAAC;aACvC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,CACvB,IAAA,uCAAoB,EAAC,eAAe,CAAC,OAAO,CAAC,CAC9C,CACJ,CAAC;QACF,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CACnC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CACzC,CAAC;QACF,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAClC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjD,YAAY,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC/B,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG;oBACpC,OAAO,EAAE,KAAK;iBACf,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjD,YAAY,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC/B,OAAO,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAA,gBAAO,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,EAAE;YAC7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;gBACpB,KAAK,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;YAC9C,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAgDD;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,EACjB,gBAAgB,GACW;QAC3B,gFAAgF;QAChF,mEAAmE;QACnE,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACjC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CAAC,gBAAmC;QAC/C,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC/C,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,8CAAc,CAAC,OAAO,EAAE,CAAC;QACvD,IAAI;YACF,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;gBACxD,MAAM,EAAE,OAAO,EAAE,GAAG,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;gBACnE,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAE5B,uDAAuD;YACvD,MAAM,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE;gBACpE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,GACjD,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;gBACjD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;gBACzC,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACjE,gCAAgC,CACjC,CAAC;gBAEF,MAAM,gBAAgB,GAAG,6BAA6B;oBACpD,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;oBACzC,CAAC,CAAC,CAAC,IAAA,uCAAoB,EAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;gBAEpD,MAAM,gBAAgB,GAAG,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC;gBAE3D,0CAA0C;gBAC1C,+EAA+E;gBAC/E,iGAAiG;gBACjG,MAAM,IAAA,2CAAwB,EAAC,GAAG,EAAE,CAClC,YAAY,EAAE,mBAAmB,EAAE,EAAE,CACtC,CAAC;gBAEF,MAAM,qBAAqB,GAAG,uBAAA,IAAI,qDAAqB;oBACrD,CAAC,CAAC,uBAAA,IAAI,0DAA0B,MAA9B,IAAI,EAA2B,gBAAgB,EAAE,eAAe,CAAC;oBACnE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAExB,IAAI,IAAA,mBAAW,EAAC,kEAAuC,EAAE,OAAO,CAAC,EAAE;oBACjE,MAAM,eAAe,GAAG,kEAAuC,CAC7D,OAAO,CACE,CAAC;oBAEZ,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAC3B,eAAe,EACf,yCAA6B,EAC7B,IAAI,wBAAY,CAAC,QAAQ,CAAC,CAC3B,CAAC;oBAEF,MAAM,cAAc,GAAG,MAAM,IAAA,2CAAwB,EACnD,GAAG,EAAE,CACH,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;wBAClC,4CAA4C;qBAC7C,CAAyB,EAC5B,KAAK,EACL,IAAK,CACN,CAAC;oBAEF,IAAI,cAAc,EAAE;wBAClB,gBAAgB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;4BAC1C,gBAAgB,CAAC,OAAO,CAAC,GAAG;gCAC1B,OAAO,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE;6BAC7C,CAAC;wBACJ,CAAC,CAAC,CAAC;qBACJ;iBACF;qBAAM;oBACL,4DAA4D;oBAC5D,MAAM,IAAA,oCAAuB,EAAe;wBAC1C,MAAM,EAAE,gBAAgB;wBACxB,SAAS,EAAE,oCAAuB;wBAClC,aAAa,EAAE,SAAS;wBACxB,SAAS,EAAE,KAAK,EAAE,aAAmB,EAAE,KAAe,EAAE,EAAE;4BACxD,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,OAAe,EAAE,EAAE;gCAC1D,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,0FAAqB,MAAzB,IAAI,EAC9B,OAAO,EACP,QAAQ,CACT,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gCAEpB,0BAA0B;gCAC1B,IAAI,aAAa,EAAE;oCACjB,gBAAgB,CAAC,OAAO,CAAC,GAAG;wCAC1B,OAAO,EAAE,aAAa;qCACvB,CAAC;iCACH;4BACH,CAAC,CAAC,CAAC;4BAEH,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;4BAC1C,OAAO,aAAa,CAAC;wBACvB,CAAC;qBACF,CAAC,CAAC;iBACJ;gBAED,MAAM,mBAAmB,GAAG,MAAM,IAAA,2CAAwB,EACxD,KAAK,IAAI,EAAE,CACT,CAAC,MAAM,qBAAqB,CAAkC,CACjE,CAAC;gBAEF,MAAM,CAAC,OAAO,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,OAAO,CAC/C,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;oBACrB,gBAAgB,CAAC,OAAO,CAAC,GAAG;wBAC1B,GAAG,gBAAgB,CAAC,OAAO,CAAC;wBAC5B,aAAa,EAAE,OAAO;qBACvB,CAAC;gBACJ,CAAC,CACF,CAAC;gBAEF,2DAA2D;gBAC3D,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;YACvC,CAAC,CAAC,CAAC;YAEH,6DAA6D;YAC7D,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAE5D,yEAAyE;YACzE,MAAM,qBAAqB,GACzB,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC1C,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC5B,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE;oBACjC,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;oBACnD,oDAAoD;oBACpD,IAAI,CAAC,IAAA,gBAAO,EAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,gBAAgB,CAAC,EAAE;wBAC9D,qBAAqB,CAAC,OAAO,CAAC,GAAG,gBAAgB,CAAC;wBAClD,UAAU,GAAG,IAAI,CAAC;qBACnB;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,4DAA4D;YAC5D,IAAI,UAAU,EAAE;gBACd,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;oBACpB,KAAK,CAAC,iBAAiB,GAAG,qBAAqB,CAAC;gBAClD,CAAC,CAAC,CAAC;aACJ;SACF;gBAAS;YACR,WAAW,EAAE,CAAC;SACf;IACH,CAAC;IAmBD;;;;;;OAMG;IACH,KAAK,CAAC,wBAAwB,CAC5B,SAAmB,EACnB,eAAiC;QAIjC,MAAM,EAAE,QAAQ,EAAE,GAAG,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;QAEpE,4DAA4D;QAC5D,OAAO,MAAM,OAAO,CAAC,GAAG,CACtB,SAAS,CAAC,GAAG,CACX,CAAC,OAAO,EAAwD,EAAE;YAChE,OAAO,IAAA,2CAAwB,EAAC,KAAK,IAAI,EAAE;gBACzC,IAAA,cAAM,EAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;gBACtC,MAAM,OAAO,GAAG,MAAM,IAAA,wBAAK,EAAC,QAAQ,EAAE,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;gBAE/D,IAAI,aAA4B,CAAC;gBACjC,IAAI,uBAAA,IAAI,qDAAqB,EAAE;oBAC7B,aAAa,GAAG,CACd,MAAM,uBAAA,IAAI,0DAA0B,MAA9B,IAAI,EAA2B,CAAC,OAAO,CAAC,EAAE,eAAe,CAAC,CACjE,CAAC,OAAO,CAAC,CAAC;iBACZ;gBACD,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CACF,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;gBAChC,IAAI,CAAC,IAAI,EAAE;oBACT,OAAO,GAAG,CAAC;iBACZ;gBAED,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,CAAC,GAAG,IAAI,CAAC;gBAC/C,OAAO;oBACL,GAAG,GAAG;oBACN,CAAC,OAAO,CAAC,EAAE;wBACT,OAAO;wBACP,aAAa;qBACd;iBACF,CAAC;YACJ,CAAC,EAAE,EAAE,CAAC,CAAC;QACT,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAxZD,4DAwZC;+VA5Q0B,eAAiC;IACxD,MAAM,uBAAuB,GAC3B,eAAe;QACf,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC;aACpD,uBAAuB,CAAC;IAC7B,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,EAC1B,QAAQ,EACR,YAAY,GACb,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,OAAO;QACL,OAAO;QACP,QAAQ;QACR,QAAQ,EAAE,IAAI,mBAAQ,CAAC,QAAQ,CAAC;QAChC,YAAY;KACb,CAAC;AACJ,CAAC;IAQC,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;IACF,OAAO,MAAM,CAAC,MAAM,CAAC,8BAA8B,CAAC,CAAC,OAAO,CAC1D,CAAC,oBAAoB,EAAE,EAAE,CACvB,oBAAoB,CAAC,YAAY,CAAC,GAAG,CACnC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,eAAe,CAC7C,CACJ,CAAC;AACJ,CAAC;AAoKD;;;;;;GAMG;AACH,KAAK,wDACH,OAAe,EACf,QAAmB;IAEnB,OAAO,MAAM,IAAA,2CAAwB,EAAC,KAAK,IAAI,EAAE;QAC/C,IAAA,cAAM,EAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;QACtC,OAAO,MAAM,IAAA,wBAAK,EAAC,QAAQ,EAAE,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC;AAsDH,kBAAe,wBAAwB,CAAC","sourcesContent":["import type { BigNumber } from '@ethersproject/bignumber';\nimport { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerSelectedEvmAccountChangeEvent,\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerStateChangeEvent,\n  ControllerGetStateAction,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  query,\n  safelyExecuteWithTimeout,\n  toChecksumHexAddress,\n} from '@metamask/controller-utils';\nimport EthQuery from '@metamask/eth-query';\nimport type {\n  NetworkClientId,\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { PreferencesControllerGetStateAction } from '@metamask/preferences-controller';\nimport { assert, hasProperty } from '@metamask/utils';\nimport { Mutex } from 'async-mutex';\nimport { cloneDeep, isEqual } from 'lodash';\nimport abiSingleCallBalancesContract from 'single-call-balance-checker-abi';\n\nimport {\n  SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID,\n  type AssetsContractController,\n  type StakedBalance,\n} from './AssetsContractController';\nimport { reduceInBatchesSerially, TOKEN_PRICES_BATCH_SIZE } from './assetsUtil';\n\n/**\n * The name of the {@link AccountTrackerController}.\n */\nconst controllerName = 'AccountTrackerController';\n\n/**\n * AccountInformation\n *\n * Account information object\n *\n * balance - Hex string of an account balance in wei\n *\n * stakedBalance - Hex string of an account staked balance in wei\n */\nexport type AccountInformation = {\n  balance: string;\n  stakedBalance?: string;\n};\n\n/**\n * AccountTrackerControllerState\n *\n * Account tracker controller state\n *\n * accountsByChainId - Map of addresses to account information by chain\n */\nexport type AccountTrackerControllerState = {\n  accountsByChainId: Record<string, { [address: string]: AccountInformation }>;\n};\n\nconst accountTrackerMetadata = {\n  accountsByChainId: {\n    persist: true,\n    anonymous: false,\n  },\n};\n\n/**\n * The action that can be performed to get the state of the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  AccountTrackerControllerState\n>;\n\n/**\n * The actions that can be performed using the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerActions =\n  AccountTrackerControllerGetStateAction;\n\n/**\n * The messenger of the {@link AccountTrackerController} for communication.\n */\nexport type AllowedActions =\n  | AccountsControllerListAccountsAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | NetworkControllerGetStateAction\n  | NetworkControllerGetNetworkClientByIdAction;\n\n/**\n * The event that {@link AccountTrackerController} can emit.\n */\nexport type AccountTrackerControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    AccountTrackerControllerState\n  >;\n\n/**\n * The events that {@link AccountTrackerController} can emit.\n */\nexport type AccountTrackerControllerEvents =\n  AccountTrackerControllerStateChangeEvent;\n\n/**\n * The external events available to the {@link AccountTrackerController}.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedEvmAccountChangeEvent\n  | AccountsControllerSelectedAccountChangeEvent;\n\n/**\n * The messenger of the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  AccountTrackerControllerActions | AllowedActions,\n  AccountTrackerControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/** The input to start polling for the {@link AccountTrackerController} */\ntype AccountTrackerPollingInput = {\n  networkClientIds: NetworkClientId[];\n};\n\n/**\n * Controller that tracks the network balances for all user accounts.\n */\nexport class AccountTrackerController extends StaticIntervalPollingController<AccountTrackerPollingInput>()<\n  typeof controllerName,\n  AccountTrackerControllerState,\n  AccountTrackerControllerMessenger\n> {\n  readonly #refreshMutex = new Mutex();\n\n  readonly #includeStakedAssets: boolean;\n\n  readonly #getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n\n  /**\n   * Creates an AccountTracker instance.\n   *\n   * @param options - The controller options.\n   * @param options.interval - Polling interval used to fetch new account balances.\n   * @param options.state - Initial state to set on this controller.\n   * @param options.messenger - The controller messaging system.\n   * @param options.getStakedBalanceForChain - The function to get the staked native asset balance for a chain.\n   * @param options.includeStakedAssets - Whether to include staked assets in the account balances.\n   */\n  constructor({\n    interval = 10000,\n    state,\n    messenger,\n    getStakedBalanceForChain,\n    includeStakedAssets = false,\n  }: {\n    interval?: number;\n    state?: Partial<AccountTrackerControllerState>;\n    messenger: AccountTrackerControllerMessenger;\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n    includeStakedAssets?: boolean;\n  }) {\n    const { selectedNetworkClientId } = messenger.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = messenger.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    super({\n      name: controllerName,\n      messenger,\n      state: {\n        accountsByChainId: {\n          [chainId]: {},\n        },\n        ...state,\n      },\n      metadata: accountTrackerMetadata,\n    });\n    this.#getStakedBalanceForChain = getStakedBalanceForChain;\n\n    this.#includeStakedAssets = includeStakedAssets;\n\n    this.setIntervalLength(interval);\n\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedEvmAccountChange',\n      (newAddress, prevAddress) => {\n        if (newAddress !== prevAddress) {\n          // Making an async call for this new event\n          // eslint-disable-next-line @typescript-eslint/no-floating-promises\n          this.refresh(this.#getNetworkClientIds());\n        }\n      },\n      (event): string => event.address,\n    );\n  }\n\n  private syncAccounts(newChainIds: string[]) {\n    const accountsByChainId = cloneDeep(this.state.accountsByChainId);\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId: currentChainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    const existing = Object.keys(accountsByChainId?.[currentChainId] ?? {});\n\n    // Initialize new chain IDs if they don't exist\n    newChainIds.forEach((newChainId) => {\n      if (!accountsByChainId[newChainId]) {\n        accountsByChainId[newChainId] = {};\n        existing.forEach((address) => {\n          accountsByChainId[newChainId][address] = { balance: '0x0' };\n        });\n      }\n    });\n\n    // Note: The address from the preferences controller are checksummed\n    // The addresses from the accounts controller are lowercased\n    const addresses = Object.values(\n      this.messagingSystem\n        .call('AccountsController:listAccounts')\n        .map((internalAccount) =>\n          toChecksumHexAddress(internalAccount.address),\n        ),\n    );\n    const newAddresses = addresses.filter(\n      (address) => !existing.includes(address),\n    );\n    const oldAddresses = existing.filter(\n      (address) => !addresses.includes(address),\n    );\n    Object.keys(accountsByChainId).forEach((chainId) => {\n      newAddresses.forEach((address) => {\n        accountsByChainId[chainId][address] = {\n          balance: '0x0',\n        };\n      });\n    });\n\n    Object.keys(accountsByChainId).forEach((chainId) => {\n      oldAddresses.forEach((address) => {\n        delete accountsByChainId[chainId][address];\n      });\n    });\n\n    if (!isEqual(this.state.accountsByChainId, accountsByChainId)) {\n      this.update((state) => {\n        state.accountsByChainId = accountsByChainId;\n      });\n    }\n  }\n\n  /**\n   * Resolves a networkClientId to a network client config\n   * or globally selected network config if not provided\n   *\n   * @param networkClientId - Optional networkClientId to fetch a network client with\n   * @returns network client config\n   */\n  #getCorrectNetworkClient(networkClientId?: NetworkClientId) {\n    const selectedNetworkClientId =\n      networkClientId ??\n      this.messagingSystem.call('NetworkController:getState')\n        .selectedNetworkClientId;\n    const {\n      configuration: { chainId },\n      provider,\n      blockTracker,\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    return {\n      chainId,\n      provider,\n      ethQuery: new EthQuery(provider),\n      blockTracker,\n    };\n  }\n\n  /**\n   * Retrieves the list of network client IDs.\n   *\n   * @returns An array of network client IDs.\n   */\n  #getNetworkClientIds(): NetworkClientId[] {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    return Object.values(networkConfigurationsByChainId).flatMap(\n      (networkConfiguration) =>\n        networkConfiguration.rpcEndpoints.map(\n          (rpcEndpoint) => rpcEndpoint.networkClientId,\n        ),\n    );\n  }\n\n  /**\n   * Refreshes the balances of the accounts using the networkClientId\n   *\n   * @param input - The input for the poll.\n   * @param input.networkClientIds - The network client IDs used to get balances.\n   */\n  async _executePoll({\n    networkClientIds,\n  }: AccountTrackerPollingInput): Promise<void> {\n    // TODO: Either fix this lint violation or explain why it's necessary to ignore.\n    // eslint-disable-next-line @typescript-eslint/no-floating-promises\n    this.refresh(networkClientIds);\n  }\n\n  /**\n   * Refreshes the balances of the accounts depending on the multi-account setting.\n   * If multi-account is disabled, only updates the selected account balance.\n   * If multi-account is enabled, updates balances for all accounts.\n   *\n   * @param networkClientIds - Optional network client IDs to fetch a network client with\n   */\n  async refresh(networkClientIds: NetworkClientId[]) {\n    const selectedAccount = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const releaseLock = await this.#refreshMutex.acquire();\n    try {\n      const chainIds = networkClientIds.map((networkClientId) => {\n        const { chainId } = this.#getCorrectNetworkClient(networkClientId);\n        return chainId;\n      });\n\n      this.syncAccounts(chainIds);\n\n      // Create an array of promises for each networkClientId\n      const updatePromises = networkClientIds.map(async (networkClientId) => {\n        const { chainId, ethQuery, provider, blockTracker } =\n          this.#getCorrectNetworkClient(networkClientId);\n        const { accountsByChainId } = this.state;\n        const { isMultiAccountBalancesEnabled } = this.messagingSystem.call(\n          'PreferencesController:getState',\n        );\n\n        const accountsToUpdate = isMultiAccountBalancesEnabled\n          ? Object.keys(accountsByChainId[chainId])\n          : [toChecksumHexAddress(selectedAccount.address)];\n\n        const accountsForChain = { ...accountsByChainId[chainId] };\n\n        // Force fresh block data before multicall\n        // TODO: This is a temporary fix to ensure that the block number is up to date.\n        // We should remove this once we have a better solution for this on the block tracker controller.\n        await safelyExecuteWithTimeout(() =>\n          blockTracker?.checkForLatestBlock?.(),\n        );\n\n        const stakedBalancesPromise = this.#includeStakedAssets\n          ? this.#getStakedBalanceForChain(accountsToUpdate, networkClientId)\n          : Promise.resolve({});\n\n        if (hasProperty(SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID, chainId)) {\n          const contractAddress = SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID[\n            chainId\n          ] as string;\n\n          const contract = new Contract(\n            contractAddress,\n            abiSingleCallBalancesContract,\n            new Web3Provider(provider),\n          );\n\n          const nativeBalances = await safelyExecuteWithTimeout(\n            () =>\n              contract.balances(accountsToUpdate, [\n                '0x0000000000000000000000000000000000000000',\n              ]) as Promise<BigNumber[]>,\n            false,\n            3_000, // 3s max call for multicall contract call\n          );\n\n          if (nativeBalances) {\n            accountsToUpdate.forEach((address, index) => {\n              accountsForChain[address] = {\n                balance: nativeBalances[index].toHexString(),\n              };\n            });\n          }\n        } else {\n          // Process accounts in batches using reduceInBatchesSerially\n          await reduceInBatchesSerially<string, void>({\n            values: accountsToUpdate,\n            batchSize: TOKEN_PRICES_BATCH_SIZE,\n            initialResult: undefined,\n            eachBatch: async (workingResult: void, batch: string[]) => {\n              const balancePromises = batch.map(async (address: string) => {\n                const balanceResult = await this.#getBalanceFromChain(\n                  address,\n                  ethQuery,\n                ).catch(() => null);\n\n                // Update account balances\n                if (balanceResult) {\n                  accountsForChain[address] = {\n                    balance: balanceResult,\n                  };\n                }\n              });\n\n              await Promise.allSettled(balancePromises);\n              return workingResult;\n            },\n          });\n        }\n\n        const stakedBalanceResult = await safelyExecuteWithTimeout(\n          async () =>\n            (await stakedBalancesPromise) as Record<string, StakedBalance>,\n        );\n\n        Object.entries(stakedBalanceResult ?? {}).forEach(\n          ([address, balance]) => {\n            accountsForChain[address] = {\n              ...accountsForChain[address],\n              stakedBalance: balance,\n            };\n          },\n        );\n\n        // After all batches are processed, return the updated data\n        return { chainId, accountsForChain };\n      });\n\n      // Wait for all networkClientId updates to settle in parallel\n      const allResults = await Promise.allSettled(updatePromises);\n\n      // Build a _copy_ of the current state and track whether anything changed\n      const nextAccountsByChainId: AccountTrackerControllerState['accountsByChainId'] =\n        cloneDeep(this.state.accountsByChainId);\n      let hasChanges = false;\n\n      allResults.forEach((result) => {\n        if (result.status === 'fulfilled') {\n          const { chainId, accountsForChain } = result.value;\n          // Only mark as changed if the incoming data differs\n          if (!isEqual(nextAccountsByChainId[chainId], accountsForChain)) {\n            nextAccountsByChainId[chainId] = accountsForChain;\n            hasChanges = true;\n          }\n        }\n      });\n\n      // 👇🏻 call `update` only when something is new / different\n      if (hasChanges) {\n        this.update((state) => {\n          state.accountsByChainId = nextAccountsByChainId;\n        });\n      }\n    } finally {\n      releaseLock();\n    }\n  }\n\n  /**\n   * Fetches the balance of a given address from the blockchain.\n   *\n   * @param address - The account address to fetch the balance for.\n   * @param ethQuery - The EthQuery instance to query getBalnce with.\n   * @returns A promise that resolves to the balance in a hex string format.\n   */\n  async #getBalanceFromChain(\n    address: string,\n    ethQuery?: EthQuery,\n  ): Promise<string | undefined> {\n    return await safelyExecuteWithTimeout(async () => {\n      assert(ethQuery, 'Provider not set.');\n      return await query(ethQuery, 'getBalance', [address]);\n    });\n  }\n\n  /**\n   * Sync accounts balances with some additional addresses.\n   *\n   * @param addresses - the additional addresses, may be hardware wallet addresses.\n   * @param networkClientId - Optional networkClientId to fetch a network client with.\n   * @returns accounts - addresses with synced balance\n   */\n  async syncBalanceWithAddresses(\n    addresses: string[],\n    networkClientId?: NetworkClientId,\n  ): Promise<\n    Record<string, { balance: string; stakedBalance?: StakedBalance }>\n  > {\n    const { ethQuery } = this.#getCorrectNetworkClient(networkClientId);\n\n    // TODO: This should use multicall when enabled by the user.\n    return await Promise.all(\n      addresses.map(\n        (address): Promise<[string, string, StakedBalance] | undefined> => {\n          return safelyExecuteWithTimeout(async () => {\n            assert(ethQuery, 'Provider not set.');\n            const balance = await query(ethQuery, 'getBalance', [address]);\n\n            let stakedBalance: StakedBalance;\n            if (this.#includeStakedAssets) {\n              stakedBalance = (\n                await this.#getStakedBalanceForChain([address], networkClientId)\n              )[address];\n            }\n            return [address, balance, stakedBalance];\n          });\n        },\n      ),\n    ).then((value) => {\n      return value.reduce((obj, item) => {\n        if (!item) {\n          return obj;\n        }\n\n        const [address, balance, stakedBalance] = item;\n        return {\n          ...obj,\n          [address]: {\n            balance,\n            stakedBalance,\n          },\n        };\n      }, {});\n    });\n  }\n}\n\nexport default AccountTrackerController;\n"]}
+{"version":3,"file":"AccountTrackerController.cjs","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AACA,wDAAoD;AACpD,wDAAwD;AAYxD,iEAIoC;AACpC,oEAA2C;AAM3C,qEAA+E;AAE/E,2CAAgE;AAChE,6CAAoC;AACpC,mCAA4C;AAC5C,sGAA4E;AAE5E,6EAIoC;AACpC,iDAAgF;AAEhF;;GAEG;AACH,MAAM,cAAc,GAAG,0BAA0B,CAAC;AA2BlD,MAAM,sBAAsB,GAAG;IAC7B,iBAAiB,EAAE;QACjB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAkFF;;GAEG;AACH,MAAa,wBAAyB,SAAQ,IAAA,oDAA+B,GAI5E;IAOC;;;;;;;;;OASG;IACH,YAAY,EACV,QAAQ,GAAG,KAAK,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAAmB,GAAG,KAAK,GAO5B;QACC,MAAM,EAAE,uBAAuB,EAAE,GAAG,SAAS,CAAC,IAAI,CAChD,4BAA4B,CAC7B,CAAC;QACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,SAAS,CAAC,IAAI,CAChB,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;QACF,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,SAAS;YACT,KAAK,EAAE;gBACL,iBAAiB,EAAE;oBACjB,CAAC,OAAO,CAAC,EAAE,EAAE;iBACd;gBACD,GAAG,KAAK;aACT;YACD,QAAQ,EAAE,sBAAsB;SACjC,CAAC,CAAC;;QAhDI,iDAAgB,IAAI,mBAAK,EAAE,EAAC;QAE5B,gEAA8B;QAE9B,qEAAgF;QA6CvF,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAE1D,uBAAA,IAAI,iDAAwB,mBAAmB,MAAA,CAAC;QAEhD,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,6CAA6C,EAC7C,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE;YAC1B,IAAI,UAAU,KAAK,WAAW,EAAE;gBAC9B,0CAA0C;gBAC1C,mEAAmE;gBACnE,IAAI,CAAC,OAAO,CAAC,uBAAA,IAAI,0FAAqB,MAAzB,IAAI,CAAuB,CAAC,CAAC;aAC3C;QACH,CAAC,EACD,CAAC,KAAK,EAAU,EAAE,CAAC,KAAK,CAAC,OAAO,CACjC,CAAC;QAEF,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,CAA2B,CAAC;IAClC,CAAC;IAEO,YAAY,CAAC,WAAqB;QACxC,MAAM,iBAAiB,GAAG,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAClE,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,GAC3C,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAExE,+CAA+C;QAC/C,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YACjC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE;gBAClC,iBAAiB,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;gBACnC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC3B,iBAAiB,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC9D,CAAC,CAAC,CAAC;aACJ;QACH,CAAC,CAAC,CAAC;QAEH,oEAAoE;QACpE,4DAA4D;QAC5D,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAC7B,IAAI,CAAC,eAAe;aACjB,IAAI,CAAC,iCAAiC,CAAC;aACvC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,CACvB,IAAA,uCAAoB,EAAC,eAAe,CAAC,OAAO,CAAC,CAC9C,CACJ,CAAC;QACF,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CACnC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CACzC,CAAC;QACF,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAClC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjD,YAAY,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC/B,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG;oBACpC,OAAO,EAAE,KAAK;iBACf,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjD,YAAY,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC/B,OAAO,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAA,gBAAO,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,EAAE;YAC7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;gBACpB,KAAK,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;YAC9C,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAgDD;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,EACjB,gBAAgB,GACW;QAC3B,gFAAgF;QAChF,mEAAmE;QACnE,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACjC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CAAC,gBAAmC;QAC/C,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC/C,uCAAuC,CACxC,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,8CAAc,CAAC,OAAO,EAAE,CAAC;QACvD,IAAI;YACF,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;gBACxD,MAAM,EAAE,OAAO,EAAE,GAAG,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;gBACnE,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAE5B,uDAAuD;YACvD,MAAM,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE;gBACpE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,GACjD,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;gBACjD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;gBACzC,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACjE,gCAAgC,CACjC,CAAC;gBAEF,MAAM,gBAAgB,GAAG,6BAA6B;oBACpD,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;oBACzC,CAAC,CAAC,CAAC,IAAA,uCAAoB,EAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;gBAEpD,MAAM,gBAAgB,GAAG,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC;gBAE3D,0CAA0C;gBAC1C,+EAA+E;gBAC/E,iGAAiG;gBACjG,MAAM,IAAA,2CAAwB,EAAC,GAAG,EAAE,CAClC,YAAY,EAAE,mBAAmB,EAAE,EAAE,CACtC,CAAC;gBAEF,MAAM,qBAAqB,GAAG,uBAAA,IAAI,qDAAqB;oBACrD,CAAC,CAAC,uBAAA,IAAI,0DAA0B,MAA9B,IAAI,EAA2B,gBAAgB,EAAE,eAAe,CAAC;oBACnE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAExB,IAAI,IAAA,mBAAW,EAAC,kEAAuC,EAAE,OAAO,CAAC,EAAE;oBACjE,MAAM,eAAe,GAAG,kEAAuC,CAC7D,OAAO,CACE,CAAC;oBAEZ,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAC3B,eAAe,EACf,yCAA6B,EAC7B,IAAI,wBAAY,CAAC,QAAQ,CAAC,CAC3B,CAAC;oBAEF,MAAM,cAAc,GAAG,MAAM,IAAA,2CAAwB,EACnD,GAAG,EAAE,CACH,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE;wBAClC,4CAA4C;qBAC7C,CAAyB,EAC5B,KAAK,EACL,IAAK,CACN,CAAC;oBAEF,IAAI,cAAc,EAAE;wBAClB,gBAAgB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;4BAC1C,gBAAgB,CAAC,OAAO,CAAC,GAAG;gCAC1B,OAAO,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE;6BAC7C,CAAC;wBACJ,CAAC,CAAC,CAAC;qBACJ;iBACF;qBAAM;oBACL,4DAA4D;oBAC5D,MAAM,IAAA,oCAAuB,EAAe;wBAC1C,MAAM,EAAE,gBAAgB;wBACxB,SAAS,EAAE,oCAAuB;wBAClC,aAAa,EAAE,SAAS;wBACxB,SAAS,EAAE,KAAK,EAAE,aAAmB,EAAE,KAAe,EAAE,EAAE;4BACxD,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,OAAe,EAAE,EAAE;gCAC1D,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,0FAAqB,MAAzB,IAAI,EAC9B,OAAO,EACP,QAAQ,CACT,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gCAEpB,0BAA0B;gCAC1B,IAAI,aAAa,EAAE;oCACjB,gBAAgB,CAAC,OAAO,CAAC,GAAG;wCAC1B,OAAO,EAAE,aAAa;qCACvB,CAAC;iCACH;4BACH,CAAC,CAAC,CAAC;4BAEH,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;4BAC1C,OAAO,aAAa,CAAC;wBACvB,CAAC;qBACF,CAAC,CAAC;iBACJ;gBAED,MAAM,mBAAmB,GAAG,MAAM,IAAA,2CAAwB,EACxD,KAAK,IAAI,EAAE,CACT,CAAC,MAAM,qBAAqB,CAAkC,CACjE,CAAC;gBAEF,MAAM,CAAC,OAAO,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,OAAO,CAC/C,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;oBACrB,gBAAgB,CAAC,OAAO,CAAC,GAAG;wBAC1B,GAAG,gBAAgB,CAAC,OAAO,CAAC;wBAC5B,aAAa,EAAE,OAAO;qBACvB,CAAC;gBACJ,CAAC,CACF,CAAC;gBAEF,2DAA2D;gBAC3D,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;YACvC,CAAC,CAAC,CAAC;YAEH,6DAA6D;YAC7D,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAE5D,yEAAyE;YACzE,MAAM,qBAAqB,GACzB,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC1C,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC5B,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE;oBACjC,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;oBACnD,oDAAoD;oBACpD,IAAI,CAAC,IAAA,gBAAO,EAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,gBAAgB,CAAC,EAAE;wBAC9D,qBAAqB,CAAC,OAAO,CAAC,GAAG,gBAAgB,CAAC;wBAClD,UAAU,GAAG,IAAI,CAAC;qBACnB;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,4DAA4D;YAC5D,IAAI,UAAU,EAAE;gBACd,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;oBACpB,KAAK,CAAC,iBAAiB,GAAG,qBAAqB,CAAC;gBAClD,CAAC,CAAC,CAAC;aACJ;SACF;gBAAS;YACR,WAAW,EAAE,CAAC;SACf;IACH,CAAC;IAmBD;;;;;;OAMG;IACH,KAAK,CAAC,wBAAwB,CAC5B,SAAmB,EACnB,eAAiC;QAIjC,MAAM,EAAE,QAAQ,EAAE,GAAG,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,eAAe,CAAC,CAAC;QAEpE,4DAA4D;QAC5D,OAAO,MAAM,OAAO,CAAC,GAAG,CACtB,SAAS,CAAC,GAAG,CACX,CAAC,OAAO,EAAwD,EAAE;YAChE,OAAO,IAAA,2CAAwB,EAAC,KAAK,IAAI,EAAE;gBACzC,IAAA,cAAM,EAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;gBACtC,MAAM,OAAO,GAAG,MAAM,IAAA,wBAAK,EAAC,QAAQ,EAAE,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;gBAE/D,IAAI,aAA4B,CAAC;gBACjC,IAAI,uBAAA,IAAI,qDAAqB,EAAE;oBAC7B,aAAa,GAAG,CACd,MAAM,uBAAA,IAAI,0DAA0B,MAA9B,IAAI,EAA2B,CAAC,OAAO,CAAC,EAAE,eAAe,CAAC,CACjE,CAAC,OAAO,CAAC,CAAC;iBACZ;gBACD,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CACF,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;gBAChC,IAAI,CAAC,IAAI,EAAE;oBACT,OAAO,GAAG,CAAC;iBACZ;gBAED,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,CAAC,GAAG,IAAI,CAAC;gBAC/C,OAAO;oBACL,GAAG,GAAG;oBACN,CAAC,OAAO,CAAC,EAAE;wBACT,OAAO;wBACP,aAAa;qBACd;iBACF,CAAC;YACJ,CAAC,EAAE,EAAE,CAAC,CAAC;QACT,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,oBAAoB,CAClB,QAA8D;QAE9D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;gBACjD,yCAAyC;gBACzC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE;oBACrC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;iBACvC;gBAED,2CAA2C;gBAC3C,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE;oBAC9C,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;iBAChE;gBAED,qBAAqB;gBACrB,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,GAAG,OAAO,CAAC;YAC9D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,oBAAoB,CAClB,cAIG;QAEH,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,cAAc,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC7D,yCAAyC;gBACzC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE;oBACrC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;iBACvC;gBAED,2CAA2C;gBAC3C,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE;oBAC9C,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;iBAChE;gBAED,4BAA4B;gBAC5B,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,GAAG,aAAa,CAAC;YAC1E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CAaF;AAleD,4DAkeC;+VApV0B,eAAiC;IACxD,MAAM,uBAAuB,GAC3B,eAAe;QACf,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC;aACpD,uBAAuB,CAAC;IAC7B,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,EAC1B,QAAQ,EACR,YAAY,GACb,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,OAAO;QACL,OAAO;QACP,QAAQ;QACR,QAAQ,EAAE,IAAI,mBAAQ,CAAC,QAAQ,CAAC;QAChC,YAAY;KACb,CAAC;AACJ,CAAC;IAQC,MAAM,EAAE,8BAA8B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClE,4BAA4B,CAC7B,CAAC;IACF,OAAO,MAAM,CAAC,MAAM,CAAC,8BAA8B,CAAC,CAAC,OAAO,CAC1D,CAAC,oBAAoB,EAAE,EAAE,CACvB,oBAAoB,CAAC,YAAY,CAAC,GAAG,CACnC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,eAAe,CAC7C,CACJ,CAAC;AACJ,CAAC;AAoKD;;;;;;GAMG;AACH,KAAK,wDACH,OAAe,EACf,QAAmB;IAEnB,OAAO,MAAM,IAAA,2CAAwB,EAAC,KAAK,IAAI,EAAE;QAC/C,IAAA,cAAM,EAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;QACtC,OAAO,MAAM,IAAA,wBAAK,EAAC,QAAQ,EAAE,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC;IAkHC,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,GAAG,cAAc,uBAAgC,EACjD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;IAEF,IAAI,CAAC,eAAe,CAAC,qBAAqB,CACxC,GAAG,cAAc,uBAAgC,EACjD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CACrC,CAAC;AACJ,CAAC;AAGH,kBAAe,wBAAwB,CAAC","sourcesContent":["import type { BigNumber } from '@ethersproject/bignumber';\nimport { Contract } from '@ethersproject/contracts';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerSelectedEvmAccountChangeEvent,\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerListAccountsAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerStateChangeEvent,\n  ControllerGetStateAction,\n  RestrictedMessenger,\n} from '@metamask/base-controller';\nimport {\n  query,\n  safelyExecuteWithTimeout,\n  toChecksumHexAddress,\n} from '@metamask/controller-utils';\nimport EthQuery from '@metamask/eth-query';\nimport type {\n  NetworkClientId,\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { PreferencesControllerGetStateAction } from '@metamask/preferences-controller';\nimport { assert, hasProperty, type Hex } from '@metamask/utils';\nimport { Mutex } from 'async-mutex';\nimport { cloneDeep, isEqual } from 'lodash';\nimport abiSingleCallBalancesContract from 'single-call-balance-checker-abi';\n\nimport {\n  SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID,\n  type AssetsContractController,\n  type StakedBalance,\n} from './AssetsContractController';\nimport { reduceInBatchesSerially, TOKEN_PRICES_BATCH_SIZE } from './assetsUtil';\n\n/**\n * The name of the {@link AccountTrackerController}.\n */\nconst controllerName = 'AccountTrackerController';\n\n/**\n * AccountInformation\n *\n * Account information object\n *\n * balance - Hex string of an account balance in wei\n *\n * stakedBalance - Hex string of an account staked balance in wei\n */\nexport type AccountInformation = {\n  balance: string;\n  stakedBalance?: string;\n};\n\n/**\n * AccountTrackerControllerState\n *\n * Account tracker controller state\n *\n * accountsByChainId - Map of addresses to account information by chain\n */\nexport type AccountTrackerControllerState = {\n  accountsByChainId: Record<string, { [address: string]: AccountInformation }>;\n};\n\nconst accountTrackerMetadata = {\n  accountsByChainId: {\n    persist: true,\n    anonymous: false,\n  },\n};\n\n/**\n * The action that can be performed to get the state of the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  AccountTrackerControllerState\n>;\n\n/**\n * The action that can be performed to update multiple native token balances in batch.\n */\nexport type AccountTrackerUpdateNativeBalancesAction = {\n  type: `${typeof controllerName}:updateNativeBalances`;\n  handler: AccountTrackerController['updateNativeBalances'];\n};\n\n/**\n * The action that can be performed to update multiple staked balances in batch.\n */\nexport type AccountTrackerUpdateStakedBalancesAction = {\n  type: `${typeof controllerName}:updateStakedBalances`;\n  handler: AccountTrackerController['updateStakedBalances'];\n};\n\n/**\n * The actions that can be performed using the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerActions =\n  | AccountTrackerControllerGetStateAction\n  | AccountTrackerUpdateNativeBalancesAction\n  | AccountTrackerUpdateStakedBalancesAction;\n\n/**\n * The messenger of the {@link AccountTrackerController} for communication.\n */\nexport type AllowedActions =\n  | AccountsControllerListAccountsAction\n  | PreferencesControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction\n  | NetworkControllerGetStateAction\n  | NetworkControllerGetNetworkClientByIdAction;\n\n/**\n * The event that {@link AccountTrackerController} can emit.\n */\nexport type AccountTrackerControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    AccountTrackerControllerState\n  >;\n\n/**\n * The events that {@link AccountTrackerController} can emit.\n */\nexport type AccountTrackerControllerEvents =\n  AccountTrackerControllerStateChangeEvent;\n\n/**\n * The external events available to the {@link AccountTrackerController}.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedEvmAccountChangeEvent\n  | AccountsControllerSelectedAccountChangeEvent;\n\n/**\n * The messenger of the {@link AccountTrackerController}.\n */\nexport type AccountTrackerControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  AccountTrackerControllerActions | AllowedActions,\n  AccountTrackerControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/** The input to start polling for the {@link AccountTrackerController} */\ntype AccountTrackerPollingInput = {\n  networkClientIds: NetworkClientId[];\n};\n\n/**\n * Controller that tracks the network balances for all user accounts.\n */\nexport class AccountTrackerController extends StaticIntervalPollingController<AccountTrackerPollingInput>()<\n  typeof controllerName,\n  AccountTrackerControllerState,\n  AccountTrackerControllerMessenger\n> {\n  readonly #refreshMutex = new Mutex();\n\n  readonly #includeStakedAssets: boolean;\n\n  readonly #getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n\n  /**\n   * Creates an AccountTracker instance.\n   *\n   * @param options - The controller options.\n   * @param options.interval - Polling interval used to fetch new account balances.\n   * @param options.state - Initial state to set on this controller.\n   * @param options.messenger - The controller messaging system.\n   * @param options.getStakedBalanceForChain - The function to get the staked native asset balance for a chain.\n   * @param options.includeStakedAssets - Whether to include staked assets in the account balances.\n   */\n  constructor({\n    interval = 10000,\n    state,\n    messenger,\n    getStakedBalanceForChain,\n    includeStakedAssets = false,\n  }: {\n    interval?: number;\n    state?: Partial<AccountTrackerControllerState>;\n    messenger: AccountTrackerControllerMessenger;\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n    includeStakedAssets?: boolean;\n  }) {\n    const { selectedNetworkClientId } = messenger.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = messenger.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    super({\n      name: controllerName,\n      messenger,\n      state: {\n        accountsByChainId: {\n          [chainId]: {},\n        },\n        ...state,\n      },\n      metadata: accountTrackerMetadata,\n    });\n    this.#getStakedBalanceForChain = getStakedBalanceForChain;\n\n    this.#includeStakedAssets = includeStakedAssets;\n\n    this.setIntervalLength(interval);\n\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedEvmAccountChange',\n      (newAddress, prevAddress) => {\n        if (newAddress !== prevAddress) {\n          // Making an async call for this new event\n          // eslint-disable-next-line @typescript-eslint/no-floating-promises\n          this.refresh(this.#getNetworkClientIds());\n        }\n      },\n      (event): string => event.address,\n    );\n\n    this.#registerMessageHandlers();\n  }\n\n  private syncAccounts(newChainIds: string[]) {\n    const accountsByChainId = cloneDeep(this.state.accountsByChainId);\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId: currentChainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    const existing = Object.keys(accountsByChainId?.[currentChainId] ?? {});\n\n    // Initialize new chain IDs if they don't exist\n    newChainIds.forEach((newChainId) => {\n      if (!accountsByChainId[newChainId]) {\n        accountsByChainId[newChainId] = {};\n        existing.forEach((address) => {\n          accountsByChainId[newChainId][address] = { balance: '0x0' };\n        });\n      }\n    });\n\n    // Note: The address from the preferences controller are checksummed\n    // The addresses from the accounts controller are lowercased\n    const addresses = Object.values(\n      this.messagingSystem\n        .call('AccountsController:listAccounts')\n        .map((internalAccount) =>\n          toChecksumHexAddress(internalAccount.address),\n        ),\n    );\n    const newAddresses = addresses.filter(\n      (address) => !existing.includes(address),\n    );\n    const oldAddresses = existing.filter(\n      (address) => !addresses.includes(address),\n    );\n    Object.keys(accountsByChainId).forEach((chainId) => {\n      newAddresses.forEach((address) => {\n        accountsByChainId[chainId][address] = {\n          balance: '0x0',\n        };\n      });\n    });\n\n    Object.keys(accountsByChainId).forEach((chainId) => {\n      oldAddresses.forEach((address) => {\n        delete accountsByChainId[chainId][address];\n      });\n    });\n\n    if (!isEqual(this.state.accountsByChainId, accountsByChainId)) {\n      this.update((state) => {\n        state.accountsByChainId = accountsByChainId;\n      });\n    }\n  }\n\n  /**\n   * Resolves a networkClientId to a network client config\n   * or globally selected network config if not provided\n   *\n   * @param networkClientId - Optional networkClientId to fetch a network client with\n   * @returns network client config\n   */\n  #getCorrectNetworkClient(networkClientId?: NetworkClientId) {\n    const selectedNetworkClientId =\n      networkClientId ??\n      this.messagingSystem.call('NetworkController:getState')\n        .selectedNetworkClientId;\n    const {\n      configuration: { chainId },\n      provider,\n      blockTracker,\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    return {\n      chainId,\n      provider,\n      ethQuery: new EthQuery(provider),\n      blockTracker,\n    };\n  }\n\n  /**\n   * Retrieves the list of network client IDs.\n   *\n   * @returns An array of network client IDs.\n   */\n  #getNetworkClientIds(): NetworkClientId[] {\n    const { networkConfigurationsByChainId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    return Object.values(networkConfigurationsByChainId).flatMap(\n      (networkConfiguration) =>\n        networkConfiguration.rpcEndpoints.map(\n          (rpcEndpoint) => rpcEndpoint.networkClientId,\n        ),\n    );\n  }\n\n  /**\n   * Refreshes the balances of the accounts using the networkClientId\n   *\n   * @param input - The input for the poll.\n   * @param input.networkClientIds - The network client IDs used to get balances.\n   */\n  async _executePoll({\n    networkClientIds,\n  }: AccountTrackerPollingInput): Promise<void> {\n    // TODO: Either fix this lint violation or explain why it's necessary to ignore.\n    // eslint-disable-next-line @typescript-eslint/no-floating-promises\n    this.refresh(networkClientIds);\n  }\n\n  /**\n   * Refreshes the balances of the accounts depending on the multi-account setting.\n   * If multi-account is disabled, only updates the selected account balance.\n   * If multi-account is enabled, updates balances for all accounts.\n   *\n   * @param networkClientIds - Optional network client IDs to fetch a network client with\n   */\n  async refresh(networkClientIds: NetworkClientId[]) {\n    const selectedAccount = this.messagingSystem.call(\n      'AccountsController:getSelectedAccount',\n    );\n    const releaseLock = await this.#refreshMutex.acquire();\n    try {\n      const chainIds = networkClientIds.map((networkClientId) => {\n        const { chainId } = this.#getCorrectNetworkClient(networkClientId);\n        return chainId;\n      });\n\n      this.syncAccounts(chainIds);\n\n      // Create an array of promises for each networkClientId\n      const updatePromises = networkClientIds.map(async (networkClientId) => {\n        const { chainId, ethQuery, provider, blockTracker } =\n          this.#getCorrectNetworkClient(networkClientId);\n        const { accountsByChainId } = this.state;\n        const { isMultiAccountBalancesEnabled } = this.messagingSystem.call(\n          'PreferencesController:getState',\n        );\n\n        const accountsToUpdate = isMultiAccountBalancesEnabled\n          ? Object.keys(accountsByChainId[chainId])\n          : [toChecksumHexAddress(selectedAccount.address)];\n\n        const accountsForChain = { ...accountsByChainId[chainId] };\n\n        // Force fresh block data before multicall\n        // TODO: This is a temporary fix to ensure that the block number is up to date.\n        // We should remove this once we have a better solution for this on the block tracker controller.\n        await safelyExecuteWithTimeout(() =>\n          blockTracker?.checkForLatestBlock?.(),\n        );\n\n        const stakedBalancesPromise = this.#includeStakedAssets\n          ? this.#getStakedBalanceForChain(accountsToUpdate, networkClientId)\n          : Promise.resolve({});\n\n        if (hasProperty(SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID, chainId)) {\n          const contractAddress = SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID[\n            chainId\n          ] as string;\n\n          const contract = new Contract(\n            contractAddress,\n            abiSingleCallBalancesContract,\n            new Web3Provider(provider),\n          );\n\n          const nativeBalances = await safelyExecuteWithTimeout(\n            () =>\n              contract.balances(accountsToUpdate, [\n                '0x0000000000000000000000000000000000000000',\n              ]) as Promise<BigNumber[]>,\n            false,\n            3_000, // 3s max call for multicall contract call\n          );\n\n          if (nativeBalances) {\n            accountsToUpdate.forEach((address, index) => {\n              accountsForChain[address] = {\n                balance: nativeBalances[index].toHexString(),\n              };\n            });\n          }\n        } else {\n          // Process accounts in batches using reduceInBatchesSerially\n          await reduceInBatchesSerially<string, void>({\n            values: accountsToUpdate,\n            batchSize: TOKEN_PRICES_BATCH_SIZE,\n            initialResult: undefined,\n            eachBatch: async (workingResult: void, batch: string[]) => {\n              const balancePromises = batch.map(async (address: string) => {\n                const balanceResult = await this.#getBalanceFromChain(\n                  address,\n                  ethQuery,\n                ).catch(() => null);\n\n                // Update account balances\n                if (balanceResult) {\n                  accountsForChain[address] = {\n                    balance: balanceResult,\n                  };\n                }\n              });\n\n              await Promise.allSettled(balancePromises);\n              return workingResult;\n            },\n          });\n        }\n\n        const stakedBalanceResult = await safelyExecuteWithTimeout(\n          async () =>\n            (await stakedBalancesPromise) as Record<string, StakedBalance>,\n        );\n\n        Object.entries(stakedBalanceResult ?? {}).forEach(\n          ([address, balance]) => {\n            accountsForChain[address] = {\n              ...accountsForChain[address],\n              stakedBalance: balance,\n            };\n          },\n        );\n\n        // After all batches are processed, return the updated data\n        return { chainId, accountsForChain };\n      });\n\n      // Wait for all networkClientId updates to settle in parallel\n      const allResults = await Promise.allSettled(updatePromises);\n\n      // Build a _copy_ of the current state and track whether anything changed\n      const nextAccountsByChainId: AccountTrackerControllerState['accountsByChainId'] =\n        cloneDeep(this.state.accountsByChainId);\n      let hasChanges = false;\n\n      allResults.forEach((result) => {\n        if (result.status === 'fulfilled') {\n          const { chainId, accountsForChain } = result.value;\n          // Only mark as changed if the incoming data differs\n          if (!isEqual(nextAccountsByChainId[chainId], accountsForChain)) {\n            nextAccountsByChainId[chainId] = accountsForChain;\n            hasChanges = true;\n          }\n        }\n      });\n\n      // 👇🏻 call `update` only when something is new / different\n      if (hasChanges) {\n        this.update((state) => {\n          state.accountsByChainId = nextAccountsByChainId;\n        });\n      }\n    } finally {\n      releaseLock();\n    }\n  }\n\n  /**\n   * Fetches the balance of a given address from the blockchain.\n   *\n   * @param address - The account address to fetch the balance for.\n   * @param ethQuery - The EthQuery instance to query getBalnce with.\n   * @returns A promise that resolves to the balance in a hex string format.\n   */\n  async #getBalanceFromChain(\n    address: string,\n    ethQuery?: EthQuery,\n  ): Promise<string | undefined> {\n    return await safelyExecuteWithTimeout(async () => {\n      assert(ethQuery, 'Provider not set.');\n      return await query(ethQuery, 'getBalance', [address]);\n    });\n  }\n\n  /**\n   * Sync accounts balances with some additional addresses.\n   *\n   * @param addresses - the additional addresses, may be hardware wallet addresses.\n   * @param networkClientId - Optional networkClientId to fetch a network client with.\n   * @returns accounts - addresses with synced balance\n   */\n  async syncBalanceWithAddresses(\n    addresses: string[],\n    networkClientId?: NetworkClientId,\n  ): Promise<\n    Record<string, { balance: string; stakedBalance?: StakedBalance }>\n  > {\n    const { ethQuery } = this.#getCorrectNetworkClient(networkClientId);\n\n    // TODO: This should use multicall when enabled by the user.\n    return await Promise.all(\n      addresses.map(\n        (address): Promise<[string, string, StakedBalance] | undefined> => {\n          return safelyExecuteWithTimeout(async () => {\n            assert(ethQuery, 'Provider not set.');\n            const balance = await query(ethQuery, 'getBalance', [address]);\n\n            let stakedBalance: StakedBalance;\n            if (this.#includeStakedAssets) {\n              stakedBalance = (\n                await this.#getStakedBalanceForChain([address], networkClientId)\n              )[address];\n            }\n            return [address, balance, stakedBalance];\n          });\n        },\n      ),\n    ).then((value) => {\n      return value.reduce((obj, item) => {\n        if (!item) {\n          return obj;\n        }\n\n        const [address, balance, stakedBalance] = item;\n        return {\n          ...obj,\n          [address]: {\n            balance,\n            stakedBalance,\n          },\n        };\n      }, {});\n    });\n  }\n\n  /**\n   * Updates the balances of multiple native tokens in a single batch operation.\n   * This is more efficient than calling updateNativeToken multiple times as it\n   * triggers only one state update.\n   *\n   * @param balances - Array of balance updates, each containing address, chainId, and balance.\n   */\n  updateNativeBalances(\n    balances: { address: string; chainId: Hex; balance: string }[],\n  ) {\n    this.update((state) => {\n      balances.forEach(({ address, chainId, balance }) => {\n        // Ensure the chainId exists in the state\n        if (!state.accountsByChainId[chainId]) {\n          state.accountsByChainId[chainId] = {};\n        }\n\n        // Ensure the address exists for this chain\n        if (!state.accountsByChainId[chainId][address]) {\n          state.accountsByChainId[chainId][address] = { balance: '0x0' };\n        }\n\n        // Update the balance\n        state.accountsByChainId[chainId][address].balance = balance;\n      });\n    });\n  }\n\n  /**\n   * Updates the staked balances of multiple accounts in a single batch operation.\n   * This is more efficient than updating staked balances individually as it\n   * triggers only one state update.\n   *\n   * @param stakedBalances - Array of staked balance updates, each containing address, chainId, and stakedBalance.\n   */\n  updateStakedBalances(\n    stakedBalances: {\n      address: string;\n      chainId: Hex;\n      stakedBalance: StakedBalance;\n    }[],\n  ) {\n    this.update((state) => {\n      stakedBalances.forEach(({ address, chainId, stakedBalance }) => {\n        // Ensure the chainId exists in the state\n        if (!state.accountsByChainId[chainId]) {\n          state.accountsByChainId[chainId] = {};\n        }\n\n        // Ensure the address exists for this chain\n        if (!state.accountsByChainId[chainId][address]) {\n          state.accountsByChainId[chainId][address] = { balance: '0x0' };\n        }\n\n        // Update the staked balance\n        state.accountsByChainId[chainId][address].stakedBalance = stakedBalance;\n      });\n    });\n  }\n\n  #registerMessageHandlers() {\n    this.messagingSystem.registerActionHandler(\n      `${controllerName}:updateNativeBalances` as const,\n      this.updateNativeBalances.bind(this),\n    );\n\n    this.messagingSystem.registerActionHandler(\n      `${controllerName}:updateStakedBalances` as const,\n      this.updateStakedBalances.bind(this),\n    );\n  }\n}\n\nexport default AccountTrackerController;\n"]}

package/dist/AccountTrackerController.d.cts

@@ -2,6 +2,7 @@ import type { AccountsControllerSelectedEvmAccountChangeEvent, AccountsControlle
 import type { ControllerStateChangeEvent, ControllerGetStateAction, RestrictedMessenger } from "@metamask/base-controller";
 import type { NetworkClientId, NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction } from "@metamask/network-controller";
 import type { PreferencesControllerGetStateAction } from "@metamask/preferences-controller";
+import { type Hex } from "@metamask/utils";
 import { type AssetsContractController, type StakedBalance } from "./AssetsContractController.cjs";
 /**
  * The name of the {@link AccountTrackerController}.
@@ -36,10 +37,24 @@ export type AccountTrackerControllerState = {
  * The action that can be performed to get the state of the {@link AccountTrackerController}.
  */
 export type AccountTrackerControllerGetStateAction = ControllerGetStateAction<typeof controllerName, AccountTrackerControllerState>;
+/**
+ * The action that can be performed to update multiple native token balances in batch.
+ */
+export type AccountTrackerUpdateNativeBalancesAction = {
+    type: `${typeof controllerName}:updateNativeBalances`;
+    handler: AccountTrackerController['updateNativeBalances'];
+};
+/**
+ * The action that can be performed to update multiple staked balances in batch.
+ */
+export type AccountTrackerUpdateStakedBalancesAction = {
+    type: `${typeof controllerName}:updateStakedBalances`;
+    handler: AccountTrackerController['updateStakedBalances'];
+};
 /**
  * The actions that can be performed using the {@link AccountTrackerController}.
  */
-export type AccountTrackerControllerActions = AccountTrackerControllerGetStateAction;
+export type AccountTrackerControllerActions = AccountTrackerControllerGetStateAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 /**
  * The messenger of the {@link AccountTrackerController} for communication.
  */
@@ -128,6 +143,30 @@ export declare class AccountTrackerController extends AccountTrackerController_b
         balance: string;
         stakedBalance?: StakedBalance;
     }>>;
+    /**
+     * Updates the balances of multiple native tokens in a single batch operation.
+     * This is more efficient than calling updateNativeToken multiple times as it
+     * triggers only one state update.
+     *
+     * @param balances - Array of balance updates, each containing address, chainId, and balance.
+     */
+    updateNativeBalances(balances: {
+        address: string;
+        chainId: Hex;
+        balance: string;
+    }[]): void;
+    /**
+     * Updates the staked balances of multiple accounts in a single batch operation.
+     * This is more efficient than updating staked balances individually as it
+     * triggers only one state update.
+     *
+     * @param stakedBalances - Array of staked balance updates, each containing address, chainId, and stakedBalance.
+     */
+    updateStakedBalances(stakedBalances: {
+        address: string;
+        chainId: Hex;
+        stakedBalance: StakedBalance;
+    }[]): void;
 }
 export default AccountTrackerController;
 //# sourceMappingURL=AccountTrackerController.d.cts.map

package/dist/AccountTrackerController.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"AccountTrackerController.d.cts","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,+CAA+C,EAC/C,0CAA0C,EAC1C,oCAAoC,EACpC,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EACV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAM5F,OAAO,EAEL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAGpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD;;;;;;;;GAQG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,CAAC,OAAO,EAAE,MAAM,GAAG,kBAAkB,CAAA;KAAE,CAAC,CAAC;CAC9E,CAAC;AASF;;GAEG;AACH,MAAM,MAAM,sCAAsC,GAAG,wBAAwB,CAC3E,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,+BAA+B,GACzC,sCAAsC,CAAC;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,oCAAoC,GACpC,mCAAmC,GACnC,0CAA0C,GAC1C,+BAA+B,GAC/B,2CAA2C,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAClD,0BAA0B,CACxB,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,8BAA8B,GACxC,wCAAwC,CAAC;AAE3C;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,+CAA+C,GAC/C,4CAA4C,CAAC;AAEjD;;GAEG;AACH,MAAM,MAAM,iCAAiC,GAAG,mBAAmB,CACjE,OAAO,cAAc,EACrB,+BAA+B,GAAG,cAAc,EAChD,8BAA8B,GAAG,aAAa,EAC9C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,0EAA0E;AAC1E,KAAK,0BAA0B,GAAG;IAChC,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,wBAAyB,SAAQ,8BAC5C,OAAO,cAAc,EACrB,6BAA6B,EAC7B,iCAAiC,CAClC;;IAOC;;;;;;;;;OASG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,GAC5B,EAAE;QACD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAC/C,SAAS,EAAE,iCAAiC,CAAC;QAC7C,wBAAwB,EAAE,wBAAwB,CAAC,0BAA0B,CAAC,CAAC;QAC/E,mBAAmB,CAAC,EAAE,OAAO,CAAC;KAC/B;IAwCD,OAAO,CAAC,YAAY;IA0GpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA8JjD;;;;;;OAMG;IACG,wBAAwB,CAC5B,SAAS,EAAE,MAAM,EAAE,EACnB,eAAe,CAAC,EAAE,eAAe,GAChC,OAAO,CACR,MAAM,CAAC,MAAM,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,CAAC,CACnE;CAsCF;AAED,eAAe,wBAAwB,CAAC"}
+{"version":3,"file":"AccountTrackerController.d.cts","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,+CAA+C,EAC/C,0CAA0C,EAC1C,oCAAoC,EACpC,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EACV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAKhE,OAAO,EAEL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAGpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD;;;;;;;;GAQG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,CAAC,OAAO,EAAE,MAAM,GAAG,kBAAkB,CAAA;KAAE,CAAC,CAAC;CAC9E,CAAC;AASF;;GAEG;AACH,MAAM,MAAM,sCAAsC,GAAG,wBAAwB,CAC3E,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAAG;IACrD,IAAI,EAAE,GAAG,OAAO,cAAc,uBAAuB,CAAC;IACtD,OAAO,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAAG;IACrD,IAAI,EAAE,GAAG,OAAO,cAAc,uBAAuB,CAAC;IACtD,OAAO,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,+BAA+B,GACvC,sCAAsC,GACtC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,oCAAoC,GACpC,mCAAmC,GACnC,0CAA0C,GAC1C,+BAA+B,GAC/B,2CAA2C,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAClD,0BAA0B,CACxB,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,8BAA8B,GACxC,wCAAwC,CAAC;AAE3C;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,+CAA+C,GAC/C,4CAA4C,CAAC;AAEjD;;GAEG;AACH,MAAM,MAAM,iCAAiC,GAAG,mBAAmB,CACjE,OAAO,cAAc,EACrB,+BAA+B,GAAG,cAAc,EAChD,8BAA8B,GAAG,aAAa,EAC9C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,0EAA0E;AAC1E,KAAK,0BAA0B,GAAG;IAChC,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,wBAAyB,SAAQ,8BAC5C,OAAO,cAAc,EACrB,6BAA6B,EAC7B,iCAAiC,CAClC;;IAOC;;;;;;;;;OASG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,GAC5B,EAAE;QACD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAC/C,SAAS,EAAE,iCAAiC,CAAC;QAC7C,wBAAwB,EAAE,wBAAwB,CAAC,0BAA0B,CAAC,CAAC;QAC/E,mBAAmB,CAAC,EAAE,OAAO,CAAC;KAC/B;IA0CD,OAAO,CAAC,YAAY;IA0GpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA8JjD;;;;;;OAMG;IACG,wBAAwB,CAC5B,SAAS,EAAE,MAAM,EAAE,EACnB,eAAe,CAAC,EAAE,eAAe,GAChC,OAAO,CACR,MAAM,CAAC,MAAM,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,CAAC,CACnE;IAuCD;;;;;;OAMG;IACH,oBAAoB,CAClB,QAAQ,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,GAAG,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EAAE;IAoBhE;;;;;;OAMG;IACH,oBAAoB,CAClB,cAAc,EAAE;QACd,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,GAAG,CAAC;QACb,aAAa,EAAE,aAAa,CAAC;KAC9B,EAAE;CA+BN;AAED,eAAe,wBAAwB,CAAC"}

package/dist/AccountTrackerController.d.mts

@@ -2,6 +2,7 @@ import type { AccountsControllerSelectedEvmAccountChangeEvent, AccountsControlle
 import type { ControllerStateChangeEvent, ControllerGetStateAction, RestrictedMessenger } from "@metamask/base-controller";
 import type { NetworkClientId, NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction } from "@metamask/network-controller";
 import type { PreferencesControllerGetStateAction } from "@metamask/preferences-controller";
+import { type Hex } from "@metamask/utils";
 import { type AssetsContractController, type StakedBalance } from "./AssetsContractController.mjs";
 /**
  * The name of the {@link AccountTrackerController}.
@@ -36,10 +37,24 @@ export type AccountTrackerControllerState = {
  * The action that can be performed to get the state of the {@link AccountTrackerController}.
  */
 export type AccountTrackerControllerGetStateAction = ControllerGetStateAction<typeof controllerName, AccountTrackerControllerState>;
+/**
+ * The action that can be performed to update multiple native token balances in batch.
+ */
+export type AccountTrackerUpdateNativeBalancesAction = {
+    type: `${typeof controllerName}:updateNativeBalances`;
+    handler: AccountTrackerController['updateNativeBalances'];
+};
+/**
+ * The action that can be performed to update multiple staked balances in batch.
+ */
+export type AccountTrackerUpdateStakedBalancesAction = {
+    type: `${typeof controllerName}:updateStakedBalances`;
+    handler: AccountTrackerController['updateStakedBalances'];
+};
 /**
  * The actions that can be performed using the {@link AccountTrackerController}.
  */
-export type AccountTrackerControllerActions = AccountTrackerControllerGetStateAction;
+export type AccountTrackerControllerActions = AccountTrackerControllerGetStateAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 /**
  * The messenger of the {@link AccountTrackerController} for communication.
  */
@@ -128,6 +143,30 @@ export declare class AccountTrackerController extends AccountTrackerController_b
         balance: string;
         stakedBalance?: StakedBalance;
     }>>;
+    /**
+     * Updates the balances of multiple native tokens in a single batch operation.
+     * This is more efficient than calling updateNativeToken multiple times as it
+     * triggers only one state update.
+     *
+     * @param balances - Array of balance updates, each containing address, chainId, and balance.
+     */
+    updateNativeBalances(balances: {
+        address: string;
+        chainId: Hex;
+        balance: string;
+    }[]): void;
+    /**
+     * Updates the staked balances of multiple accounts in a single batch operation.
+     * This is more efficient than updating staked balances individually as it
+     * triggers only one state update.
+     *
+     * @param stakedBalances - Array of staked balance updates, each containing address, chainId, and stakedBalance.
+     */
+    updateStakedBalances(stakedBalances: {
+        address: string;
+        chainId: Hex;
+        stakedBalance: StakedBalance;
+    }[]): void;
 }
 export default AccountTrackerController;
 //# sourceMappingURL=AccountTrackerController.d.mts.map

package/dist/AccountTrackerController.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"AccountTrackerController.d.mts","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,+CAA+C,EAC/C,0CAA0C,EAC1C,oCAAoC,EACpC,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EACV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAM5F,OAAO,EAEL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAGpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD;;;;;;;;GAQG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,CAAC,OAAO,EAAE,MAAM,GAAG,kBAAkB,CAAA;KAAE,CAAC,CAAC;CAC9E,CAAC;AASF;;GAEG;AACH,MAAM,MAAM,sCAAsC,GAAG,wBAAwB,CAC3E,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,+BAA+B,GACzC,sCAAsC,CAAC;AAEzC;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,oCAAoC,GACpC,mCAAmC,GACnC,0CAA0C,GAC1C,+BAA+B,GAC/B,2CAA2C,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAClD,0BAA0B,CACxB,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,8BAA8B,GACxC,wCAAwC,CAAC;AAE3C;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,+CAA+C,GAC/C,4CAA4C,CAAC;AAEjD;;GAEG;AACH,MAAM,MAAM,iCAAiC,GAAG,mBAAmB,CACjE,OAAO,cAAc,EACrB,+BAA+B,GAAG,cAAc,EAChD,8BAA8B,GAAG,aAAa,EAC9C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,0EAA0E;AAC1E,KAAK,0BAA0B,GAAG;IAChC,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,wBAAyB,SAAQ,8BAC5C,OAAO,cAAc,EACrB,6BAA6B,EAC7B,iCAAiC,CAClC;;IAOC;;;;;;;;;OASG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,GAC5B,EAAE;QACD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAC/C,SAAS,EAAE,iCAAiC,CAAC;QAC7C,wBAAwB,EAAE,wBAAwB,CAAC,0BAA0B,CAAC,CAAC;QAC/E,mBAAmB,CAAC,EAAE,OAAO,CAAC;KAC/B;IAwCD,OAAO,CAAC,YAAY;IA0GpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA8JjD;;;;;;OAMG;IACG,wBAAwB,CAC5B,SAAS,EAAE,MAAM,EAAE,EACnB,eAAe,CAAC,EAAE,eAAe,GAChC,OAAO,CACR,MAAM,CAAC,MAAM,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,CAAC,CACnE;CAsCF;AAED,eAAe,wBAAwB,CAAC"}
+{"version":3,"file":"AccountTrackerController.d.mts","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,+CAA+C,EAC/C,0CAA0C,EAC1C,oCAAoC,EACpC,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACpB,kCAAkC;AAOnC,OAAO,KAAK,EACV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAKhE,OAAO,EAEL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAGpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD;;;;;;;;GAQG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,CAAC,OAAO,EAAE,MAAM,GAAG,kBAAkB,CAAA;KAAE,CAAC,CAAC;CAC9E,CAAC;AASF;;GAEG;AACH,MAAM,MAAM,sCAAsC,GAAG,wBAAwB,CAC3E,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAAG;IACrD,IAAI,EAAE,GAAG,OAAO,cAAc,uBAAuB,CAAC;IACtD,OAAO,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAAG;IACrD,IAAI,EAAE,GAAG,OAAO,cAAc,uBAAuB,CAAC;IACtD,OAAO,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;CAC3D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,+BAA+B,GACvC,sCAAsC,GACtC,wCAAwC,GACxC,wCAAwC,CAAC;AAE7C;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,oCAAoC,GACpC,mCAAmC,GACnC,0CAA0C,GAC1C,+BAA+B,GAC/B,2CAA2C,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,wCAAwC,GAClD,0BAA0B,CACxB,OAAO,cAAc,EACrB,6BAA6B,CAC9B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,8BAA8B,GACxC,wCAAwC,CAAC;AAE3C;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,+CAA+C,GAC/C,4CAA4C,CAAC;AAEjD;;GAEG;AACH,MAAM,MAAM,iCAAiC,GAAG,mBAAmB,CACjE,OAAO,cAAc,EACrB,+BAA+B,GAAG,cAAc,EAChD,8BAA8B,GAAG,aAAa,EAC9C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAEF,0EAA0E;AAC1E,KAAK,0BAA0B,GAAG;IAChC,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,wBAAyB,SAAQ,8BAC5C,OAAO,cAAc,EACrB,6BAA6B,EAC7B,iCAAiC,CAClC;;IAOC;;;;;;;;;OASG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,GAC5B,EAAE;QACD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAC/C,SAAS,EAAE,iCAAiC,CAAC;QAC7C,wBAAwB,EAAE,wBAAwB,CAAC,0BAA0B,CAAC,CAAC;QAC/E,mBAAmB,CAAC,EAAE,OAAO,CAAC;KAC/B;IA0CD,OAAO,CAAC,YAAY;IA0GpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA8JjD;;;;;;OAMG;IACG,wBAAwB,CAC5B,SAAS,EAAE,MAAM,EAAE,EACnB,eAAe,CAAC,EAAE,eAAe,GAChC,OAAO,CACR,MAAM,CAAC,MAAM,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,CAAC,CACnE;IAuCD;;;;;;OAMG;IACH,oBAAoB,CAClB,QAAQ,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,GAAG,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EAAE;IAoBhE;;;;;;OAMG;IACH,oBAAoB,CAClB,cAAc,EAAE;QACd,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,GAAG,CAAC;QACb,aAAa,EAAE,aAAa,CAAC;KAC9B,EAAE;CA+BN;AAED,eAAe,wBAAwB,CAAC"}

package/dist/AccountTrackerController.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 _AccountTrackerController_instances, _AccountTrackerController_refreshMutex, _AccountTrackerController_includeStakedAssets, _AccountTrackerController_getStakedBalanceForChain, _AccountTrackerController_getCorrectNetworkClient, _AccountTrackerController_getNetworkClientIds, _AccountTrackerController_getBalanceFromChain;
+var _AccountTrackerController_instances, _AccountTrackerController_refreshMutex, _AccountTrackerController_includeStakedAssets, _AccountTrackerController_getStakedBalanceForChain, _AccountTrackerController_getCorrectNetworkClient, _AccountTrackerController_getNetworkClientIds, _AccountTrackerController_getBalanceFromChain, _AccountTrackerController_registerMessageHandlers;
 function $importDefault(module) {
     if (module?.__esModule) {
         return module.default;
@@ -82,6 +82,7 @@ export class AccountTrackerController extends StaticIntervalPollingController()
                 this.refresh(__classPrivateFieldGet(this, _AccountTrackerController_instances, "m", _AccountTrackerController_getNetworkClientIds).call(this));
             }
         }, (event) => event.address);
+        __classPrivateFieldGet(this, _AccountTrackerController_instances, "m", _AccountTrackerController_registerMessageHandlers).call(this);
     }
     syncAccounts(newChainIds) {
         const accountsByChainId = cloneDeep(this.state.accountsByChainId);
@@ -272,6 +273,52 @@ export class AccountTrackerController extends StaticIntervalPollingController()
             }, {});
         });
     }
+    /**
+     * Updates the balances of multiple native tokens in a single batch operation.
+     * This is more efficient than calling updateNativeToken multiple times as it
+     * triggers only one state update.
+     *
+     * @param balances - Array of balance updates, each containing address, chainId, and balance.
+     */
+    updateNativeBalances(balances) {
+        this.update((state) => {
+            balances.forEach(({ address, chainId, balance }) => {
+                // Ensure the chainId exists in the state
+                if (!state.accountsByChainId[chainId]) {
+                    state.accountsByChainId[chainId] = {};
+                }
+                // Ensure the address exists for this chain
+                if (!state.accountsByChainId[chainId][address]) {
+                    state.accountsByChainId[chainId][address] = { balance: '0x0' };
+                }
+                // Update the balance
+                state.accountsByChainId[chainId][address].balance = balance;
+            });
+        });
+    }
+    /**
+     * Updates the staked balances of multiple accounts in a single batch operation.
+     * This is more efficient than updating staked balances individually as it
+     * triggers only one state update.
+     *
+     * @param stakedBalances - Array of staked balance updates, each containing address, chainId, and stakedBalance.
+     */
+    updateStakedBalances(stakedBalances) {
+        this.update((state) => {
+            stakedBalances.forEach(({ address, chainId, stakedBalance }) => {
+                // Ensure the chainId exists in the state
+                if (!state.accountsByChainId[chainId]) {
+                    state.accountsByChainId[chainId] = {};
+                }
+                // Ensure the address exists for this chain
+                if (!state.accountsByChainId[chainId][address]) {
+                    state.accountsByChainId[chainId][address] = { balance: '0x0' };
+                }
+                // Update the staked balance
+                state.accountsByChainId[chainId][address].stakedBalance = stakedBalance;
+            });
+        });
+    }
 }
 _AccountTrackerController_refreshMutex = new WeakMap(), _AccountTrackerController_includeStakedAssets = new WeakMap(), _AccountTrackerController_getStakedBalanceForChain = new WeakMap(), _AccountTrackerController_instances = new WeakSet(), _AccountTrackerController_getCorrectNetworkClient = function _AccountTrackerController_getCorrectNetworkClient(networkClientId) {
     const selectedNetworkClientId = networkClientId ??
@@ -300,6 +347,9 @@ async function _AccountTrackerController_getBalanceFromChain(address, ethQuery)
         assert(ethQuery, 'Provider not set.');
         return await query(ethQuery, 'getBalance', [address]);
     });
+}, _AccountTrackerController_registerMessageHandlers = function _AccountTrackerController_registerMessageHandlers() {
+    this.messagingSystem.registerActionHandler(`${controllerName}:updateNativeBalances`, this.updateNativeBalances.bind(this));
+    this.messagingSystem.registerActionHandler(`${controllerName}:updateStakedBalances`, this.updateStakedBalances.bind(this));
 };
 export default AccountTrackerController;
 //# sourceMappingURL=AccountTrackerController.mjs.map