CLI support for Arbitrum bridge + L2 rewards distribution

arbitrum-addresses.json

@@ -0,0 +1,42 @@
+{
+  "source": "https://github.com/OffchainLabs/arbitrum/tree/f54baf10871ee86aedca4880796342ef9bd0b0ab/packages/",
+  "1": {
+    "L1GatewayRouter": {
+      "address": "0x72Ce9c846789fdB6fC1f34aC4AD25Dd9ef7031ef"
+    },
+    "IInbox": {
+      "address": "0x4Dbd4fc535Ac27206064B68FfCf827b0A60BAB3f"
+    }
+  },
+  "4": {
+    "L1GatewayRouter": {
+      "address": "0x70C143928eCfFaf9F5b406f7f4fC28Dc43d68380"
+    },
+    "IInbox": {
+      "address": "0x578BAde599406A8fE3d24Fd7f7211c0911F5B29e"
+    }
+  },
+  "5": {
+    "L1GatewayRouter": {
+      "address": "0x8BDFa67ace22cE2BFb2fFebe72f0c91CDA694d4b"
+    },
+    "IInbox": {
+      "address": "0x1FdBBcC914e84aF593884bf8e8Dd6877c29035A2"
+    }
+  },
+  "42161": {
+    "L2GatewayRouter": {
+      "address": "0x5288c571Fd7aD117beA99bF60FE0846C4E84F933"
+    }
+  },
+  "421611": {
+    "L2GatewayRouter": {
+      "address": "0x9413AD42910c1eA60c737dB5f58d1C504498a3cD"
+    }
+  },
+  "421612": {
+    "L2GatewayRouter": {
+      "address": "0xC502Ded1EE1d616B43F7f20Ebde83Be1A275ca3c"
+    }
+  }
+}

cli/address-book.ts

