import os, ospaths, strutils, chronicles, chronos, blscurve, nimcrypto, json_serialization, serialization, web3, stint, spec/[datatypes, digest, crypto], conf, ssz, interop contract(DepositContract): proc deposit(pubkey: Bytes48, withdrawalCredentials: Bytes32, signature: Bytes96, deposit_data_root: FixedBytes[32]) proc writeTextFile(filename: string, contents: string) = writeFile(filename, contents) # echo "Wrote ", filename proc writeFile(filename: string, value: auto) = Json.saveFile(filename, value, pretty = true) # echo "Wrote ", filename proc ethToWei(eth: UInt256): UInt256 = eth * 1000000000000000000.u256 proc generateDeposits*(totalValidators: int, outputDir: string, randomKeys: bool, firstIdx = 0): seq[Deposit] = info "Generating deposits", totalValidators, outputDir, randomKeys for i in 0 ..< totalValidators: let v = validatorFileBaseName(firstIdx + i) depositFn = outputDir / v & ".deposit.json" privKeyFn = outputDir / v & ".privkey" if existsFile(depositFn) and existsFile(privKeyFn): try: result.add Json.loadFile(depositFn, Deposit) continue except SerializationError as err: debug "Rewriting unreadable deposit", err = err.formatMsg(depositFn) discard let privKey = if randomKeys: ValidatorPrivKey.random else: makeInteropPrivKey(i) pubKey = privKey.pubKey() let dp = makeDeposit(pubKey, privKey) writeTextFile(privKeyFn, $privKey) writeFile(depositFn, dp) result.add(dp) proc sendDeposits*( deposits: seq[Deposit], depositWeb3Url, depositContractAddress: string) {.async.} = let web3 = await newWeb3(depositWeb3Url) contractAddress = Address.fromHex(depositContractAddress) eth1Addresses = await web3.provider.eth_accounts() for i, dp in deposits: web3.defaultAccount = eth1Addresses[i] let depositContract = web3.contractSender(DepositContract, contractAddress) let tx = await depositContract.deposit( Bytes48(dp.data.pubKey.getBytes()), Bytes32(dp.data.withdrawal_credentials.data), Bytes96(dp.data.signature.getBytes()), FixedBytes[32](hash_tree_root(dp.data).data)).send(value = 32.u256.ethToWei) when isMainModule: import confutils cli do (totalValidators: int = 125000, outputDir: string = "validators", randomKeys: bool = false, depositWeb3Url: string = "", depositContractAddress: string = ""): let deposits = generateDeposits(totalValidators, outputDir, randomKeys) if depositWeb3Url.len() > 0 and depositContractAddress.len() > 0: echo "Sending deposits to eth1..." waitFor sendDeposits(deposits, depositWeb3Url, depositContractAddress) echo "Done"