@aztec/bot 0.0.0-test.1 → 0.0.1-commit.001888fc

package/dest/factory.js

@@ -1,229 +1,470 @@
-import { getSchnorrAccount } from '@aztec/accounts/schnorr';
-import { getDeployedTestAccountsWallets, getInitialTestAccounts } from '@aztec/accounts/testing';
-import { BatchCall, FeeJuicePaymentMethodWithClaim, L1FeeJuicePortalManager, createLogger, createPXEClient, retryUntil } from '@aztec/aztec.js';
-import { createEthereumChain, createL1Clients } from '@aztec/ethereum';
-import { Fr } from '@aztec/foundation/fields';
-import { EasyPrivateTokenContract } from '@aztec/noir-contracts.js/EasyPrivateToken';
+import { getInitialTestAccountsData } from '@aztec/accounts/testing';
+import { AztecAddress } from '@aztec/aztec.js/addresses';
+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 { waitForTx } from '@aztec/aztec.js/node';
+import { createEthereumChain } from '@aztec/ethereum/chain';
+import { createExtendedL1Client } from '@aztec/ethereum/client';
+import { RollupContract } from '@aztec/ethereum/contracts';
+import { Fr } from '@aztec/foundation/curves/bn254';
+import { EthAddress } from '@aztec/foundation/eth-address';
+import { Timer } from '@aztec/foundation/timer';
+import { AMMContract } from '@aztec/noir-contracts.js/AMM';
+import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
 import { TokenContract } from '@aztec/noir-contracts.js/Token';
+import { TestContract } from '@aztec/noir-test-contracts.js/Test';
+import { GasSettings } from '@aztec/stdlib/gas';
 import { deriveSigningKey } from '@aztec/stdlib/keys';
