@metamask/assets-controllers 76.0.0 → 77.0.1

package/dist/TokenBalancesController.cjs

@@ -359,50 +359,73 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
             : [selected];
         const prev = this.state;
         const next = draft(prev, (d) => {
-            // First, initialize all tokens from allTokens state with balance 0
-            // for the accounts and chains we're processing
+            var _a, _b;
+            // Initialize account and chain structures if they don't exist, but preserve existing balances
             for (const chainId of targetChains) {
                 for (const account of accountsToProcess) {
-                    // Initialize tokens from allTokens
+                    // Ensure the nested structure exists without overwriting existing balances
+                    (_a = d.tokenBalances)[account] ?? (_a[account] = {});
+                    (_b = d.tokenBalances[account])[chainId] ?? (_b[chainId] = {});
+                    // Initialize tokens from allTokens only if they don't exist yet
                     const chainTokens = __classPrivateFieldGet(this, _TokenBalancesController_allTokens, "f")[chainId];
                     if (chainTokens?.[account]) {
                         Object.values(chainTokens[account]).forEach((token) => {
-                            var _a, _b;
                             const tokenAddress = checksum(token.address);
-                            ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] = '0x0';
+                            // Only initialize if the token balance doesn't exist yet
+                            if (!(tokenAddress in d.tokenBalances[account][chainId])) {
+                                d.tokenBalances[account][chainId][tokenAddress] = '0x0';
+                            }
                         });
                     }
-                    // Initialize tokens from allDetectedTokens
+                    // Initialize tokens from allDetectedTokens only if they don't exist yet
                     const detectedChainTokens = __classPrivateFieldGet(this, _TokenBalancesController_detectedTokens, "f")[chainId];
                     if (detectedChainTokens?.[account]) {
                         Object.values(detectedChainTokens[account]).forEach((token) => {
-                            var _a, _b;
                             const tokenAddress = checksum(token.address);
-                            ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] = '0x0';
+                            // Only initialize if the token balance doesn't exist yet
+                            if (!(tokenAddress in d.tokenBalances[account][chainId])) {
+                                d.tokenBalances[account][chainId][tokenAddress] = '0x0';
+                            }
                         });
                     }
                 }
             }
-            // Then update with actual fetched balances where available
+            // Update with actual fetched balances only if the value has changed
             aggregated.forEach(({ success, value, account, token, chainId }) => {
                 var _a, _b;
                 if (success && value !== undefined) {
-                    ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[checksum(token)] =
-                        (0, controller_utils_1.toHex)(value);
+                    const newBalance = (0, controller_utils_1.toHex)(value);
+                    const tokenAddress = checksum(token);
+                    const currentBalance = d.tokenBalances[account]?.[chainId]?.[tokenAddress];
+                    // Only update if the balance has actually changed
+                    if (currentBalance !== newBalance) {
+                        ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] =
+                            newBalance;
+                    }
                 }
             });
         });
         if (!(0, lodash_1.isEqual)(prev, next)) {
             this.update(() => next);
             const nativeBalances = aggregated.filter((r) => r.success && r.token === ZERO_ADDRESS);
-            // Update native token balances in a single batch operation for better performance
+            // Get current AccountTracker state to compare existing balances
+            const accountTrackerState = this.messagingSystem.call('AccountTrackerController:getState');
+            // Update native token balances only if they have changed
             if (nativeBalances.length > 0) {
-                const balanceUpdates = nativeBalances.map((balance) => ({
+                const balanceUpdates = nativeBalances
+                    .map((balance) => ({
                     address: balance.account,
                     chainId: balance.chainId,
                     balance: balance.value ? (0, controller_utils_1.BNToHex)(balance.value) : '0x0',
-                }));
-                this.messagingSystem.call('AccountTrackerController:updateNativeBalances', balanceUpdates);
+                }))
+                    .filter((update) => {
+                    const currentBalance = accountTrackerState.accountsByChainId[update.chainId]?.[checksum(update.address)]?.balance;
+                    // Only include if the balance has actually changed
+                    return currentBalance !== update.balance;
+                });
+                if (balanceUpdates.length > 0) {
+                    this.messagingSystem.call('AccountTrackerController:updateNativeBalances', balanceUpdates);
+                }
             }
             // Get staking contract addresses for filtering
             const stakingContractAddresses = Object.values(AssetsContractController_1.STAKING_CONTRACT_ADDRESS_BY_CHAINID).map((addr) => addr.toLowerCase());
@@ -413,12 +436,20 @@ class TokenBalancesController extends (0, polling_controller_1.StaticIntervalPol
                     stakingContractAddresses.includes(r.token.toLowerCase()));
             });
             if (stakedBalances.length > 0) {
-                const stakedBalanceUpdates = stakedBalances.map((balance) => ({
+                const stakedBalanceUpdates = stakedBalances
+                    .map((balance) => ({
                     address: balance.account,
                     chainId: balance.chainId,
                     stakedBalance: balance.value ? (0, controller_utils_1.toHex)(balance.value) : '0x0',
-                }));
-                this.messagingSystem.call('AccountTrackerController:updateStakedBalances', stakedBalanceUpdates);
+                }))
+                    .filter((update) => {
+                    const currentStakedBalance = accountTrackerState.accountsByChainId[update.chainId]?.[checksum(update.address)]?.stakedBalance;
+                    // Only include if the staked balance has actually changed
+                    return currentStakedBalance !== update.stakedBalance;
+                });
+                if (stakedBalanceUpdates.length > 0) {
+                    this.messagingSystem.call('AccountTrackerController:updateStakedBalances', stakedBalanceUpdates);
+                }
             }
         }
     }

