/*
 * Copyright (c) 2022, OP Labs PBC (MIT License)
 * https://github.com/ethereum-optimism/optimism
 */

import { task, types } from 'hardhat/config'
import { providers, utils, Wallet, Event } from 'ethers'
import dotenv from 'dotenv'
import 'hardhat-deploy'
import '@nomiclabs/hardhat-ethers'
import { DepositTx } from '@eth-optimism/core-utils'

dotenv.config()

const sleep = async (ms: number) => {
  return new Promise((resolve) => setTimeout(resolve, ms))
}

task('deposit', 'Deposits funds onto L2.')
  .addParam(
    'l1ProviderUrl',
    'L1 provider URL.',
    'http://localhost:8545',
    types.string
  )
  .addParam(
    'l2ProviderUrl',
    'L2 provider URL.',
    'http://localhost:9545',
    types.string
  )
  .addParam('to', 'Recipient address.', null, types.string)
  .addParam('amountEth', 'Amount in ETH to send.', null, types.string)
  .addOptionalParam(
    'privateKey',
    'Private key to send transaction',
    process.env.PRIVATE_KEY,
    types.string
  )
  .setAction(async (args, hre) => {
    const { l1ProviderUrl, l2ProviderUrl, to, amountEth, privateKey } = args
    const proxy = await hre.deployments.get('OptimismPortalProxy')

    const OptimismPortal = await hre.ethers.getContractAt(
      'OptimismPortal',
      proxy.address
    )

    const l1Provider = new providers.JsonRpcProvider(l1ProviderUrl)
    const l2Provider = new providers.JsonRpcProvider(l2ProviderUrl)

    let l1Wallet: Wallet | providers.JsonRpcSigner
    if (privateKey) {
      l1Wallet = new Wallet(privateKey, l1Provider)
    } else {
      l1Wallet = l1Provider.getSigner()
    }

    const from = await l1Wallet.getAddress()
    console.log(`Sending from ${from}`)
    const balance = await l1Wallet.getBalance()
    if (balance.eq(0)) {
      throw new Error(`${from} has no balance`)
    }

    const amountWei = utils.parseEther(amountEth)
    const value = amountWei.add(utils.parseEther('0.01'))
    console.log(`Depositing ${amountEth} ETH to ${to}`)

    const preL2Balance = await l2Provider.getBalance(to)
    console.log(`${to} has ${utils.formatEther(preL2Balance)} ETH on L2`)

    // Below adds 0.01 ETH to account for gas.
    const tx = await OptimismPortal.depositTransaction(
      to,
      amountWei,
      '3000000',
      false,
      [],
      { value }
    )
    console.log(`Got TX hash ${tx.hash}. Waiting...`)
    const receipt = await tx.wait()
    console.log(`Included in block ${receipt.blockHash}`)

    // find the transaction deposited event and derive
    // the deposit transaction from it
    const event = receipt.events.find(
      (e: Event) => e.event === 'TransactionDeposited'
    )
    console.log(`Deposit has log index ${event.logIndex}`)
    const l2tx = DepositTx.fromL1Event(event)
    const hash = l2tx.hash()
    console.log(`Waiting for L2 TX hash ${hash}`)

    let i = 0
    while (true) {
      const expected = await l2Provider.send('eth_getTransactionByHash', [hash])
      if (expected) {
        console.log('Deposit success')
        console.log(JSON.stringify(expected, null, 2))
        break
      }
      const postL2Balance = await l2Provider.getBalance(to)
      if (postL2Balance.gt(preL2Balance)) {
        console.log(
          `Unexpected balance increase without detecting deposit transaction`
        )
      }

      if (i % 100 === 0) {
        const block = await l2Provider.getBlock('latest')
        console.log(`latest block ${block.number}:${block.hash}`)
      }

      await sleep(500)
      i++
    }
  })
