import { expect } from 'chai'
import { BigNumber } from 'ethers'
import { MixedRouteQuoterV1 } from '../typechain'

import hre, { ethers } from 'hardhat'
import { encodePath } from './shared/path'
import { expandTo18Decimals, expandToNDecimals } from './shared/expandTo18Decimals'
import { FeeAmount, V2_FEE_PLACEHOLDER } from './shared/constants'

const V3_FACTORY = '0x1F98431c8aD98523631AE4a59f267346ea31F984'
const V2_FACTORY = '0x5C69bEe701ef814a2B6a3EDD4B1652CB9cc5aA6f'

const USDC = '0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48'
const USDT = '0xdAC17F958D2ee523a2206206994597C13D831ec7'
const WETH = '0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2'
const UNI = '0x1f9840a85d5af5bf1d1762f925bdaddc4201f984'
const DAI = '0x6B175474E89094C44Da98b954EedeAC495271d0F'

/// @dev basic V2 routes
const DAI_V2_UNI_V2_WETH = encodePath([DAI, UNI, WETH], [V2_FEE_PLACEHOLDER, V2_FEE_PLACEHOLDER])
const USDC_V2_UNI_V2_WETH = encodePath([USDC, UNI, WETH], [V2_FEE_PLACEHOLDER, V2_FEE_PLACEHOLDER])

/// @dev basic V3 routes
const USDC_V3_USDT = encodePath([USDC, USDT], [FeeAmount.LOW])
const UNI_V3_WETH = encodePath([UNI, WETH], [FeeAmount.MEDIUM])

/// @dev stablecoin IL routes
const USDT_V3_DAI_V2_USDC = encodePath([USDT, DAI, USDC], [FeeAmount.LOW, V2_FEE_PLACEHOLDER])
const DAI_V3_USDC_V2_USDT = encodePath([DAI, USDC, USDT], [100, V2_FEE_PLACEHOLDER])

/// @dev erc20 IL routes
// V3 - V2
const UNI_V3_WETH_V2_DAI = encodePath([UNI, WETH, DAI], [FeeAmount.MEDIUM, V2_FEE_PLACEHOLDER])
const USDC_V3_UNI_V2_WETH = encodePath([USDC, UNI, WETH], [FeeAmount.MEDIUM, V2_FEE_PLACEHOLDER])
// V2 - V3
const UNI_V2_WETH_V3_DAI = encodePath([UNI, WETH, DAI], [V2_FEE_PLACEHOLDER, FeeAmount.MEDIUM])

/// @dev complex IL routes
// (use two V3 pools)
const DAI_V3_3000_UNI_V2_USDT_V3_3000_WETH = encodePath(
  [DAI, UNI, USDT, WETH],
  [FeeAmount.MEDIUM, V2_FEE_PLACEHOLDER, FeeAmount.MEDIUM]
)
// (use two V2 pools)
const DAI_V3_3000_UNI_V2_USDT_V2_WETH = encodePath(
  [DAI, UNI, USDT, WETH],
  [FeeAmount.MEDIUM, V2_FEE_PLACEHOLDER, V2_FEE_PLACEHOLDER]
)

