@metamask-previews/assets-controllers 25.0.0-preview.7fd401b → 25.0.0-preview.867b00ee

package/dist/TokenDetectionController.js

@@ -19,13 +19,49 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _TokenDetectionController_instances, _TokenDetectionController_intervalId, _TokenDetectionController_chainId, _TokenDetectionController_selectedAddress, _TokenDetectionController_networkClientId, _TokenDetectionController_disabled, _TokenDetectionController_isDetectionEnabledFromPreferences, _TokenDetectionController_isDetectionEnabledForNetwork, _TokenDetectionController_addDetectedTokens, _TokenDetectionController_getBalancesInSingleCall, _TokenDetectionController_getTokensState, _TokenDetectionController_stopPolling, _TokenDetectionController_startPolling, _TokenDetectionController_getCorrectChainId;
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _TokenDetectionController_instances, _TokenDetectionController_intervalId, _TokenDetectionController_chainId, _TokenDetectionController_selectedAddress, _TokenDetectionController_networkClientId, _TokenDetectionController_disabled, _TokenDetectionController_isUnlocked, _TokenDetectionController_isDetectionEnabledFromPreferences, _TokenDetectionController_isDetectionEnabledForNetwork, _TokenDetectionController_getBalancesInSingleCall, _TokenDetectionController_trackMetaMetricsEvent, _TokenDetectionController_registerEventListeners, _TokenDetectionController_stopPolling, _TokenDetectionController_startPolling, _TokenDetectionController_getCorrectChainIdAndNetworkClientId, _TokenDetectionController_restartTokenDetection;
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TokenDetectionController = exports.controllerName = void 0;
+exports.TokenDetectionController = exports.controllerName = exports.STATIC_MAINNET_TOKEN_LIST = exports.isEqualCaseInsensitive = void 0;
+const contract_metadata_1 = __importDefault(require("@metamask/contract-metadata"));
 const controller_utils_1 = require("@metamask/controller-utils");
 const polling_controller_1 = require("@metamask/polling-controller");
 const assetsUtil_1 = require("./assetsUtil");
 const DEFAULT_INTERVAL = 180000;
+/**
+ * Compare 2 given strings and return boolean
+ * eg: "foo" and "FOO" => true
+ * eg: "foo" and "bar" => false
+ * eg: "foo" and 123 => false
+ *
+ * @param value1 - first string to compare
+ * @param value2 - first string to compare
+ * @returns true if 2 strings are identical when they are lowercase
+ */
+function isEqualCaseInsensitive(value1, value2) {
+    if (typeof value1 !== 'string' || typeof value2 !== 'string') {
+        return false;
+    }
+    return value1.toLowerCase() === value2.toLowerCase();
+}
+exports.isEqualCaseInsensitive = isEqualCaseInsensitive;
+exports.STATIC_MAINNET_TOKEN_LIST = Object.entries(contract_metadata_1.default).reduce((acc, [base, contract]) => {
+    const { logo } = contract, tokenMetadata = __rest(contract, ["logo"]);
+    return Object.assign(Object.assign({}, acc), { [base.toLowerCase()]: Object.assign(Object.assign({}, tokenMetadata), { address: base.toLowerCase(), iconUrl: `images/contract/${logo}`, aggregators: [] }) });
+}, {});
 exports.controllerName = 'TokenDetectionController';
 /**
  * Controller that passively polls on a set interval for Tokens auto detection
@@ -34,6 +70,7 @@ exports.controllerName = 'TokenDetectionController';
  * @property selectedAddress - Vault selected address
  * @property networkClientId - The network client ID of the current selected network
  * @property disabled - Boolean to track if network requests are blocked
+ * @property isUnlocked - Boolean to track if the keyring state is unlocked
  * @property isDetectionEnabledFromPreferences - Boolean to track if detection is enabled from PreferencesController
  * @property isDetectionEnabledForNetwork - Boolean to track if detected is enabled for current network
  */
@@ -47,14 +84,10 @@ class TokenDetectionController extends polling_controller_1.StaticIntervalPollin
      * @param options.interval - Polling interval used to fetch new token rates
      * @param options.networkClientId - The selected network client ID of the current network
      * @param options.selectedAddress - Vault selected address
-     * @param options.onPreferencesStateChange - Allows subscribing to preferences controller state changes.
-     * @param options.addDetectedTokens - Add a list of detected tokens.
      * @param options.getBalancesInSingleCall - Gets the balances of a list of tokens for the given address.
-     * @param options.getTokensState - Gets the current state of the Tokens controller.
-     * @param options.getPreferencesState - Gets the state of the preferences controller.
+     * @param options.trackMetaMetricsEvent - Sets options for MetaMetrics event tracking.
      */
-    constructor({ networkClientId, selectedAddress = '', interval = DEFAULT_INTERVAL, disabled = true, onPreferencesStateChange, getBalancesInSingleCall, addDetectedTokens, getPreferencesState, getTokensState, messenger, }) {
-        const { useTokenDetection: defaultUseTokenDetection } = getPreferencesState();
+    constructor({ networkClientId, selectedAddress, interval = DEFAULT_INTERVAL, disabled = true, getBalancesInSingleCall, trackMetaMetricsEvent, messenger, }) {
         super({
             name: exports.controllerName,
             messenger,
@@ -67,47 +100,26 @@ class TokenDetectionController extends polling_controller_1.StaticIntervalPollin
         _TokenDetectionController_selectedAddress.set(this, void 0);
         _TokenDetectionController_networkClientId.set(this, void 0);
         _TokenDetectionController_disabled.set(this, void 0);
+        _TokenDetectionController_isUnlocked.set(this, void 0);
         _TokenDetectionController_isDetectionEnabledFromPreferences.set(this, void 0);
         _TokenDetectionController_isDetectionEnabledForNetwork.set(this, void 0);
-        _TokenDetectionController_addDetectedTokens.set(this, void 0);
         _TokenDetectionController_getBalancesInSingleCall.set(this, void 0);
-        _TokenDetectionController_getTokensState.set(this, void 0);
+        _TokenDetectionController_trackMetaMetricsEvent.set(this, void 0);
         __classPrivateFieldSet(this, _TokenDetectionController_disabled, disabled, "f");
         this.setIntervalLength(interval);
         __classPrivateFieldSet(this, _TokenDetectionController_networkClientId, networkClientId, "f");
-        __classPrivateFieldSet(this, _TokenDetectionController_selectedAddress, selectedAddress, "f");
-        __classPrivateFieldSet(this, _TokenDetectionController_chainId, __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainId).call(this, networkClientId), "f");
+        __classPrivateFieldSet(this, _TokenDetectionController_selectedAddress, selectedAddress !== null && selectedAddress !== void 0 ? selectedAddress : this.messagingSystem.call('AccountsController:getSelectedAccount')
+            .address, "f");
+        const { chainId } = __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainIdAndNetworkClientId).call(this, networkClientId);
+        __classPrivateFieldSet(this, _TokenDetectionController_chainId, chainId, "f");
+        const { useTokenDetection: defaultUseTokenDetection } = this.messagingSystem.call('PreferencesController:getState');
         __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, defaultUseTokenDetection, "f");
         __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledForNetwork, (0, assetsUtil_1.isTokenDetectionSupportedForNetwork)(__classPrivateFieldGet(this, _TokenDetectionController_chainId, "f")), "f");
-        __classPrivateFieldSet(this, _TokenDetectionController_addDetectedTokens, addDetectedTokens, "f");
         __classPrivateFieldSet(this, _TokenDetectionController_getBalancesInSingleCall, getBalancesInSingleCall, "f");
-        __classPrivateFieldSet(this, _TokenDetectionController_getTokensState, getTokensState, "f");
-        this.messagingSystem.subscribe('TokenListController:stateChange', ({ tokenList }) => __awaiter(this, void 0, void 0, function* () {
-            const hasTokens = Object.keys(tokenList).length;
-            if (hasTokens) {
-                yield this.detectTokens();
-            }
-        }));
-        onPreferencesStateChange(({ selectedAddress: newSelectedAddress, useTokenDetection }) => __awaiter(this, void 0, void 0, function* () {
-            const isSelectedAddressChanged = __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f") !== newSelectedAddress;
-            const isDetectionChangedFromPreferences = __classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f") !== useTokenDetection;
-            __classPrivateFieldSet(this, _TokenDetectionController_selectedAddress, newSelectedAddress, "f");
-            __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, useTokenDetection, "f");
-            if (useTokenDetection &&
-                (isSelectedAddressChanged || isDetectionChangedFromPreferences)) {
-                yield this.detectTokens();
-            }
-        }));
-        this.messagingSystem.subscribe('NetworkController:networkDidChange', ({ selectedNetworkClientId }) => __awaiter(this, void 0, void 0, function* () {
-            __classPrivateFieldSet(this, _TokenDetectionController_networkClientId, selectedNetworkClientId, "f");
-            const newChainId = __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainId).call(this, selectedNetworkClientId);
-            const isChainIdChanged = __classPrivateFieldGet(this, _TokenDetectionController_chainId, "f") !== newChainId;
-            __classPrivateFieldSet(this, _TokenDetectionController_chainId, newChainId, "f");
-            __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledForNetwork, (0, assetsUtil_1.isTokenDetectionSupportedForNetwork)(newChainId), "f");
-            if (__classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledForNetwork, "f") && isChainIdChanged) {
-                yield this.detectTokens();
-            }
-        }));
+        __classPrivateFieldSet(this, _TokenDetectionController_trackMetaMetricsEvent, trackMetaMetricsEvent, "f");
+        const { isUnlocked } = this.messagingSystem.call('KeyringController:getState');
+        __classPrivateFieldSet(this, _TokenDetectionController_isUnlocked, isUnlocked, "f");
+        __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_registerEventListeners).call(this);
     }
     /**
      * Allows controller to make active and passive polling requests
@@ -121,6 +133,14 @@ class TokenDetectionController extends polling_controller_1.StaticIntervalPollin
     disable() {
         __classPrivateFieldSet(this, _TokenDetectionController_disabled, true, "f");
     }
+    /**
+     * Internal isActive state
+     *
+     * @type {object}
+     */
+    get isActive() {
+        return !__classPrivateFieldGet(this, _TokenDetectionController_disabled, "f") && __classPrivateFieldGet(this, _TokenDetectionController_isUnlocked, "f");
+    }
     /**
      * Start polling for detected tokens.
      */
@@ -139,81 +159,101 @@ class TokenDetectionController extends polling_controller_1.StaticIntervalPollin
     }
     _executePoll(networkClientId, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            if (__classPrivateFieldGet(this, _TokenDetectionController_disabled, "f")) {
+            if (!this.isActive) {
                 return;
             }
             yield this.detectTokens({
                 networkClientId,
-                accountAddress: options.address,
+                selectedAddress: options.address,
             });
         });
     }
     /**
-     * Triggers asset ERC20 token auto detection for each contract address in contract metadata on mainnet.
+     * For each token in the token list provided by the TokenListController, checks the token's balance for the selected account address on the active network.
+     * On mainnet, if token detection is disabled in preferences, ERC20 token auto detection will be triggered for each contract address in the legacy token list from the @metamask/contract-metadata repo.
      *
-     * @param options - Options to detect tokens.
+     * @param options - Options for token detection.
      * @param options.networkClientId - The ID of the network client to use.
-     * @param options.accountAddress - The account address to use.
+     * @param options.selectedAddress - the selectedAddress against which to detect for token balances.
      */
