import React, { useState, useContext } from 'react'; import PropTypes from 'prop-types'; import { useSelector, useDispatch } from 'react-redux'; import { chain } from 'lodash'; import { importTokens, ignoreTokens, setNewTokensImported, } from '../../../store/actions'; import { getDetectedTokensInCurrentNetwork } from '../../../selectors'; import { MetaMetricsContext } from '../../../contexts/metametrics'; import { TOKEN_STANDARDS } from '../../../helpers/constants/common'; import { ASSET_TYPES } from '../../../../shared/constants/transaction'; import { EVENT, EVENT_NAMES } from '../../../../shared/constants/metametrics'; import DetectedTokenSelectionPopover from './detected-token-selection-popover/detected-token-selection-popover'; import DetectedTokenIgnoredPopover from './detected-token-ignored-popover/detected-token-ignored-popover'; const sortingBasedOnTokenSelection = (tokensDetected) => { return ( chain(tokensDetected) // get the values .values() // create a new object with keys 'selected', 'deselected' and group the tokens .groupBy((token) => (token.selected ? 'selected' : 'deselected')) // ditch the 'selected' property and get just the tokens' .mapValues((group) => group.map(({ token }) => token)) // Exit the chain and get the underlying value, an object. .value() ); }; const DetectedToken = ({ setShowDetectedTokens }) => { const dispatch = useDispatch(); const trackEvent = useContext(MetaMetricsContext); const detectedTokens = useSelector(getDetectedTokensInCurrentNetwork); const [tokensListDetected, setTokensListDetected] = useState(() => detectedTokens.reduce((tokenObj, token) => { tokenObj[token.address] = { token, selected: true }; return tokenObj; }, {}), ); const [ showDetectedTokenIgnoredPopover, setShowDetectedTokenIgnoredPopover, ] = useState(false); const importSelectedTokens = async (selectedTokens) => { selectedTokens.forEach((importedToken) => { trackEvent({ event: EVENT_NAMES.TOKEN_ADDED, category: EVENT.CATEGORIES.WALLET, sensitiveProperties: { token_symbol: importedToken.symbol, token_contract_address: importedToken.address, token_decimal_precision: importedToken.decimals, source: EVENT.SOURCE.TOKEN.DETECTED, token_standard: TOKEN_STANDARDS.ERC20, asset_type: ASSET_TYPES.TOKEN, }, }); }); await dispatch(importTokens(selectedTokens)); const tokenSymbols = selectedTokens.map(({ symbol }) => symbol); dispatch(setNewTokensImported(tokenSymbols.join(', '))); }; const handleClearTokensSelection = async () => { const { selected: selectedTokens = [], deselected: deSelectedTokens = [], } = sortingBasedOnTokenSelection(tokensListDetected); if (deSelectedTokens.length < detectedTokens.length) { await importSelectedTokens(selectedTokens); } const tokensDetailsList = deSelectedTokens.map( ({ symbol, address }) => `${symbol} - ${address}`, ); trackEvent({ event: EVENT_NAMES.TOKEN_HIDDEN, category: EVENT.CATEGORIES.WALLET, sensitiveProperties: { tokens: tokensDetailsList, location: EVENT.LOCATION.TOKEN_DETECTION, token_standard: TOKEN_STANDARDS.ERC20, asset_type: ASSET_TYPES.TOKEN, }, }); await dispatch(ignoreTokens(deSelectedTokens)); setShowDetectedTokens(false); }; const handleTokenSelection = (token) => { setTokensListDetected((prevState) => ({ ...prevState, [token.address]: { ...prevState[token.address], selected: !prevState[token.address].selected, }, })); }; const onImport = async () => { const { selected: selectedTokens = [] } = sortingBasedOnTokenSelection( tokensListDetected, ); if (selectedTokens.length < detectedTokens.length) { setShowDetectedTokenIgnoredPopover(true); } else { await importSelectedTokens(selectedTokens); setShowDetectedTokens(false); } }; const onIgnoreAll = () => { const newTokensListDetected = { ...tokensListDetected }; for (const tokenAddress of Object.keys(tokensListDetected)) { newTokensListDetected[tokenAddress].selected = false; } setTokensListDetected(newTokensListDetected); setShowDetectedTokenIgnoredPopover(true); }; const onCancelIgnore = () => { setShowDetectedTokenIgnoredPopover(false); }; return ( <> {showDetectedTokenIgnoredPopover && ( )} ); }; DetectedToken.propTypes = { setShowDetectedTokens: PropTypes.func.isRequired, }; export default DetectedToken;