@aztec/bot 0.0.1-commit.b655e406 → 0.0.1-commit.b6e433891

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (49) hide show
  1. package/dest/amm_bot.d.ts +6 -7
  2. package/dest/amm_bot.d.ts.map +1 -1
  3. package/dest/amm_bot.js +27 -16
  4. package/dest/base_bot.d.ts +6 -6
  5. package/dest/base_bot.d.ts.map +1 -1
  6. package/dest/base_bot.js +14 -15
  7. package/dest/bot.d.ts +6 -6
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +8 -4
  10. package/dest/config.d.ts +73 -58
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +44 -17
  13. package/dest/cross_chain_bot.d.ts +54 -0
  14. package/dest/cross_chain_bot.d.ts.map +1 -0
  15. package/dest/cross_chain_bot.js +134 -0
  16. package/dest/factory.d.ts +18 -26
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +261 -79
  19. package/dest/index.d.ts +2 -1
  20. package/dest/index.d.ts.map +1 -1
  21. package/dest/index.js +1 -0
  22. package/dest/interface.d.ts +1 -1
  23. package/dest/l1_to_l2_seeding.d.ts +8 -0
  24. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  25. package/dest/l1_to_l2_seeding.js +63 -0
  26. package/dest/rpc.d.ts +1 -1
  27. package/dest/runner.d.ts +3 -3
  28. package/dest/runner.d.ts.map +1 -1
  29. package/dest/runner.js +429 -31
  30. package/dest/store/bot_store.d.ts +30 -5
  31. package/dest/store/bot_store.d.ts.map +1 -1
  32. package/dest/store/bot_store.js +38 -7
  33. package/dest/store/index.d.ts +2 -2
  34. package/dest/store/index.d.ts.map +1 -1
  35. package/dest/utils.d.ts +1 -1
  36. package/dest/utils.js +3 -3
  37. package/package.json +19 -15
  38. package/src/amm_bot.ts +26 -21
  39. package/src/base_bot.ts +13 -27
  40. package/src/bot.ts +10 -8
  41. package/src/config.ts +90 -60
  42. package/src/cross_chain_bot.ts +203 -0
  43. package/src/factory.ts +264 -75
  44. package/src/index.ts +1 -0
  45. package/src/l1_to_l2_seeding.ts +79 -0
  46. package/src/runner.ts +18 -5
  47. package/src/store/bot_store.ts +61 -6
  48. package/src/store/index.ts +1 -1
  49. package/src/utils.ts +3 -3
package/src/factory.ts CHANGED
@@ -1,4 +1,3 @@
1
- import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
2
1
  import { getInitialTestAccountsData } from '@aztec/accounts/testing';
3
2
  import { AztecAddress } from '@aztec/aztec.js/addresses';