-    detectTokens({ networkClientId, accountAddress, } = {}) {
+    detectTokens({ networkClientId, selectedAddress, } = {}) {
+        var _a, _b;
         return __awaiter(this, void 0, void 0, function* () {
-            if (__classPrivateFieldGet(this, _TokenDetectionController_disabled, "f") ||
-                !__classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledForNetwork, "f") ||
-                !__classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f")) {
+            if (!this.isActive) {
                 return;
             }
-            const { tokens } = __classPrivateFieldGet(this, _TokenDetectionController_getTokensState, "f").call(this);
-            const selectedAddress = accountAddress || __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f");
-            const chainId = __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainId).call(this, networkClientId);
-            const tokensAddresses = tokens.map(
-            /* istanbul ignore next*/ (token) => token.address.toLowerCase());
-            const { tokenList } = this.messagingSystem.call('TokenListController:getState');
-            const tokensToDetect = [];
-            for (const address of Object.keys(tokenList)) {
-                if (!tokensAddresses.includes(address)) {
-                    tokensToDetect.push(address);
-                }
+            const addressAgainstWhichToDetect = selectedAddress !== null && selectedAddress !== void 0 ? selectedAddress : __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f");
+            const { chainId: chainIdAgainstWhichToDetect, networkClientId: networkClientIdAgainstWhichToDetect, } = __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainIdAndNetworkClientId).call(this, networkClientId);
+            if (!(0, assetsUtil_1.isTokenDetectionSupportedForNetwork)(chainIdAgainstWhichToDetect)) {
+                return;
             }
-            const sliceOfTokensToDetect = [];
-            sliceOfTokensToDetect[0] = tokensToDetect.slice(0, 1000);
-            sliceOfTokensToDetect[1] = tokensToDetect.slice(1000, tokensToDetect.length - 1);
-            /* istanbul ignore else */
-            if (!selectedAddress) {
+            if (!__classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f") &&
+                chainIdAgainstWhichToDetect !== controller_utils_1.ChainId.mainnet) {
                 return;
             }
-            for (const tokensSlice of sliceOfTokensToDetect) {
+            const isTokenDetectionInactiveInMainnet = !__classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f") &&
+                chainIdAgainstWhichToDetect === controller_utils_1.ChainId.mainnet;
+            const { tokensChainsCache } = this.messagingSystem.call('TokenListController:getState');
+            const tokenList = (_b = (_a = tokensChainsCache[chainIdAgainstWhichToDetect]) === null || _a === void 0 ? void 0 : _a.data) !== null && _b !== void 0 ? _b : {};
+            const tokenListUsed = isTokenDetectionInactiveInMainnet
+                ? exports.STATIC_MAINNET_TOKEN_LIST
+                : tokenList;
+            const { allTokens, allDetectedTokens, allIgnoredTokens } = this.messagingSystem.call('TokensController:getState');
+            const [tokensAddresses, detectedTokensAddresses, ignoredTokensAddresses] = [
+                allTokens,
+                allDetectedTokens,
+                allIgnoredTokens,
+            ].map((tokens) => {
+                var _a, _b;
+                return ((_b = (_a = tokens[chainIdAgainstWhichToDetect]) === null || _a === void 0 ? void 0 : _a[addressAgainstWhichToDetect]) !== null && _b !== void 0 ? _b : []).map((value) => (typeof value === 'string' ? value : value.address));
+            });
+            const tokensToDetect = [];
+            for (const tokenAddress of Object.keys(tokenListUsed)) {
+                if ([
+                    tokensAddresses,
+                    detectedTokensAddresses,
+                    ignoredTokensAddresses,
+                ].every((addresses) => !addresses.find((address) => isEqualCaseInsensitive(address, tokenAddress)))) {
+                    tokensToDetect.push(tokenAddress);
+                }
+            }
+            const slicesOfTokensToDetect = [];
+            slicesOfTokensToDetect[0] = tokensToDetect.slice(0, 1000);
+            slicesOfTokensToDetect[1] = tokensToDetect.slice(1000, tokensToDetect.length - 1);
+            for (const tokensSlice of slicesOfTokensToDetect) {
                 if (tokensSlice.length === 0) {
                     break;
                 }
                 yield (0, controller_utils_1.safelyExecute)(() => __awaiter(this, void 0, void 0, function* () {
-                    var _a;
-                    const balances = yield __classPrivateFieldGet(this, _TokenDetectionController_getBalancesInSingleCall, "f").call(this, selectedAddress, tokensSlice);
-                    const tokensToAdd = [];
-                    for (const tokenAddress of Object.keys(balances)) {
-                        let ignored;
-                        /* istanbul ignore else */
-                        const { ignoredTokens } = __classPrivateFieldGet(this, _TokenDetectionController_getTokensState, "f").call(this);
-                        if (ignoredTokens.length) {
-                            ignored = ignoredTokens.find((ignoredTokenAddress) => ignoredTokenAddress === (0, controller_utils_1.toChecksumHexAddress)(tokenAddress));
-                        }
-                        const caseInsensitiveTokenKey = (_a = Object.keys(tokenList).find((i) => i.toLowerCase() === tokenAddress.toLowerCase())) !== null && _a !== void 0 ? _a : '';
-                        if (ignored === undefined) {
-                            const { decimals, symbol, aggregators, iconUrl, name } = tokenList[caseInsensitiveTokenKey];
-                            tokensToAdd.push({
-                                address: tokenAddress,
-                                decimals,
-                                symbol,
-                                aggregators,
-                                image: iconUrl,
-                                isERC721: false,
-                                name,
-                            });
-                        }
+                    const balances = yield __classPrivateFieldGet(this, _TokenDetectionController_getBalancesInSingleCall, "f").call(this, addressAgainstWhichToDetect, tokensSlice, networkClientIdAgainstWhichToDetect);
+                    const tokensWithBalance = [];
+                    const eventTokensDetails = [];
+                    for (const nonZeroTokenAddress of Object.keys(balances)) {
+                        const { decimals, symbol, aggregators, iconUrl, name } = tokenListUsed[nonZeroTokenAddress];
+                        eventTokensDetails.push(`${symbol} - ${nonZeroTokenAddress}`);
+                        tokensWithBalance.push({
+                            address: nonZeroTokenAddress,
+                            decimals,
+                            symbol,
+                            aggregators,
+                            image: iconUrl,
+                            isERC721: false,
+                            name,
+                        });
                     }
-                    if (tokensToAdd.length) {
-                        yield __classPrivateFieldGet(this, _TokenDetectionController_addDetectedTokens, "f").call(this, tokensToAdd, {
-                            selectedAddress,
-                            chainId,
+                    if (tokensWithBalance.length) {
+                        __classPrivateFieldGet(this, _TokenDetectionController_trackMetaMetricsEvent, "f").call(this, {
+                            event: 'Token Detected',
+                            category: 'Wallet',
+                            properties: {
+                                tokens: eventTokensDetails,
+                                token_standard: 'ERC20',
+                                asset_type: 'TOKEN',
+                            },
+                        });
+                        yield this.messagingSystem.call('TokensController:addDetectedTokens', tokensWithBalance, {
+                            selectedAddress: addressAgainstWhichToDetect,
+                            chainId: chainIdAgainstWhichToDetect,
                         });
                     }
                 }));
@@ -222,13 +262,62 @@ class TokenDetectionController extends polling_controller_1.StaticIntervalPollin
     }
 }
 exports.TokenDetectionController = TokenDetectionController;
-_TokenDetectionController_intervalId = new WeakMap(), _TokenDetectionController_chainId = new WeakMap(), _TokenDetectionController_selectedAddress = new WeakMap(), _TokenDetectionController_networkClientId = new WeakMap(), _TokenDetectionController_disabled = new WeakMap(), _TokenDetectionController_isDetectionEnabledFromPreferences = new WeakMap(), _TokenDetectionController_isDetectionEnabledForNetwork = new WeakMap(), _TokenDetectionController_addDetectedTokens = new WeakMap(), _TokenDetectionController_getBalancesInSingleCall = new WeakMap(), _TokenDetectionController_getTokensState = new WeakMap(), _TokenDetectionController_instances = new WeakSet(), _TokenDetectionController_stopPolling = function _TokenDetectionController_stopPolling() {
+_TokenDetectionController_intervalId = new WeakMap(), _TokenDetectionController_chainId = new WeakMap(), _TokenDetectionController_selectedAddress = new WeakMap(), _TokenDetectionController_networkClientId = new WeakMap(), _TokenDetectionController_disabled = new WeakMap(), _TokenDetectionController_isUnlocked = new WeakMap(), _TokenDetectionController_isDetectionEnabledFromPreferences = new WeakMap(), _TokenDetectionController_isDetectionEnabledForNetwork = new WeakMap(), _TokenDetectionController_getBalancesInSingleCall = new WeakMap(), _TokenDetectionController_trackMetaMetricsEvent = new WeakMap(), _TokenDetectionController_instances = new WeakSet(), _TokenDetectionController_registerEventListeners = function _TokenDetectionController_registerEventListeners() {
+    this.messagingSystem.subscribe('KeyringController:unlock', () => __awaiter(this, void 0, void 0, function* () {
+        __classPrivateFieldSet(this, _TokenDetectionController_isUnlocked, true, "f");
+        yield __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_restartTokenDetection).call(this);
+    }));
+    this.messagingSystem.subscribe('KeyringController:lock', () => {
+        __classPrivateFieldSet(this, _TokenDetectionController_isUnlocked, false, "f");
+        __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_stopPolling).call(this);
+    });
+    this.messagingSystem.subscribe('TokenListController:stateChange', ({ tokenList }) => __awaiter(this, void 0, void 0, function* () {
+        const hasTokens = Object.keys(tokenList).length;
+        if (hasTokens) {
+            yield __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_restartTokenDetection).call(this);
+        }
+    }));
+    this.messagingSystem.subscribe('PreferencesController:stateChange', ({ selectedAddress: newSelectedAddress, useTokenDetection }) => __awaiter(this, void 0, void 0, function* () {
+        const isSelectedAddressChanged = __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f") !== newSelectedAddress;
+        const isDetectionChangedFromPreferences = __classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f") !== useTokenDetection;
+        __classPrivateFieldSet(this, _TokenDetectionController_selectedAddress, newSelectedAddress, "f");
+        __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, useTokenDetection, "f");
+        if (useTokenDetection &&
+            (isSelectedAddressChanged || isDetectionChangedFromPreferences)) {
+            yield __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_restartTokenDetection).call(this, {
+                selectedAddress: __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f"),
+            });
+        }
+    }));
+    this.messagingSystem.subscribe('AccountsController:selectedAccountChange', ({ address: newSelectedAddress }) => __awaiter(this, void 0, void 0, function* () {
+        const isSelectedAddressChanged = __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f") !== newSelectedAddress;
+        if (isSelectedAddressChanged &&
+            __classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledFromPreferences, "f")) {
+            __classPrivateFieldSet(this, _TokenDetectionController_selectedAddress, newSelectedAddress, "f");
+            yield __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_restartTokenDetection).call(this, {
+                selectedAddress: __classPrivateFieldGet(this, _TokenDetectionController_selectedAddress, "f"),
+            });
+        }
+    }));
+    this.messagingSystem.subscribe('NetworkController:networkDidChange', ({ selectedNetworkClientId }) => __awaiter(this, void 0, void 0, function* () {
+        const isNetworkClientIdChanged = __classPrivateFieldGet(this, _TokenDetectionController_networkClientId, "f") !== selectedNetworkClientId;
+        const { chainId: newChainId } = __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_getCorrectChainIdAndNetworkClientId).call(this, selectedNetworkClientId);
+        __classPrivateFieldSet(this, _TokenDetectionController_chainId, newChainId, "f");
+        __classPrivateFieldSet(this, _TokenDetectionController_isDetectionEnabledForNetwork, (0, assetsUtil_1.isTokenDetectionSupportedForNetwork)(newChainId), "f");
+        if (isNetworkClientIdChanged && __classPrivateFieldGet(this, _TokenDetectionController_isDetectionEnabledForNetwork, "f")) {
+            __classPrivateFieldSet(this, _TokenDetectionController_networkClientId, selectedNetworkClientId, "f");
+            yield __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_restartTokenDetection).call(this, {
+                networkClientId: __classPrivateFieldGet(this, _TokenDetectionController_networkClientId, "f"),
+            });
+        }
+    }));
+}, _TokenDetectionController_stopPolling = function _TokenDetectionController_stopPolling() {
     if (__classPrivateFieldGet(this, _TokenDetectionController_intervalId, "f")) {
         clearInterval(__classPrivateFieldGet(this, _TokenDetectionController_intervalId, "f"));
     }
 }, _TokenDetectionController_startPolling = function _TokenDetectionController_startPolling() {
     return __awaiter(this, void 0, void 0, function* () {
-        if (__classPrivateFieldGet(this, _TokenDetectionController_disabled, "f")) {
+        if (!this.isActive) {
             return;
         }
         __classPrivateFieldGet(this, _TokenDetectionController_instances, "m", _TokenDetectionController_stopPolling).call(this);
@@ -237,10 +326,30 @@ _TokenDetectionController_intervalId = new WeakMap(), _TokenDetectionController_
             yield this.detectTokens();
         }), this.getIntervalLength()), "f");
     });
