@aztec/bot 0.0.1-commit.6d3c34e → 0.0.1-commit.7035c9bd6

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 (46) 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 +12 -13
  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 +32 -16
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +37 -10
  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 +20 -10
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +247 -74
  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/l1_to_l2_seeding.d.ts +8 -0
  23. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  24. package/dest/l1_to_l2_seeding.js +63 -0
  25. package/dest/runner.d.ts +3 -3
  26. package/dest/runner.d.ts.map +1 -1
  27. package/dest/runner.js +17 -1
  28. package/dest/store/bot_store.d.ts +31 -6
  29. package/dest/store/bot_store.d.ts.map +1 -1
  30. package/dest/store/bot_store.js +37 -6
  31. package/dest/store/index.d.ts +2 -2
  32. package/dest/store/index.d.ts.map +1 -1
  33. package/dest/utils.js +3 -3
  34. package/package.json +17 -14
  35. package/src/amm_bot.ts +26 -21
  36. package/src/base_bot.ts +11 -25
  37. package/src/bot.ts +10 -8
  38. package/src/config.ts +40 -12
  39. package/src/cross_chain_bot.ts +203 -0
  40. package/src/factory.ts +235 -70
  41. package/src/index.ts +1 -0
  42. package/src/l1_to_l2_seeding.ts +79 -0
  43. package/src/runner.ts +18 -5
  44. package/src/store/bot_store.ts +60 -5
  45. package/src/store/index.ts +1 -1
  46. 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,27 +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
14
  import { deriveKeys } from '@aztec/aztec.js/keys';
15
15
  import { createLogger } from '@aztec/aztec.js/log';
16
16
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
17
+ import { waitForTx } from '@aztec/aztec.js/node';
17
18
  import { createEthereumChain } from '@aztec/ethereum/chain';
18
19
  import { createExtendedL1Client } from '@aztec/ethereum/client';
20
+ import { RollupContract } from '@aztec/ethereum/contracts';
21
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
19
22
  import { Fr } from '@aztec/foundation/curves/bn254';
23
+ import { EthAddress } from '@aztec/foundation/eth-address';
20
24
  import { Timer } from '@aztec/foundation/timer';
21
25
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
22
26
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
23
27
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
28
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
24
29
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
25
- import { GasSettings } from '@aztec/stdlib/gas';
30
+ import { GasFees, GasSettings } from '@aztec/stdlib/gas';
26
31
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
27
32
  import { deriveSigningKey } from '@aztec/stdlib/keys';
28
- import { TestWallet } from '@aztec/test-wallet/server';
33
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
29
34
 
30
35
  import { type BotConfig, SupportedTokenContracts } from './config.js';
36
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
31
37
  import type { BotStore } from './store/index.js';
32
38
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
33
39
 
@@ -39,32 +45,36 @@ export class BotFactory {
39
45
 
40
46
  constructor(
41
47
  private readonly config: BotConfig,
42
- private readonly wallet: TestWallet,
48
+ private readonly wallet: EmbeddedWallet,
43
49
  private readonly store: BotStore,
44
50
  private readonly aztecNode: AztecNode,
45
51
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
46
- ) {}
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
+ }
47
57
 
48
58
  /**
49
59
  * Initializes a new bot by setting up the sender account, registering the recipient,
50
60
  * deploying the token contract, and minting tokens if necessary.
51
61
  */
52
62
  public async setup(): Promise<{
53
- wallet: TestWallet;
63
+ wallet: EmbeddedWallet;
54
64
  defaultAccountAddress: AztecAddress;
55
65
  token: TokenContract | PrivateTokenContract;
56
66
  node: AztecNode;
57
67
  recipient: AztecAddress;
58
68
  }> {
59
- const recipient = (await this.wallet.createAccount()).address;
60
69
  const defaultAccountAddress = await this.setupAccount();
70
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
61
71
  const token = await this.setupToken(defaultAccountAddress);
62
72
  await this.mintTokens(token, defaultAccountAddress);
63
73
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
64
74
  }
65
75
 