4
3
  import {
@@ -7,24 +6,34 @@ import {
7
6
  ContractFunctionInteraction,
8
7
  type DeployMethod,
9
8
  type DeployOptions,
9
+ NO_WAIT,
10
10
  } from '@aztec/aztec.js/contracts';
11
- import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
12
11
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
12
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
13
13
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
14
+ import { deriveKeys } from '@aztec/aztec.js/keys';
14
15
  import { createLogger } from '@aztec/aztec.js/log';
15
16
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
16
- import { createEthereumChain, createExtendedL1Client } from '@aztec/ethereum';
17
- import { Fr } from '@aztec/foundation/fields';
17
+ import { waitForTx } from '@aztec/aztec.js/node';
18
+ import { createEthereumChain } from '@aztec/ethereum/chain';
19
+ import { createExtendedL1Client } from '@aztec/ethereum/client';
20
+ import { RollupContract } from '@aztec/ethereum/contracts';
21
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
22
+ import { Fr } from '@aztec/foundation/curves/bn254';
23
+ import { EthAddress } from '@aztec/foundation/eth-address';
18
24
  import { Timer } from '@aztec/foundation/timer';
19
25
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
20
26
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
21
27
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
22
- import { GasSettings } from '@aztec/stdlib/gas';
28
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
29
+ import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
30
+ import { GasFees, GasSettings } from '@aztec/stdlib/gas';
23
31
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
24
32
  import { deriveSigningKey } from '@aztec/stdlib/keys';
25
- import { TestWallet } from '@aztec/test-wallet/server';
33
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
26
34
 
27
35
  import { type BotConfig, SupportedTokenContracts } from './config.js';
36
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
28
37
  import type { BotStore } from './store/index.js';
29
38
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
30
39
 
@@ -36,25 +45,42 @@ export class BotFactory {
36
45
 
37
46
  constructor(
38
47
  private readonly config: BotConfig,
39
- private readonly wallet: TestWallet,
48
+ private readonly wallet: EmbeddedWallet,
40
49
  private readonly store: BotStore,
41
50
  private readonly aztecNode: AztecNode,
42
51
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
43
- ) {}
52
+ ) {
53
+ // Set fee padding on the wallet so that all transactions during setup
54
+ // (token deploy, minting, etc.) use the configured padding, not the default.
55
+ this.wallet.setMinFeePadding(config.minFeePadding);
56
+ }
44
57
 
45
58
  /**
46
59
  * Initializes a new bot by setting up the sender account, registering the recipient,
47
60
  * deploying the token contract, and minting tokens if necessary.
48
61
  */
49
- public async setup() {
50
- const recipient = (await this.wallet.createAccount()).address;
62
+ public async setup(): Promise<{
63
+ wallet: EmbeddedWallet;
64
+ defaultAccountAddress: AztecAddress;
65
+ token: TokenContract | PrivateTokenContract;
66
+ node: AztecNode;
67
+ recipient: AztecAddress;
68
+ }> {
51
69
  const defaultAccountAddress = await this.setupAccount();
70
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
52
71
  const token = await this.setupToken(defaultAccountAddress);
53
72
  await this.mintTokens(token, defaultAccountAddress);
54
73
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
55
74
  }
56
75
 
57
- public async setupAmm() {
76
+ public async setupAmm(): Promise<{
77
+ wallet: EmbeddedWallet;
78
+ defaultAccountAddress: AztecAddress;
79
+ amm: AMMContract;
80
+ token0: TokenContract;
81
+ token1: TokenContract;
82
+ node: AztecNode;
83
+ }> {
58
84
  const defaultAccountAddress = await this.setupAccount();
59
85
  const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
60
86
  const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
@@ -78,6 +104,89 @@ export class BotFactory {
78
104
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
79
105
  }
80
106
 
107
+ /**
108
+ * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
109
+ * seeding initial L1→L2 messages, and waiting for the first to be ready.
110
+ */
111
+ public async setupCrossChain(): Promise<{
112
+ wallet: EmbeddedWallet;
113
+ defaultAccountAddress: AztecAddress;
114
+ contract: TestContract;
115
+ node: AztecNode;
116
+ l1Client: ExtendedViemWalletClient;
117
+ rollupVersion: bigint;
118
+ }> {
119
+ const defaultAccountAddress = await this.setupAccount();
120
+
121
+ // Create L1 client (same pattern as bridgeL1FeeJuice)
122
+ const l1RpcUrls = this.config.l1RpcUrls;
123
+ if (!l1RpcUrls?.length) {
124
+ throw new Error('L1 RPC URLs required for cross-chain bot');
125
+ }
126
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
127
+ if (!mnemonicOrPrivateKey) {
128
+ throw new Error('L1 mnemonic or private key required for cross-chain bot');
129
+ }
130
+ const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
131
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
132
+ const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
133
+
134
+ // Fetch Rollup version (needed for Inbox L2Actor struct)
135
+ const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
136
+ const rollupVersion = await rollupContract.getVersion();
137
+
138
+ // Deploy TestContract
139
+ const contract = await this.setupTestContract(defaultAccountAddress);
140
+
141
+ // Recover any pending messages from store (clean up stale ones first)
142
+ await this.store.cleanupOldPendingMessages();
143
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
144
+
145
+ // Seed initial L1→L2 messages if pipeline is empty
146
+ const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
147
+ for (let i = 0; i < seedCount; i++) {
148
+ await seedL1ToL2Message(
149
+ l1Client,
150
+ EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()),
151
+ contract.address,
152
+ rollupVersion,
153
+ this.store,
154
+ this.log,
155
+ );
156
+ }
157
+
158
+ // Block until at least one message is ready
159
+ const allMessages = await this.store.getUnconsumedL1ToL2Messages();
160
+ if (allMessages.length > 0) {
161
+ this.log.info(`Waiting for first L1→L2 message to be ready...`);
162
+ const firstMsg = allMessages[0];
163
+ await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
164
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
165
+ });
166
+ this.log.info(`First L1→L2 message is ready`);
167
+ }
168
+
169
+ return {
170
+ wallet: this.wallet,
171
+ defaultAccountAddress,
172
+ contract,
173
+ node: this.aztecNode,
174
+ l1Client,
175
+ rollupVersion,
176
+ };
177
+ }
178
+
179
+ private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
180
+ const deployOpts: DeployOptions = {
181
+ from: deployer,
182
+ contractAddressSalt: this.config.tokenSalt,
183
+ universalDeploy: true,
184
+ };
185
+ const deploy = TestContract.deploy(this.wallet);
186
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
187
+ return TestContract.at(instance.address, this.wallet);
188
+ }
189
+
81
190
  /**
82
191
  * Checks if the sender account contract is initialized, and initializes it if necessary.
83
192
  * @returns The sender wallet.
@@ -96,14 +205,9 @@ export class BotFactory {
96
205
  private async setupAccountWithPrivateKey(secret: Fr) {
97
206
  const salt = this.config.senderSalt ?? Fr.ONE;
98
207
  const signingKey = deriveSigningKey(secret);
99
- const accountData = {
100
- secret,
101
- salt,
102
- contract: new SchnorrAccountContract(signingKey!),
103
- };
104
- const accountManager = await this.wallet.createAccount(accountData);
105
- const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
106
- if (isInit) {
208
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
209
+ const metadata = await this.wallet.getContractMetadata(accountManager.address);
210
+ if (metadata.isContractInitialized) {
107
211
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
108
212
  const timer = new Timer();
109
213
  const address = accountManager.address;
@@ -118,12 +222,23 @@ export class BotFactory {
118
222
 
119
223
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
120
224
  const deployMethod = await accountManager.getDeployMethod();
121
- const maxFeesPerGas = (await this.aztecNode.getCurrentBaseFees()).mul(1 + this.config.baseFeePadding);
122
- const gasSettings = GasSettings.default({ maxFeesPerGas });
123
- const sentTx = deployMethod.send({ from: AztecAddress.ZERO, fee: { gasSettings, paymentMethod } });
124
- const txHash = await sentTx.getTxHash();
125
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
126
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
225
+ const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
226
+
227
+ const { estimatedGas } = await deployMethod.simulate({
228
+ from: AztecAddress.ZERO,
229
+ fee: { estimateGas: true, paymentMethod },
230
+ });
231
+ const gasSettings = GasSettings.from({ ...estimatedGas!, maxFeesPerGas, maxPriorityFeesPerGas: GasFees.empty() });
232
+
233
+ await this.withNoMinTxsPerBlock(async () => {
234
+ const { txHash } = await deployMethod.send({
235
+ from: AztecAddress.ZERO,
236
+ fee: { gasSettings, paymentMethod },
237
+ wait: NO_WAIT,
238
+ });
239
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`, { gasSettings });
240
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
241
+ });
127
242
  this.log.info(`Account deployed at ${address}`);
128
243
 
129
244
  // Clean up the consumed bridge claim
@@ -135,12 +250,11 @@ export class BotFactory {
135
250
 
136
251
  private async setupTestAccount() {
137
252
  const [initialAccountData] = await getInitialTestAccountsData();
138
- const accountData = {
139
- secret: initialAccountData.secret,
140
- salt: initialAccountData.salt,
141
- contract: new SchnorrAccountContract(initialAccountData.signingKey),
142
- };
143
- const accountManager = await this.wallet.createAccount(accountData);
253
+ const accountManager = await this.wallet.createSchnorrAccount(
254
+ initialAccountData.secret,
255
+ initialAccountData.salt,
256
+ initialAccountData.signingKey,
257
+ );
144
258
  return accountManager.address;
145
259
  }
146
260
 
@@ -151,33 +265,52 @@ export class BotFactory {
151
265
  */