-}, _TokenDetectionController_getCorrectChainId = function _TokenDetectionController_getCorrectChainId(networkClientId) {
-    var _a;
-    const { chainId } = (_a = this.messagingSystem.call('NetworkController:getNetworkConfigurationByNetworkClientId', networkClientId !== null && networkClientId !== void 0 ? networkClientId : __classPrivateFieldGet(this, _TokenDetectionController_networkClientId, "f"))) !== null && _a !== void 0 ? _a : {};
-    return chainId !== null && chainId !== void 0 ? chainId : __classPrivateFieldGet(this, _TokenDetectionController_chainId, "f");
+}, _TokenDetectionController_getCorrectChainIdAndNetworkClientId = function _TokenDetectionController_getCorrectChainIdAndNetworkClientId(networkClientId) {
+    if (networkClientId) {
+        const networkConfiguration = this.messagingSystem.call('NetworkController:getNetworkConfigurationByNetworkClientId', networkClientId);
+        if (networkConfiguration) {
+            return {
+                chainId: networkConfiguration.chainId,
+                networkClientId,
+            };
+        }
+    }
+    const { chainId } = this.messagingSystem.call('NetworkController:getProviderConfig');
+    const newNetworkClientId = this.messagingSystem.call('NetworkController:findNetworkClientIdByChainId', chainId);
+    return {
+        chainId,
+        networkClientId: newNetworkClientId,
+    };
+}, _TokenDetectionController_restartTokenDetection = function _TokenDetectionController_restartTokenDetection({ selectedAddress, networkClientId, } = {}) {
+    return __awaiter(this, void 0, void 0, function* () {
+        yield this.detectTokens({
+            networkClientId,
+            selectedAddress,
+        });
+        this.setIntervalLength(DEFAULT_INTERVAL);
+    });
 };
 exports.default = TokenDetectionController;
 //# sourceMappingURL=TokenDetectionController.js.map

package/dist/TokenDetectionController.js.map