-import { makeTracedFetch } from '@aztec/telemetry-client';
-import { SupportedTokenContracts, getVersions } from './config.js';
+import { SupportedTokenContracts } from './config.js';
+import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
 import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
 const MINT_BALANCE = 1e12;
 const MIN_BALANCE = 1e3;
 export class BotFactory {
     config;
-    pxe;
-    node;
+    wallet;
+    store;
+    aztecNode;
+    aztecNodeAdmin;
     log;
-    constructor(config, dependencies = {}){
+    constructor(config, wallet, store, aztecNode, aztecNodeAdmin){
         this.config = config;
+        this.wallet = wallet;
+        this.store = store;
+        this.aztecNode = aztecNode;
+        this.aztecNodeAdmin = aztecNodeAdmin;
         this.log = createLogger('bot');
-        if (config.flushSetupTransactions && !dependencies.node) {
-            throw new Error(`Either a node client or node url must be provided if transaction flushing is requested`);
-        }
-        if (config.senderPrivateKey && !dependencies.node) {
-            throw new Error(`Either a node client or node url must be provided for bridging L1 fee juice to deploy an account with private key`);
-        }
-        if (!dependencies.pxe && !config.pxeUrl) {
-            throw new Error(`Either a PXE client or a PXE URL must be provided`);
-        }
-        this.node = dependencies.node;
-        if (dependencies.pxe) {
-            this.log.info(`Using local PXE`);
-            this.pxe = dependencies.pxe;
-            return;
-        }
-        this.log.info(`Using remote PXE at ${config.pxeUrl}`);
-        this.pxe = createPXEClient(config.pxeUrl, getVersions(), makeTracedFetch([
-            1,
-            2,
-            3
-        ], false));
+        // Set fee padding on the wallet so that all transactions during setup
+        // (token deploy, minting, etc.) use the configured padding, not the default.
+        this.wallet.setMinFeePadding(config.minFeePadding);
     }
     /**
    * 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.registerRecipient();
-        const wallet = await this.setupAccount();
-        const token = await this.setupToken(wallet);
-        await this.mintTokens(token);
+        const defaultAccountAddress = await this.setupAccount();
+        const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
+        const token = await this.setupToken(defaultAccountAddress);
+        await this.mintTokens(token, defaultAccountAddress);
         return {
-            wallet,
+            wallet: this.wallet,
+            defaultAccountAddress,
             token,
-            pxe: this.pxe,
+            node: this.aztecNode,
             recipient
         };
     }
+    async setupAmm() {
+        const defaultAccountAddress = await this.setupAccount();
+        const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
+        const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
+        const liquidityToken = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotLPToken', 'BOTLP');
+        const amm = await this.setupAmmContract(defaultAccountAddress, this.config.tokenSalt, token0, token1, liquidityToken);
+        await this.fundAmm(defaultAccountAddress, defaultAccountAddress, amm, token0, token1, liquidityToken);
+        this.log.info(`AMM initialized and funded`);
+        return {
+            wallet: this.wallet,
+            defaultAccountAddress,
+            amm,
+            token0,
+            token1,
+            node: this.aztecNode
+        };
+    }
+    /**
+   * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
+   * seeding initial L1→L2 messages, and waiting for the first to be ready.
+   */ async setupCrossChain() {
+        const defaultAccountAddress = await this.setupAccount();
+        // Create L1 client (same pattern as bridgeL1FeeJuice)
+        const l1RpcUrls = this.config.l1RpcUrls;
+        if (!l1RpcUrls?.length) {
+            throw new Error('L1 RPC URLs required for cross-chain bot');
+        }
+        const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
+        if (!mnemonicOrPrivateKey) {
+            throw new Error('L1 mnemonic or private key required for cross-chain bot');
+        }
+        const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
+        const chain = createEthereumChain(l1RpcUrls, l1ChainId);
+        const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
+        // Fetch Rollup version (needed for Inbox L2Actor struct)
+        const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
+        const rollupVersion = await rollupContract.getVersion();
+        // Deploy TestContract
+        const contract = await this.setupTestContract(defaultAccountAddress);
+        // Recover any pending messages from store (clean up stale ones first)
+        await this.store.cleanupOldPendingMessages();
+        const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
+        // Seed initial L1→L2 messages if pipeline is empty
+        const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
+        for(let i = 0; i < seedCount; i++){
+            await seedL1ToL2Message(l1Client, EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()), contract.address, rollupVersion, this.store, this.log);
+        }
+        // Block until at least one message is ready
+        const allMessages = await this.store.getUnconsumedL1ToL2Messages();
+        if (allMessages.length > 0) {
+            this.log.info(`Waiting for first L1→L2 message to be ready...`);
+            const firstMsg = allMessages[0];
+            await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
+                timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
+            });
+            this.log.info(`First L1→L2 message is ready`);
+        }
+        return {
+            wallet: this.wallet,
+            defaultAccountAddress,
+            contract,
+            node: this.aztecNode,
+            l1Client,
+            rollupVersion
+        };
+    }
+    async setupTestContract(deployer) {
+        const deployOpts = {
+            from: deployer,
+            contractAddressSalt: this.config.tokenSalt,
+            universalDeploy: true
+        };
+        const deploy = TestContract.deploy(this.wallet);
+        const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
+        return TestContract.at(instance.address, this.wallet);
+    }
     /**
    * Checks if the sender account contract is initialized, and initializes it if necessary.
    * @returns The sender wallet.
    */ async setupAccount() {
-        if (this.config.senderPrivateKey) {
-            return await this.setupAccountWithPrivateKey(this.config.senderPrivateKey);
+        const privateKey = this.config.senderPrivateKey?.getValue();
+        if (privateKey) {
+            this.log.info(`Setting up account with provided private key`);
+            return await this.setupAccountWithPrivateKey(privateKey);
         } else {
+            this.log.info(`Setting up test account`);
             return await this.setupTestAccount();
         }
     }
-    async setupAccountWithPrivateKey(privateKey) {
-        const salt = Fr.ONE;
-        const signingKey = deriveSigningKey(privateKey);
-        const account = await getSchnorrAccount(this.pxe, privateKey, signingKey, salt);
-        const isInit = (await this.pxe.getContractMetadata(account.getAddress())).isContractInitialized;
-        if (isInit) {
-            this.log.info(`Account at ${account.getAddress().toString()} already initialized`);
-            const wallet = await account.register();
-            return wallet;
+    async setupAccountWithPrivateKey(secret) {
+        const salt = this.config.senderSalt ?? Fr.ONE;
+        const signingKey = deriveSigningKey(secret);
+        const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
+        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;
+            this.log.info(`Account at ${address} registered. duration=${timer.ms()}`);
+            await this.store.deleteBridgeClaim(address);
+            return address;
         } else {
-            const address = account.getAddress();
+            const address = accountManager.address;
             this.log.info(`Deploying account at ${address}`);
-            const claim = await this.bridgeL1FeeJuice(address, 10n ** 22n);
-            const wallet = await account.getWallet();
-            const paymentMethod = new FeeJuicePaymentMethodWithClaim(wallet, claim);
-            const sentTx = account.deploy({
-                fee: {
-                    paymentMethod
-                }
+            const claim = await this.getOrCreateBridgeClaim(address);
+            const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
+            const deployMethod = await accountManager.getDeployMethod();
+            const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
+            const gasSettings = GasSettings.default({
+                maxFeesPerGas
             });
-            const txHash = await sentTx.getTxHash();
-            this.log.info(`Sent tx with hash ${txHash.toString()}`);
-            await this.tryFlushTxs();
-            this.log.verbose('Waiting for account deployment to settle');
-            await sentTx.wait({
-                timeout: this.config.txMinedWaitSeconds
+            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}`);
-            return wallet;
+            // Clean up the consumed bridge claim
+            await this.store.deleteBridgeClaim(address);
+            return accountManager.address;
         }
     }
     async setupTestAccount() {
-        let [wallet] = await getDeployedTestAccountsWallets(this.pxe);
-        if (wallet) {
-            this.log.info(`Using funded test account: ${wallet.getAddress()}`);
-        } else {
-            this.log.info('Registering funded test account');
-            const [account] = await getInitialTestAccounts();
-            const manager = await getSchnorrAccount(this.pxe, account.secret, account.signingKey, account.salt);
-            wallet = await manager.register();
-            this.log.info(`Funded test account registered: ${wallet.getAddress()}`);
-        }
-        return wallet;
-    }
-    /**
-   * Registers the recipient for txs in the pxe.
-   */ async registerRecipient() {
-        const recipient = await this.pxe.registerAccount(this.config.recipientEncryptionSecret, Fr.ONE);
-        return recipient.address;
+        const [initialAccountData] = await getInitialTestAccountsData();
+        const accountManager = await this.wallet.createSchnorrAccount(initialAccountData.secret, initialAccountData.salt, initialAccountData.signingKey);
+        return accountManager.address;
     }
     /**
    * 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.
-   */ async setupToken(wallet) {
+   */ 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(wallet, wallet.getAddress(), 'BotToken', 'BOT', 18);
-        } else if (this.config.contract === SupportedTokenContracts.EasyPrivateTokenContract) {
-            deploy = EasyPrivateTokenContract.deploy(wallet, MINT_BALANCE, wallet.getAddress());
-            deployOpts.skipPublicDeployment = true;
-            deployOpts.skipClassRegistration = true;
+            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) {
+            // 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;
-            deployOpts.skipPublicSimulation = true;
+            // 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);
+            // The contract constructor initializes private storage vars that need the contract's own nullifier key.
+            deployOpts.additionalScopes = [
+                tokenInstance.address
+            ];
         } else {
             throw new Error(`Unsupported token contract type: ${this.config.contract}`);
         }
-        const address = (await deploy.getInstance(deployOpts)).address;
-        if ((await this.pxe.getContractMetadata(address)).isContractPubliclyDeployed) {
+        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();
-            this.log.info(`Sent tx with hash ${txHash.toString()}`);
-            await this.tryFlushTxs();
-            this.log.verbose('Waiting for token setup to settle');
-            return sentTx.deployed({
+            const { txHash } = await deploy.send({
+                ...deployOpts,
+                wait: NO_WAIT
+            });
+            this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
+            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.
+   */ 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);
+        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 = {
+            from: deployer,
+            contractAddressSalt,
+            universalDeploy: true
+        };
+        const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
+        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 { receipt: 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;
+    }
+    async fundAmm(defaultAccountAddress, liquidityProvider, amm, token0, token1, lpToken) {
+        const getPrivateBalances = ()=>Promise.all([
+                token0.methods.balance_of_private(liquidityProvider).simulate({
+                    from: liquidityProvider
+                }).then((r)=>r.result),
+                token1.methods.balance_of_private(liquidityProvider).simulate({
+                    from: liquidityProvider
+                }).then((r)=>r.result),
+                lpToken.methods.balance_of_private(liquidityProvider).simulate({
+                    from: liquidityProvider
+                }).then((r)=>r.result)
+            ]);
+        const authwitNonce = Fr.random();
+        // keep some tokens for swapping
+        const amount0Max = MINT_BALANCE / 2;
+        const amount0Min = MINT_BALANCE / 4;
+        const amount1Max = MINT_BALANCE / 2;
+        const amount1Min = MINT_BALANCE / 4;
+        const [t0Bal, t1Bal, lpBal] = await getPrivateBalances();
+        this.log.info(`Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1. Current private balances of ${liquidityProvider}: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`);
+        // Add authwitnesses for the transfers in AMM::add_liquidity function
+        const token0Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
+            caller: amm.address,
+            call: await token0.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount0Max, authwitNonce).getFunctionCall()
+        });
+        const token1Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
+            caller: amm.address,
+            call: await token1.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount1Max, authwitNonce).getFunctionCall()
+        });
+        const { receipt: 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,
+            wait: {
+                timeout: this.config.txMinedWaitSeconds
+            }
+        });
+        this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
+        const { receipt: addLiquidityReceipt } = await amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce).send({
+            from: liquidityProvider,
+            authWitnesses: [
+                token0Authwit,
+                token1Authwit
+            ],
+            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 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`);
+            await deploy.register();
+        } else {
+            this.log.info(`Deploying contract ${name} at ${address.toString()}`);
+            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.
    * @param token - Token contract.
-   */ async mintTokens(token) {
-        const sender = token.wallet.getAddress();
+   */ async mintTokens(token, minter) {
         const isStandardToken = isStandardTokenContract(token);
         let privateBalance = 0n;
         let publicBalance = 0n;
         if (isStandardToken) {
-            ({ privateBalance, publicBalance } = await getBalances(token, sender));
+            ({ privateBalance, publicBalance } = await getBalances(token, minter));
         } else {
-            privateBalance = await getPrivateBalance(token, sender);
+            privateBalance = await getPrivateBalance(token, minter);
         }
         const calls = [];
         if (privateBalance < MIN_BALANCE) {
-            this.log.info(`Minting private tokens for ${sender.toString()}`);
-            const from = sender; // we are setting from to sender here because we need a sender to calculate the tag
-            calls.push(isStandardToken ? await token.methods.mint_to_private(from, sender, MINT_BALANCE).request() : await token.methods.mint(MINT_BALANCE, sender).request());
+            this.log.info(`Minting private tokens for ${minter.toString()}`);
+            calls.push(isStandardToken ? token.methods.mint_to_private(minter, MINT_BALANCE) : token.methods.mint(MINT_BALANCE, minter));
         }
         if (isStandardToken && publicBalance < MIN_BALANCE) {
-            this.log.info(`Minting public tokens for ${sender.toString()}`);
-            calls.push(await token.methods.mint_to_public(sender, MINT_BALANCE).request());
+            this.log.info(`Minting public tokens for ${minter.toString()}`);
+            calls.push(token.methods.mint_to_public(minter, MINT_BALANCE));
         }
         if (calls.length === 0) {
-            this.log.info(`Skipping minting as ${sender.toString()} has enough tokens`);
+            this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
             return;
         }
-        const sentTx = new BatchCall(token.wallet, calls).send();
-        const txHash = await sentTx.getTxHash();
-        this.log.info(`Sent tx with hash ${txHash.toString()}`);
-        await this.tryFlushTxs();
-        this.log.verbose('Waiting for token mint to settle');
-        await sentTx.wait({
-            timeout: this.config.txMinedWaitSeconds
+        // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
+        const additionalScopes = isStandardToken ? undefined : [
+            token.address
+        ];
+        await this.withNoMinTxsPerBlock(async ()=>{
+            const { txHash } = await new BatchCall(token.wallet, calls).send({
+                from: minter,
+                additionalScopes,
+                wait: NO_WAIT
+            });
+            this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
+            return waitForTx(this.aztecNode, txHash, {
+                timeout: this.config.txMinedWaitSeconds
+            });
         });
     }
-    async bridgeL1FeeJuice(recipient, amount) {
+    /**
+   * Gets or creates a bridge claim for the recipient.
+   * Checks if a claim already exists in the store and reuses it if valid.
+   * Only creates a new bridge if fee juice balance is below threshold.
+   */ async getOrCreateBridgeClaim(recipient) {
+        // Check if we have an existing claim in the store
+        const existingClaim = await this.store.getBridgeClaim(recipient);
+        if (existingClaim) {
+            this.log.info(`Found existing bridge claim for ${recipient.toString()}, checking validity...`);
+            // Check if the message is ready on L2
+            try {
+                const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
+                await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
+                        timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
+                    }));
+                return existingClaim.claim;
+            } catch (err) {
+                this.log.warn(`Failed to verify existing claim, creating new one: ${err}`);
+                await this.store.deleteBridgeClaim(recipient);
+            }
+        }
+        const claim = await this.bridgeL1FeeJuice(recipient);
+        await this.store.saveBridgeClaim(recipient, claim);
+        return claim;
+    }
+    async bridgeL1FeeJuice(recipient) {
         const l1RpcUrls = this.config.l1RpcUrls;
         if (!l1RpcUrls?.length) {
             throw new Error('L1 Rpc url is required to bridge the fee juice to fund the deployment of the account.');
         }
-        const mnemonicOrPrivateKey = this.config.l1PrivateKey || this.config.l1Mnemonic;
+        const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
         if (!mnemonicOrPrivateKey) {
             throw new Error('Either a mnemonic or private key of an L1 account is required to bridge the fee juice to fund the deployment of the account.');
         }
-        const { l1ChainId } = await this.pxe.getNodeInfo();
+        const { l1ChainId } = await this.aztecNode.getNodeInfo();
         const chain = createEthereumChain(l1RpcUrls, l1ChainId);
-        const { publicClient, walletClient } = createL1Clients(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
-        const portal = await L1FeeJuicePortalManager.new(this.pxe, publicClient, walletClient, this.log);
-        const claim = await portal.bridgeTokensPublic(recipient, amount, true);
-        const isSynced = async ()=>await this.pxe.isL1ToL2MessageSynced(Fr.fromHexString(claim.messageHash));
-        await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 1);
-        this.log.info(`Created a claim for ${amount} L1 fee juice to ${recipient}.`, claim);
-        // Progress by 2 L2 blocks so that the l1ToL2Message added above will be available to use on L2.
-        await this.advanceL2Block();
-        await this.advanceL2Block();
+        const extendedClient = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
+        const portal = await L1FeeJuicePortalManager.new(this.aztecNode, extendedClient, this.log);
+        const mintAmount = await portal.getTokenManager().getMintAmount();
+        const claim = await portal.bridgeTokensPublic(recipient, mintAmount, true);
+        await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
+                timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
+            }));
+        this.log.info(`Created a claim for ${mintAmount} L1 fee juice to ${recipient}.`, claim);
         return claim;
     }
-    async advanceL2Block() {
-        const initialBlockNumber = await this.node.getBlockNumber();
-        await this.tryFlushTxs();
-        await retryUntil(async ()=>await this.node.getBlockNumber() >= initialBlockNumber + 1);
-    }
-    async tryFlushTxs() {
-        if (this.config.flushSetupTransactions) {
-            this.log.verbose('Flushing transactions');
-            try {
-                await this.node.flushTxs();
-            } catch (err) {
-                this.log.error(`Failed to flush transactions: ${err}`);
-            }
+    async withNoMinTxsPerBlock(fn) {
+        if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
+            this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
+            return fn();
+        }
+        const { minTxsPerBlock } = await this.aztecNodeAdmin.getConfig();
+        this.log.warn(`Setting sequencer minTxsPerBlock to 0 from ${minTxsPerBlock} to flush setup transactions`);
+        await this.aztecNodeAdmin.setConfig({
+            minTxsPerBlock: 0
+        });
+        try {
+            return await fn();
+        } finally{
+            this.log.warn(`Restoring sequencer minTxsPerBlock to ${minTxsPerBlock}`);
+            await this.aztecNodeAdmin.setConfig({
+                minTxsPerBlock
+            });
         }
     }
 }

package/dest/index.d.ts

@@ -1,6 +1,9 @@
 export { Bot } from './bot.js';
+export { AmmBot } from './amm_bot.js';
+export { CrossChainBot } from './cross_chain_bot.js';
 export { BotRunner } from './runner.js';
+export { BotStore } from './store/bot_store.js';
 export { type BotConfig, getBotConfigFromEnv, getBotDefaultConfig, botConfigMappings, SupportedTokenContracts, } from './config.js';
-export { createBotRunnerRpcServer, getBotRunnerApiHandler } from './rpc.js';
+export { getBotRunnerApiHandler } from './rpc.js';
 export * from './interface.js';
-//# sourceMappingURL=index.d.ts.map
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsR0FBRyxFQUFFLE1BQU0sVUFBVSxDQUFDO0FBQy9CLE9BQU8sRUFBRSxNQUFNLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFDdEMsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ3JELE9BQU8sRUFBRSxTQUFTLEVBQUUsTUFBTSxhQUFhLENBQUM7QUFDeEMsT0FBTyxFQUFFLFFBQVEsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ2hELE9BQU8sRUFDTCxLQUFLLFNBQVMsRUFDZCxtQkFBbUIsRUFDbkIsbUJBQW1CLEVBQ25CLGlCQUFpQixFQUNqQix1QkFBdUIsR0FDeEIsTUFBTSxhQUFhLENBQUM7QUFDckIsT0FBTyxFQUFFLHNCQUFzQixFQUFFLE1BQU0sVUFBVSxDQUFDO0FBQ2xELGNBQWMsZ0JBQWdCLENBQUMifQ==

package/dest/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EACL,KAAK,SAAS,EACd,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAC5E,cAAc,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EACL,KAAK,SAAS,EACd,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,cAAc,gBAAgB,CAAC"}

package/dest/index.js

@@ -1,5 +1,8 @@
 export { Bot } from './bot.js';
+export { AmmBot } from './amm_bot.js';
+export { CrossChainBot } from './cross_chain_bot.js';
 export { BotRunner } from './runner.js';
+export { BotStore } from './store/bot_store.js';
 export { getBotConfigFromEnv, getBotDefaultConfig, botConfigMappings, SupportedTokenContracts } from './config.js';
-export { createBotRunnerRpcServer, getBotRunnerApiHandler } from './rpc.js';
+export { getBotRunnerApiHandler } from './rpc.js';
 export * from './interface.js';