152
266
  private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
153
267
  let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
268
+ let tokenInstance: ContractInstanceWithAddress | undefined;
154
269
  const deployOpts: DeployOptions = {
155
270
  from: sender,
156
271
  contractAddressSalt: this.config.tokenSalt,
157
272
  universalDeploy: true,
158
273
  };
274
+ let token: TokenContract | PrivateTokenContract;
159
275
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
160
276
  deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
277
+ tokenInstance = await deploy.getInstance(deployOpts);
278
+ token = TokenContract.at(tokenInstance.address, this.wallet);
161
279
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
162
- deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender);
280
+ // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
281
+ const tokenSecretKey = Fr.random();
282
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
283
+ deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
163
284
  deployOpts.skipInstancePublication = true;
164
285
  deployOpts.skipClassPublication = true;
165
286
  deployOpts.skipInitialization = false;
287
+
288
+ // Register the contract with the secret key before deployment
289
+ tokenInstance = await deploy.getInstance(deployOpts);
290
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
291
+ await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
292
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
293
+ deployOpts.additionalScopes = [tokenInstance.address];
166
294
  } else {
167
295
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
168
296
  }
169
297
 
170
- const address = (await deploy.getInstance(deployOpts)).address;
171
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
298
+ const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
299
+ const metadata = await this.wallet.getContractMetadata(address);
300
+ if (metadata.isContractPublished) {
172
301
  this.log.info(`Token at ${address.toString()} already deployed`);
173
- return deploy.register();
302
+ await deploy.register();
174
303
  } else {
175
304
  this.log.info(`Deploying token contract at ${address.toString()}`);
176
- const sentTx = deploy.send(deployOpts);
177
- const txHash = await sentTx.getTxHash();
178
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
179
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
305
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
306
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
307
+ this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`, { estimatedGas });
308
+ await this.withNoMinTxsPerBlock(async () => {
309
+ await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
310
+ return token;
311
+ });
180
312
  }
313
+ return token;
181
314
  }
182
315
 
183
316
  /**
@@ -185,7 +318,7 @@ export class BotFactory {
185
318
  * @param wallet - Wallet to deploy the token contract from.
186
319
  * @returns The TokenContract instance.
187
320
  */
188
- private setupTokenContract(
321
+ private async setupTokenContract(
189
322
  deployer: AztecAddress,
190
323
  contractAddressSalt: Fr,
191
324
  name: string,
@@ -194,7 +327,8 @@ export class BotFactory {
194
327
  ): Promise<TokenContract> {
195
328
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
196
329
  const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
197
- return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
330
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
331
+ return TokenContract.at(instance.address, this.wallet);
198
332
  }
199
333
 
200
334
  private async setupAmmContract(
@@ -206,12 +340,23 @@ export class BotFactory {
206
340
  ): Promise<AMMContract> {
207
341
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
208
342
  const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
209
- const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
343
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
344
+ const amm = AMMContract.at(instance.address, this.wallet);
210
345
 
211
346
  this.log.info(`AMM deployed at ${amm.address}`);
212
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({ from: deployer });
213
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
214
- await minterTx.wait({ timeout: this.config.txMinedWaitSeconds });
347
+ const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
348
+ const { estimatedGas: setMinterGas } = await setMinterInteraction.simulate({
349
+ from: deployer,
350
+ fee: { estimateGas: true },
351
+ });
352
+ const { receipt: minterReceipt } = await setMinterInteraction.send({
353
+ from: deployer,
354
+ fee: { gasSettings: setMinterGas },
355
+ wait: { timeout: this.config.txMinedWaitSeconds },
356
+ });
357
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`, {
358
+ estimatedGas: setMinterGas,
359
+ });
215
360
  this.log.info(`Liquidity token initialized`);
