@metamask/assets-controllers 74.1.0 → 74.1.1

package/CHANGELOG.md

@@ -7,6 +7,18 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [74.1.1]
+
+### Changed
+
+- Improve balance fetching performance and resilience by parallelizing multi-chain operations and moving timeout handling to fetchers ([#6390](https://github.com/MetaMask/core/pull/6390))
+
+  - Replace sequential `for` loops with `Promise.allSettled` in `RpcBalanceFetcher` and `AccountTrackerController` for parallel chain processing
+  - Move timeout handling from controller-level `Promise.race` to fetcher-level `safelyExecuteWithTimeout` for better error isolation
+  - Add `safelyExecuteWithTimeout` to both `RpcBalanceFetcher` and `AccountsApiBalanceFetcher` to prevent individual chain timeouts from blocking other chains
+  - Remove redundant timeout wrappers from `TokenBalancesController` and `AccountTrackerController`
+  - Improve test coverage for timeout and error handling scenarios in all balance fetchers
+
 ## [74.1.0]
 
 ### Added
@@ -1914,7 +1926,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.1.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.1.1...HEAD
+[74.1.1]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.1.0...@metamask/assets-controllers@74.1.1
 [74.1.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@74.0.0...@metamask/assets-controllers@74.1.0
 [74.0.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.3.0...@metamask/assets-controllers@74.0.0
 [73.3.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@73.2.0...@metamask/assets-controllers@73.3.0

package/dist/AccountTrackerController.cjs

@@ -33,7 +33,6 @@ const api_balance_fetcher_1 = require("./multi-chain-accounts-service/api-balanc
  * The name of the {@link AccountTrackerController}.
  */
 const controllerName = 'AccountTrackerController';
-const DEFAULT_TIMEOUT_MS = 15000;
 const ZERO_ADDRESS = '0x0000000000000000000000000000000000000000';
 /**
  * RPC-based balance fetcher for AccountTrackerController.
@@ -55,13 +54,14 @@ class AccountTrackerRpcBalanceFetcher {
         return true; // fallback – supports every chain
     }
     async fetch({ chainIds, queryAllAccounts, selectedAccount, allAccounts, }) {
-        const results = [];
-        for (const chainId of chainIds) {
+        // Process all chains in parallel for better performance
+        const chainProcessingPromises = chainIds.map(async (chainId) => {
             const accountsToUpdate = queryAllAccounts
                 ? Object.values(allAccounts).map((account) => (0, controller_utils_1.toChecksumHexAddress)(account.address))
                 : [selectedAccount];
             const { provider, blockTracker } = __classPrivateFieldGet(this, _AccountTrackerRpcBalanceFetcher_getNetworkClient, "f").call(this, chainId);
             const ethQuery = new eth_query_1.default(provider);
+            const chainResults = [];
             // Force fresh block data before multicall
             await (0, controller_utils_1.safelyExecuteWithTimeout)(() => blockTracker?.checkForLatestBlock?.());
             // Fetch native balances
@@ -71,7 +71,7 @@ class AccountTrackerRpcBalanceFetcher {
                 const nativeBalances = await (0, controller_utils_1.safelyExecuteWithTimeout)(() => contract.balances(accountsToUpdate, [ZERO_ADDRESS]), false, 3000);
                 if (nativeBalances) {
                     accountsToUpdate.forEach((address, index) => {
-                        results.push({
+                        chainResults.push({
                             success: true,
                             value: new bn_js_1.default(nativeBalances[index].toString()),
                             account: address,
@@ -91,7 +91,7 @@ class AccountTrackerRpcBalanceFetcher {
                         const balancePromises = batch.map(async (address) => {
                             const balanceResult = await __classPrivateFieldGet(this, _AccountTrackerRpcBalanceFetcher_instances, "m", _AccountTrackerRpcBalanceFetcher_getBalanceFromChain).call(this, address, ethQuery).catch(() => null);
                             if (balanceResult) {
-                                results.push({
+                                chainResults.push({
                                     success: true,
                                     value: new bn_js_1.default(balanceResult.replace('0x', ''), 16),
                                     account: address,
@@ -100,7 +100,7 @@ class AccountTrackerRpcBalanceFetcher {
                                 });
                             }
                             else {
-                                results.push({
+                                chainResults.push({
                                     success: false,
                                     account: address,
                                     token: ZERO_ADDRESS,
@@ -122,7 +122,7 @@ class AccountTrackerRpcBalanceFetcher {
                     const stakingContractAddress = AssetsContractController_1.STAKING_CONTRACT_ADDRESS_BY_CHAINID[chainId];
                     if (stakingContractAddress) {
                         Object.entries(stakedBalanceResult).forEach(([address, balance]) => {
-                            results.push({
+                            chainResults.push({
                                 success: true,
                                 value: balance
                                     ? new bn_js_1.default(balance.replace('0x', ''), 16)
@@ -135,7 +135,20 @@ class AccountTrackerRpcBalanceFetcher {
                     }
                 }
             }
-        }
+            return chainResults;
+        });
+        // Wait for all chains to complete (or fail) and collect results
+        const chainResultsArray = await Promise.allSettled(chainProcessingPromises);
+        const results = [];
+        chainResultsArray.forEach((chainResult) => {
+            if (chainResult.status === 'fulfilled') {
+                results.push(...chainResult.value);
+            }
+            else {
+                // Log error but continue with other chains
+                console.warn('Chain processing failed:', chainResult.reason);
+            }
+        });
         return results;
     }
 }
@@ -304,17 +317,12 @@ class AccountTrackerController extends (0, polling_controller_1.StaticIntervalPo
                     continue;
                 }
                 try {
-                    const balances = await Promise.race([
-                        fetcher.fetch({
-                            chainIds: supportedChains,
-                            queryAllAccounts: isMultiAccountBalancesEnabled,
-                            selectedAccount: (0, controller_utils_1.toChecksumHexAddress)(selectedAccount.address),
-                            allAccounts,
-                        }),
-                        new Promise((_resolve, reject) => setTimeout(() => {
-                            reject(new Error(`Timeout after ${DEFAULT_TIMEOUT_MS}ms`));
-                        }, DEFAULT_TIMEOUT_MS)),
-                    ]);
+                    const balances = await fetcher.fetch({
+                        chainIds: supportedChains,
+                        queryAllAccounts: isMultiAccountBalancesEnabled,
+                        selectedAccount: (0, controller_utils_1.toChecksumHexAddress)(selectedAccount.address),
+                        allAccounts,
+                    });
                     if (balances && balances.length > 0) {
                         aggregated.push(...balances);
                         // Remove chains that were successfully processed

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;AAO3C,qEAA+E;AAE/E,2CAAgE;AAChE,6CAAoC;AACpC,kDAAuB;AACvB,mCAA4C;AAC5C,sGAA4E;AAE5E,6EAKoC;AACpC,iDAAgF;AAChF,gGAI4D;AAE5D;;GAEG;AACH,MAAM,cAAc,GAAG,0BAA0B,CAAC;AAKlD,MAAM,kBAAkB,GAAG,KAAK,CAAC;AACjC,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE;;;GAGG;AACH,MAAM,+BAA+B;IASnC,YACE,WAA2C,EAC3C,gBAAiD,EACjD,mBAA4B,EAC5B,wBAA8E;;QAZvE,+DAA6C;QAE7C,oEAAmD;QAEnD,uEAA8B;QAE9B,4EAAgF;QAQvF,uBAAA,IAAI,gDAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,qDAAqB,gBAAgB,MAAA,CAAC;QAC1C,uBAAA,IAAI,wDAAwB,mBAAmB,MAAA,CAAC;QAChD,uBAAA,IAAI,6DAA6B,wBAAwB,MAAA,CAAC;IAC5D,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,CAAC,kCAAkC;IACjD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,EACV,QAAQ,EACR,gBAAgB,EAChB,eAAe,EACf,WAAW,GAC4B;QACvC,MAAM,OAAO,GAAuB,EAAE,CAAC;QAEvC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,MAAM,gBAAgB,GAAG,gBAAgB;gBACvC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAC5B,CAAC,OAAO,EAAE,EAAE,CACV,IAAA,uCAAoB,EAAC,OAAO,CAAC,OAAO,CAAoB,CAC3D;gBACH,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YAEtB,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,uBAAA,IAAI,yDAAkB,MAAtB,IAAI,EAAmB,OAAO,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,IAAI,mBAAQ,CAAC,QAAQ,CAAC,CAAC;YAExC,0CAA0C;YAC1C,MAAM,IAAA,2CAAwB,EAAC,GAAG,EAAE,CAClC,YAAY,EAAE,mBAAmB,EAAE,EAAE,CACtC,CAAC;YAEF,wBAAwB;YACxB,IAAI,IAAA,mBAAW,EAAC,kEAAuC,EAAE,OAAO,CAAC,EAAE;gBACjE,MAAM,eAAe,GAAG,kEAAuC,CAC7D,OAAO,CACE,CAAC;gBAEZ,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAC3B,eAAe,EACf,yCAA6B,EAC7B,uBAAA,IAAI,oDAAa,MAAjB,IAAI,EAAc,OAAO,CAAC,CAC3B,CAAC;gBAEF,MAAM,cAAc,GAAG,MAAM,IAAA,2CAAwB,EACnD,GAAG,EAAE,CACH,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,YAAY,CAAC,CAEjD,EACH,KAAK,EACL,IAAK,CACN,CAAC;gBAEF,IAAI,cAAc,EAAE;oBAClB,gBAAgB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;wBAC1C,OAAO,CAAC,IAAI,CAAC;4BACX,OAAO,EAAE,IAAI;4BACb,KAAK,EAAE,IAAI,eAAE,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;4BAC/C,OAAO,EAAE,OAAO;4BAChB,KAAK,EAAE,YAAY;4BACnB,OAAO;yBACR,CAAC,CAAC;oBACL,CAAC,CAAC,CAAC;iBACJ;aACF;iBAAM;gBACL,4DAA4D;gBAC5D,MAAM,IAAA,oCAAuB,EAAe;oBAC1C,MAAM,EAAE,gBAAgB;oBACxB,SAAS,EAAE,oCAAuB;oBAClC,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,KAAK,EAAE,aAAmB,EAAE,KAAe,EAAE,EAAE;wBACxD,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,OAAe,EAAE,EAAE;4BAC1D,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,wGAAqB,MAAzB,IAAI,EAC9B,OAAO,EACP,QAAQ,CACT,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;4BAEpB,IAAI,aAAa,EAAE;gCACjB,OAAO,CAAC,IAAI,CAAC;oCACX,OAAO,EAAE,IAAI;oCACb,KAAK,EAAE,IAAI,eAAE,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oCAClD,OAAO,EAAE,OAA0B;oCACnC,KAAK,EAAE,YAAY;oCACnB,OAAO;iCACR,CAAC,CAAC;6BACJ;iCAAM;gCACL,OAAO,CAAC,IAAI,CAAC;oCACX,OAAO,EAAE,KAAK;oCACd,OAAO,EAAE,OAA0B;oCACnC,KAAK,EAAE,YAAY;oCACnB,OAAO;iCACR,CAAC,CAAC;6BACJ;wBACH,CAAC,CAAC,CAAC;wBAEH,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;wBAC1C,OAAO,aAAa,CAAC;oBACvB,CAAC;iBACF,CAAC,CAAC;aACJ;YAED,mCAAmC;YACnC,IAAI,uBAAA,IAAI,4DAAqB,EAAE;gBAC7B,MAAM,qBAAqB,GAAG,uBAAA,IAAI,iEAA0B,MAA9B,IAAI,EAChC,gBAAgB,EAChB,OAAO,CACR,CAAC;gBAEF,MAAM,mBAAmB,GAAG,MAAM,IAAA,2CAAwB,EACxD,KAAK,IAAI,EAAE,CACT,CAAC,MAAM,qBAAqB,CAAkC,CACjE,CAAC;gBAEF,IAAI,mBAAmB,EAAE;oBACvB,mDAAmD;oBACnD,MAAM,sBAAsB,GAC1B,8DAAmC,CACjC,OAA2D,CAC5D,CAAC;oBAEJ,IAAI,sBAAsB,EAAE;wBAC1B,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,OAAO,CACzC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;4BACrB,OAAO,CAAC,IAAI,CAAC;gCACX,OAAO,EAAE,IAAI;gCACb,KAAK,EAAE,OAAO;oCACZ,CAAC,CAAC,IAAI,eAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oCACvC,CAAC,CAAC,IAAI,eAAE,CAAC,GAAG,CAAC;gCACf,OAAO,EAAE,OAA0B;gCACnC,KAAK,EAAE,IAAA,uCAAoB,EACzB,sBAAsB,CACJ;gCACpB,OAAO;6BACR,CAAC,CAAC;wBACL,CAAC,CACF,CAAC;qBACH;iBACF;aACF;SACF;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CAkBF;;AAhBC;;;;;;GAMG;AACH,KAAK,+DACH,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;AA4BH,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;IASC;;;;;;;;;;;OAWG;IACH,YAAY,EACV,QAAQ,GAAG,KAAK,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAAmB,GAAG,KAAK,EAC3B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GASnC;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;;QAxDI,iDAAgB,IAAI,mBAAK,EAAE,EAAC;QAE5B,gEAA8B;QAE9B,qEAAgF;QAEhF,4DAAmC;QAiJnC,gDAAe,CAAC,OAAY,EAAgB,EAAE;YACrD,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,qDAAoB,CAAC,OAAY,EAAE,EAAE;YAC5C,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;QArHA,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAE1D,uBAAA,IAAI,iDAAwB,mBAAmB,MAAA,CAAC;QAEhD,6EAA6E;QAC7E,uBAAA,IAAI,6CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,+CAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,6CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,+BAA+B,CACjC,uBAAA,IAAI,6CAAa,EACjB,uBAAA,IAAI,kDAAkB,EACtB,mBAAmB,EACnB,wBAAwB,CACzB;SACF,MAAA,CAAC;QAEF,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;IAyED;;;;;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,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QACF,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACjE,gCAAgC,CACjC,CAAC;QAEF,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,8CAA8C;YAC9C,MAAM,UAAU,GAAuB,EAAE,CAAC;YAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAiB,CAAC;YAEpD,oEAAoE;YACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,iDAAiB,EAAE;gBAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;gBACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;oBAC3B,SAAS;iBACV;gBAED,IAAI;oBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;wBAClC,OAAO,CAAC,KAAK,CAAC;4BACZ,QAAQ,EAAE,eAAe;4BACzB,gBAAgB,EAAE,6BAA6B;4BAC/C,eAAe,EAAE,IAAA,uCAAoB,EACnC,eAAe,CAAC,OAAO,CACL;4BACpB,WAAW;yBACZ,CAAC;wBACF,IAAI,OAAO,CAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,CACtC,UAAU,CAAC,GAAG,EAAE;4BACd,MAAM,CAAC,IAAI,KAAK,CAAC,iBAAiB,kBAAkB,IAAI,CAAC,CAAC,CAAC;wBAC7D,CAAC,EAAE,kBAAkB,CAAC,CACvB;qBACF,CAAC,CAAC;oBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;wBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;wBAC7B,iDAAiD;wBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;wBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;qBACH;iBACF;gBAAC,OAAO,KAAK,EAAE;oBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;oBACF,sCAAsC;iBACvC;gBAED,iDAAiD;gBACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;oBAChC,MAAM;iBACP;aACF;YAED,yEAAyE;YACzE,MAAM,qBAAqB,GACzB,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC1C,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,+BAA+B,GAGjC,EAAE,CAAC;YAEP,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,MAAM,QAAQ,GAAG,KAAK,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;oBAE3C,IAAI,KAAK,KAAK,YAAY,EAAE;wBAC1B,iBAAiB;wBACjB,IAAI,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE;4BAChE,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,GAAG,QAAQ,CAAC;4BAC3D,UAAU,GAAG,IAAI,CAAC;yBACnB;qBACF;yBAAM;wBACL,iDAAiD;wBACjD,IAAI,CAAC,+BAA+B,CAAC,OAAO,CAAC,EAAE;4BAC7C,+BAA+B,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;yBAC/C;wBACD,+BAA+B,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,wBAAwB;YACxB,MAAM,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC,OAAO,CACrD,CAAC,CAAC,OAAO,EAAE,iBAAiB,CAAC,EAAE,EAAE;gBAC/B,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,CACvC,CAAC,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,EAAE;oBAC3B,IACE,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa;wBACrD,aAAa,EACb;wBACA,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa;4BACnD,aAAa,CAAC;wBAChB,UAAU,GAAG,IAAI,CAAC;qBACnB;gBACH,CAAC,CACF,CAAC;YACJ,CAAC,CACF,CAAC;YAEF,yCAAyC;YACzC,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;IAED;;;;;;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;AArfD,4DAqfC;6gBAzT0B,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;IA0QC,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  NetworkClient,\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 BN from 'bn.js';\nimport { cloneDeep, isEqual } from 'lodash';\nimport abiSingleCallBalancesContract from 'single-call-balance-checker-abi';\n\nimport {\n  SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID,\n  STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n  type AssetsContractController,\n  type StakedBalance,\n} from './AssetsContractController';\nimport { reduceInBatchesSerially, TOKEN_PRICES_BATCH_SIZE } from './assetsUtil';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\n\n/**\n * The name of the {@link AccountTrackerController}.\n */\nconst controllerName = 'AccountTrackerController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst DEFAULT_TIMEOUT_MS = 15000;\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\n/**\n * RPC-based balance fetcher for AccountTrackerController.\n * Fetches only native balances and staked balances (no token balances).\n */\nclass AccountTrackerRpcBalanceFetcher implements BalanceFetcher {\n  readonly #getProvider: (chainId: Hex) => Web3Provider;\n\n  readonly #getNetworkClient: (chainId: Hex) => NetworkClient;\n\n  readonly #includeStakedAssets: boolean;\n\n  readonly #getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n\n  constructor(\n    getProvider: (chainId: Hex) => Web3Provider,\n    getNetworkClient: (chainId: Hex) => NetworkClient,\n    includeStakedAssets: boolean,\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'],\n  ) {\n    this.#getProvider = getProvider;\n    this.#getNetworkClient = getNetworkClient;\n    this.#includeStakedAssets = includeStakedAssets;\n    this.#getStakedBalanceForChain = getStakedBalanceForChain;\n  }\n\n  supports(): boolean {\n    return true; // fallback – supports every chain\n  }\n\n  async fetch({\n    chainIds,\n    queryAllAccounts,\n    selectedAccount,\n    allAccounts,\n  }: Parameters<BalanceFetcher['fetch']>[0]): Promise<ProcessedBalance[]> {\n    const results: ProcessedBalance[] = [];\n\n    for (const chainId of chainIds) {\n      const accountsToUpdate = queryAllAccounts\n        ? Object.values(allAccounts).map(\n            (account) =>\n              toChecksumHexAddress(account.address) as ChecksumAddress,\n          )\n        : [selectedAccount];\n\n      const { provider, blockTracker } = this.#getNetworkClient(chainId);\n      const ethQuery = new EthQuery(provider);\n\n      // Force fresh block data before multicall\n      await safelyExecuteWithTimeout(() =>\n        blockTracker?.checkForLatestBlock?.(),\n      );\n\n      // Fetch native balances\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          this.#getProvider(chainId),\n        );\n\n        const nativeBalances = await safelyExecuteWithTimeout(\n          () =>\n            contract.balances(accountsToUpdate, [ZERO_ADDRESS]) as Promise<\n              BigNumber[]\n            >,\n          false,\n          3_000, // 3s max call for multicall contract call\n        );\n\n        if (nativeBalances) {\n          accountsToUpdate.forEach((address, index) => {\n            results.push({\n              success: true,\n              value: new BN(nativeBalances[index].toString()),\n              account: address,\n              token: ZERO_ADDRESS,\n              chainId,\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              if (balanceResult) {\n                results.push({\n                  success: true,\n                  value: new BN(balanceResult.replace('0x', ''), 16),\n                  account: address as ChecksumAddress,\n                  token: ZERO_ADDRESS,\n                  chainId,\n                });\n              } else {\n                results.push({\n                  success: false,\n                  account: address as ChecksumAddress,\n                  token: ZERO_ADDRESS,\n                  chainId,\n                });\n              }\n            });\n\n            await Promise.allSettled(balancePromises);\n            return workingResult;\n          },\n        });\n      }\n\n      // Fetch staked balances if enabled\n      if (this.#includeStakedAssets) {\n        const stakedBalancesPromise = this.#getStakedBalanceForChain(\n          accountsToUpdate,\n          chainId,\n        );\n\n        const stakedBalanceResult = await safelyExecuteWithTimeout(\n          async () =>\n            (await stakedBalancesPromise) as Record<string, StakedBalance>,\n        );\n\n        if (stakedBalanceResult) {\n          // Find the staking contract address for this chain\n          const stakingContractAddress =\n            STAKING_CONTRACT_ADDRESS_BY_CHAINID[\n              chainId as keyof typeof STAKING_CONTRACT_ADDRESS_BY_CHAINID\n            ];\n\n          if (stakingContractAddress) {\n            Object.entries(stakedBalanceResult).forEach(\n              ([address, balance]) => {\n                results.push({\n                  success: true,\n                  value: balance\n                    ? new BN(balance.replace('0x', ''), 16)\n                    : new BN('0'),\n                  account: address as ChecksumAddress,\n                  token: toChecksumHexAddress(\n                    stakingContractAddress,\n                  ) as ChecksumAddress,\n                  chainId,\n                });\n              },\n            );\n          }\n        }\n      }\n    }\n\n    return results;\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 getBalance 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/**\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  readonly #balanceFetchers: BalanceFetcher[];\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   * @param options.useAccountsAPI - Enable Accounts‑API strategy (if supported chain).\n   * @param options.allowExternalServices - Disable external HTTP calls (privacy / offline mode).\n   */\n  constructor({\n    interval = 10000,\n    state,\n    messenger,\n    getStakedBalanceForChain,\n    includeStakedAssets = false,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: {\n    interval?: number;\n    state?: Partial<AccountTrackerControllerState>;\n    messenger: AccountTrackerControllerMessenger;\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n    includeStakedAssets?: boolean;\n    useAccountsAPI?: boolean;\n    allowExternalServices?: () => 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    // Initialize balance fetchers - Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new AccountTrackerRpcBalanceFetcher(\n        this.#getProvider,\n        this.#getNetworkClient,\n        includeStakedAssets,\n        getStakedBalanceForChain,\n      ),\n    ];\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  readonly #getProvider = (chainId: Hex): 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: Hex) => {\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   * 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 allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n    const { isMultiAccountBalancesEnabled } = this.messagingSystem.call(\n      'PreferencesController:getState',\n    );\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      // Use balance fetchers with fallback strategy\n      const aggregated: ProcessedBalance[] = [];\n      let remainingChains = [...chainIds] as ChainIdHex[];\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 Promise.race([\n            fetcher.fetch({\n              chainIds: supportedChains,\n              queryAllAccounts: isMultiAccountBalancesEnabled,\n              selectedAccount: toChecksumHexAddress(\n                selectedAccount.address,\n              ) as ChecksumAddress,\n              allAccounts,\n            }),\n            new Promise<never>((_resolve, reject) =>\n              setTimeout(() => {\n                reject(new Error(`Timeout after ${DEFAULT_TIMEOUT_MS}ms`));\n              }, DEFAULT_TIMEOUT_MS),\n            ),\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      // 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      // Process the aggregated balance results\n      const stakedBalancesByChainAndAddress: Record<\n        string,\n        Record<string, string>\n      > = {};\n\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          const hexValue = `0x${value.toString(16)}`;\n\n          if (token === ZERO_ADDRESS) {\n            // Native balance\n            if (nextAccountsByChainId[chainId][account].balance !== hexValue) {\n              nextAccountsByChainId[chainId][account].balance = hexValue;\n              hasChanges = true;\n            }\n          } else {\n            // Staked balance (from staking contract address)\n            if (!stakedBalancesByChainAndAddress[chainId]) {\n              stakedBalancesByChainAndAddress[chainId] = {};\n            }\n            stakedBalancesByChainAndAddress[chainId][account] = hexValue;\n          }\n        }\n      });\n\n      // Apply staked balances\n      Object.entries(stakedBalancesByChainAndAddress).forEach(\n        ([chainId, balancesByAddress]) => {\n          Object.entries(balancesByAddress).forEach(\n            ([address, stakedBalance]) => {\n              if (\n                nextAccountsByChainId[chainId][address].stakedBalance !==\n                stakedBalance\n              ) {\n                nextAccountsByChainId[chainId][address].stakedBalance =\n                  stakedBalance;\n                hasChanges = true;\n              }\n            },\n          );\n        },\n      );\n\n      // Only update state if something changed\n      if (hasChanges) {\n        this.update((state) => {\n          state.accountsByChainId = nextAccountsByChainId;\n        });\n      }\n    } finally {\n      releaseLock();\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"]}
+{"version":3,"file":"AccountTrackerController.cjs","sourceRoot":"","sources":["../src/AccountTrackerController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AACA,wDAAoD;AACpD,wDAAwD;AAYxD,iEAIoC;AACpC,oEAA2C;AAO3C,qEAA+E;AAE/E,2CAAgE;AAChE,6CAAoC;AACpC,kDAAuB;AACvB,mCAA4C;AAC5C,sGAA4E;AAE5E,6EAKoC;AACpC,iDAAgF;AAChF,gGAI4D;AAE5D;;GAEG;AACH,MAAM,cAAc,GAAG,0BAA0B,CAAC;AAKlD,MAAM,YAAY,GAChB,4CAA+D,CAAC;AAElE;;;GAGG;AACH,MAAM,+BAA+B;IASnC,YACE,WAA2C,EAC3C,gBAAiD,EACjD,mBAA4B,EAC5B,wBAA8E;;QAZvE,+DAA6C;QAE7C,oEAAmD;QAEnD,uEAA8B;QAE9B,4EAAgF;QAQvF,uBAAA,IAAI,gDAAgB,WAAW,MAAA,CAAC;QAChC,uBAAA,IAAI,qDAAqB,gBAAgB,MAAA,CAAC;QAC1C,uBAAA,IAAI,wDAAwB,mBAAmB,MAAA,CAAC;QAChD,uBAAA,IAAI,6DAA6B,wBAAwB,MAAA,CAAC;IAC5D,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,CAAC,kCAAkC;IACjD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,EACV,QAAQ,EACR,gBAAgB,EAChB,eAAe,EACf,WAAW,GAC4B;QACvC,wDAAwD;QACxD,MAAM,uBAAuB,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YAC7D,MAAM,gBAAgB,GAAG,gBAAgB;gBACvC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAC5B,CAAC,OAAO,EAAE,EAAE,CACV,IAAA,uCAAoB,EAAC,OAAO,CAAC,OAAO,CAAoB,CAC3D;gBACH,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YAEtB,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,uBAAA,IAAI,yDAAkB,MAAtB,IAAI,EAAmB,OAAO,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,IAAI,mBAAQ,CAAC,QAAQ,CAAC,CAAC;YACxC,MAAM,YAAY,GAAuB,EAAE,CAAC;YAE5C,0CAA0C;YAC1C,MAAM,IAAA,2CAAwB,EAAC,GAAG,EAAE,CAClC,YAAY,EAAE,mBAAmB,EAAE,EAAE,CACtC,CAAC;YAEF,wBAAwB;YACxB,IAAI,IAAA,mBAAW,EAAC,kEAAuC,EAAE,OAAO,CAAC,EAAE;gBACjE,MAAM,eAAe,GAAG,kEAAuC,CAC7D,OAAO,CACE,CAAC;gBAEZ,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAC3B,eAAe,EACf,yCAA6B,EAC7B,uBAAA,IAAI,oDAAa,MAAjB,IAAI,EAAc,OAAO,CAAC,CAC3B,CAAC;gBAEF,MAAM,cAAc,GAAG,MAAM,IAAA,2CAAwB,EACnD,GAAG,EAAE,CACH,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,YAAY,CAAC,CAEjD,EACH,KAAK,EACL,IAAK,CACN,CAAC;gBAEF,IAAI,cAAc,EAAE;oBAClB,gBAAgB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;wBAC1C,YAAY,CAAC,IAAI,CAAC;4BAChB,OAAO,EAAE,IAAI;4BACb,KAAK,EAAE,IAAI,eAAE,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;4BAC/C,OAAO,EAAE,OAAO;4BAChB,KAAK,EAAE,YAAY;4BACnB,OAAO;yBACR,CAAC,CAAC;oBACL,CAAC,CAAC,CAAC;iBACJ;aACF;iBAAM;gBACL,4DAA4D;gBAC5D,MAAM,IAAA,oCAAuB,EAAe;oBAC1C,MAAM,EAAE,gBAAgB;oBACxB,SAAS,EAAE,oCAAuB;oBAClC,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,KAAK,EAAE,aAAmB,EAAE,KAAe,EAAE,EAAE;wBACxD,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,OAAe,EAAE,EAAE;4BAC1D,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,wGAAqB,MAAzB,IAAI,EAC9B,OAAO,EACP,QAAQ,CACT,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;4BAEpB,IAAI,aAAa,EAAE;gCACjB,YAAY,CAAC,IAAI,CAAC;oCAChB,OAAO,EAAE,IAAI;oCACb,KAAK,EAAE,IAAI,eAAE,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oCAClD,OAAO,EAAE,OAA0B;oCACnC,KAAK,EAAE,YAAY;oCACnB,OAAO;iCACR,CAAC,CAAC;6BACJ;iCAAM;gCACL,YAAY,CAAC,IAAI,CAAC;oCAChB,OAAO,EAAE,KAAK;oCACd,OAAO,EAAE,OAA0B;oCACnC,KAAK,EAAE,YAAY;oCACnB,OAAO;iCACR,CAAC,CAAC;6BACJ;wBACH,CAAC,CAAC,CAAC;wBAEH,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;wBAC1C,OAAO,aAAa,CAAC;oBACvB,CAAC;iBACF,CAAC,CAAC;aACJ;YAED,mCAAmC;YACnC,IAAI,uBAAA,IAAI,4DAAqB,EAAE;gBAC7B,MAAM,qBAAqB,GAAG,uBAAA,IAAI,iEAA0B,MAA9B,IAAI,EAChC,gBAAgB,EAChB,OAAO,CACR,CAAC;gBAEF,MAAM,mBAAmB,GAAG,MAAM,IAAA,2CAAwB,EACxD,KAAK,IAAI,EAAE,CACT,CAAC,MAAM,qBAAqB,CAAkC,CACjE,CAAC;gBAEF,IAAI,mBAAmB,EAAE;oBACvB,mDAAmD;oBACnD,MAAM,sBAAsB,GAC1B,8DAAmC,CACjC,OAA2D,CAC5D,CAAC;oBAEJ,IAAI,sBAAsB,EAAE;wBAC1B,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,OAAO,CACzC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;4BACrB,YAAY,CAAC,IAAI,CAAC;gCAChB,OAAO,EAAE,IAAI;gCACb,KAAK,EAAE,OAAO;oCACZ,CAAC,CAAC,IAAI,eAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oCACvC,CAAC,CAAC,IAAI,eAAE,CAAC,GAAG,CAAC;gCACf,OAAO,EAAE,OAA0B;gCACnC,KAAK,EAAE,IAAA,uCAAoB,EACzB,sBAAsB,CACJ;gCACpB,OAAO;6BACR,CAAC,CAAC;wBACL,CAAC,CACF,CAAC;qBACH;iBACF;aACF;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,gEAAgE;QAChE,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC;QAC5E,MAAM,OAAO,GAAuB,EAAE,CAAC;QAEvC,iBAAiB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;YACxC,IAAI,WAAW,CAAC,MAAM,KAAK,WAAW,EAAE;gBACtC,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;aACpC;iBAAM;gBACL,2CAA2C;gBAC3C,OAAO,CAAC,IAAI,CAAC,0BAA0B,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;aAC9D;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;CAkBF;;AAhBC;;;;;;GAMG;AACH,KAAK,+DACH,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;AA4BH,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;IASC;;;;;;;;;;;OAWG;IACH,YAAY,EACV,QAAQ,GAAG,KAAK,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAAmB,GAAG,KAAK,EAC3B,cAAc,GAAG,KAAK,EACtB,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,GASnC;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;;QAxDI,iDAAgB,IAAI,mBAAK,EAAE,EAAC;QAE5B,gEAA8B;QAE9B,qEAAgF;QAEhF,4DAAmC;QAiJnC,gDAAe,CAAC,OAAY,EAAgB,EAAE;YACrD,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,qDAAoB,CAAC,OAAY,EAAE,EAAE;YAC5C,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;QArHA,uBAAA,IAAI,sDAA6B,wBAAwB,MAAA,CAAC;QAE1D,uBAAA,IAAI,iDAAwB,mBAAmB,MAAA,CAAC;QAEhD,6EAA6E;QAC7E,uBAAA,IAAI,6CAAoB;YACtB,GAAG,CAAC,cAAc,IAAI,qBAAqB,EAAE;gBAC3C,CAAC,CAAC,CAAC,IAAI,+CAAyB,CAAC,WAAW,EAAE,uBAAA,IAAI,6CAAa,CAAC,CAAC;gBACjE,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,+BAA+B,CACjC,uBAAA,IAAI,6CAAa,EACjB,uBAAA,IAAI,kDAAkB,EACtB,mBAAmB,EACnB,wBAAwB,CACzB;SACF,MAAA,CAAC;QAEF,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;IAyED;;;;;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,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,iCAAiC,CAClC,CAAC;QACF,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACjE,gCAAgC,CACjC,CAAC;QAEF,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,8CAA8C;YAC9C,MAAM,UAAU,GAAuB,EAAE,CAAC;YAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAiB,CAAC;YAEpD,oEAAoE;YACpE,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,iDAAiB,EAAE;gBAC3C,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB,CAAC;gBACF,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;oBAC3B,SAAS;iBACV;gBAED,IAAI;oBACF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;wBACnC,QAAQ,EAAE,eAAe;wBACzB,gBAAgB,EAAE,6BAA6B;wBAC/C,eAAe,EAAE,IAAA,uCAAoB,EACnC,eAAe,CAAC,OAAO,CACL;wBACpB,WAAW;qBACZ,CAAC,CAAC;oBAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;wBACnC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;wBAC7B,iDAAiD;wBACjD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;wBAChE,eAAe,GAAG,eAAe,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CACvC,CAAC;qBACH;iBACF;gBAAC,OAAO,KAAK,EAAE;oBACd,OAAO,CAAC,IAAI,CACV,qCAAqC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CACpF,CAAC;oBACF,sCAAsC;iBACvC;gBAED,iDAAiD;gBACjD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE;oBAChC,MAAM;iBACP;aACF;YAED,yEAAyE;YACzE,MAAM,qBAAqB,GACzB,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC1C,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,yCAAyC;YACzC,MAAM,+BAA+B,GAGjC,EAAE,CAAC;YAEP,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,MAAM,QAAQ,GAAG,KAAK,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;oBAE3C,IAAI,KAAK,KAAK,YAAY,EAAE;wBAC1B,iBAAiB;wBACjB,IAAI,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE;4BAChE,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,GAAG,QAAQ,CAAC;4BAC3D,UAAU,GAAG,IAAI,CAAC;yBACnB;qBACF;yBAAM;wBACL,iDAAiD;wBACjD,IAAI,CAAC,+BAA+B,CAAC,OAAO,CAAC,EAAE;4BAC7C,+BAA+B,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;yBAC/C;wBACD,+BAA+B,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YAEH,wBAAwB;YACxB,MAAM,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC,OAAO,CACrD,CAAC,CAAC,OAAO,EAAE,iBAAiB,CAAC,EAAE,EAAE;gBAC/B,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,CACvC,CAAC,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,EAAE;oBAC3B,IACE,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa;wBACrD,aAAa,EACb;wBACA,qBAAqB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa;4BACnD,aAAa,CAAC;wBAChB,UAAU,GAAG,IAAI,CAAC;qBACnB;gBACH,CAAC,CACF,CAAC;YACJ,CAAC,CACF,CAAC;YAEF,yCAAyC;YACzC,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;IAED;;;;;;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;AA9eD,4DA8eC;6gBAlT0B,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;IAmQC,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  NetworkClient,\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 BN from 'bn.js';\nimport { cloneDeep, isEqual } from 'lodash';\nimport abiSingleCallBalancesContract from 'single-call-balance-checker-abi';\n\nimport {\n  SINGLE_CALL_BALANCES_ADDRESS_BY_CHAINID,\n  STAKING_CONTRACT_ADDRESS_BY_CHAINID,\n  type AssetsContractController,\n  type StakedBalance,\n} from './AssetsContractController';\nimport { reduceInBatchesSerially, TOKEN_PRICES_BATCH_SIZE } from './assetsUtil';\nimport {\n  AccountsApiBalanceFetcher,\n  type BalanceFetcher,\n  type ProcessedBalance,\n} from './multi-chain-accounts-service/api-balance-fetcher';\n\n/**\n * The name of the {@link AccountTrackerController}.\n */\nconst controllerName = 'AccountTrackerController';\n\nexport type ChainIdHex = Hex;\nexport type ChecksumAddress = Hex;\n\nconst ZERO_ADDRESS =\n  '0x0000000000000000000000000000000000000000' as ChecksumAddress;\n\n/**\n * RPC-based balance fetcher for AccountTrackerController.\n * Fetches only native balances and staked balances (no token balances).\n */\nclass AccountTrackerRpcBalanceFetcher implements BalanceFetcher {\n  readonly #getProvider: (chainId: Hex) => Web3Provider;\n\n  readonly #getNetworkClient: (chainId: Hex) => NetworkClient;\n\n  readonly #includeStakedAssets: boolean;\n\n  readonly #getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n\n  constructor(\n    getProvider: (chainId: Hex) => Web3Provider,\n    getNetworkClient: (chainId: Hex) => NetworkClient,\n    includeStakedAssets: boolean,\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'],\n  ) {\n    this.#getProvider = getProvider;\n    this.#getNetworkClient = getNetworkClient;\n    this.#includeStakedAssets = includeStakedAssets;\n    this.#getStakedBalanceForChain = getStakedBalanceForChain;\n  }\n\n  supports(): boolean {\n    return true; // fallback – supports every chain\n  }\n\n  async fetch({\n    chainIds,\n    queryAllAccounts,\n    selectedAccount,\n    allAccounts,\n  }: Parameters<BalanceFetcher['fetch']>[0]): Promise<ProcessedBalance[]> {\n    // Process all chains in parallel for better performance\n    const chainProcessingPromises = chainIds.map(async (chainId) => {\n      const accountsToUpdate = queryAllAccounts\n        ? Object.values(allAccounts).map(\n            (account) =>\n              toChecksumHexAddress(account.address) as ChecksumAddress,\n          )\n        : [selectedAccount];\n\n      const { provider, blockTracker } = this.#getNetworkClient(chainId);\n      const ethQuery = new EthQuery(provider);\n      const chainResults: ProcessedBalance[] = [];\n\n      // Force fresh block data before multicall\n      await safelyExecuteWithTimeout(() =>\n        blockTracker?.checkForLatestBlock?.(),\n      );\n\n      // Fetch native balances\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          this.#getProvider(chainId),\n        );\n\n        const nativeBalances = await safelyExecuteWithTimeout(\n          () =>\n            contract.balances(accountsToUpdate, [ZERO_ADDRESS]) as Promise<\n              BigNumber[]\n            >,\n          false,\n          3_000, // 3s max call for multicall contract call\n        );\n\n        if (nativeBalances) {\n          accountsToUpdate.forEach((address, index) => {\n            chainResults.push({\n              success: true,\n              value: new BN(nativeBalances[index].toString()),\n              account: address,\n              token: ZERO_ADDRESS,\n              chainId,\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              if (balanceResult) {\n                chainResults.push({\n                  success: true,\n                  value: new BN(balanceResult.replace('0x', ''), 16),\n                  account: address as ChecksumAddress,\n                  token: ZERO_ADDRESS,\n                  chainId,\n                });\n              } else {\n                chainResults.push({\n                  success: false,\n                  account: address as ChecksumAddress,\n                  token: ZERO_ADDRESS,\n                  chainId,\n                });\n              }\n            });\n\n            await Promise.allSettled(balancePromises);\n            return workingResult;\n          },\n        });\n      }\n\n      // Fetch staked balances if enabled\n      if (this.#includeStakedAssets) {\n        const stakedBalancesPromise = this.#getStakedBalanceForChain(\n          accountsToUpdate,\n          chainId,\n        );\n\n        const stakedBalanceResult = await safelyExecuteWithTimeout(\n          async () =>\n            (await stakedBalancesPromise) as Record<string, StakedBalance>,\n        );\n\n        if (stakedBalanceResult) {\n          // Find the staking contract address for this chain\n          const stakingContractAddress =\n            STAKING_CONTRACT_ADDRESS_BY_CHAINID[\n              chainId as keyof typeof STAKING_CONTRACT_ADDRESS_BY_CHAINID\n            ];\n\n          if (stakingContractAddress) {\n            Object.entries(stakedBalanceResult).forEach(\n              ([address, balance]) => {\n                chainResults.push({\n                  success: true,\n                  value: balance\n                    ? new BN(balance.replace('0x', ''), 16)\n                    : new BN('0'),\n                  account: address as ChecksumAddress,\n                  token: toChecksumHexAddress(\n                    stakingContractAddress,\n                  ) as ChecksumAddress,\n                  chainId,\n                });\n              },\n            );\n          }\n        }\n      }\n\n      return chainResults;\n    });\n\n    // Wait for all chains to complete (or fail) and collect results\n    const chainResultsArray = await Promise.allSettled(chainProcessingPromises);\n    const results: ProcessedBalance[] = [];\n\n    chainResultsArray.forEach((chainResult) => {\n      if (chainResult.status === 'fulfilled') {\n        results.push(...chainResult.value);\n      } else {\n        // Log error but continue with other chains\n        console.warn('Chain processing failed:', chainResult.reason);\n      }\n    });\n\n    return results;\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 getBalance 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/**\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  readonly #balanceFetchers: BalanceFetcher[];\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   * @param options.useAccountsAPI - Enable Accounts‑API strategy (if supported chain).\n   * @param options.allowExternalServices - Disable external HTTP calls (privacy / offline mode).\n   */\n  constructor({\n    interval = 10000,\n    state,\n    messenger,\n    getStakedBalanceForChain,\n    includeStakedAssets = false,\n    useAccountsAPI = false,\n    allowExternalServices = () => true,\n  }: {\n    interval?: number;\n    state?: Partial<AccountTrackerControllerState>;\n    messenger: AccountTrackerControllerMessenger;\n    getStakedBalanceForChain: AssetsContractController['getStakedBalanceForChain'];\n    includeStakedAssets?: boolean;\n    useAccountsAPI?: boolean;\n    allowExternalServices?: () => 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    // Initialize balance fetchers - Strategy order: API first, then RPC fallback\n    this.#balanceFetchers = [\n      ...(useAccountsAPI && allowExternalServices()\n        ? [new AccountsApiBalanceFetcher('extension', this.#getProvider)]\n        : []),\n      new AccountTrackerRpcBalanceFetcher(\n        this.#getProvider,\n        this.#getNetworkClient,\n        includeStakedAssets,\n        getStakedBalanceForChain,\n      ),\n    ];\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  readonly #getProvider = (chainId: Hex): 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: Hex) => {\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   * 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 allAccounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n    const { isMultiAccountBalancesEnabled } = this.messagingSystem.call(\n      'PreferencesController:getState',\n    );\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      // Use balance fetchers with fallback strategy\n      const aggregated: ProcessedBalance[] = [];\n      let remainingChains = [...chainIds] as ChainIdHex[];\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: isMultiAccountBalancesEnabled,\n            selectedAccount: toChecksumHexAddress(\n              selectedAccount.address,\n            ) 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      // 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      // Process the aggregated balance results\n      const stakedBalancesByChainAndAddress: Record<\n        string,\n        Record<string, string>\n      > = {};\n\n      aggregated.forEach(({ success, value, account, token, chainId }) => {\n        if (success && value !== undefined) {\n          const hexValue = `0x${value.toString(16)}`;\n\n          if (token === ZERO_ADDRESS) {\n            // Native balance\n            if (nextAccountsByChainId[chainId][account].balance !== hexValue) {\n              nextAccountsByChainId[chainId][account].balance = hexValue;\n              hasChanges = true;\n            }\n          } else {\n            // Staked balance (from staking contract address)\n            if (!stakedBalancesByChainAndAddress[chainId]) {\n              stakedBalancesByChainAndAddress[chainId] = {};\n            }\n            stakedBalancesByChainAndAddress[chainId][account] = hexValue;\n          }\n        }\n      });\n\n      // Apply staked balances\n      Object.entries(stakedBalancesByChainAndAddress).forEach(\n        ([chainId, balancesByAddress]) => {\n          Object.entries(balancesByAddress).forEach(\n            ([address, stakedBalance]) => {\n              if (\n                nextAccountsByChainId[chainId][address].stakedBalance !==\n                stakedBalance\n              ) {\n                nextAccountsByChainId[chainId][address].stakedBalance =\n                  stakedBalance;\n                hasChanges = true;\n              }\n            },\n          );\n        },\n      );\n\n      // Only update state if something changed\n      if (hasChanges) {\n        this.update((state) => {\n          state.accountsByChainId = nextAccountsByChainId;\n        });\n      }\n    } finally {\n      releaseLock();\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.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,EAEV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAMhE,OAAO,EAGL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAQpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AA6LlC;;;;;;;;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;;IASC;;;;;;;;;;;OAWG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,EAC3B,cAAsB,EACtB,qBAAkC,GACnC,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;QAC9B,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;KACvC;IAuDD,OAAO,CAAC,YAAY;IAmIpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IAmIjD;;;;;;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"}
+{"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,EAEV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAMhE,OAAO,EAGL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAQpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AA2MlC;;;;;;;;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;;IASC;;;;;;;;;;;OAWG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,EAC3B,cAAsB,EACtB,qBAAkC,GACnC,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;QAC9B,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;KACvC;IAuDD,OAAO,CAAC,YAAY;IAmIpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA4HjD;;;;;;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.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,EAEV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAMhE,OAAO,EAGL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAQpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AA6LlC;;;;;;;;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;;IASC;;;;;;;;;;;OAWG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,EAC3B,cAAsB,EACtB,qBAAkC,GACnC,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;QAC9B,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;KACvC;IAuDD,OAAO,CAAC,YAAY;IAmIpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IAmIjD;;;;;;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"}
+{"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,EAEV,eAAe,EACf,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AAEtC,OAAO,KAAK,EAAE,mCAAmC,EAAE,yCAAyC;AAC5F,OAAO,EAAuB,KAAK,GAAG,EAAE,wBAAwB;AAMhE,OAAO,EAGL,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,uCAAmC;AAQpC;;GAEG;AACH,QAAA,MAAM,cAAc,6BAA6B,CAAC;AAElD,MAAM,MAAM,UAAU,GAAG,GAAG,CAAC;AAC7B,MAAM,MAAM,eAAe,GAAG,GAAG,CAAC;AA2MlC;;;;;;;;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;;IASC;;;;;;;;;;;OAWG;gBACS,EACV,QAAgB,EAChB,KAAK,EACL,SAAS,EACT,wBAAwB,EACxB,mBAA2B,EAC3B,cAAsB,EACtB,qBAAkC,GACnC,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;QAC9B,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,qBAAqB,CAAC,EAAE,MAAM,OAAO,CAAC;KACvC;IAuDD,OAAO,CAAC,YAAY;IAmIpB;;;;;OAKG;IACG,YAAY,CAAC,EACjB,gBAAgB,GACjB,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C;;;;;;OAMG;IACG,OAAO,CAAC,gBAAgB,EAAE,eAAe,EAAE;IA4HjD;;;;;;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

@@ -37,7 +37,6 @@ import { AccountsApiBalanceFetcher } from "./multi-chain-accounts-service/api-ba
  * The name of the {@link AccountTrackerController}.
  */
 const controllerName = 'AccountTrackerController';
-const DEFAULT_TIMEOUT_MS = 15000;
 const ZERO_ADDRESS = '0x0000000000000000000000000000000000000000';
 /**
  * RPC-based balance fetcher for AccountTrackerController.
@@ -59,13 +58,14 @@ class AccountTrackerRpcBalanceFetcher {
         return true; // fallback – supports every chain
     }
     async fetch({ chainIds, queryAllAccounts, selectedAccount, allAccounts, }) {
-        const results = [];
-        for (const chainId of chainIds) {
+        // Process all chains in parallel for better performance
+        const chainProcessingPromises = chainIds.map(async (chainId) => {
             const accountsToUpdate = queryAllAccounts
                 ? Object.values(allAccounts).map((account) => toChecksumHexAddress(account.address))
                 : [selectedAccount];
             const { provider, blockTracker } = __classPrivateFieldGet(this, _AccountTrackerRpcBalanceFetcher_getNetworkClient, "f").call(this, chainId);
             const ethQuery = new EthQuery(provider);
+            const chainResults = [];
             // Force fresh block data before multicall
             await safelyExecuteWithTimeout(() => blockTracker?.checkForLatestBlock?.());
             // Fetch native balances
@@ -75,7 +75,7 @@ class AccountTrackerRpcBalanceFetcher {
                 const nativeBalances = await safelyExecuteWithTimeout(() => contract.balances(accountsToUpdate, [ZERO_ADDRESS]), false, 3000);
                 if (nativeBalances) {
                     accountsToUpdate.forEach((address, index) => {
-                        results.push({
+                        chainResults.push({
                             success: true,
                             value: new BN(nativeBalances[index].toString()),
                             account: address,
@@ -95,7 +95,7 @@ class AccountTrackerRpcBalanceFetcher {
                         const balancePromises = batch.map(async (address) => {
                             const balanceResult = await __classPrivateFieldGet(this, _AccountTrackerRpcBalanceFetcher_instances, "m", _AccountTrackerRpcBalanceFetcher_getBalanceFromChain).call(this, address, ethQuery).catch(() => null);
                             if (balanceResult) {
-                                results.push({
+                                chainResults.push({
                                     success: true,
                                     value: new BN(balanceResult.replace('0x', ''), 16),
                                     account: address,
@@ -104,7 +104,7 @@ class AccountTrackerRpcBalanceFetcher {
                                 });
                             }
                             else {
-                                results.push({
+                                chainResults.push({
                                     success: false,
                                     account: address,
                                     token: ZERO_ADDRESS,
@@ -126,7 +126,7 @@ class AccountTrackerRpcBalanceFetcher {
                     const stakingContractAddress = STAKING_CONTRACT_ADDRESS_BY_CHAINID[chainId];
                     if (stakingContractAddress) {
                         Object.entries(stakedBalanceResult).forEach(([address, balance]) => {
-                            results.push({
+                            chainResults.push({
                                 success: true,
                                 value: balance
                                     ? new BN(balance.replace('0x', ''), 16)
@@ -139,7 +139,20 @@ class AccountTrackerRpcBalanceFetcher {
                     }
                 }
             }
-        }
+            return chainResults;
+        });
+        // Wait for all chains to complete (or fail) and collect results
+        const chainResultsArray = await Promise.allSettled(chainProcessingPromises);
+        const results = [];
+        chainResultsArray.forEach((chainResult) => {
+            if (chainResult.status === 'fulfilled') {
+                results.push(...chainResult.value);
+            }
+            else {
+                // Log error but continue with other chains
+                console.warn('Chain processing failed:', chainResult.reason);
+            }
+        });
         return results;
     }
 }
@@ -308,17 +321,12 @@ export class AccountTrackerController extends StaticIntervalPollingController()
                     continue;
                 }
                 try {
-                    const balances = await Promise.race([
-                        fetcher.fetch({
-                            chainIds: supportedChains,
-                            queryAllAccounts: isMultiAccountBalancesEnabled,
-                            selectedAccount: toChecksumHexAddress(selectedAccount.address),
-                            allAccounts,
-                        }),
-                        new Promise((_resolve, reject) => setTimeout(() => {
-                            reject(new Error(`Timeout after ${DEFAULT_TIMEOUT_MS}ms`));
-                        }, DEFAULT_TIMEOUT_MS)),
-                    ]);
+                    const balances = await fetcher.fetch({
+                        chainIds: supportedChains,
+                        queryAllAccounts: isMultiAccountBalancesEnabled,
+                        selectedAccount: toChecksumHexAddress(selectedAccount.address),
+                        allAccounts,
+                    });
                     if (balances && balances.length > 0) {
                         aggregated.push(...balances);
                         // Remove chains that were successfully processed