// Import this file (test-utils/tokens/mock) at the top of a test file to use
// these predefined token lookup mocks.

jest.mock('hooks/Tokens')
jest.mock('components/AccountDrawer/MiniPortfolio/Activity/getCurrency')

import { Currency, WETH9 } from '@uniswap/sdk-core'
import { getCurrency } from 'components/AccountDrawer/MiniPortfolio/Activity/getCurrency'
import { useCurrency, useCurrencyInfo } from 'hooks/Tokens'
import {
  DAI_ARBITRUM_INFO,
  DAI_INFO,
  NATIVE_INFO,
  TEST_TOKEN_1,
  TEST_TOKEN_1_INFO,
  TEST_TOKEN_2,
  TEST_TOKEN_2_INFO,
  TEST_TOKEN_3,
  TEST_TOKEN_3_INFO,
  USDC_ARBITRUM_INFO,
  USDC_INFO,
  USDT_INFO,
  WBTC_INFO,
  WETH_INFO,
} from 'test-utils/constants'
import { mocked } from 'test-utils/mocked'
import { COMMON_BASES } from 'uniswap/src/constants/routing'
import { DAI, DAI_ARBITRUM_ONE, USDC_ARBITRUM, USDC_MAINNET, USDT, WBTC } from 'uniswap/src/constants/tokens'
import { UniverseChainId } from 'uniswap/src/features/chains/types'
import { isSameAddress } from 'utilities/src/addresses'

beforeEach(() => {
  // Global mocks for token lookups. To override in a test, use `mocked().mockImplementation(...)`.
  mocked(getCurrency).mockImplementation(async (currencyId: string, chainId: UniverseChainId) => {
    if (currencyId?.toLowerCase() === 'eth') {
      return NATIVE_INFO?.currency
    }
    if (isSameAddress(currencyId, DAI.address)) {
      return DAI_INFO?.currency
    }
    if (isSameAddress(currencyId, USDC_MAINNET.address)) {
      return USDC_INFO?.currency
    }
    if (isSameAddress(currencyId, WETH9[UniverseChainId.Mainnet].address)) {
      return WETH_INFO?.currency
    }
    if (isSameAddress(currencyId, USDT.address)) {
      return USDT_INFO?.currency
    }
    if (isSameAddress(currencyId, WBTC.address)) {
      return WBTC_INFO?.currency
    }
    if (isSameAddress(currencyId, DAI_ARBITRUM_ONE.address)) {
      return DAI_ARBITRUM_INFO?.currency
    }
    if (isSameAddress(currencyId, USDC_ARBITRUM.address)) {
      return USDC_ARBITRUM_INFO?.currency
    }
    if (isSameAddress(currencyId, TEST_TOKEN_1.address)) {
      return TEST_TOKEN_1_INFO?.currency
    }
    if (isSameAddress(currencyId, TEST_TOKEN_2.address)) {
      return TEST_TOKEN_2_INFO?.currency
    }
    if (isSameAddress(currencyId, TEST_TOKEN_3.address)) {
      return TEST_TOKEN_3_INFO?.currency
    }
    return COMMON_BASES[chainId ?? UniverseChainId.Mainnet]?.find((base) =>
      base.currency.isNative ? base.currency.symbol === 'ETH' : base.currency.address === currencyId,
    )?.currency
  })
  mocked(useCurrency).mockImplementation((address?: string, chainId?: UniverseChainId) => {
    if (address?.toLowerCase() === 'eth') {
      return NATIVE_INFO?.currency
    }
    if (isSameAddress(address, DAI.address)) {
      return DAI_INFO?.currency
    }
    if (isSameAddress(address, USDC_MAINNET.address)) {
      return USDC_INFO?.currency
    }
    if (isSameAddress(address, WETH9[UniverseChainId.Mainnet].address)) {
      return WETH_INFO?.currency
    }
    if (isSameAddress(address, USDT.address)) {
      return USDT_INFO?.currency
    }
    if (isSameAddress(address, WBTC.address)) {
      return WBTC_INFO?.currency
    }
    if (isSameAddress(address, DAI_ARBITRUM_ONE.address)) {
      return DAI_ARBITRUM_INFO?.currency
    }
    if (isSameAddress(address, USDC_ARBITRUM.address)) {
      return USDC_ARBITRUM_INFO?.currency
    }
    if (isSameAddress(address, TEST_TOKEN_1.address)) {
      return TEST_TOKEN_1_INFO?.currency
    }
    if (isSameAddress(address, TEST_TOKEN_2.address)) {
      return TEST_TOKEN_2_INFO?.currency
    }
    if (isSameAddress(address, TEST_TOKEN_3.address)) {
      return TEST_TOKEN_3_INFO?.currency
    }
    return COMMON_BASES[chainId ?? UniverseChainId.Mainnet]?.find((base) =>
      base.currency.isNative ? base.currency.symbol === 'ETH' : base.currency.address === address,
    )?.currency
  })
  mocked(useCurrencyInfo).mockImplementation((currency?: Currency | string) => {
    if (typeof currency !== 'string' && currency?.isNative) {
      return NATIVE_INFO
    }
    const address = typeof currency === 'string' ? currency : currency?.address
    if (isSameAddress(address, DAI.address)) {
      return DAI_INFO
    }
    if (isSameAddress(address, USDC_MAINNET.address)) {
      return USDC_INFO
    }
    if (isSameAddress(address, WETH9[UniverseChainId.Mainnet].address)) {
      return WETH_INFO
    }
    if (isSameAddress(address, USDT.address)) {
      return USDT_INFO
    }
    if (isSameAddress(address, WBTC.address)) {
      return WBTC_INFO
    }
    if (isSameAddress(address, DAI_ARBITRUM_ONE.address)) {
      return DAI_ARBITRUM_INFO
    }
    if (isSameAddress(address, USDC_ARBITRUM.address)) {
      return USDC_ARBITRUM_INFO
    }
    if (isSameAddress(address, TEST_TOKEN_1.address)) {
      return TEST_TOKEN_1_INFO
    }
    if (isSameAddress(address, TEST_TOKEN_2.address)) {
      return TEST_TOKEN_2_INFO
    }
    if (isSameAddress(address, TEST_TOKEN_3.address)) {
      return TEST_TOKEN_3_INFO
    }
    return undefined
  })
})