216
361
 
217
362
  return amm;
@@ -227,9 +372,18 @@ export class BotFactory {
227
372
  ): Promise<void> {
228
373
  const getPrivateBalances = () =>
229
374
  Promise.all([
230
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
231
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
232
- lpToken.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
375
+ token0.methods
376
+ .balance_of_private(liquidityProvider)
377
+ .simulate({ from: liquidityProvider })
378
+ .then(r => r.result),
379
+ token1.methods
380
+ .balance_of_private(liquidityProvider)
381
+ .simulate({ from: liquidityProvider })
382
+ .then(r => r.result),
383
+ lpToken.methods
384
+ .balance_of_private(liquidityProvider)
385
+ .simulate({ from: liquidityProvider })
386
+ .then(r => r.result),
233
387
  ]);
234
388
 
235
389
  const authwitNonce = Fr.random();
@@ -270,23 +424,44 @@ export class BotFactory {
270
424
  .getFunctionCall(),
271
425
  });
272
426
 
273
- const mintTx = new BatchCall(this.wallet, [
427
+ const mintBatch = new BatchCall(this.wallet, [
274
428
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
275
429
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
276
- ]).send({ from: liquidityProvider });
430
+ ]);
431
+ const { estimatedGas: mintGas } = await mintBatch.simulate({
432
+ from: liquidityProvider,
433
+ fee: { estimateGas: true },
434
+ });
435
+ const { receipt: mintReceipt } = await mintBatch.send({
436
+ from: liquidityProvider,
437
+ fee: { gasSettings: mintGas },
438
+ wait: { timeout: this.config.txMinedWaitSeconds },
439
+ });
277
440
 
