116 lines
4.2 KiB
JavaScript
116 lines
4.2 KiB
JavaScript
// We require the Hardhat Runtime Environment explicitly here. This is optional
|
|
// but useful for running the script in a standalone fashion through `node <script>`.
|
|
//
|
|
// When running the script with `hardhat run <script>` you'll find the Hardhat
|
|
// Runtime Environment's members available in the global scope.
|
|
const hre = require('hardhat')
|
|
const { toFixedHex, poseidonHash2 } = require('../src/utils')
|
|
const MerkleTree = require('fixed-merkle-tree')
|
|
const abi = new hre.ethers.utils.AbiCoder()
|
|
const instances = [
|
|
'0x1111000000000000000000000000000000001111',
|
|
'0x2222000000000000000000000000000000002222',
|
|
'0x3333000000000000000000000000000000003333',
|
|
'0x4444000000000000000000000000000000004444',
|
|
]
|
|
|
|
const blocks = ['0xaaaaaaaa', '0xbbbbbbbb', '0xcccccccc', '0xdddddddd']
|
|
const CHUNK_TREE_HEIGHT = 2
|
|
const levels = 20
|
|
|
|
const nonRandomBN = (nonce = 0) =>
|
|
hre.ethers.BigNumber.from('0x004d51bffaafdb3eed0661c1cfd76c8cd6ec1456b80b24bbb855f3a141ebf0be').sub(nonce)
|
|
|
|
async function main() {
|
|
const governance = '0x5efda50f22d34F262c29268506C5Fa42cB56A1Ce'
|
|
const [tornadoProxy] = await hre.ethers.getSigners()
|
|
console.log('deployer/tornadoProxy acc: ', tornadoProxy.address)
|
|
|
|
const tree = new MerkleTree(levels, [], { hashFunction: poseidonHash2 })
|
|
const TornadoTreesV1Mock = await hre.ethers.getContractFactory('TornadoTreesV1Mock')
|
|
const tornadoTreesV1Mock = await TornadoTreesV1Mock.deploy(0, 0, tree.root(), tree.root())
|
|
await tornadoTreesV1Mock.deployed()
|
|
console.log('tornadoTreesV1Mock deployed to:', tornadoTreesV1Mock.address)
|
|
|
|
const BatchTreeUpdateVerifier = await hre.ethers.getContractFactory('BatchTreeUpdateVerifier')
|
|
const verifier = await BatchTreeUpdateVerifier.deploy()
|
|
await verifier.deployed()
|
|
console.log('Verifier deployed to:', verifier.address)
|
|
|
|
const TornadoTrees = await hre.ethers.getContractFactory('TornadoTreesMock')
|
|
const tornadoTrees = await TornadoTrees.deploy(
|
|
governance,
|
|
tornadoProxy.address,
|
|
tornadoTreesV1Mock.address,
|
|
verifier.address,
|
|
{
|
|
depositsFrom: 0,
|
|
depositsStep: 0,
|
|
withdrawalsFrom: 0,
|
|
withdrawalsStep: 0,
|
|
},
|
|
)
|
|
await tornadoTrees.deployed()
|
|
|
|
const notes = []
|
|
const depositEvents = {}
|
|
const withdrawalEvents = {}
|
|
for (let i = 0; i < 2 ** CHUNK_TREE_HEIGHT; i++) {
|
|
const note = {
|
|
instance: instances[i % instances.length],
|
|
depositBlock: blocks[i % blocks.length],
|
|
withdrawalBlock: 2 + i + i * 4 * 60 * 24,
|
|
commitment: nonRandomBN(i),
|
|
nullifierHash: nonRandomBN(i + instances.length),
|
|
}
|
|
|
|
await tornadoTrees.register(
|
|
note.instance,
|
|
toFixedHex(note.commitment),
|
|
toFixedHex(note.nullifierHash),
|
|
note.depositBlock,
|
|
note.withdrawalBlock,
|
|
{ gasLimit: 200000 },
|
|
)
|
|
const encodedData = abi.encode(
|
|
['address', 'bytes32', 'uint256'],
|
|
[note.instance, toFixedHex(note.commitment), note.depositBlock],
|
|
)
|
|
const leaf = hre.ethers.utils.keccak256(encodedData)
|
|
depositEvents[leaf] = {
|
|
hash: toFixedHex(note.commitment),
|
|
instance: toFixedHex(note.instance, 20),
|
|
block: toFixedHex(note.depositBlock, 4),
|
|
}
|
|
const encodedDataW = abi.encode(
|
|
['address', 'bytes32', 'uint256'],
|
|
[note.instance, toFixedHex(note.nullifierHash), note.withdrawalBlock],
|
|
)
|
|
const leafW = hre.ethers.utils.keccak256(encodedDataW)
|
|
withdrawalEvents[leafW] = {
|
|
hash: toFixedHex(note.nullifierHash),
|
|
instance: toFixedHex(note.instance, 20),
|
|
block: toFixedHex(note.withdrawalBlock, 4),
|
|
}
|
|
notes[i] = note
|
|
}
|
|
console.log(`Registered ${notes.length} new deposits and withdrawals in tornadoTreesV1Mock`)
|
|
console.log(JSON.stringify(depositEvents, null, 2))
|
|
console.log(JSON.stringify(withdrawalEvents, null, 2))
|
|
|
|
console.log('tornadoTrees deployed to:', tornadoTrees.address)
|
|
console.log('You can use the same private key to register new deposits in the tornadoTrees')
|
|
|
|
console.log(`\nTORNADO_TREES_V1=${tornadoTreesV1Mock.address}`)
|
|
console.log(`TORNADO_TREES=${tornadoTrees.address}`)
|
|
}
|
|
|
|
// We recommend this pattern to be able to use async/await everywhere
|
|
// and properly handle errors.
|
|
main()
|
|
.then(() => process.exit(0))
|
|
.catch((error) => {
|
|
console.error(error)
|
|
process.exit(1)
|
|
})
|