@aztec/bot 0.0.1-commit.d3ec352c → 0.0.1-commit.e3c1de76

package/dest/factory.js

@@ -1,13 +1,16 @@
 import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
 import { getInitialTestAccountsData } from '@aztec/accounts/testing';
 import { AztecAddress } from '@aztec/aztec.js/addresses';
-import { BatchCall } from '@aztec/aztec.js/contracts';
+import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
 import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
 import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
+import { deriveKeys } from '@aztec/aztec.js/keys';
 import { createLogger } from '@aztec/aztec.js/log';
 import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
-import { createEthereumChain, createExtendedL1Client } from '@aztec/ethereum';
-import { Fr } from '@aztec/foundation/fields';
+import { waitForTx } from '@aztec/aztec.js/node';
+import { createEthereumChain } from '@aztec/ethereum/chain';
+import { createExtendedL1Client } from '@aztec/ethereum/client';
+import { Fr } from '@aztec/foundation/curves/bn254';
 import { Timer } from '@aztec/foundation/timer';
 import { AMMContract } from '@aztec/noir-contracts.js/AMM';
 import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
@@ -37,8 +40,8 @@ export class BotFactory {
    * Initializes a new bot by setting up the sender account, registering the recipient,
    * deploying the token contract, and minting tokens if necessary.
    */ async setup() {
-        const recipient = (await this.wallet.createAccount()).address;
         const defaultAccountAddress = await this.setupAccount();
+        const recipient = (await this.wallet.createAccount()).address;
         const token = await this.setupToken(defaultAccountAddress);
         await this.mintTokens(token, defaultAccountAddress);
         return {
@@ -88,8 +91,8 @@ export class BotFactory {
             contract: new SchnorrAccountContract(signingKey)
         };
         const accountManager = await this.wallet.createAccount(accountData);
-        const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
-        if (isInit) {
+        const metadata = await this.wallet.getContractMetadata(accountManager.address);
+        if (metadata.isContractInitialized) {
             this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
             const timer = new Timer();
             const address = accountManager.address;
@@ -102,22 +105,24 @@ export class BotFactory {
             const claim = await this.getOrCreateBridgeClaim(address);
             const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
             const deployMethod = await accountManager.getDeployMethod();
-            const maxFeesPerGas = (await this.aztecNode.getCurrentBaseFees()).mul(1 + this.config.baseFeePadding);
+            const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
             const gasSettings = GasSettings.default({
                 maxFeesPerGas
             });
-            const sentTx = deployMethod.send({
-                from: AztecAddress.ZERO,
-                fee: {
-                    gasSettings,
-                    paymentMethod
-                }
-            });
-            const txHash = await sentTx.getTxHash();
-            this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
-            await this.withNoMinTxsPerBlock(()=>sentTx.wait({
+            await this.withNoMinTxsPerBlock(async ()=>{
+                const txHash = await deployMethod.send({
+                    from: AztecAddress.ZERO,
+                    fee: {
+                        gasSettings,
+                        paymentMethod
+                    },
+                    wait: NO_WAIT
+                });
+                this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
+                return waitForTx(this.aztecNode, txHash, {
                     timeout: this.config.txMinedWaitSeconds
-                }));
+                });
+            });
             this.log.info(`Account deployed at ${address}`);
             // Clean up the consumed bridge claim
             await this.store.deleteBridgeClaim(address);
@@ -140,47 +145,66 @@ export class BotFactory {
    * @returns The TokenContract instance.
    */ async setupToken(sender) {
         let deploy;
+        let tokenInstance;
         const deployOpts = {
             from: sender,
             contractAddressSalt: this.config.tokenSalt,
             universalDeploy: true
         };
+        let token;
         if (this.config.contract === SupportedTokenContracts.TokenContract) {
             deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
+            tokenInstance = await deploy.getInstance(deployOpts);
+            token = TokenContract.at(tokenInstance.address, this.wallet);
         } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
-            deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender);
+            // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
+            const tokenSecretKey = Fr.random();
+            const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
+            deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
             deployOpts.skipInstancePublication = true;
             deployOpts.skipClassPublication = true;
             deployOpts.skipInitialization = false;
+            // Register the contract with the secret key before deployment
+            tokenInstance = await deploy.getInstance(deployOpts);
+            token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
+            await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
         } else {
             throw new Error(`Unsupported token contract type: ${this.config.contract}`);
         }
-        const address = (await deploy.getInstance(deployOpts)).address;
-        if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
+        const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
+        const metadata = await this.wallet.getContractMetadata(address);
+        if (metadata.isContractPublished) {
             this.log.info(`Token at ${address.toString()} already deployed`);
-            return deploy.register();
+            await deploy.register();
         } else {
             this.log.info(`Deploying token contract at ${address.toString()}`);
-            const sentTx = deploy.send(deployOpts);
-            const txHash = await sentTx.getTxHash();
+            const txHash = await deploy.send({
+                ...deployOpts,
+                wait: NO_WAIT
+            });
             this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
-            return this.withNoMinTxsPerBlock(()=>sentTx.deployed({
+            await this.withNoMinTxsPerBlock(async ()=>{
+                await waitForTx(this.aztecNode, txHash, {
                     timeout: this.config.txMinedWaitSeconds
-                }));
+                });
+                return token;
+            });
         }
+        return token;
     }
     /**
    * Checks if the token contract is deployed and deploys it if necessary.
    * @param wallet - Wallet to deploy the token contract from.
    * @returns The TokenContract instance.
-   */ setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals = 18) {
+   */ async setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals = 18) {
         const deployOpts = {
             from: deployer,
             contractAddressSalt,
             universalDeploy: true
         };
         const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
-        return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
+        const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
+        return TokenContract.at(instance.address, this.wallet);
     }
     async setupAmmContract(deployer, contractAddressSalt, token0, token1, lpToken) {
         const deployOpts = {
@@ -189,15 +213,16 @@ export class BotFactory {
             universalDeploy: true
         };
         const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
-        const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
+        const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
+        const amm = AMMContract.at(instance.address, this.wallet);
         this.log.info(`AMM deployed at ${amm.address}`);
-        const minterTx = lpToken.methods.set_minter(amm.address, true).send({
-            from: deployer
-        });
-        this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
-        await minterTx.wait({
-            timeout: this.config.txMinedWaitSeconds
+        const minterReceipt = await lpToken.methods.set_minter(amm.address, true).send({
+            from: deployer,
+            wait: {
+                timeout: this.config.txMinedWaitSeconds
+            }
         });
+        this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
         this.log.info(`Liquidity token initialized`);
         return amm;
     }
@@ -230,45 +255,52 @@ export class BotFactory {
             caller: amm.address,
             call: await token1.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount1Max, authwitNonce).getFunctionCall()
         });
-        const mintTx = new BatchCall(this.wallet, [
+        const mintReceipt = await new BatchCall(this.wallet, [
             token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
             token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE)
         ]).send({
-            from: liquidityProvider
-        });
-        this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
-        await mintTx.wait({
-            timeout: this.config.txMinedWaitSeconds
+            from: liquidityProvider,
+            wait: {
+                timeout: this.config.txMinedWaitSeconds
+            }
         });
-        const addLiquidityTx = amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce).send({
+        this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
+        const addLiquidityReceipt = await amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce).send({
             from: liquidityProvider,
             authWitnesses: [
                 token0Authwit,
                 token1Authwit
-            ]
-        });
-        this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
-        await addLiquidityTx.wait({
-            timeout: this.config.txMinedWaitSeconds
+            ],
+            wait: {
+                timeout: this.config.txMinedWaitSeconds
+            }
         });
+        this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
         this.log.info(`Liquidity added`);
         const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
         this.log.info(`Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${newT0Bal}, token1=${newT1Bal}, lp=${newLPBal}`);
     }
     async registerOrDeployContract(name, deploy, deployOpts) {
-        const address = (await deploy.getInstance(deployOpts)).address;
-        if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
+        const instance = await deploy.getInstance(deployOpts);
+        const address = instance.address;
+        const metadata = await this.wallet.getContractMetadata(address);
+        if (metadata.isContractPublished) {
             this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
-            return deploy.register();
+            await deploy.register();
         } else {
             this.log.info(`Deploying contract ${name} at ${address.toString()}`);
-            const sentTx = deploy.send(deployOpts);
-            const txHash = await sentTx.getTxHash();
-            this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
-            return this.withNoMinTxsPerBlock(()=>sentTx.deployed({
+            await this.withNoMinTxsPerBlock(async ()=>{
+                const txHash = await deploy.send({
+                    ...deployOpts,
+                    wait: NO_WAIT
+                });
+                this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
+                return waitForTx(this.aztecNode, txHash, {
                     timeout: this.config.txMinedWaitSeconds
-                }));
+                });
+            });
         }
+        return instance;
     }
     /**
    * Mints private and public tokens for the sender if their balance is below the minimum.
@@ -295,14 +327,16 @@ export class BotFactory {
             this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
             return;
         }
-        const sentTx = new BatchCall(token.wallet, calls).send({
-            from: minter
-        });
-        const txHash = await sentTx.getTxHash();
-        this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
-        await this.withNoMinTxsPerBlock(()=>sentTx.wait({
+        await this.withNoMinTxsPerBlock(async ()=>{
+            const txHash = await new BatchCall(token.wallet, calls).send({
+                from: minter,
+                wait: NO_WAIT
+            });
+            this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
+            return waitForTx(this.aztecNode, txHash, {
                 timeout: this.config.txMinedWaitSeconds
-            }));
+            });
+        });
     }
     /**
    * Gets or creates a bridge claim for the recipient.