278
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
279
- await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
441
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`, { estimatedGas: mintGas });
280
442
 
281
- const addLiquidityTx = amm.methods
282
- .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
283
- .send({
284
- from: liquidityProvider,
285
- authWitnesses: [token0Authwit, token1Authwit],
286
- });
443
+ const addLiquidityInteraction = amm.methods.add_liquidity(
444
+ amount0Max,
445
+ amount1Max,
446
+ amount0Min,
447
+ amount1Min,
448
+ authwitNonce,
449
+ );
450
+ const { estimatedGas: addLiquidityGas } = await addLiquidityInteraction.simulate({
451
+ from: liquidityProvider,
452
+ fee: { estimateGas: true },
453
+ authWitnesses: [token0Authwit, token1Authwit],
454
+ });
455
+ const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
456
+ from: liquidityProvider,
457
+ fee: { gasSettings: addLiquidityGas },
458
+ authWitnesses: [token0Authwit, token1Authwit],
459
+ wait: { timeout: this.config.txMinedWaitSeconds },
460
+ });
287
461
 
288
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
289
- await addLiquidityTx.wait({ timeout: this.config.txMinedWaitSeconds });
462
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`, {
463
+ estimatedGas: addLiquidityGas,
464
+ });
290
465
  this.log.info(`Liquidity added`);
291
466
 
292
467
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -299,18 +474,23 @@ export class BotFactory {
299
474
  name: string,
300
475
  deploy: DeployMethod<T>,
301
476
  deployOpts: DeployOptions,
302
- ): Promise<T> {
303
- const address = (await deploy.getInstance(deployOpts)).address;
304
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
477
+ ): Promise<ContractInstanceWithAddress> {
478
+ const instance = await deploy.getInstance(deployOpts);
479
+ const address = instance.address;
480
+ const metadata = await this.wallet.getContractMetadata(address);
481
+ if (metadata.isContractPublished) {
305
482
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
306
- return deploy.register();
483
+ await deploy.register();
307
484
  } else {
308
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
309
- const sentTx = deploy.send(deployOpts);
310
- const txHash = await sentTx.getTxHash();
311
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
312
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
485
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
486
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
487
+ await this.withNoMinTxsPerBlock(async () => {
488
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
489
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
490
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
491
+ });
313
492
  }
493
+ return instance;
314
494
  }