package/dist/TokenBalancesController.cjs.map

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

package/dist/TokenBalancesController.d.cts

@@ -4,7 +4,7 @@ import type { KeyringControllerAccountRemovedEvent } from "@metamask/keyring-con
 import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction, NetworkControllerStateChangeEvent } from "@metamask/network-controller";
 import type { PreferencesControllerGetStateAction, PreferencesControllerStateChangeEvent } from "@metamask/preferences-controller";
 import type { Hex } from "@metamask/utils";
-import type { AccountTrackerUpdateNativeBalancesAction, AccountTrackerUpdateStakedBalancesAction } from "./AccountTrackerController.cjs";
+import type { AccountTrackerControllerGetStateAction, AccountTrackerUpdateNativeBalancesAction, AccountTrackerUpdateStakedBalancesAction } from "./AccountTrackerController.cjs";
 import type { TokensControllerGetStateAction, TokensControllerStateChangeEvent } from "./TokensController.cjs";
 export type ChainIdHex = Hex;
 export type ChecksumAddress = Hex;
@@ -29,7 +29,7 @@ export type NativeBalanceEvent = {
     payload: unknown[];
 };
 export type TokenBalancesControllerEvents = TokenBalancesControllerStateChangeEvent | NativeBalanceEvent;
-export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
+export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerControllerGetStateAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 export type AllowedEvents = TokensControllerStateChangeEvent | PreferencesControllerStateChangeEvent | NetworkControllerStateChangeEvent | KeyringControllerAccountRemovedEvent;
 export type TokenBalancesControllerMessenger = RestrictedMessenger<typeof CONTROLLER, TokenBalancesControllerActions | AllowedActions, TokenBalancesControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
 export type ChainPollingConfig = {

package/dist/TokenBalancesController.d.cts.map

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

package/dist/TokenBalancesController.d.mts

@@ -4,7 +4,7 @@ import type { KeyringControllerAccountRemovedEvent } from "@metamask/keyring-con
 import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction, NetworkControllerStateChangeEvent } from "@metamask/network-controller";
 import type { PreferencesControllerGetStateAction, PreferencesControllerStateChangeEvent } from "@metamask/preferences-controller";
 import type { Hex } from "@metamask/utils";
-import type { AccountTrackerUpdateNativeBalancesAction, AccountTrackerUpdateStakedBalancesAction } from "./AccountTrackerController.mjs";
+import type { AccountTrackerControllerGetStateAction, AccountTrackerUpdateNativeBalancesAction, AccountTrackerUpdateStakedBalancesAction } from "./AccountTrackerController.mjs";
 import type { TokensControllerGetStateAction, TokensControllerStateChangeEvent } from "./TokensController.mjs";
 export type ChainIdHex = Hex;
 export type ChecksumAddress = Hex;
@@ -29,7 +29,7 @@ export type NativeBalanceEvent = {
     payload: unknown[];
 };
 export type TokenBalancesControllerEvents = TokenBalancesControllerStateChangeEvent | NativeBalanceEvent;