@@ -28,7 +28,7 @@ export interface AddressBook {
 }
 
 export const getAddressBook = (path: string, chainId: string): AddressBook => {
-  if (!path) throw new Error(`A path the the address book file is required.`)
+  if (!path) throw new Error(`A path to the address book file is required.`)
   if (!chainId) throw new Error(`A chainId is required.`)
 
   const addressBook = JSON.parse(fs.readFileSync(path, 'utf8') || '{}') as AddressBookJson

cli/cli.ts

@@ -8,6 +8,7 @@ import { proxyCommand } from './commands/proxy'
 import { protocolCommand } from './commands/protocol'
 import { contractsCommand } from './commands/contracts'
 import { airdropCommand } from './commands/airdrop'
+import { bridgeCommand } from './commands/bridge'
 
 import { cliOpts } from './defaults'
 
@@ -27,11 +28,13 @@ yargs
   .option('m', cliOpts.mnemonic)
   .option('p', cliOpts.providerUrl)
   .option('n', cliOpts.accountNumber)
+  .option('r', cliOpts.arbitrumAddressBook)
   .command(deployCommand)
   .command(migrateCommand)
   .command(proxyCommand)
   .command(protocolCommand)
   .command(contractsCommand)
   .command(airdropCommand)
+  .command(bridgeCommand)
   .demandCommand(1, 'Choose a command from the above list')
   .help().argv

cli/commands/bridge/index.ts

@@ -0,0 +1,22 @@
+import yargs, { Argv } from 'yargs'
+
+import { redeemSendToL2Command, sendToL2Command } from './to-l2'
+import { startSendToL1Command, finishSendToL1Command, waitFinishSendToL1Command } from './to-l1'
+import { cliOpts } from '../../defaults'
+
+export const bridgeCommand = {
+  command: 'bridge',
+  describe: 'Graph token bridge actions.',
+  builder: (yargs: Argv): yargs.Argv => {
+    return yargs
+      .option('-l', cliOpts.l2ProviderUrl)
+      .command(sendToL2Command)
+      .command(redeemSendToL2Command)
+      .command(startSendToL1Command)
+      .command(finishSendToL1Command)
+      .command(waitFinishSendToL1Command)
+  },
+  handler: (): void => {
+    yargs.showHelp()
+  },
+}

cli/commands/bridge/to-l1.ts

@@ -0,0 +1,217 @@
+import { loadEnv, CLIArgs, CLIEnvironment } from '../../env'
+import { logger } from '../../logging'
+import { getAddressBook } from '../../address-book'
+import { getProvider, sendTransaction, toGRT } from '../../network'
+import { chainIdIsL2 } from '../../utils'
+import { loadAddressBookContract } from '../../contracts'
+import {
+  L2TransactionReceipt,
+  getL2Network,
+  L2ToL1MessageStatus,
+  L2ToL1MessageWriter,
+} from '@arbitrum/sdk'
+import { L2GraphTokenGateway } from '../../../build/types/L2GraphTokenGateway'
+import { BigNumber } from 'ethers'
+import { JsonRpcProvider } from '@ethersproject/providers'
+
+const FOURTEEN_DAYS_IN_SECONDS = 24 * 3600 * 14
+
+const BLOCK_SEARCH_THRESHOLD = 6 * 3600
+const searchForArbBlockByTimestamp = async (
+  l2Provider: JsonRpcProvider,
+  timestamp: number,
+): Promise<number> => {
+  let step = 131072
+  let block = await l2Provider.getBlock('latest')
+  while (block.timestamp > timestamp) {
+    while (block.number - step < 0) {
+      step = Math.round(step / 2)
+    }
+    block = await l2Provider.getBlock(block.number - step)
+  }
+  while (step > 1 && Math.abs(block.timestamp - timestamp) > BLOCK_SEARCH_THRESHOLD) {
+    step = Math.round(step / 2)
+    if (block.timestamp - timestamp > 0) {
+      block = await l2Provider.getBlock(block.number - step)
+    } else {
+      block = await l2Provider.getBlock(block.number + step)
+    }
+  }
+  return block.number
+}
+
+const wait = (ms: number): Promise<void> => {
+  return new Promise((res) => setTimeout(res, ms))
+}
+
+const waitUntilOutboxEntryCreatedWithCb = async (
+  msg: L2ToL1MessageWriter,
+  retryDelay: number,
+  callback: () => void,
+) => {
+  let done = false
+  while (!done) {
+    const status = await msg.status(null)
+    if (status == L2ToL1MessageStatus.CONFIRMED) {
+      done = true
+    } else {
+      callback()
+      await wait(retryDelay)
+    }
+  }
+}
+
+export const startSendToL1 = async (cli: CLIEnvironment, cliArgs: CLIArgs): Promise<void> => {
+  logger.info(`>>> Sending tokens to L1 <<<\n`)
+  const l2Provider = getProvider(cliArgs.l2ProviderUrl)
+  const l2ChainId = (await l2Provider.getNetwork()).chainId
+  if (chainIdIsL2(cli.chainId) || !chainIdIsL2(l2ChainId)) {
+    throw new Error(
+      'Please use an L1 provider in --provider-url, and an L2 provider in --l2-provider-url',
+    )
+  }
+
+  const l1GRT = cli.contracts['GraphToken']
+  const l1GRTAddress = l1GRT.address
+  const amount = toGRT(cliArgs.amount)
+  const recipient = cliArgs.recipient ? cliArgs.recipient : cli.wallet.address
+  const l2Wallet = cli.wallet.connect(l2Provider)
+  const l2AddressBook = getAddressBook(cliArgs.addressBook, l2ChainId.toString())
+
+  const gateway = loadAddressBookContract('L2GraphTokenGateway', l2AddressBook, l2Wallet)
+  const l2GRT = loadAddressBookContract('L2GraphToken', l2AddressBook, l2Wallet)
+
+  const l1Gateway = cli.contracts['L1GraphTokenGateway']
+  logger.info(`Will send ${cliArgs.amount} GRT to ${recipient}`)
+  logger.info(`Using L2 gateway ${gateway.address} and L1 gateway ${l1Gateway.address}`)
+
+  const params = [l1GRTAddress, recipient, amount, '0x']
+  logger.info('Approving token transfer')
+  await sendTransaction(l2Wallet, l2GRT, 'approve', [gateway.address, amount])
+  logger.info('Sending outbound transfer transaction')
+  const receipt = await sendTransaction(
+    l2Wallet,
+    gateway,
+    'outboundTransfer(address,address,uint256,bytes)',
+    params,
+  )
+  const l2Receipt = new L2TransactionReceipt(receipt)
+  const l2ToL1Message = (
+    await l2Receipt.getL2ToL1Messages(cli.wallet, await getL2Network(l2Provider))
+  )[0]
+
+  logger.info(
+    `The transaction generated an outbox message with batch number ${l2ToL1Message.batchNumber}`,
+  )
+  logger.info(`and index in batch ${l2ToL1Message.indexInBatch}.`)
+  logger.info(
+    `After the dispute period is finalized (in ~1 week), you can finalize this by calling`,
+  )
+  logger.info(`finish-send-to-l1 with the following txhash:`)
+  logger.info(l2Receipt.transactionHash)
+}
+
+export const finishSendToL1 = async (
+  cli: CLIEnvironment,
+  cliArgs: CLIArgs,
+  wait: boolean,
+): Promise<void> => {
+  logger.info(`>>> Finishing transaction sending tokens to L1 <<<\n`)
+  const l2Provider = getProvider(cliArgs.l2ProviderUrl)
+  const l2ChainId = (await l2Provider.getNetwork()).chainId
+  if (chainIdIsL2(cli.chainId) || !chainIdIsL2(l2ChainId)) {
+    throw new Error(
+      'Please use an L1 provider in --provider-url, and an L2 provider in --l2-provider-url',
+    )
+  }
+
+  const l2AddressBook = getAddressBook(cliArgs.addressBook, l2ChainId.toString())
+
+  const gateway = loadAddressBookContract(
+    'L2GraphTokenGateway',
+    l2AddressBook,
+    l2Provider,
+  ) as L2GraphTokenGateway
+  let txHash: string
+  if (cliArgs.txHash) {
+    txHash = cliArgs.txHash
+  } else {
+    logger.info(
+      `Looking for withdrawals initiated by ${cli.wallet.address} in roughly the last 14 days`,
+    )
+    const fromBlock = await searchForArbBlockByTimestamp(
+      l2Provider,
+      Math.round(Date.now() / 1000) - FOURTEEN_DAYS_IN_SECONDS,
+    )
+    const filt = gateway.filters.WithdrawalInitiated(null, cli.wallet.address)
+    const allEvents = await gateway.queryFilter(filt, BigNumber.from(fromBlock).toHexString())
+    if (allEvents.length == 0) {
+      throw new Error('No withdrawals found')
+    }
+    txHash = allEvents[allEvents.length - 1].transactionHash
+  }
+  logger.info(`Getting receipt from transaction ${txHash}`)
+  const receipt = await l2Provider.getTransactionReceipt(txHash)
+
+  const l2Receipt = new L2TransactionReceipt(receipt)
+  logger.info(`Getting L2 to L1 message...`)
+  const l2ToL1Message = (
+    await l2Receipt.getL2ToL1Messages(cli.wallet, await getL2Network(l2Provider))
+  )[0]
+
+  if (wait) {
+    const retryDelayMs = cliArgs.retryDelaySeconds ? cliArgs.retryDelaySeconds * 1000 : 60000
+    logger.info('Waiting for outbox entry to be created, this can take a full week...')
+    await waitUntilOutboxEntryCreatedWithCb(l2ToL1Message, retryDelayMs, () => {
+      logger.info('Still waiting...')
+    })
+  } else {
+    logger.info('Checking if L2 to L1 message is confirmed...')
+    const status = await l2ToL1Message.status(null)
+    if (status != L2ToL1MessageStatus.CONFIRMED) {
+      throw new Error(
+        `Transaction is not confirmed, status is ${status} when it should be ${L2ToL1MessageStatus.CONFIRMED}. Has the dispute period passed?`,
+      )
+    }
+  }
+  logger.info('Getting proof to execute message')
+  const proofInfo = await l2ToL1Message.tryGetProof(l2Provider)
+
+  if (await l2ToL1Message.hasExecuted(proofInfo)) {
+    throw new Error('Message already executed!')
+  }
+
+  logger.info('Executing outbox transaction')
+  const tx = await l2ToL1Message.execute(proofInfo)
+  const outboxExecuteReceipt = await tx.wait()
+  logger.info('Transaction succeeded! tx hash:')
+  logger.info(outboxExecuteReceipt.transactionHash)
+}
+
+export const startSendToL1Command = {
+  command: 'start-send-to-l1 <amount> [recipient]',
+  describe: 'Start an L2-to-L1 Graph Token transaction',
+  handler: async (argv: CLIArgs): Promise<void> => {
+    return startSendToL1(await loadEnv(argv), argv)
+  },
+}
+
+export const finishSendToL1Command = {
+  command: 'finish-send-to-l1 [txHash]',
+  describe:
+    'Finish an L2-to-L1 Graph Token transaction. L2 dispute period must have completed. ' +
+    'If txHash is not specified, the last withdrawal from the main account in the past 14 days will be redeemed.',
+  handler: async (argv: CLIArgs): Promise<void> => {
+    return finishSendToL1(await loadEnv(argv), argv, false)
+  },
+}
+
+export const waitFinishSendToL1Command = {
+  command: 'wait-finish-send-to-l1 [txHash] [retryDelaySeconds]',
+  describe:
+    "Wait for an L2-to-L1 Graph Token transaction's dispute period to complete (which takes about a week), and then finalize it. " +
+    'If txHash is not specified, the last withdrawal from the main account in the past 14 days will be redeemed.',
+  handler: async (argv: CLIArgs): Promise<void> => {
+    return finishSendToL1(await loadEnv(argv), argv, true)
+  },
+}