315
495
 
316
496
  /**
@@ -346,10 +526,21 @@ export class BotFactory {
346
526
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
347
527
  return;
348
528
  }
349
- const sentTx = new BatchCall(token.wallet, calls).send({ from: minter });
350
- const txHash = await sentTx.getTxHash();
351
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
352
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
529
+
530
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
531
+ const additionalScopes = isStandardToken ? undefined : [token.address];
532
+ const mintBatch = new BatchCall(token.wallet, calls);
533
+ const { estimatedGas } = await mintBatch.simulate({ from: minter, fee: { estimateGas: true }, additionalScopes });
534
+ await this.withNoMinTxsPerBlock(async () => {
535
+ const { txHash } = await mintBatch.send({
536
+ from: minter,
537
+ additionalScopes,
538
+ fee: { gasSettings: estimatedGas },
539
+ wait: NO_WAIT,
540
+ });
541
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`, { estimatedGas });
542
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
543
+ });
353
544
  }
354
545
 
355
546
  /**
@@ -369,7 +560,6 @@ export class BotFactory {
369
560
  await this.withNoMinTxsPerBlock(() =>
370
561
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
371
562
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
372
- forPublicConsumption: false,
373
563
  }),
374
564
  );
375
565
  return existingClaim.claim;
@@ -408,7 +598,6 @@ export class BotFactory {
408
598
  await this.withNoMinTxsPerBlock(() =>
409
599
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
410
600
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
411
- forPublicConsumption: false,
412
601
  }),
413
602
  );
414
603
 
package/src/index.ts CHANGED
@@ -1,5 +1,6 @@
1
1
  export { Bot } from './bot.js';
2
2
  export { AmmBot } from './amm_bot.js';
3
+ export { CrossChainBot } from './cross_chain_bot.js';
3
4
  export { BotRunner } from './runner.js';
4
5
  export { BotStore } from './store/bot_store.js';
5
6
  export {
@@ -0,0 +1,79 @@
1
+ import { generateClaimSecret } from '@aztec/aztec.js/ethereum';
2
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
3
+ import { compactArray } from '@aztec/foundation/collection';
4
+ import { Fr } from '@aztec/foundation/curves/bn254';
5
+ import { EthAddress } from '@aztec/foundation/eth-address';
6
+ import type { Logger } from '@aztec/foundation/log';
7
+ import { InboxAbi } from '@aztec/l1-artifacts';
8
+ import type { AztecAddress } from '@aztec/stdlib/aztec-address';
9
+
10
+ import { decodeEventLog, getContract } from 'viem';
11
+
12
+ import type { BotStore, PendingL1ToL2Message } from './store/index.js';
13
+
14
+ /** Sends an L1→L2 message via the Inbox contract and stores it. */
15
+ export async function seedL1ToL2Message(
16
+ l1Client: ExtendedViemWalletClient,
17
+ inboxAddress: EthAddress,
18
+ l2Recipient: AztecAddress,
19
+ rollupVersion: bigint,
20
+ store: BotStore,
21
+ log: Logger,
22
+ ): Promise<PendingL1ToL2Message> {
23
+ log.info('Seeding L1→L2 message');
24
+ const [secret, secretHash] = await generateClaimSecret(log);
25
+ const content = Fr.random();
26
+
27
+ const inbox = getContract({
28
+ address: inboxAddress.toString(),
29
+ abi: InboxAbi,
30
+ client: l1Client,
31
+ });
32
+
33
+ const txHash = await inbox.write.sendL2Message(
34
+ [{ actor: l2Recipient.toString(), version: rollupVersion }, content.toString(), secretHash.toString()],
35
+ { gas: 1_000_000n },
36
+ );
37
+ log.info(`L1→L2 message sent in tx ${txHash}`);
38
+
39
+ const txReceipt = await l1Client.waitForTransactionReceipt({ hash: txHash });
40
+ if (txReceipt.status !== 'success') {
41
+ throw new Error(`L1→L2 message tx failed: ${txHash}`);
42
+ }
43
+
44
+ // Extract MessageSent event
45
+ const messageSentLogs = compactArray(
46
+ txReceipt.logs
47
+ .filter(l => l.address.toLowerCase() === inboxAddress.toString().toLowerCase())
48
+ .map(l => {
49
+ try {
50
+ return decodeEventLog({ abi: InboxAbi, eventName: 'MessageSent', data: l.data, topics: l.topics });
51
+ } catch {
52
+ return undefined;
53
+ }
54
+ }),
55
+ );
56
+
57
+ if (messageSentLogs.length !== 1) {
58
+ throw new Error(`Expected 1 MessageSent event, got ${messageSentLogs.length}`);
59
+ }
60
+
61
+ const event = messageSentLogs[0];
62
+
63
+ const msgHash = event.args.hash;
64
+ const globalLeafIndex = event.args.index;
65
+
66
+ const msg: PendingL1ToL2Message = {
67
+ content: content.toString(),
68
+ secret: secret.toString(),
69
+ secretHash: secretHash.toString(),
70
+ msgHash,
71
+ sender: l1Client.account!.address,
72
+ globalLeafIndex: globalLeafIndex.toString(),
73
+ timestamp: Date.now(),
74
+ };
75
+
76
+ await store.savePendingL1ToL2Message(msg);
77
+ log.info(`Seeded L1→L2 message msgHash=${msg.msgHash}`);
78
+ return msg;
79
+ }
package/src/runner.ts CHANGED
@@ -4,12 +4,13 @@ import { omit } from '@aztec/foundation/collection';
4
4
  import { RunningPromise } from '@aztec/foundation/running-promise';
