require('dotenv').config()

const config = require('../config')

const hre = require('hardhat')

const { ethers } = hre

const { createInterface } = require('readline')

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

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')
  }
}

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

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

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

function verifiedMessage(name, address) {
  return `\n${name} @ ${address} 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 deploy() {
  if (process.env.STAKING_PROXY_ADDRESS === '') throw Error('Missing STAKING_PROXY_ADDRESS.')

  const signer = await ethers.getSigner()

  const _network = await ethers.provider.getNetwork()

  const stakingProxyAddress = `${process.env.STAKING_PROXY_ADDRESS}`

  const registryImplementationDeployer = (await ethers.getContractFactory('RelayerRegistry')).connect(signer)

  let registry

  if (await prompt(promptMessageBase('Continuing to RelayerRegistry (Implementation) deployment.'))) {
    registry = await registryImplementationDeployer.deploy(
      config.TORN,
      config.governance,
      config.ens,
      stakingProxyAddress,
      config.feeManager,
    )
    console.log(deployedMessage('RelayerRegistry (Implementation)', _network.chainId, registry.address))
  } else {
    return '\nDecided to stop at RelayerRegistry (Implementation) deployment.\n'
  }

  if (await prompt(promptMessageBase('Continuing to contract verification.'))) {
    console.log('\nWaiting 20 seconds.\n')
    await timeout(20)

    await hre.run('verify:verify', {
      address: registry.address,
      constructorArguments: [
        config.TORN,
        config.governance,
        config.ens,
        stakingProxyAddress,
        config.feeManager,
      ],
    })
    console.log(verifiedMessage('RelayerRegistry (Implementation)', registry.address))
  } else {
    return '\nDecided to stop at contract verification.\n'
  }
}

deploy()