-
Notifications
You must be signed in to change notification settings - Fork 52
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Script for automating SpokePool deposits
This is useful for making deposits to the zkSync SpokePool(s) because zkSync's block explorer doesn't yet support making transactions via proxy contracts.
- Loading branch information
Showing
1 changed file
with
183 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,183 @@ | ||
import * as contracts from "@across-protocol/contracts-v2"; | ||
import { ethers, Contract, Wallet } from "ethers"; | ||
import minimist from "minimist"; | ||
import { config } from "dotenv"; | ||
import { getDeployedContract, getNodeUrlList } from "../src/utils"; | ||
|
||
type ERC20 = { | ||
address: string; | ||
decimals: number; | ||
symbol: string; | ||
}; | ||
|
||
const testChains = [5, 280]; | ||
const chains = [1, 10, 137, 324, 42161]; | ||
|
||
const padding = 20; | ||
|
||
function getTokenAddress(symbol: string, chainId: number): ERC20 { | ||
const token = contracts.TOKEN_SYMBOLS_MAP[symbol]; | ||
if (token === undefined) { | ||
throw new Error(`Token ${symbol} unrecognised`); | ||
} | ||
|
||
const { addresses, decimals } = token; | ||
const address = addresses[chainId]; | ||
if (address === undefined) { | ||
throw new Error(`Token ${symbol} not supported on chain ID ${chainId}`); | ||
} | ||
|
||
return { address: addresses[chainId], decimals, symbol }; | ||
} | ||
|
||
async function getHubPoolContract(chainId: number): Promise<Contract> { | ||
const contractName = "HubPool"; | ||
const hubPoolChainId = chains.includes(chainId) ? 1 : 5; | ||
if (!testChains.includes(chainId)) { | ||
throw new Error(`Unsupported chainId: ${chainId}`); | ||
} | ||
|
||
const hubPool = getDeployedContract(contractName, hubPoolChainId); | ||
const provider = new ethers.providers.StaticJsonRpcProvider(getNodeUrlList(hubPoolChainId, 1)[0]); | ||
return hubPool.connect(provider); | ||
} | ||
|
||
async function getSpokePoolContract(chainId: number): Promise<Contract> { | ||
const hubPool = await getHubPoolContract(chainId); | ||
const spokePoolAddr = (await hubPool.crossChainContracts(chainId))[1]; | ||
|
||
const contract = new Contract(spokePoolAddr, contracts.SpokePool__factory.abi); | ||
return contract; | ||
} | ||
|
||
async function deposit(args: Record<string, number | string>, signer: Wallet): Promise<boolean> { | ||
const depositor = await signer.getAddress(); | ||
const fromChainId = Number(args.from); | ||
const toChainId = Number(args.to); | ||
const recipient = args.recipient ?? depositor; | ||
const baseAmount = Number(args.amount); | ||
const tokenSymbol = (args.token as string).toUpperCase(); | ||
|
||
const provider = new ethers.providers.StaticJsonRpcProvider(getNodeUrlList(fromChainId, 1)[0]); | ||
signer = signer.connect(provider); | ||
const spokePool = (await getSpokePoolContract(fromChainId)).connect(signer); | ||
|
||
const token = getTokenAddress(tokenSymbol, fromChainId); | ||
const amount = ethers.utils.parseUnits(baseAmount.toString(), token.decimals); | ||
|
||
const erc20 = new Contract(token.address, contracts.ExpandedERC20__factory.abi, signer); | ||
const allowance = await erc20.allowance(depositor, spokePool.address); | ||
if (amount.gt(allowance)) { | ||
const approvalAmount = amount.mul(5); | ||
const approval = await erc20.approve(spokePool.address, approvalAmount); | ||
console.log(`Approving SpokePool for ${approvalAmount} ${tokenSymbol}: ${approval.hash}.`); | ||
await approval.wait(); | ||
console.log("Approval complete..."); | ||
} | ||
|
||
const relayerFeePct = ethers.constants.Zero; | ||
const quoteTimestamp = Math.round(Date.now() / 1000); | ||
const maxCount = ethers.constants.MaxUint256; | ||
|
||
console.log( | ||
`Submitting deposit on chain ID ${fromChainId}\n` + | ||
`\t${"originChainId".padEnd(padding)}: ${fromChainId}\n` + | ||
`\t${"destinationChainId".padEnd(padding)}: ${toChainId}\n` + | ||
`\t${"depositor".padEnd(padding)}: ${depositor}\n` + | ||
`\t${"recipient".padEnd(padding)}: ${recipient}\n` + | ||
`\t${"relayerFeePct".padEnd(padding)}: ${relayerFeePct}\n` + | ||
`\t${"quoteTimestamp".padEnd(padding)}: ${quoteTimestamp}\n` | ||
); | ||
const deposit = await spokePool.deposit( | ||
recipient, | ||
token.address, | ||
amount, | ||
toChainId, | ||
relayerFeePct, | ||
quoteTimestamp, | ||
"0x", | ||
maxCount | ||
); | ||
console.log("Submitted deposit:"); | ||
const receipt = await deposit.wait(); | ||
receipt.logs | ||
.filter((log) => log.address === spokePool.address) | ||
.forEach((_log) => { | ||
const log = spokePool.interface.parseLog(_log); | ||
console.log( | ||
`\t${"depositId".padEnd(padding)}: ${log.args.depositId}\n` + | ||
`\t${"transactionHash".padEnd(padding)}: ${_log.transactionHash}\n` | ||
); | ||
}); | ||
|
||
return true; | ||
} | ||
|
||
export function usage(badInput?: string): boolean { | ||
let usageStr = badInput ? `\nUnrecognized input: "${badInput}".\n\n` : ""; | ||
const walletOpts = "mnemonic|privateKey"; | ||
const depositArgs = | ||
"--origin <originChainId> --destination <destinationChainId>" + " --token <tokenSymbol> --amount <amount>"; | ||
const fillArgs = "--origin <originChainId> --hash <depositHash>"; | ||
|
||
const pad = "deposit".length; | ||
usageStr += ` | ||
Usage: | ||
\tyarn ts-node ./scripts/spokepool --wallet <${walletOpts}> ${"deposit".padEnd(pad)} ${depositArgs} | ||
\tyarn ts-node ./scripts/spokepool --wallet <${walletOpts}> ${"fill".padEnd(pad)} ${fillArgs} | ||
`.slice(1); // Skip leading newline | ||
console.log(usageStr); | ||
|
||
// eslint-disable-next-line no-process-exit | ||
process.exit(badInput === undefined ? 0 : 9); | ||
// not reached | ||
} | ||
|
||
async function run(argv: string[]): Promise<boolean> { | ||
const depositOpts = ["from", "to", "token", "amount"]; | ||
const opts = { | ||
string: ["wallet", ...depositOpts], | ||
default: { | ||
wallet: "mnemonic", | ||
}, | ||
unknown: usage, | ||
}; | ||
const args = minimist(argv.slice(1), opts); | ||
|
||
config(); | ||
|
||
let signer: Wallet; | ||
switch (args.wallet) { | ||
case "mnemonic": | ||
signer = Wallet.fromMnemonic(process.env.MNEMONIC); | ||
break; | ||
case "privateKey": | ||
signer = new Wallet(process.env.PRIVATE_KEY); | ||
break; | ||
default: | ||
usage(args.wallet); // no return | ||
} | ||
["MNEMONIC", "PRIVATE_KEY"].forEach((envVar) => (process.env[envVar] = "")); | ||
|
||
if (argv[0]) { | ||
return await deposit(args, signer); | ||
} else if (args.fill) { | ||
// @todo Not supported yet... | ||
usage(); // no return | ||
} else { | ||
usage(); // no return | ||
} | ||
} | ||
|
||
if (require.main === module) { | ||
run(process.argv.slice(2)) | ||
.then(async () => { | ||
// eslint-disable-next-line no-process-exit | ||
process.exit(0); | ||
}) | ||
.catch(async (error) => { | ||
console.error("Process exited with", error); | ||
// eslint-disable-next-line no-process-exit | ||
process.exit(1); | ||
}); | ||
} |