66
76
  public async setupAmm(): Promise<{
67
- wallet: TestWallet;
77
+ wallet: EmbeddedWallet;
68
78
  defaultAccountAddress: AztecAddress;
69
79
  amm: AMMContract;
70
80
  token0: TokenContract;
@@ -94,6 +104,89 @@ export class BotFactory {
94
104
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
95
105
  }
96
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
+
97
190
  /**
98
191
  * Checks if the sender account contract is initialized, and initializes it if necessary.
99
192
  * @returns The sender wallet.
@@ -112,14 +205,9 @@ export class BotFactory {
112
205
  private async setupAccountWithPrivateKey(secret: Fr) {
113
206
  const salt = this.config.senderSalt ?? Fr.ONE;
114
207
  const signingKey = deriveSigningKey(secret);
115
- const accountData = {
116
- secret,
117
- salt,
118
- contract: new SchnorrAccountContract(signingKey!),
119
- };
120
- const accountManager = await this.wallet.createAccount(accountData);
121
- const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
122
- 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) {
123
211
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
124
212
  const timer = new Timer();
125
213
  const address = accountManager.address;
@@ -135,11 +223,22 @@ export class BotFactory {
135
223
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
136
224
  const deployMethod = await accountManager.getDeployMethod();
137
225
  const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
138
- const gasSettings = GasSettings.default({ maxFeesPerGas });
139
- const sentTx = deployMethod.send({ from: AztecAddress.ZERO, fee: { gasSettings, paymentMethod } });
140
- const txHash = await sentTx.getTxHash();
141
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
142
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
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
+ });
143
242
  this.log.info(`Account deployed at ${address}`);
144
243
 
145
244
  // Clean up the consumed bridge claim
@@ -151,12 +250,11 @@ export class BotFactory {
151
250
 
152
251
  private async setupTestAccount() {
153
252
  const [initialAccountData] = await getInitialTestAccountsData();
154
- const accountData = {
155
- secret: initialAccountData.secret,
156
- salt: initialAccountData.salt,
157
- contract: new SchnorrAccountContract(initialAccountData.signingKey),
158
- };
159
- const accountManager = await this.wallet.createAccount(accountData);
253
+ const accountManager = await this.wallet.createSchnorrAccount(
254
+ initialAccountData.secret,
255
+ initialAccountData.salt,
256
+ initialAccountData.signingKey,
257
+ );
160
258
  return accountManager.address;
161
259
  }
162
260
 
@@ -173,8 +271,11 @@ export class BotFactory {
173
271
  contractAddressSalt: this.config.tokenSalt,
174
272
  universalDeploy: true,
175
273
  };
274
+ let token: TokenContract | PrivateTokenContract;
176
275
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
177
276
  deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
277
+ tokenInstance = await deploy.getInstance(deployOpts);
278
+ token = TokenContract.at(tokenInstance.address, this.wallet);
178
279
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
179
280
  // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
180
281
  const tokenSecretKey = Fr.random();
@@ -186,22 +287,30 @@ export class BotFactory {
186
287
 
187
288
  // Register the contract with the secret key before deployment
188
289
  tokenInstance = await deploy.getInstance(deployOpts);
290
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
189
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];
190
294
  } else {
191
295
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
192
296
  }
193
297
 
194
298
  const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
195
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
299
+ const metadata = await this.wallet.getContractMetadata(address);
300
+ if (metadata.isContractPublished) {
196
301
  this.log.info(`Token at ${address.toString()} already deployed`);
197
- return deploy.register();
302
+ await deploy.register();
198
303
  } else {
199
304
  this.log.info(`Deploying token contract at ${address.toString()}`);
200
- const sentTx = deploy.send(deployOpts);
201
- const txHash = await sentTx.getTxHash();
202
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
203
- 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
+ });
204
312
  }
313
+ return token;
205
314
  }
206
315
 
207
316
  /**
@@ -209,7 +318,7 @@ export class BotFactory {
209
318
  * @param wallet - Wallet to deploy the token contract from.
210
319
  * @returns The TokenContract instance.
211
320
  */
212
- private setupTokenContract(
321
+ private async setupTokenContract(
213
322
  deployer: AztecAddress,
214
323
  contractAddressSalt: Fr,
215
324
  name: string,
@@ -218,7 +327,8 @@ export class BotFactory {
218
327
  ): Promise<TokenContract> {
219
328
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
220
329
  const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
221
- 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);
222
332
  }
223
333
 
224
334
  private async setupAmmContract(
@@ -230,12 +340,23 @@ export class BotFactory {
230
340
  ): Promise<AMMContract> {
231
341
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
232
342
  const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
233
- 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);
234
345
 
235
346
  this.log.info(`AMM deployed at ${amm.address}`);
236
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({ from: deployer });
237
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
238
- 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
+ });
239
360
  this.log.info(`Liquidity token initialized`);
240
361
 
241
362
  return amm;
@@ -251,9 +372,18 @@ export class BotFactory {
251
372
  ): Promise<void> {
252
373
  const getPrivateBalances = () =>
253
374
  Promise.all([
254
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
255
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
256
- 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),
257
387
  ]);
258
388
 
259
389
  const authwitNonce = Fr.random();
@@ -294,23 +424,44 @@ export class BotFactory {
294
424
  .getFunctionCall(),
295
425
  });
296
426
 
297
- const mintTx = new BatchCall(this.wallet, [
427
+ const mintBatch = new BatchCall(this.wallet, [
298
428
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
299
429
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
300
- ]).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
+ });
301
440
 
302
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
303
- await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
441
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`, { estimatedGas: mintGas });
304
442
 
305
- const addLiquidityTx = amm.methods
306
- .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
307
- .send({
308
- from: liquidityProvider,
309
- authWitnesses: [token0Authwit, token1Authwit],
310
- });
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
+ });
311
461
 
312
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
313
- 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
+ });
314
465
  this.log.info(`Liquidity added`);
315
466
 
316
467
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -323,18 +474,23 @@ export class BotFactory {
323
474
  name: string,
324
475
  deploy: DeployMethod<T>,
325
476
  deployOpts: DeployOptions,
326
- ): Promise<T> {
327
- const address = (await deploy.getInstance(deployOpts)).address;
328
- 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) {
329
482
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
330
- return deploy.register();
483
+ await deploy.register();
331
484
  } else {
332
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
333
- const sentTx = deploy.send(deployOpts);
334
- const txHash = await sentTx.getTxHash();
335
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
336
- 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
+ });
337
492
  }
493
+ return instance;
338
494
  }
339
495
 
340
496
  /**
@@ -370,10 +526,21 @@ export class BotFactory {
370
526
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
371
527
  return;
372
528
  }
373
- const sentTx = new BatchCall(token.wallet, calls).send({ from: minter });
374
- const txHash = await sentTx.getTxHash();
375
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
376
- 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
+ });
377
544
  }
378
545
 
379
546
  /**
@@ -393,7 +560,6 @@ export class BotFactory {
393
560
  await this.withNoMinTxsPerBlock(() =>
394
561
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
395
562
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
396
- forPublicConsumption: false,
397
563
  }),
398
564
  );
399
565
  return existingClaim.claim;
@@ -432,7 +598,6 @@ export class BotFactory {
432
598
  await this.withNoMinTxsPerBlock(() =>
433
599
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
434
600
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
435
- forPublicConsumption: false,
436
601
  }),
437
602
  );
438
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}`);