describe('MixedRouteQuoterV1 integration tests', function () {
  let mixedRouteQuoter: MixedRouteQuoterV1

  this.timeout(100000)

  before(async function () {
    if (!process.env.ARCHIVE_RPC_URL) {
      this.skip()
    }

    await hre.network.provider.request({
      method: 'hardhat_reset',
      params: [
        {
          forking: {
            jsonRpcUrl: process.env.ARCHIVE_RPC_URL,
            blockNumber: 14390000,
          },
        },
      ],
    })

    const MixedRouteQuoterV1Factory = await ethers.getContractFactory('MixedRouteQuoterV1')
    mixedRouteQuoter = (await MixedRouteQuoterV1Factory.deploy(V3_FACTORY, V2_FACTORY, WETH)) as MixedRouteQuoterV1
  })

  after(async () => {
    // Disable mainnet forking to avoid effecting other tests.
    await hre.network.provider.request({
      method: 'hardhat_reset',
      params: [],
    })
  })

  /**
   * Test values only valid starting at block 14390000
   */
  it('sets block number correctly', async () => {
    const blockNumber = BigNumber.from(
      await hre.network.provider.request({
        method: 'eth_blockNumber',
        params: [],
      })
    )
    /// @dev +1 so 14390001 since we just requested
    expect(blockNumber.eq(14390001)).to.be.true
  })

  describe('quotes stablecoin only paths correctly', () => {
    /// @dev the amount must be expanded to the decimals of the first token in the path
    it('V3-V2 stablecoin path with 6 decimal in start of path', async () => {
      const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
        'quoteExactInput(bytes,uint256)'
      ](USDT_V3_DAI_V2_USDC, expandToNDecimals(10000, 6))

      expect(amountOut).eq(BigNumber.from('9966336832'))
      expect(v3SqrtPriceX96AfterList[0].eq(BigNumber.from('0x10c6727487c45717095f'))).to.be.true
    })

    it('V3-V2 stablecoin path with 6 decimal in middle of path', async () => {
      const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
        'quoteExactInput(bytes,uint256)'
      ](DAI_V3_USDC_V2_USDT, expandTo18Decimals(10000))

      expect(amountOut).eq(BigNumber.from('9959354898'))
      expect(v3SqrtPriceX96AfterList[0].eq(BigNumber.from('0x10c715093f77e3073634'))).to.be.true
    })
  })

  describe('V2-V2 quotes', () => {
    it('quotes V2-V2 correctly', async () => {
      const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
        'quoteExactInput(bytes,uint256)'
      ](DAI_V2_UNI_V2_WETH, expandTo18Decimals(10000))

      expect(amountOut).eq(BigNumber.from('2035189623576328665'))
      expect(v3SqrtPriceX96AfterList.every((el) => el.eq(0))).to.be.true
      expect(v3InitializedTicksCrossedList.every((el) => el == 0)).to.be.true
    })

    it('quotes V2 (6 decimal stablecoin) -V2 correctly', async () => {
      const { amountOut } = await mixedRouteQuoter.callStatic['quoteExactInput(bytes,uint256)'](
        USDC_V2_UNI_V2_WETH,
        expandToNDecimals(10000, 6)
      )

      expect(amountOut).eq(BigNumber.from('1989381322826753150'))
    })
  })

  it('quotes V3-V2 erc20s with mixed decimal scales correctly', async () => {
    const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
      'quoteExactInput(bytes,uint256)'
    ](USDC_V3_UNI_V2_WETH, expandToNDecimals(10000, 6))

    expect(amountOut).eq(BigNumber.from('3801923847986895918')) // 3.801923847986895918
    expect(v3SqrtPriceX96AfterList[0].eq(BigNumber.from('0x3110863ba621ac3915fd'))).to.be.true
  })

  it('quotes V3-V2 correctly', async () => {
    const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
      'quoteExactInput(bytes,uint256)'
    ](UNI_V3_WETH_V2_DAI, expandTo18Decimals(10000))

    expect(amountOut).eq(BigNumber.from('80675538331724434694636'))
    expect(v3SqrtPriceX96AfterList[0].eq(BigNumber.from('0x0e83f285cb58c4cca14fb78b'))).to.be.true
  })

  it('quotes V3-V2-V3 correctly', async () => {
    const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
      'quoteExactInput(bytes,uint256)'
    ](DAI_V3_3000_UNI_V2_USDT_V3_3000_WETH, expandTo18Decimals(10000))

    expect(amountOut).eq(BigNumber.from('886596560223108447'))
    expect(v3SqrtPriceX96AfterList[0].eq(BigNumber.from('0xfffd8963efd1fc6a506488495d951d5263988d25'))).to.be.true
    expect(v3SqrtPriceX96AfterList[2].eq(BigNumber.from('0x034b624fce51aba62a4722'))).to.be.true
  })

  it('quotes V2-V3 correctly', async () => {
    const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
      'quoteExactInput(bytes,uint256)'
    ](UNI_V2_WETH_V3_DAI, expandTo18Decimals(10000))

    expect(amountOut).eq(BigNumber.from('81108655328627859394525'))
    expect(v3SqrtPriceX96AfterList[1].eq(BigNumber.from('0x0518b75d40eb50192903493d'))).to.be.true
  })

  it('quotes only V3 correctly', async () => {
    const { amountOut, v3SqrtPriceX96AfterList, v3InitializedTicksCrossedList } = await mixedRouteQuoter.callStatic[
      'quoteExactInput(bytes,uint256)'
    ](UNI_V3_WETH, expandTo18Decimals(10000))

    expect(amountOut.eq(BigNumber.from('32215526370828998898'))).to.be.true
  })
})