@@ -1 +1 @@
-{"version":3,"file":"TokenDetectionController.js","sourceRoot":"","sources":["../src/TokenDetectionController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAKA,iEAGoC;AAOpC,qEAA+E;AAK/E,6CAAmE;AAQnE,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEnB,QAAA,cAAc,GAAG,0BAA0B,CAAC;AAmCzD;;;;;;;;;GASG;AACH,MAAa,wBAAyB,SAAQ,oDAI7C;IAqBC;;;;;;;;;;;;;;OAcG;IACH,YAAY,EACV,eAAe,EACf,eAAe,GAAG,EAAE,EACpB,QAAQ,GAAG,gBAAgB,EAC3B,QAAQ,GAAG,IAAI,EACf,wBAAwB,EACxB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EACd,SAAS,GAcV;QACC,MAAM,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,GACnD,mBAAmB,EAAE,CAAC;QAExB,KAAK,CAAC;YACJ,IAAI,EAAE,sBAAc;YACpB,SAAS;YACT,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;;QApEL,uDAA4C;QAE5C,oDAAc;QAEd,4DAAyB;QAEzB,4DAAkC;QAElC,qDAAmB;QAEnB,8EAA4C;QAE5C,yEAAuC;QAEvC,8DAAmE;QAEnE,oEAAuF;QAEvF,2DAA4C;QAoD1C,uBAAA,IAAI,sCAAa,QAAQ,MAAA,CAAC;QAC1B,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,uBAAA,IAAI,6CAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6CAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,qCAAY,uBAAA,IAAI,wFAAmB,MAAvB,IAAI,EAAoB,eAAe,CAAC,MAAA,CAAC;QAEzD,uBAAA,IAAI,+DAAsC,wBAAwB,MAAA,CAAC;QACnE,uBAAA,IAAI,0DAAiC,IAAA,gDAAmC,EACtE,uBAAA,IAAI,yCAAS,CACd,MAAA,CAAC;QAEF,uBAAA,IAAI,+CAAsB,iBAAiB,MAAA,CAAC;QAC5C,uBAAA,IAAI,qDAA4B,uBAAuB,MAAA,CAAC;QACxD,uBAAA,IAAI,4CAAmB,cAAc,MAAA,CAAC;QAEtC,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,CAAO,EAAE,SAAS,EAAE,EAAE,EAAE;YACtB,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;YAEhD,IAAI,SAAS,EAAE;gBACb,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;aAC3B;QACH,CAAC,CAAA,CACF,CAAC;QAEF,wBAAwB,CACtB,CAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,EAAE,EAAE;YACnE,MAAM,wBAAwB,GAC5B,uBAAA,IAAI,iDAAiB,KAAK,kBAAkB,CAAC;YAC/C,MAAM,iCAAiC,GACrC,uBAAA,IAAI,mEAAmC,KAAK,iBAAiB,CAAC;YAEhE,uBAAA,IAAI,6CAAoB,kBAAkB,MAAA,CAAC;YAC3C,uBAAA,IAAI,+DAAsC,iBAAiB,MAAA,CAAC;YAE5D,IACE,iBAAiB;gBACjB,CAAC,wBAAwB,IAAI,iCAAiC,CAAC,EAC/D;gBACA,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;aAC3B;QACH,CAAC,CAAA,CACF,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,oCAAoC,EACpC,CAAO,EAAE,uBAAuB,EAAE,EAAE,EAAE;YACpC,uBAAA,IAAI,6CAAoB,uBAAuB,MAAA,CAAC;YAChD,MAAM,UAAU,GAAG,uBAAA,IAAI,wFAAmB,MAAvB,IAAI,EAAoB,uBAAuB,CAAC,CAAC;YACpE,MAAM,gBAAgB,GAAG,uBAAA,IAAI,yCAAS,KAAK,UAAU,CAAC;YACtD,uBAAA,IAAI,qCAAY,UAAU,MAAA,CAAC;YAE3B,uBAAA,IAAI,0DACF,IAAA,gDAAmC,EAAC,UAAU,CAAC,MAAA,CAAC;YAElD,IAAI,uBAAA,IAAI,8DAA8B,IAAI,gBAAgB,EAAE;gBAC1D,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;aAC3B;QACH,CAAC,CAAA,CACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,uBAAA,IAAI,sCAAa,KAAK,MAAA,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,uBAAA,IAAI,sCAAa,IAAI,MAAA,CAAC;IACxB,CAAC;IAED;;OAEG;IACG,KAAK;;YACT,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,MAAM,uBAAA,IAAI,mFAAc,MAAlB,IAAI,CAAgB,CAAC;QAC7B,CAAC;KAAA;IAED;;OAEG;IACH,IAAI;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,uBAAA,IAAI,kFAAa,MAAjB,IAAI,CAAe,CAAC;IACtB,CAAC;IA+BK,YAAY,CAChB,eAAuB,EACvB,OAA4B;;YAE5B,IAAI,uBAAA,IAAI,0CAAU,EAAE;gBAClB,OAAO;aACR;YACD,MAAM,IAAI,CAAC,YAAY,CAAC;gBACtB,eAAe;gBACf,cAAc,EAAE,OAAO,CAAC,OAAO;aAChC,CAAC,CAAC;QACL,CAAC;KAAA;IAED;;;;;;OAMG;IACG,YAAY,CAAC,EACjB,eAAe,EACf,cAAc,MAIZ,EAAE;;YACJ,IACE,uBAAA,IAAI,0CAAU;gBACd,CAAC,uBAAA,IAAI,8DAA8B;gBACnC,CAAC,uBAAA,IAAI,mEAAmC,EACxC;gBACA,OAAO;aACR;YACD,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAA,IAAI,gDAAgB,MAApB,IAAI,CAAkB,CAAC;YAC1C,MAAM,eAAe,GAAG,cAAc,IAAI,uBAAA,IAAI,iDAAiB,CAAC;YAChE,MAAM,OAAO,GAAG,uBAAA,IAAI,wFAAmB,MAAvB,IAAI,EAAoB,eAAe,CAAC,CAAC;YAEzD,MAAM,eAAe,GAAG,MAAM,CAAC,GAAG;YAChC,yBAAyB,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CACjE,CAAC;YACF,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7C,8BAA8B,CAC/B,CAAC;YACF,MAAM,cAAc,GAAa,EAAE,CAAC;YACpC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;gBAC5C,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;oBACtC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;iBAC9B;aACF;YACD,MAAM,qBAAqB,GAAG,EAAE,CAAC;YACjC,qBAAqB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YACzD,qBAAqB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,KAAK,CAC7C,IAAI,EACJ,cAAc,CAAC,MAAM,GAAG,CAAC,CAC1B,CAAC;YAEF,0BAA0B;YAC1B,IAAI,CAAC,eAAe,EAAE;gBACpB,OAAO;aACR;YAED,KAAK,MAAM,WAAW,IAAI,qBAAqB,EAAE;gBAC/C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC5B,MAAM;iBACP;gBAED,MAAM,IAAA,gCAAa,EAAC,GAAS,EAAE;;oBAC7B,MAAM,QAAQ,GAAG,MAAM,uBAAA,IAAI,yDAAyB,MAA7B,IAAI,EACzB,eAAe,EACf,WAAW,CACZ,CAAC;oBACF,MAAM,WAAW,GAAY,EAAE,CAAC;oBAChC,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;wBAChD,IAAI,OAAO,CAAC;wBACZ,0BAA0B;wBAC1B,MAAM,EAAE,aAAa,EAAE,GAAG,uBAAA,IAAI,gDAAgB,MAApB,IAAI,CAAkB,CAAC;wBACjD,IAAI,aAAa,CAAC,MAAM,EAAE;4BACxB,OAAO,GAAG,aAAa,CAAC,IAAI,CAC1B,CAAC,mBAAmB,EAAE,EAAE,CACtB,mBAAmB,KAAK,IAAA,uCAAoB,EAAC,YAAY,CAAC,CAC7D,CAAC;yBACH;wBACD,MAAM,uBAAuB,GAC3B,MAAA,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CACzB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,YAAY,CAAC,WAAW,EAAE,CACtD,mCAAI,EAAE,CAAC;wBAEV,IAAI,OAAO,KAAK,SAAS,EAAE;4BACzB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,GACpD,SAAS,CAAC,uBAAuB,CAAC,CAAC;4BACrC,WAAW,CAAC,IAAI,CAAC;gCACf,OAAO,EAAE,YAAY;gCACrB,QAAQ;gCACR,MAAM;gCACN,WAAW;gCACX,KAAK,EAAE,OAAO;gCACd,QAAQ,EAAE,KAAK;gCACf,IAAI;6BACL,CAAC,CAAC;yBACJ;qBACF;oBAED,IAAI,WAAW,CAAC,MAAM,EAAE;wBACtB,MAAM,uBAAA,IAAI,mDAAmB,MAAvB,IAAI,EAAoB,WAAW,EAAE;4BACzC,eAAe;4BACf,OAAO;yBACR,CAAC,CAAC;qBACJ;gBACH,CAAC,CAAA,CAAC,CAAC;aACJ;QACH,CAAC;KAAA;CACF;AAtTD,4DAsTC;;IA5IG,IAAI,uBAAA,IAAI,4CAAY,EAAE;QACpB,aAAa,CAAC,uBAAA,IAAI,4CAAY,CAAC,CAAC;KACjC;AACH,CAAC;;QAMC,IAAI,uBAAA,IAAI,0CAAU,EAAE;YAClB,OAAO;SACR;QACD,uBAAA,IAAI,kFAAa,MAAjB,IAAI,CAAe,CAAC;QACpB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,uBAAA,IAAI,wCAAe,WAAW,CAAC,GAAS,EAAE;YACxC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC,CAAA,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAA,CAAC;IAC/B,CAAC;sGAEkB,eAAiC;;IAClD,MAAM,EAAE,OAAO,EAAE,GACf,MAAA,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,4DAA4D,EAC5D,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,uBAAA,IAAI,iDAAiB,CACzC,mCAAI,EAAE,CAAC;IACV,OAAO,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,uBAAA,IAAI,yCAAS,CAAC;AAClC,CAAC;AAoHH,kBAAe,wBAAwB,CAAC","sourcesContent":["import type {\n  RestrictedControllerMessenger,\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n} from '@metamask/base-controller';\nimport {\n  safelyExecute,\n  toChecksumHexAddress,\n} from '@metamask/controller-utils';\nimport type {\n  NetworkClientId,\n  NetworkControllerNetworkDidChangeEvent,\n  NetworkControllerStateChangeEvent,\n  NetworkControllerGetNetworkConfigurationByNetworkClientId,\n} from '@metamask/network-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { PreferencesState } from '@metamask/preferences-controller';\nimport type { Hex } from '@metamask/utils';\n\nimport type { AssetsContractController } from './AssetsContractController';\nimport { isTokenDetectionSupportedForNetwork } from './assetsUtil';\nimport type {\n  GetTokenListState,\n  TokenListStateChange,\n} from './TokenListController';\nimport type { Token } from './TokenRatesController';\nimport type { TokensController, TokensState } from './TokensController';\n\nconst DEFAULT_INTERVAL = 180000;\n\nexport const controllerName = 'TokenDetectionController';\n\nexport type TokenDetectionState = Record<never, never>;\n\nexport type TokenDetectionControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  TokenDetectionState\n>;\n\nexport type TokenDetectionControllerActions =\n  TokenDetectionControllerGetStateAction;\n\nexport type AllowedActions =\n  | NetworkControllerGetNetworkConfigurationByNetworkClientId\n  | GetTokenListState;\n\nexport type TokenDetectionControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof controllerName, TokenDetectionState>;\n\nexport type TokenDetectionControllerEvents =\n  TokenDetectionControllerStateChangeEvent;\n\nexport type AllowedEvents =\n  | NetworkControllerStateChangeEvent\n  | NetworkControllerNetworkDidChangeEvent\n  | TokenListStateChange;\n\nexport type TokenDetectionControllerMessenger = RestrictedControllerMessenger<\n  typeof controllerName,\n  TokenDetectionControllerActions | AllowedActions,\n  TokenDetectionControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that passively polls on a set interval for Tokens auto detection\n * @property intervalId - Polling interval used to fetch new token rates\n * @property chainId - The chain ID of the current network\n * @property selectedAddress - Vault selected address\n * @property networkClientId - The network client ID of the current selected network\n * @property disabled - Boolean to track if network requests are blocked\n * @property isDetectionEnabledFromPreferences - Boolean to track if detection is enabled from PreferencesController\n * @property isDetectionEnabledForNetwork - Boolean to track if detected is enabled for current network\n */\nexport class TokenDetectionController extends StaticIntervalPollingController<\n  typeof controllerName,\n  TokenDetectionState,\n  TokenDetectionControllerMessenger\n> {\n  #intervalId?: ReturnType<typeof setTimeout>;\n\n  #chainId: Hex;\n\n  #selectedAddress: string;\n\n  #networkClientId: NetworkClientId;\n\n  #disabled: boolean;\n\n  #isDetectionEnabledFromPreferences: boolean;\n\n  #isDetectionEnabledForNetwork: boolean;\n\n  readonly #addDetectedTokens: TokensController['addDetectedTokens'];\n\n  readonly #getBalancesInSingleCall: AssetsContractController['getBalancesInSingleCall'];\n\n  readonly #getTokensState: () => TokensState;\n\n  /**\n   * Creates a TokenDetectionController instance.\n   *\n   * @param options - The controller options.\n   * @param options.messenger - The controller messaging system.\n   * @param options.disabled - If set to true, all network requests are blocked.\n   * @param options.interval - Polling interval used to fetch new token rates\n   * @param options.networkClientId - The selected network client ID of the current network\n   * @param options.selectedAddress - Vault selected address\n   * @param options.onPreferencesStateChange - Allows subscribing to preferences controller state changes.\n   * @param options.addDetectedTokens - Add a list of detected tokens.\n   * @param options.getBalancesInSingleCall - Gets the balances of a list of tokens for the given address.\n   * @param options.getTokensState - Gets the current state of the Tokens controller.\n   * @param options.getPreferencesState - Gets the state of the preferences controller.\n   */\n  constructor({\n    networkClientId,\n    selectedAddress = '',\n    interval = DEFAULT_INTERVAL,\n    disabled = true,\n    onPreferencesStateChange,\n    getBalancesInSingleCall,\n    addDetectedTokens,\n    getPreferencesState,\n    getTokensState,\n    messenger,\n  }: {\n    networkClientId: NetworkClientId;\n    selectedAddress?: string;\n    interval?: number;\n    disabled?: boolean;\n    onPreferencesStateChange: (\n      listener: (preferencesState: PreferencesState) => void,\n    ) => void;\n    addDetectedTokens: TokensController['addDetectedTokens'];\n    getBalancesInSingleCall: AssetsContractController['getBalancesInSingleCall'];\n    getTokensState: () => TokensState;\n    getPreferencesState: () => PreferencesState;\n    messenger: TokenDetectionControllerMessenger;\n  }) {\n    const { useTokenDetection: defaultUseTokenDetection } =\n      getPreferencesState();\n\n    super({\n      name: controllerName,\n      messenger,\n      state: {},\n      metadata: {},\n    });\n\n    this.#disabled = disabled;\n    this.setIntervalLength(interval);\n\n    this.#networkClientId = networkClientId;\n    this.#selectedAddress = selectedAddress;\n    this.#chainId = this.#getCorrectChainId(networkClientId);\n\n    this.#isDetectionEnabledFromPreferences = defaultUseTokenDetection;\n    this.#isDetectionEnabledForNetwork = isTokenDetectionSupportedForNetwork(\n      this.#chainId,\n    );\n\n    this.#addDetectedTokens = addDetectedTokens;\n    this.#getBalancesInSingleCall = getBalancesInSingleCall;\n    this.#getTokensState = getTokensState;\n\n    this.messagingSystem.subscribe(\n      'TokenListController:stateChange',\n      async ({ tokenList }) => {\n        const hasTokens = Object.keys(tokenList).length;\n\n        if (hasTokens) {\n          await this.detectTokens();\n        }\n      },\n    );\n\n    onPreferencesStateChange(\n      async ({ selectedAddress: newSelectedAddress, useTokenDetection }) => {\n        const isSelectedAddressChanged =\n          this.#selectedAddress !== newSelectedAddress;\n        const isDetectionChangedFromPreferences =\n          this.#isDetectionEnabledFromPreferences !== useTokenDetection;\n\n        this.#selectedAddress = newSelectedAddress;\n        this.#isDetectionEnabledFromPreferences = useTokenDetection;\n\n        if (\n          useTokenDetection &&\n          (isSelectedAddressChanged || isDetectionChangedFromPreferences)\n        ) {\n          await this.detectTokens();\n        }\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'NetworkController:networkDidChange',\n      async ({ selectedNetworkClientId }) => {\n        this.#networkClientId = selectedNetworkClientId;\n        const newChainId = this.#getCorrectChainId(selectedNetworkClientId);\n        const isChainIdChanged = this.#chainId !== newChainId;\n        this.#chainId = newChainId;\n\n        this.#isDetectionEnabledForNetwork =\n          isTokenDetectionSupportedForNetwork(newChainId);\n\n        if (this.#isDetectionEnabledForNetwork && isChainIdChanged) {\n          await this.detectTokens();\n        }\n      },\n    );\n  }\n\n  /**\n   * Allows controller to make active and passive polling requests\n   */\n  enable() {\n    this.#disabled = false;\n  }\n\n  /**\n   * Blocks controller from making network calls\n   */\n  disable() {\n    this.#disabled = true;\n  }\n\n  /**\n   * Start polling for detected tokens.\n   */\n  async start() {\n    this.enable();\n    await this.#startPolling();\n  }\n\n  /**\n   * Stop polling for detected tokens.\n   */\n  stop() {\n    this.disable();\n    this.#stopPolling();\n  }\n\n  #stopPolling() {\n    if (this.#intervalId) {\n      clearInterval(this.#intervalId);\n    }\n  }\n\n  /**\n   * Starts a new polling interval.\n   */\n  async #startPolling(): Promise<void> {\n    if (this.#disabled) {\n      return;\n    }\n    this.#stopPolling();\n    await this.detectTokens();\n    this.#intervalId = setInterval(async () => {\n      await this.detectTokens();\n    }, this.getIntervalLength());\n  }\n\n  #getCorrectChainId(networkClientId?: NetworkClientId) {\n    const { chainId } =\n      this.messagingSystem.call(\n        'NetworkController:getNetworkConfigurationByNetworkClientId',\n        networkClientId ?? this.#networkClientId,\n      ) ?? {};\n    return chainId ?? this.#chainId;\n  }\n\n  async _executePoll(\n    networkClientId: string,\n    options: { address: string },\n  ): Promise<void> {\n    if (this.#disabled) {\n      return;\n    }\n    await this.detectTokens({\n      networkClientId,\n      accountAddress: options.address,\n    });\n  }\n\n  /**\n   * Triggers asset ERC20 token auto detection for each contract address in contract metadata on mainnet.\n   *\n   * @param options - Options to detect tokens.\n   * @param options.networkClientId - The ID of the network client to use.\n   * @param options.accountAddress - The account address to use.\n   */\n  async detectTokens({\n    networkClientId,\n    accountAddress,\n  }: {\n    networkClientId?: NetworkClientId;\n    accountAddress?: string;\n  } = {}): Promise<void> {\n    if (\n      this.#disabled ||\n      !this.#isDetectionEnabledForNetwork ||\n      !this.#isDetectionEnabledFromPreferences\n    ) {\n      return;\n    }\n    const { tokens } = this.#getTokensState();\n    const selectedAddress = accountAddress || this.#selectedAddress;\n    const chainId = this.#getCorrectChainId(networkClientId);\n\n    const tokensAddresses = tokens.map(\n      /* istanbul ignore next*/ (token) => token.address.toLowerCase(),\n    );\n    const { tokenList } = this.messagingSystem.call(\n      'TokenListController:getState',\n    );\n    const tokensToDetect: string[] = [];\n    for (const address of Object.keys(tokenList)) {\n      if (!tokensAddresses.includes(address)) {\n        tokensToDetect.push(address);\n      }\n    }\n    const sliceOfTokensToDetect = [];\n    sliceOfTokensToDetect[0] = tokensToDetect.slice(0, 1000);\n    sliceOfTokensToDetect[1] = tokensToDetect.slice(\n      1000,\n      tokensToDetect.length - 1,\n    );\n\n    /* istanbul ignore else */\n    if (!selectedAddress) {\n      return;\n    }\n\n    for (const tokensSlice of sliceOfTokensToDetect) {\n      if (tokensSlice.length === 0) {\n        break;\n      }\n\n      await safelyExecute(async () => {\n        const balances = await this.#getBalancesInSingleCall(\n          selectedAddress,\n          tokensSlice,\n        );\n        const tokensToAdd: Token[] = [];\n        for (const tokenAddress of Object.keys(balances)) {\n          let ignored;\n          /* istanbul ignore else */\n          const { ignoredTokens } = this.#getTokensState();\n          if (ignoredTokens.length) {\n            ignored = ignoredTokens.find(\n              (ignoredTokenAddress) =>\n                ignoredTokenAddress === toChecksumHexAddress(tokenAddress),\n            );\n          }\n          const caseInsensitiveTokenKey =\n            Object.keys(tokenList).find(\n              (i) => i.toLowerCase() === tokenAddress.toLowerCase(),\n            ) ?? '';\n\n          if (ignored === undefined) {\n            const { decimals, symbol, aggregators, iconUrl, name } =\n              tokenList[caseInsensitiveTokenKey];\n            tokensToAdd.push({\n              address: tokenAddress,\n              decimals,\n              symbol,\n              aggregators,\n              image: iconUrl,\n              isERC721: false,\n              name,\n            });\n          }\n        }\n\n        if (tokensToAdd.length) {\n          await this.#addDetectedTokens(tokensToAdd, {\n            selectedAddress,\n            chainId,\n          });\n        }\n      });\n    }\n  }\n}\n\nexport default TokenDetectionController;\n"]}
+{"version":3,"file":"TokenDetectionController.js","sourceRoot":"","sources":["../src/TokenDetectionController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AASA,oFAAsD;AACtD,iEAAoE;AAapE,qEAA+E;AAQ/E,6CAAmE;AAYnE,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEhC;;;;;;;;;GASG;AACH,SAAgB,sBAAsB,CACpC,MAAc,EACd,MAAc;IAEd,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;QAC5D,OAAO,KAAK,CAAC;KACd;IACD,OAAO,MAAM,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,WAAW,EAAE,CAAC;AACvD,CAAC;AARD,wDAQC;AAYY,QAAA,yBAAyB,GAAG,MAAM,CAAC,OAAO,CACrD,2BAAW,CACZ,CAAC,MAAM,CAKN,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE;IAC1B,MAAM,EAAE,IAAI,KAAuB,QAAQ,EAA1B,aAAa,UAAK,QAAQ,EAArC,QAA0B,CAAW,CAAC;IAC5C,uCACK,GAAG,KACN,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,kCACf,aAAa,KAChB,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE,EAC3B,OAAO,EAAE,mBAAmB,IAAI,EAAE,EAClC,WAAW,EAAE,EAAE,OAEjB;AACJ,CAAC,EAAE,EAAE,CAAC,CAAC;AAEM,QAAA,cAAc,GAAG,0BAA0B,CAAC;AA6CzD;;;;;;;;;;GAUG;AACH,MAAa,wBAAyB,SAAQ,oDAI7C;IA6BC;;;;;;;;;;;OAWG;IACH,YAAY,EACV,eAAe,EACf,eAAe,EACf,QAAQ,GAAG,gBAAgB,EAC3B,QAAQ,GAAG,IAAI,EACf,uBAAuB,EACvB,qBAAqB,EACrB,SAAS,GAiBV;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,sBAAc;YACpB,SAAS;YACT,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;;QAtEL,uDAA4C;QAE5C,oDAAc;QAEd,4DAAyB;QAEzB,4DAAkC;QAElC,qDAAmB;QAEnB,uDAAqB;QAErB,8EAA4C;QAE5C,yEAAuC;QAEvC,oEAAuF;QAEvF,kEAQW;QA8CT,uBAAA,IAAI,sCAAa,QAAQ,MAAA,CAAC;QAC1B,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjC,uBAAA,IAAI,6CAAoB,eAAe,MAAA,CAAC;QACxC,uBAAA,IAAI,6CACF,eAAe,aAAf,eAAe,cAAf,eAAe,GACf,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,CAAC;aAC/D,OAAO,MAAA,CAAC;QACb,MAAM,EAAE,OAAO,EAAE,GACf,uBAAA,IAAI,0GAAqC,MAAzC,IAAI,EAAsC,eAAe,CAAC,CAAC;QAC7D,uBAAA,IAAI,qCAAY,OAAO,MAAA,CAAC;QAExB,MAAM,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,GACnD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAC9D,uBAAA,IAAI,+DAAsC,wBAAwB,MAAA,CAAC;QACnE,uBAAA,IAAI,0DAAiC,IAAA,gDAAmC,EACtE,uBAAA,IAAI,yCAAS,CACd,MAAA,CAAC;QAEF,uBAAA,IAAI,qDAA4B,uBAAuB,MAAA,CAAC;QAExD,uBAAA,IAAI,mDAA0B,qBAAqB,MAAA,CAAC;QAEpD,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9C,4BAA4B,CAC7B,CAAC;QACF,uBAAA,IAAI,wCAAe,UAAU,MAAA,CAAC;QAE9B,uBAAA,IAAI,6FAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IAwFD;;OAEG;IACH,MAAM;QACJ,uBAAA,IAAI,sCAAa,KAAK,MAAA,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,uBAAA,IAAI,sCAAa,IAAI,MAAA,CAAC;IACxB,CAAC;IAED;;;;OAIG;IACH,IAAI,QAAQ;QACV,OAAO,CAAC,uBAAA,IAAI,0CAAU,IAAI,uBAAA,IAAI,4CAAY,CAAC;IAC7C,CAAC;IAED;;OAEG;IACG,KAAK;;YACT,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,MAAM,uBAAA,IAAI,mFAAc,MAAlB,IAAI,CAAgB,CAAC;QAC7B,CAAC;KAAA;IAED;;OAEG;IACH,IAAI;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,uBAAA,IAAI,kFAAa,MAAjB,IAAI,CAAe,CAAC;IACtB,CAAC;IAmDK,YAAY,CAChB,eAAuB,EACvB,OAA4B;;YAE5B,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAClB,OAAO;aACR;YACD,MAAM,IAAI,CAAC,YAAY,CAAC;gBACtB,eAAe;gBACf,eAAe,EAAE,OAAO,CAAC,OAAO;aACjC,CAAC,CAAC;QACL,CAAC;KAAA;IAqBD;;;;;;;OAOG;IACG,YAAY,CAAC,EACjB,eAAe,EACf,eAAe,MAIb,EAAE;;;YACJ,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAClB,OAAO;aACR;YAED,MAAM,2BAA2B,GAC/B,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,uBAAA,IAAI,iDAAiB,CAAC;YAC3C,MAAM,EACJ,OAAO,EAAE,2BAA2B,EACpC,eAAe,EAAE,mCAAmC,GACrD,GAAG,uBAAA,IAAI,0GAAqC,MAAzC,IAAI,EAAsC,eAAe,CAAC,CAAC;YAE/D,IAAI,CAAC,IAAA,gDAAmC,EAAC,2BAA2B,CAAC,EAAE;gBACrE,OAAO;aACR;YAED,IACE,CAAC,uBAAA,IAAI,mEAAmC;gBACxC,2BAA2B,KAAK,0BAAO,CAAC,OAAO,EAC/C;gBACA,OAAO;aACR;YACD,MAAM,iCAAiC,GACrC,CAAC,uBAAA,IAAI,mEAAmC;gBACxC,2BAA2B,KAAK,0BAAO,CAAC,OAAO,CAAC;YAClD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACrD,8BAA8B,CAC/B,CAAC;YACF,MAAM,SAAS,GACb,MAAA,MAAA,iBAAiB,CAAC,2BAA2B,CAAC,0CAAE,IAAI,mCAAI,EAAE,CAAC;YAC7D,MAAM,aAAa,GAAG,iCAAiC;gBACrD,CAAC,CAAC,iCAAyB;gBAC3B,CAAC,CAAC,SAAS,CAAC;YAEd,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,GACtD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YACzD,MAAM,CAAC,eAAe,EAAE,uBAAuB,EAAE,sBAAsB,CAAC,GAAG;gBACzE,SAAS;gBACT,iBAAiB;gBACjB,gBAAgB;aACjB,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;;gBACf,OAAA,CACE,MAAA,MAAA,MAAM,CAAC,2BAA2B,CAAC,0CAAG,2BAA2B,CAAC,mCAAI,EAAE,CACzE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAA;aAAA,CACtE,CAAC;YACF,MAAM,cAAc,GAAa,EAAE,CAAC;YACpC,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;gBACrD,IACE;oBACE,eAAe;oBACf,uBAAuB;oBACvB,sBAAsB;iBACvB,CAAC,KAAK,CACL,CAAC,SAAS,EAAE,EAAE,CACZ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1B,sBAAsB,CAAC,OAAO,EAAE,YAAY,CAAC,CAC9C,CACJ,EACD;oBACA,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;iBACnC;aACF;YACD,MAAM,sBAAsB,GAAG,EAAE,CAAC;YAClC,sBAAsB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAC1D,sBAAsB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,KAAK,CAC9C,IAAI,EACJ,cAAc,CAAC,MAAM,GAAG,CAAC,CAC1B,CAAC;YACF,KAAK,MAAM,WAAW,IAAI,sBAAsB,EAAE;gBAChD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC5B,MAAM;iBACP;gBAED,MAAM,IAAA,gCAAa,EAAC,GAAS,EAAE;oBAC7B,MAAM,QAAQ,GAAG,MAAM,uBAAA,IAAI,yDAAyB,MAA7B,IAAI,EACzB,2BAA2B,EAC3B,WAAW,EACX,mCAAmC,CACpC,CAAC;oBACF,MAAM,iBAAiB,GAAY,EAAE,CAAC;oBACtC,MAAM,kBAAkB,GAAa,EAAE,CAAC;oBACxC,KAAK,MAAM,mBAAmB,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;wBACvD,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,GACpD,aAAa,CAAC,mBAAmB,CAAC,CAAC;wBACrC,kBAAkB,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,mBAAmB,EAAE,CAAC,CAAC;wBAC9D,iBAAiB,CAAC,IAAI,CAAC;4BACrB,OAAO,EAAE,mBAAmB;4BAC5B,QAAQ;4BACR,MAAM;4BACN,WAAW;4BACX,KAAK,EAAE,OAAO;4BACd,QAAQ,EAAE,KAAK;4BACf,IAAI;yBACL,CAAC,CAAC;qBACJ;oBACD,IAAI,iBAAiB,CAAC,MAAM,EAAE;wBAC5B,uBAAA,IAAI,uDAAuB,MAA3B,IAAI,EAAwB;4BAC1B,KAAK,EAAE,gBAAgB;4BACvB,QAAQ,EAAE,QAAQ;4BAClB,UAAU,EAAE;gCACV,MAAM,EAAE,kBAAkB;gCAC1B,cAAc,EAAE,OAAO;gCACvB,UAAU,EAAE,OAAO;6BACpB;yBACF,CAAC,CAAC;wBACH,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7B,oCAAoC,EACpC,iBAAiB,EACjB;4BACE,eAAe,EAAE,2BAA2B;4BAC5C,OAAO,EAAE,2BAA2B;yBACrC,CACF,CAAC;qBACH;gBACH,CAAC,CAAA,CAAC,CAAC;aACJ;;KACF;CACF;AA7bD,4DA6bC;;IA7UG,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,0BAA0B,EAAE,GAAS,EAAE;QACpE,uBAAA,IAAI,wCAAe,IAAI,MAAA,CAAC;QACxB,MAAM,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,CAAyB,CAAC;IACtC,CAAC,CAAA,CAAC,CAAC;IAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAC5D,uBAAA,IAAI,wCAAe,KAAK,MAAA,CAAC;QACzB,uBAAA,IAAI,kFAAa,MAAjB,IAAI,CAAe,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,CAAO,EAAE,SAAS,EAAE,EAAE,EAAE;QACtB,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;QAEhD,IAAI,SAAS,EAAE;YACb,MAAM,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,CAAyB,CAAC;SACrC;IACH,CAAC,CAAA,CACF,CAAC;IAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,mCAAmC,EACnC,CAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,EAAE,EAAE;QACnE,MAAM,wBAAwB,GAC5B,uBAAA,IAAI,iDAAiB,KAAK,kBAAkB,CAAC;QAC/C,MAAM,iCAAiC,GACrC,uBAAA,IAAI,mEAAmC,KAAK,iBAAiB,CAAC;QAEhE,uBAAA,IAAI,6CAAoB,kBAAkB,MAAA,CAAC;QAC3C,uBAAA,IAAI,+DAAsC,iBAAiB,MAAA,CAAC;QAE5D,IACE,iBAAiB;YACjB,CAAC,wBAAwB,IAAI,iCAAiC,CAAC,EAC/D;YACA,MAAM,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB;gBAChC,eAAe,EAAE,uBAAA,IAAI,iDAAiB;aACvC,CAAC,CAAC;SACJ;IACH,CAAC,CAAA,CACF,CAAC;IAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,0CAA0C,EAC1C,CAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,EAAE,EAAE;QACxC,MAAM,wBAAwB,GAC5B,uBAAA,IAAI,iDAAiB,KAAK,kBAAkB,CAAC;QAC/C,IACE,wBAAwB;YACxB,uBAAA,IAAI,mEAAmC,EACvC;YACA,uBAAA,IAAI,6CAAoB,kBAAkB,MAAA,CAAC;YAC3C,MAAM,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB;gBAChC,eAAe,EAAE,uBAAA,IAAI,iDAAiB;aACvC,CAAC,CAAC;SACJ;IACH,CAAC,CAAA,CACF,CAAC;IAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,oCAAoC,EACpC,CAAO,EAAE,uBAAuB,EAAE,EAAE,EAAE;QACpC,MAAM,wBAAwB,GAC5B,uBAAA,IAAI,iDAAiB,KAAK,uBAAuB,CAAC;QAEpD,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAC3B,uBAAA,IAAI,0GAAqC,MAAzC,IAAI,EAAsC,uBAAuB,CAAC,CAAC;QACrE,uBAAA,IAAI,qCAAY,UAAU,MAAA,CAAC;QAC3B,uBAAA,IAAI,0DACF,IAAA,gDAAmC,EAAC,UAAU,CAAC,MAAA,CAAC;QAElD,IAAI,wBAAwB,IAAI,uBAAA,IAAI,8DAA8B,EAAE;YAClE,uBAAA,IAAI,6CAAoB,uBAAuB,MAAA,CAAC;YAChD,MAAM,uBAAA,IAAI,4FAAuB,MAA3B,IAAI,EAAwB;gBAChC,eAAe,EAAE,uBAAA,IAAI,iDAAiB;aACvC,CAAC,CAAC;SACJ;IACH,CAAC,CAAA,CACF,CAAC;AACJ,CAAC;IA0CC,IAAI,uBAAA,IAAI,4CAAY,EAAE;QACpB,aAAa,CAAC,uBAAA,IAAI,4CAAY,CAAC,CAAC;KACjC;AACH,CAAC;;QAMC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YAClB,OAAO;SACR;QACD,uBAAA,IAAI,kFAAa,MAAjB,IAAI,CAAe,CAAC;QACpB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,uBAAA,IAAI,wCAAe,WAAW,CAAC,GAAS,EAAE;YACxC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC,CAAA,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAA,CAAC;IAC/B,CAAC;0IAEoC,eAAiC;IAIpE,IAAI,eAAe,EAAE;QACnB,MAAM,oBAAoB,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACpD,4DAA4D,EAC5D,eAAe,CAChB,CAAC;QACF,IAAI,oBAAoB,EAAE;YACxB,OAAO;gBACL,OAAO,EAAE,oBAAoB,CAAC,OAAO;gBACrC,eAAe;aAChB,CAAC;SACH;KACF;IACD,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,qCAAqC,CACtC,CAAC;IACF,MAAM,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAClD,gDAAgD,EAChD,OAAO,CACR,CAAC;IACF,OAAO;QACL,OAAO;QACP,eAAe,EAAE,kBAAkB;KACpC,CAAC;AACJ,CAAC,6GAuB4B,EAC3B,eAAe,EACf,eAAe,MAC2C,EAAE;;QAC5D,MAAM,IAAI,CAAC,YAAY,CAAC;YACtB,eAAe;YACf,eAAe;SAChB,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;IAC3C,CAAC;;AAuIH,kBAAe,wBAAwB,CAAC","sourcesContent":["import type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  RestrictedControllerMessenger,\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n} from '@metamask/base-controller';\nimport contractMap from '@metamask/contract-metadata';\nimport { ChainId, safelyExecute } from '@metamask/controller-utils';\nimport type {\n  KeyringControllerGetStateAction,\n  KeyringControllerLockEvent,\n  KeyringControllerUnlockEvent,\n} from '@metamask/keyring-controller';\nimport type {\n  NetworkClientId,\n  NetworkControllerFindNetworkClientIdByChainIdAction,\n  NetworkControllerGetNetworkConfigurationByNetworkClientId,\n  NetworkControllerGetProviderConfigAction,\n  NetworkControllerNetworkDidChangeEvent,\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';\n\nimport type { AssetsContractController } from './AssetsContractController';\nimport { isTokenDetectionSupportedForNetwork } from './assetsUtil';\nimport type {\n  GetTokenListState,\n  TokenListStateChange,\n  TokenListToken,\n} from './TokenListController';\nimport type { Token } from './TokenRatesController';\nimport type {\n  TokensControllerAddDetectedTokensAction,\n  TokensControllerGetStateAction,\n} from './TokensController';\n\nconst DEFAULT_INTERVAL = 180000;\n\n/**\n * Compare 2 given strings and return boolean\n * eg: \"foo\" and \"FOO\" => true\n * eg: \"foo\" and \"bar\" => false\n * eg: \"foo\" and 123 => false\n *\n * @param value1 - first string to compare\n * @param value2 - first string to compare\n * @returns true if 2 strings are identical when they are lowercase\n */\nexport function isEqualCaseInsensitive(\n  value1: string,\n  value2: string,\n): boolean {\n  if (typeof value1 !== 'string' || typeof value2 !== 'string') {\n    return false;\n  }\n  return value1.toLowerCase() === value2.toLowerCase();\n}\n\ntype LegacyToken = Omit<\n  Token,\n  'aggregators' | 'image' | 'balanceError' | 'isERC721'\n> & {\n  name: string;\n  logo: string;\n  erc20?: boolean;\n  erc721?: boolean;\n};\n\nexport const STATIC_MAINNET_TOKEN_LIST = Object.entries<LegacyToken>(\n  contractMap,\n).reduce<\n  Record<\n    string,\n    Partial<TokenListToken> & Pick<Token, 'address' | 'symbol' | 'decimals'>\n  >\n>((acc, [base, contract]) => {\n  const { logo, ...tokenMetadata } = contract;\n  return {\n    ...acc,\n    [base.toLowerCase()]: {\n      ...tokenMetadata,\n      address: base.toLowerCase(),\n      iconUrl: `images/contract/${logo}`,\n      aggregators: [],\n    },\n  };\n}, {});\n\nexport const controllerName = 'TokenDetectionController';\n\nexport type TokenDetectionState = Record<never, never>;\n\nexport type TokenDetectionControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  TokenDetectionState\n>;\n\nexport type TokenDetectionControllerActions =\n  TokenDetectionControllerGetStateAction;\n\nexport type AllowedActions =\n  | AccountsControllerGetSelectedAccountAction\n  | NetworkControllerFindNetworkClientIdByChainIdAction\n  | NetworkControllerGetNetworkConfigurationByNetworkClientId\n  | NetworkControllerGetProviderConfigAction\n  | GetTokenListState\n  | KeyringControllerGetStateAction\n  | PreferencesControllerGetStateAction\n  | TokensControllerGetStateAction\n  | TokensControllerAddDetectedTokensAction;\n\nexport type TokenDetectionControllerStateChangeEvent =\n  ControllerStateChangeEvent<typeof controllerName, TokenDetectionState>;\n\nexport type TokenDetectionControllerEvents =\n  TokenDetectionControllerStateChangeEvent;\n\nexport type AllowedEvents =\n  | AccountsControllerSelectedAccountChangeEvent\n  | NetworkControllerNetworkDidChangeEvent\n  | TokenListStateChange\n  | KeyringControllerLockEvent\n  | KeyringControllerUnlockEvent\n  | PreferencesControllerStateChangeEvent;\n\nexport type TokenDetectionControllerMessenger = RestrictedControllerMessenger<\n  typeof controllerName,\n  TokenDetectionControllerActions | AllowedActions,\n  TokenDetectionControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that passively polls on a set interval for Tokens auto detection\n * @property intervalId - Polling interval used to fetch new token rates\n * @property chainId - The chain ID of the current network\n * @property selectedAddress - Vault selected address\n * @property networkClientId - The network client ID of the current selected network\n * @property disabled - Boolean to track if network requests are blocked\n * @property isUnlocked - Boolean to track if the keyring state is unlocked\n * @property isDetectionEnabledFromPreferences - Boolean to track if detection is enabled from PreferencesController\n * @property isDetectionEnabledForNetwork - Boolean to track if detected is enabled for current network\n */\nexport class TokenDetectionController extends StaticIntervalPollingController<\n  typeof controllerName,\n  TokenDetectionState,\n  TokenDetectionControllerMessenger\n> {\n  #intervalId?: ReturnType<typeof setTimeout>;\n\n  #chainId: Hex;\n\n  #selectedAddress: string;\n\n  #networkClientId: NetworkClientId;\n\n  #disabled: boolean;\n\n  #isUnlocked: boolean;\n\n  #isDetectionEnabledFromPreferences: boolean;\n\n  #isDetectionEnabledForNetwork: boolean;\n\n  readonly #getBalancesInSingleCall: AssetsContractController['getBalancesInSingleCall'];\n\n  readonly #trackMetaMetricsEvent: (options: {\n    event: string;\n    category: string;\n    properties: {\n      tokens: string[];\n      token_standard: string;\n      asset_type: string;\n    };\n  }) => void;\n\n  /**\n   * Creates a TokenDetectionController instance.\n   *\n   * @param options - The controller options.\n   * @param options.messenger - The controller messaging system.\n   * @param options.disabled - If set to true, all network requests are blocked.\n   * @param options.interval - Polling interval used to fetch new token rates\n   * @param options.networkClientId - The selected network client ID of the current network\n   * @param options.selectedAddress - Vault selected address\n   * @param options.getBalancesInSingleCall - Gets the balances of a list of tokens for the given address.\n   * @param options.trackMetaMetricsEvent - Sets options for MetaMetrics event tracking.\n   */\n  constructor({\n    networkClientId,\n    selectedAddress,\n    interval = DEFAULT_INTERVAL,\n    disabled = true,\n    getBalancesInSingleCall,\n    trackMetaMetricsEvent,\n    messenger,\n  }: {\n    networkClientId: NetworkClientId;\n    selectedAddress?: string;\n    interval?: number;\n    disabled?: boolean;\n    getBalancesInSingleCall: AssetsContractController['getBalancesInSingleCall'];\n    trackMetaMetricsEvent: (options: {\n      event: string;\n      category: string;\n      properties: {\n        tokens: string[];\n        token_standard: string;\n        asset_type: string;\n      };\n    }) => void;\n    messenger: TokenDetectionControllerMessenger;\n  }) {\n    super({\n      name: controllerName,\n      messenger,\n      state: {},\n      metadata: {},\n    });\n\n    this.#disabled = disabled;\n    this.setIntervalLength(interval);\n\n    this.#networkClientId = networkClientId;\n    this.#selectedAddress =\n      selectedAddress ??\n      this.messagingSystem.call('AccountsController:getSelectedAccount')\n        .address;\n    const { chainId } =\n      this.#getCorrectChainIdAndNetworkClientId(networkClientId);\n    this.#chainId = chainId;\n\n    const { useTokenDetection: defaultUseTokenDetection } =\n      this.messagingSystem.call('PreferencesController:getState');\n    this.#isDetectionEnabledFromPreferences = defaultUseTokenDetection;\n    this.#isDetectionEnabledForNetwork = isTokenDetectionSupportedForNetwork(\n      this.#chainId,\n    );\n\n    this.#getBalancesInSingleCall = getBalancesInSingleCall;\n\n    this.#trackMetaMetricsEvent = trackMetaMetricsEvent;\n\n    const { isUnlocked } = this.messagingSystem.call(\n      'KeyringController:getState',\n    );\n    this.#isUnlocked = isUnlocked;\n\n    this.#registerEventListeners();\n  }\n\n  /**\n   * Constructor helper for registering this controller's messaging system subscriptions to controller events.\n   */\n  #registerEventListeners() {\n    this.messagingSystem.subscribe('KeyringController:unlock', async () => {\n      this.#isUnlocked = true;\n      await this.#restartTokenDetection();\n    });\n\n    this.messagingSystem.subscribe('KeyringController:lock', () => {\n      this.#isUnlocked = false;\n      this.#stopPolling();\n    });\n\n    this.messagingSystem.subscribe(\n      'TokenListController:stateChange',\n      async ({ tokenList }) => {\n        const hasTokens = Object.keys(tokenList).length;\n\n        if (hasTokens) {\n          await this.#restartTokenDetection();\n        }\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'PreferencesController:stateChange',\n      async ({ selectedAddress: newSelectedAddress, useTokenDetection }) => {\n        const isSelectedAddressChanged =\n          this.#selectedAddress !== newSelectedAddress;\n        const isDetectionChangedFromPreferences =\n          this.#isDetectionEnabledFromPreferences !== useTokenDetection;\n\n        this.#selectedAddress = newSelectedAddress;\n        this.#isDetectionEnabledFromPreferences = useTokenDetection;\n\n        if (\n          useTokenDetection &&\n          (isSelectedAddressChanged || isDetectionChangedFromPreferences)\n        ) {\n          await this.#restartTokenDetection({\n            selectedAddress: this.#selectedAddress,\n          });\n        }\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedAccountChange',\n      async ({ address: newSelectedAddress }) => {\n        const isSelectedAddressChanged =\n          this.#selectedAddress !== newSelectedAddress;\n        if (\n          isSelectedAddressChanged &&\n          this.#isDetectionEnabledFromPreferences\n        ) {\n          this.#selectedAddress = newSelectedAddress;\n          await this.#restartTokenDetection({\n            selectedAddress: this.#selectedAddress,\n          });\n        }\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'NetworkController:networkDidChange',\n      async ({ selectedNetworkClientId }) => {\n        const isNetworkClientIdChanged =\n          this.#networkClientId !== selectedNetworkClientId;\n\n        const { chainId: newChainId } =\n          this.#getCorrectChainIdAndNetworkClientId(selectedNetworkClientId);\n        this.#chainId = newChainId;\n        this.#isDetectionEnabledForNetwork =\n          isTokenDetectionSupportedForNetwork(newChainId);\n\n        if (isNetworkClientIdChanged && this.#isDetectionEnabledForNetwork) {\n          this.#networkClientId = selectedNetworkClientId;\n          await this.#restartTokenDetection({\n            networkClientId: this.#networkClientId,\n          });\n        }\n      },\n    );\n  }\n\n  /**\n   * Allows controller to make active and passive polling requests\n   */\n  enable() {\n    this.#disabled = false;\n  }\n\n  /**\n   * Blocks controller from making network calls\n   */\n  disable() {\n    this.#disabled = true;\n  }\n\n  /**\n   * Internal isActive state\n   *\n   * @type {object}\n   */\n  get isActive() {\n    return !this.#disabled && this.#isUnlocked;\n  }\n\n  /**\n   * Start polling for detected tokens.\n   */\n  async start() {\n    this.enable();\n    await this.#startPolling();\n  }\n\n  /**\n   * Stop polling for detected tokens.\n   */\n  stop() {\n    this.disable();\n    this.#stopPolling();\n  }\n\n  #stopPolling() {\n    if (this.#intervalId) {\n      clearInterval(this.#intervalId);\n    }\n  }\n\n  /**\n   * Starts a new polling interval.\n   */\n  async #startPolling(): Promise<void> {\n    if (!this.isActive) {\n      return;\n    }\n    this.#stopPolling();\n    await this.detectTokens();\n    this.#intervalId = setInterval(async () => {\n      await this.detectTokens();\n    }, this.getIntervalLength());\n  }\n\n  #getCorrectChainIdAndNetworkClientId(networkClientId?: NetworkClientId): {\n    chainId: Hex;\n    networkClientId: NetworkClientId;\n  } {\n    if (networkClientId) {\n      const networkConfiguration = this.messagingSystem.call(\n        'NetworkController:getNetworkConfigurationByNetworkClientId',\n        networkClientId,\n      );\n      if (networkConfiguration) {\n        return {\n          chainId: networkConfiguration.chainId,\n          networkClientId,\n        };\n      }\n    }\n    const { chainId } = this.messagingSystem.call(\n      'NetworkController:getProviderConfig',\n    );\n    const newNetworkClientId = this.messagingSystem.call(\n      'NetworkController:findNetworkClientIdByChainId',\n      chainId,\n    );\n    return {\n      chainId,\n      networkClientId: newNetworkClientId,\n    };\n  }\n\n  async _executePoll(\n    networkClientId: string,\n    options: { address: string },\n  ): Promise<void> {\n    if (!this.isActive) {\n      return;\n    }\n    await this.detectTokens({\n      networkClientId,\n      selectedAddress: options.address,\n    });\n  }\n\n  /**\n   * Restart token detection polling period and call detectNewTokens\n   * in case of address change or user session initialization.\n   *\n   * @param options - Options for restart token detection.\n   * @param options.selectedAddress - the selectedAddress against which to detect for token balances\n   * @param options.networkClientId - The ID of the network client to use.\n   */\n  async #restartTokenDetection({\n    selectedAddress,\n    networkClientId,\n  }: { selectedAddress?: string; networkClientId?: string } = {}) {\n    await this.detectTokens({\n      networkClientId,\n      selectedAddress,\n    });\n    this.setIntervalLength(DEFAULT_INTERVAL);\n  }\n\n  /**\n   * For each token in the token list provided by the TokenListController, checks the token's balance for the selected account address on the active network.\n   * On mainnet, if token detection is disabled in preferences, ERC20 token auto detection will be triggered for each contract address in the legacy token list from the @metamask/contract-metadata repo.\n   *\n   * @param options - Options for token detection.\n   * @param options.networkClientId - The ID of the network client to use.\n   * @param options.selectedAddress - the selectedAddress against which to detect for token balances.\n   */\n  async detectTokens({\n    networkClientId,\n    selectedAddress,\n  }: {\n    networkClientId?: NetworkClientId;\n    selectedAddress?: string;\n  } = {}): Promise<void> {\n    if (!this.isActive) {\n      return;\n    }\n\n    const addressAgainstWhichToDetect =\n      selectedAddress ?? this.#selectedAddress;\n    const {\n      chainId: chainIdAgainstWhichToDetect,\n      networkClientId: networkClientIdAgainstWhichToDetect,\n    } = this.#getCorrectChainIdAndNetworkClientId(networkClientId);\n\n    if (!isTokenDetectionSupportedForNetwork(chainIdAgainstWhichToDetect)) {\n      return;\n    }\n\n    if (\n      !this.#isDetectionEnabledFromPreferences &&\n      chainIdAgainstWhichToDetect !== ChainId.mainnet\n    ) {\n      return;\n    }\n    const isTokenDetectionInactiveInMainnet =\n      !this.#isDetectionEnabledFromPreferences &&\n      chainIdAgainstWhichToDetect === ChainId.mainnet;\n    const { tokensChainsCache } = this.messagingSystem.call(\n      'TokenListController:getState',\n    );\n    const tokenList =\n      tokensChainsCache[chainIdAgainstWhichToDetect]?.data ?? {};\n    const tokenListUsed = isTokenDetectionInactiveInMainnet\n      ? STATIC_MAINNET_TOKEN_LIST\n      : tokenList;\n\n    const { allTokens, allDetectedTokens, allIgnoredTokens } =\n      this.messagingSystem.call('TokensController:getState');\n    const [tokensAddresses, detectedTokensAddresses, ignoredTokensAddresses] = [\n      allTokens,\n      allDetectedTokens,\n      allIgnoredTokens,\n    ].map((tokens) =>\n      (\n        tokens[chainIdAgainstWhichToDetect]?.[addressAgainstWhichToDetect] ?? []\n      ).map((value) => (typeof value === 'string' ? value : value.address)),\n    );\n    const tokensToDetect: string[] = [];\n    for (const tokenAddress of Object.keys(tokenListUsed)) {\n      if (\n        [\n          tokensAddresses,\n          detectedTokensAddresses,\n          ignoredTokensAddresses,\n        ].every(\n          (addresses) =>\n            !addresses.find((address) =>\n              isEqualCaseInsensitive(address, tokenAddress),\n            ),\n        )\n      ) {\n        tokensToDetect.push(tokenAddress);\n      }\n    }\n    const slicesOfTokensToDetect = [];\n    slicesOfTokensToDetect[0] = tokensToDetect.slice(0, 1000);\n    slicesOfTokensToDetect[1] = tokensToDetect.slice(\n      1000,\n      tokensToDetect.length - 1,\n    );\n    for (const tokensSlice of slicesOfTokensToDetect) {\n      if (tokensSlice.length === 0) {\n        break;\n      }\n\n      await safelyExecute(async () => {\n        const balances = await this.#getBalancesInSingleCall(\n          addressAgainstWhichToDetect,\n          tokensSlice,\n          networkClientIdAgainstWhichToDetect,\n        );\n        const tokensWithBalance: Token[] = [];\n        const eventTokensDetails: string[] = [];\n        for (const nonZeroTokenAddress of Object.keys(balances)) {\n          const { decimals, symbol, aggregators, iconUrl, name } =\n            tokenListUsed[nonZeroTokenAddress];\n          eventTokensDetails.push(`${symbol} - ${nonZeroTokenAddress}`);\n          tokensWithBalance.push({\n            address: nonZeroTokenAddress,\n            decimals,\n            symbol,\n            aggregators,\n            image: iconUrl,\n            isERC721: false,\n            name,\n          });\n        }\n        if (tokensWithBalance.length) {\n          this.#trackMetaMetricsEvent({\n            event: 'Token Detected',\n            category: 'Wallet',\n            properties: {\n              tokens: eventTokensDetails,\n              token_standard: 'ERC20',\n              asset_type: 'TOKEN',\n            },\n          });\n          await this.messagingSystem.call(\n            'TokensController:addDetectedTokens',\n            tokensWithBalance,\n            {\n              selectedAddress: addressAgainstWhichToDetect,\n              chainId: chainIdAgainstWhichToDetect,\n            },\n          );\n        }\n      });\n    }\n  }\n}\n\nexport default TokenDetectionController;\n"]}

package/dist/TokensController.d.ts

@@ -2,13 +2,13 @@
 import { Contract } from '@ethersproject/contracts';
 import { Web3Provider } from '@ethersproject/providers';
 import type { AddApprovalRequest } from '@metamask/approval-controller';
-import type { BaseConfig, BaseState, RestrictedControllerMessenger } from '@metamask/base-controller';
+import type { BaseConfig, BaseState, RestrictedControllerMessenger, ControllerGetStateAction, ControllerStateChangeEvent } from '@metamask/base-controller';
 import { BaseControllerV1 } from '@metamask/base-controller';
-import type { NetworkClientId, NetworkController, NetworkState } from '@metamask/network-controller';
-import type { PreferencesState } from '@metamask/preferences-controller';
+import type { NetworkClientId, NetworkControllerGetNetworkClientByIdAction, NetworkControllerNetworkDidChangeEvent, Provider } from '@metamask/network-controller';
+import type { PreferencesControllerStateChangeEvent } from '@metamask/preferences-controller';
 import type { Hex } from '@metamask/utils';
 import { EventEmitter } from 'events';
-import type { TokenListState } from './TokenListController';
+import type { TokenListStateChange } from './TokenListController';
 import type { Token } from './TokenRatesController';
 /**
  * @type TokensConfig
@@ -19,7 +19,7 @@ import type { Token } from './TokenRatesController';
 export interface TokensConfig extends BaseConfig {
     selectedAddress: string;
     chainId: Hex;
-    provider: any;
+    provider: Provider | undefined;
 }
 /**
  * @type SuggestedAssetMeta
@@ -49,7 +49,7 @@ declare type SuggestedAssetMeta = {
  * @property allIgnoredTokens - Object containing hidden/ignored tokens by network and account
  * @property allDetectedTokens - Object containing tokens detected with non-zero balances
  */
-export interface TokensState extends BaseState {
+export declare type TokensState = {
     tokens: Token[];
     ignoredTokens: string[];
     detectedTokens: Token[];
@@ -68,24 +68,33 @@ export interface TokensState extends BaseState {
             [key: string]: Token[];
         };
     };
-}
+};
 /**
  * The name of the {@link TokensController}.
  */
 declare const controllerName = "TokensController";
+export declare type TokensControllerActions = TokensControllerGetStateAction | TokensControllerAddDetectedTokensAction;
+export declare type TokensControllerGetStateAction = ControllerGetStateAction<typeof controllerName, TokensState>;
+export declare type TokensControllerAddDetectedTokensAction = {
+    type: `${typeof controllerName}:addDetectedTokens`;
+    handler: TokensController['addDetectedTokens'];
+};
 /**
  * The external actions available to the {@link TokensController}.
  */
-declare type AllowedActions = AddApprovalRequest;
+export declare type AllowedActions = AddApprovalRequest | NetworkControllerGetNetworkClientByIdAction;
+export declare type TokensControllerStateChangeEvent = ControllerStateChangeEvent<typeof controllerName, TokensState>;
+export declare type TokensControllerEvents = TokensControllerStateChangeEvent;
+export declare type AllowedEvents = NetworkControllerNetworkDidChangeEvent | PreferencesControllerStateChangeEvent | TokenListStateChange;
 /**
  * The messenger of the {@link TokensController}.
  */
-export declare type TokensControllerMessenger = RestrictedControllerMessenger<typeof controllerName, AllowedActions, never, AllowedActions['type'], never>;
+export declare type TokensControllerMessenger = RestrictedControllerMessenger<typeof controllerName, TokensControllerActions | AllowedActions, TokensControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
 export declare const getDefaultTokensState: () => TokensState;
 /**
  * Controller that stores assets and exposes convenience methods
  */
-export declare class TokensController extends BaseControllerV1<TokensConfig, TokensState> {
+export declare class TokensController extends BaseControllerV1<TokensConfig, TokensState & BaseState> {
     private readonly mutex;
     private abortController;
     private readonly messagingSystem;
@@ -104,26 +113,17 @@ export declare class TokensController extends BaseControllerV1<TokensConfig, Tok
      * Name of this controller used during composition
      */
     name: string;
-    private readonly getNetworkClientById;
     /**
      * Creates a TokensController instance.
      *
      * @param options - The controller options.
      * @param options.chainId - The chain ID of the current network.
-     * @param options.onPreferencesStateChange - Allows subscribing to preference controller state changes.
-     * @param options.onNetworkDidChange - Allows subscribing to network controller networkDidChange events.
-     * @param options.onTokenListStateChange - Allows subscribing to token list controller state changes.
-     * @param options.getNetworkClientById - Gets the network client with the given id from the NetworkController.
      * @param options.config - Initial options used to configure this controller.
      * @param options.state - Initial state to set on this controller.
      * @param options.messenger - The controller messenger.
      */
-    constructor({ chainId: initialChainId, onPreferencesStateChange, onNetworkDidChange, onTokenListStateChange, getNetworkClientById, config, state, messenger, }: {
+    constructor({ chainId: initialChainId, config, state, messenger, }: {
         chainId: Hex;
-        onPreferencesStateChange: (listener: (preferencesState: PreferencesState) => void) => void;
-        onNetworkDidChange: (listener: (networkState: NetworkState) => void) => void;
-        onTokenListStateChange: (listener: (tokenListState: TokenListState) => void) => void;
-        getNetworkClientById: NetworkController['getNetworkClientById'];
         config?: Partial<TokensConfig>;
         state?: Partial<TokensState>;
         messenger: TokensControllerMessenger;