require('dotenv').config()

const hre = require('hardhat')
const { ethers } = hre
const config = require('../config')
const { createInterface } = require('readline')

function idToNetwork(id) {
  switch (id) {
    case 1:
      return 'Mainnet'
    case 3:
      return 'Ropsten'
    case 4:
      return 'Rinkeby'
    case 5:
      return 'Goerli'
    case 11155111:
      return 'Sepolia'
    default:
      throw Error('\nChain Id could not be recognized. What network are you using?\n')
  }
}

const prompter = createInterface({ input: process.stdin, output: process.stdout })

function _prompt(prompt, resolve) {
  prompter.question(prompt, (answer) => {
    if (answer == 'y') {
      resolve(true)
    } else if (answer == 'n') {
      resolve(false)
    } else _prompt('', resolve)
  })
}

function prompt(prompt) {
  return new Promise((resolve) => _prompt(prompt, resolve))
}

function happyDeployedMessage(name, chainId, address) {
  return `\n${name} successfully deployed on ${idToNetwork(chainId)} @ ${address} 🥳\n`
}

function happyVerifiedMessage(name, address) {
  return `\n${name} @ ${address} successfully verified on Etherscan! 🥳\n`
}

function timeout(seconds) {
  return new Promise((resolve) => setTimeout(resolve, seconds * 1000))
}

const promptMessageBase = (middle) => `\n${middle}\n\nAre you sure you would like to continue? 🧐 (y/n): `

async function main() {
  const minimalFactoryFactory = await ethers.getContractFactory('MinimalInstanceFactory')
  const chainId = (await ethers.provider.getNetwork()).chainId

  let minimalFactory, ethImplAddr, ERC20ImplAddr

  if (await prompt(promptMessageBase('Continuing to MinimalInstanceFactory deployment.'))) {
    minimalFactory = await minimalFactoryFactory.deploy(
      config.verifier,
      config.hasher,
      config.router,
      config.merkleTreeHeight,
    )

    console.log(happyDeployedMessage('MinimalInstanceFactory', chainId, minimalFactory.address))

    await hre.run('verify:verify', {
      address: minimalFactory.address,
      constructorArguments: [config.verifier, config.hasher, config.router, config.merkleTreeHeight],
    })
  } else {
    return '\nDecided to stop at MinimalInstanceFactory deployment.\n'
  }

  console.log(happyDeployedMessage('MinimalInstanceFactory', chainId, minimalFactory.address))

  ethImplAddr = await minimalFactory.ethImpl()
  ERC20ImplAddr = await minimalFactory.ERC20Impl()

  console.log(happyDeployedMessage('ETHTornadoCloneable', chainId, ethImplAddr))
  console.log(happyDeployedMessage('ERC20TornadoCloneable', chainId, ERC20ImplAddr))

  if (await prompt(promptMessageBase('Continuing to impl verifications.'))) {
    console.log('\nWaiting 3 seconds.\n')
    await timeout(3)

    await hre.run('verify:verify', {
      address: ethImplAddr,
      constructorArguments: [config.verifier, config.hasher],
    })

    console.log(happyVerifiedMessage('ETHTornadoCloneable', ethImplAddr))
    console.log('\nWaiting 5 seconds.\n')
    await timeout(5)

    await hre.run('verify:verify', {
      address: ERC20ImplAddr,
      constructorArguments: [config.verifier, config.hasher],
    })

    console.log(happyVerifiedMessage('ERC20TornadoCloneable', ERC20ImplAddr))
  } else {
    return '\nDecided to stop at contract verification.\n'
  }
}

main().then((res) => {
  console.log(res ?? '\nScript succesfully finished.\n')
})