-export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
+export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | TokensControllerGetStateAction | PreferencesControllerGetStateAction | AccountsControllerGetSelectedAccountAction | AccountsControllerListAccountsAction | AccountTrackerControllerGetStateAction | AccountTrackerUpdateNativeBalancesAction | AccountTrackerUpdateStakedBalancesAction;
 export type AllowedEvents = TokensControllerStateChangeEvent | PreferencesControllerStateChangeEvent | NetworkControllerStateChangeEvent | KeyringControllerAccountRemovedEvent;
 export type TokenBalancesControllerMessenger = RestrictedMessenger<typeof CONTROLLER, TokenBalancesControllerActions | AllowedActions, TokenBalancesControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
 export type ChainPollingConfig = {

package/dist/TokenBalancesController.d.mts.map

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

package/dist/TokenBalancesController.mjs

@@ -357,50 +357,73 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
             : [selected];
         const prev = this.state;
         const next = draft(prev, (d) => {
-            // First, initialize all tokens from allTokens state with balance 0
-            // for the accounts and chains we're processing
+            var _a, _b;
+            // Initialize account and chain structures if they don't exist, but preserve existing balances
             for (const chainId of targetChains) {
                 for (const account of accountsToProcess) {
-                    // Initialize tokens from allTokens
+                    // Ensure the nested structure exists without overwriting existing balances
+                    (_a = d.tokenBalances)[account] ?? (_a[account] = {});
+                    (_b = d.tokenBalances[account])[chainId] ?? (_b[chainId] = {});
+                    // Initialize tokens from allTokens only if they don't exist yet
                     const chainTokens = __classPrivateFieldGet(this, _TokenBalancesController_allTokens, "f")[chainId];
                     if (chainTokens?.[account]) {
                         Object.values(chainTokens[account]).forEach((token) => {
-                            var _a, _b;
                             const tokenAddress = checksum(token.address);
-                            ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] = '0x0';
+                            // Only initialize if the token balance doesn't exist yet
+                            if (!(tokenAddress in d.tokenBalances[account][chainId])) {
+                                d.tokenBalances[account][chainId][tokenAddress] = '0x0';
+                            }
                         });
                     }
-                    // Initialize tokens from allDetectedTokens
+                    // Initialize tokens from allDetectedTokens only if they don't exist yet
                     const detectedChainTokens = __classPrivateFieldGet(this, _TokenBalancesController_detectedTokens, "f")[chainId];
                     if (detectedChainTokens?.[account]) {
                         Object.values(detectedChainTokens[account]).forEach((token) => {
-                            var _a, _b;
                             const tokenAddress = checksum(token.address);
-                            ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] = '0x0';
+                            // Only initialize if the token balance doesn't exist yet
+                            if (!(tokenAddress in d.tokenBalances[account][chainId])) {
+                                d.tokenBalances[account][chainId][tokenAddress] = '0x0';
+                            }
                         });
                     }
                 }
             }
-            // Then update with actual fetched balances where available
+            // Update with actual fetched balances only if the value has changed
             aggregated.forEach(({ success, value, account, token, chainId }) => {
                 var _a, _b;
                 if (success && value !== undefined) {
-                    ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[checksum(token)] =
-                        toHex(value);
+                    const newBalance = toHex(value);
+                    const tokenAddress = checksum(token);
+                    const currentBalance = d.tokenBalances[account]?.[chainId]?.[tokenAddress];
+                    // Only update if the balance has actually changed
+                    if (currentBalance !== newBalance) {
+                        ((_b = ((_a = d.tokenBalances)[account] ?? (_a[account] = {})))[chainId] ?? (_b[chainId] = {}))[tokenAddress] =
+                            newBalance;
+                    }
                 }
             });
         });
         if (!isEqual(prev, next)) {
             this.update(() => next);
             const nativeBalances = aggregated.filter((r) => r.success && r.token === ZERO_ADDRESS);
-            // Update native token balances in a single batch operation for better performance
+            // Get current AccountTracker state to compare existing balances
+            const accountTrackerState = this.messagingSystem.call('AccountTrackerController:getState');
+            // Update native token balances only if they have changed
             if (nativeBalances.length > 0) {
-                const balanceUpdates = nativeBalances.map((balance) => ({
+                const balanceUpdates = nativeBalances
+                    .map((balance) => ({
                     address: balance.account,
                     chainId: balance.chainId,
                     balance: balance.value ? BNToHex(balance.value) : '0x0',
-                }));
-                this.messagingSystem.call('AccountTrackerController:updateNativeBalances', balanceUpdates);
+                }))
+                    .filter((update) => {
+                    const currentBalance = accountTrackerState.accountsByChainId[update.chainId]?.[checksum(update.address)]?.balance;
+                    // Only include if the balance has actually changed
+                    return currentBalance !== update.balance;
+                });
+                if (balanceUpdates.length > 0) {
+                    this.messagingSystem.call('AccountTrackerController:updateNativeBalances', balanceUpdates);
+                }
             }
             // Get staking contract addresses for filtering
             const stakingContractAddresses = Object.values(STAKING_CONTRACT_ADDRESS_BY_CHAINID).map((addr) => addr.toLowerCase());
@@ -411,12 +434,20 @@ export class TokenBalancesController extends StaticIntervalPollingController() {
                     stakingContractAddresses.includes(r.token.toLowerCase()));
             });
             if (stakedBalances.length > 0) {
-                const stakedBalanceUpdates = stakedBalances.map((balance) => ({
+                const stakedBalanceUpdates = stakedBalances
+                    .map((balance) => ({
                     address: balance.account,
                     chainId: balance.chainId,
                     stakedBalance: balance.value ? toHex(balance.value) : '0x0',
-                }));
-                this.messagingSystem.call('AccountTrackerController:updateStakedBalances', stakedBalanceUpdates);
+                }))
+                    .filter((update) => {
+                    const currentStakedBalance = accountTrackerState.accountsByChainId[update.chainId]?.[checksum(update.address)]?.stakedBalance;
+                    // Only include if the staked balance has actually changed
+                    return currentStakedBalance !== update.stakedBalance;
+                });
+                if (stakedBalanceUpdates.length > 0) {
+                    this.messagingSystem.call('AccountTrackerController:updateStakedBalances', stakedBalanceUpdates);
+                }
             }
         }
     }