5
5
  import type { AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
6
6
  import { type TelemetryClient, type Traceable, type Tracer, trackSpan } from '@aztec/telemetry-client';
7
- import type { TestWallet } from '@aztec/test-wallet/server';
7
+ import type { EmbeddedWallet } from '@aztec/wallets/embedded';
8
8
 
9
9
  import { AmmBot } from './amm_bot.js';
10
10
  import type { BaseBot } from './base_bot.js';
11
11
  import { Bot } from './bot.js';
12
12
  import type { BotConfig } from './config.js';
13
+ import { CrossChainBot } from './cross_chain_bot.js';
13
14
  import type { BotInfo, BotRunnerApi } from './interface.js';
14
15
  import { BotStore } from './store/index.js';
15
16
 
@@ -24,7 +25,7 @@ export class BotRunner implements BotRunnerApi, Traceable {
24
25
 
25
26
  public constructor(
26
27
  private config: BotConfig,
27
- private readonly wallet: TestWallet,
28
+ private readonly wallet: EmbeddedWallet,
28
29
  private readonly aztecNode: AztecNode,
29
30
  private readonly telemetry: TelemetryClient,
30
31
  private readonly aztecNodeAdmin: AztecNodeAdmin | undefined,
@@ -146,9 +147,21 @@ export class BotRunner implements BotRunnerApi, Traceable {
146
147
 
147
148
  async #createBot() {
148
149
  try {
149
- this.bot = this.config.ammTxs
150
- ? AmmBot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store)
151
- : Bot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
150
+ switch (this.config.botMode) {
151
+ case 'crosschain':
152
+ this.bot = CrossChainBot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
153
+ break;
154
+ case 'amm':
155
+ this.bot = AmmBot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
156
+ break;
157
+ case 'transfer':
158
+ this.bot = Bot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
159
+ break;
160
+ default: {
161
+ const _exhaustive: never = this.config.botMode;
162
+ throw new Error(`Unsupported bot mode: [${_exhaustive}]`);
163
+ }
164
+ }
152
165
  await this.bot;
153
166
  } catch (err) {
154
167
  this.log.error(`Error setting up bot: ${err}`);