diff --git a/packages/assets-controllers/src/TokenDetectionController.ts b/packages/assets-controllers/src/TokenDetectionController.ts index 52a84bfc01..e0af458854 100644 --- a/packages/assets-controllers/src/TokenDetectionController.ts +++ b/packages/assets-controllers/src/TokenDetectionController.ts @@ -147,7 +147,7 @@ export type TokenDetectionControllerMessenger = RestrictedControllerMessenger< /** The input to start polling for the {@link TokenDetectionController} */ type TokenDetectionPollingInput = { - networkClientId: NetworkClientId; + chainIds: Hex[]; address: string; }; @@ -394,28 +394,6 @@ export class TokenDetectionController extends StaticIntervalPollingController { - const isNetworkClientIdChanged = - this.#networkClientId !== selectedNetworkClientId; - - const { chainId: newChainId } = - this.#getCorrectChainIdAndNetworkClientId(selectedNetworkClientId); - this.#isDetectionEnabledForNetwork = - isTokenDetectionSupportedForNetwork(newChainId); - - if (isNetworkClientIdChanged && this.#isDetectionEnabledForNetwork) { - this.#networkClientId = selectedNetworkClientId; - await this.#restartTokenDetection({ - networkClientId: this.#networkClientId, - }); - } - }, - ); } /** @@ -501,6 +479,60 @@ export class TokenDetectionController extends StaticIntervalPollingController { + // Get the network configuration for the specified chainId + const configuration = networkConfigurationsByChainId[chainId]; + + // Check if a configuration exists for the specified chainId + if (!configuration) { + console.error(`No configuration found for chainId: ${chainId}`); + return { chainId, networkClientId: '' }; // TODO: better handle this case + } + + // Get the rpcEndpoints array from the configuration + const { rpcEndpoints } = configuration; + + // Check if selectedNetworkClientId exists in rpcEndpoints + const matchingEndpoint = rpcEndpoints.find( + (endpoint) => endpoint.networkClientId === selectedNetworkClientId, + ); + + // Return an object with both chainId and networkClientId + return { + chainId, + networkClientId: matchingEndpoint + ? matchingEndpoint.networkClientId + : rpcEndpoints[0].networkClientId, + }; + }); + } + #getCorrectChainIdAndNetworkClientId(networkClientId?: NetworkClientId): { chainId: Hex; networkClientId: NetworkClientId; @@ -533,14 +565,14 @@ export class TokenDetectionController extends StaticIntervalPollingController { if (!this.isActive) { return; } await this.detectTokens({ - networkClientId, + chainIds, selectedAddress: address, }); } @@ -551,17 +583,17 @@ export class TokenDetectionController extends StaticIntervalPollingController { await this.detectTokens({ - networkClientId, + chainIds, selectedAddress, }); this.setIntervalLength(DEFAULT_INTERVAL); @@ -572,72 +604,92 @@ export class TokenDetectionController extends StaticIntervalPollingController { if (!this.isActive) { return; } + // Address used by the user const addressAgainstWhichToDetect = selectedAddress ?? this.#getSelectedAddress(); - const { chainId, networkClientId: selectedNetworkClientId } = - this.#getCorrectChainIdAndNetworkClientId(networkClientId); - const chainIdAgainstWhichToDetect = chainId; - const networkClientIdAgainstWhichToDetect = selectedNetworkClientId; - if (!isTokenDetectionSupportedForNetwork(chainIdAgainstWhichToDetect)) { - return; - } - if ( - !this.#isDetectionEnabledFromPreferences && - chainIdAgainstWhichToDetect !== ChainId.mainnet - ) { + // Will be an array of objects in the form [{chainId : "0x1", networkClientId: "8e8b86f2-f949-4f29-90bc-f7409f0832d5"}] + const clientNetworksIdByChainId = + this.#getCorrectNetworkClientIdByChainId(chainIds); + + if (!clientNetworksIdByChainId) { return; } - const isTokenDetectionInactiveInMainnet = - !this.#isDetectionEnabledFromPreferences && - chainIdAgainstWhichToDetect === ChainId.mainnet; - const { tokensChainsCache } = this.messagingSystem.call( - 'TokenListController:getState', - ); - this.#tokensChainsCache = isTokenDetectionInactiveInMainnet - ? this.#getConvertedStaticMainnetTokenList() - : tokensChainsCache ?? {}; - const tokenCandidateSlices = this.#getSlicesOfTokensToDetect({ + // Execute the rest in a loop for each pair of chainId and networkClientId in the result array + for (const { chainId: chainIdAgainstWhichToDetect, - selectedAddress: addressAgainstWhichToDetect, - }); + networkClientId: networkClientIdAgainstWhichToDetect, + } of clientNetworksIdByChainId) { + // Check if token detection is supported for this chainId + if (!isTokenDetectionSupportedForNetwork(chainIdAgainstWhichToDetect)) { + continue; + } - // Attempt Accounts API Detection - const accountAPIResult = await this.#addDetectedTokensViaAPI({ - chainId: chainIdAgainstWhichToDetect, - selectedAddress: addressAgainstWhichToDetect, - tokenCandidateSlices, - }); - if (accountAPIResult?.result === 'success') { - return; - } + // Check if detection is enabled based on preferences and chain ID + if ( + !this.#isDetectionEnabledFromPreferences && + chainIdAgainstWhichToDetect !== ChainId.mainnet + ) { + continue; + } - // Attempt RPC Detection - const tokenDetectionPromises = tokenCandidateSlices.map((tokensSlice) => - this.#addDetectedTokens({ - tokensSlice, + const isTokenDetectionInactiveInMainnet = + !this.#isDetectionEnabledFromPreferences && + chainIdAgainstWhichToDetect === ChainId.mainnet; + + // Get tokens chains cache from TokenListController or initialize mainnet tokens if inactive + const { tokensChainsCache } = this.messagingSystem.call( + 'TokenListController:getState', + ); + this.#tokensChainsCache = isTokenDetectionInactiveInMainnet + ? this.#getConvertedStaticMainnetTokenList() + : tokensChainsCache ?? {}; + + // Get slices of tokens to detect for the given chainId and selected address + const tokenCandidateSlices = this.#getSlicesOfTokensToDetect({ + chainId: chainIdAgainstWhichToDetect, selectedAddress: addressAgainstWhichToDetect, - networkClientId: networkClientIdAgainstWhichToDetect, + }); + + // Attempt Accounts API Detection + const accountAPIResult = await this.#addDetectedTokensViaAPI({ chainId: chainIdAgainstWhichToDetect, - }), - ); + selectedAddress: addressAgainstWhichToDetect, + tokenCandidateSlices, + }); + + if (accountAPIResult?.result === 'success') { + continue; // Skip to the next iteration if detection via API was successful + } + + // Attempt RPC Detection if API detection wasn't successful + const tokenDetectionPromises = tokenCandidateSlices.map((tokensSlice) => + this.#addDetectedTokens({ + tokensSlice, + selectedAddress: addressAgainstWhichToDetect, + networkClientId: networkClientIdAgainstWhichToDetect, + chainId: chainIdAgainstWhichToDetect, + }), + ); - await Promise.all(tokenDetectionPromises); + // Await all token detection promises for this chainId + await Promise.all(tokenDetectionPromises); + } } #getSlicesOfTokensToDetect({