@aztec/bot 0.0.1-commit.d431d1c → 0.0.1-commit.db765a8

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 +30 -14
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +36 -8
  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 +141 -0
  16. package/dest/factory.d.ts +20 -10
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +161 -69
  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 +30 -5
  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 +16 -13
  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 +39 -10
  39. package/src/cross_chain_bot.ts +210 -0
  40. package/src/factory.ts +184 -55
  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
30
  import { 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
69
  const defaultAccountAddress = await this.setupAccount();
60
- const recipient = (await this.wallet.createAccount()).address;
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,94 @@ 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
+ // Use forPublicConsumption: false so we wait until the message is in the current world
166
+ // state. With true, it returns one block early which causes gas estimation simulation to
167
+ // fail since it runs against the current state.
168
+ // See https://linear.app/aztec-labs/issue/A-548 for details.
169
+ forPublicConsumption: false,
170
+ });
171
+ this.log.info(`First L1→L2 message is ready`);
172
+ }
173
+
174
+ return {
175
+ wallet: this.wallet,
176
+ defaultAccountAddress,
177
+ contract,
178
+ node: this.aztecNode,
179
+ l1Client,
180
+ rollupVersion,
181
+ };
182
+ }
183
+
184
+ private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
185
+ const deployOpts: DeployOptions = {
186
+ from: deployer,
187
+ contractAddressSalt: this.config.tokenSalt,
188
+ universalDeploy: true,
189
+ };
190
+ const deploy = TestContract.deploy(this.wallet);
191
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
192
+ return TestContract.at(instance.address, this.wallet);
193
+ }
194
+
97
195
  /**
98
196
  * Checks if the sender account contract is initialized, and initializes it if necessary.
99
197
  * @returns The sender wallet.
@@ -112,12 +210,7 @@ export class BotFactory {
112
210
  private async setupAccountWithPrivateKey(secret: Fr) {
113
211
  const salt = this.config.senderSalt ?? Fr.ONE;
114
212
  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);
213
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
121
214
  const metadata = await this.wallet.getContractMetadata(accountManager.address);
122
215
  if (metadata.isContractInitialized) {
123
216
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
@@ -136,10 +229,16 @@ export class BotFactory {
136
229
  const deployMethod = await accountManager.getDeployMethod();
137
230
  const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
138
231
  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 }));
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()}`);
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,7 +287,10 @@ 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
  }
@@ -195,14 +299,17 @@ export class BotFactory {
195
299
  const metadata = await this.wallet.getContractMetadata(address);
196
300
  if (metadata.isContractPublished) {
197
301
  this.log.info(`Token at ${address.toString()} already deployed`);
198
- return deploy.register();
302
+ await deploy.register();
199
303
  } else {
200
304
  this.log.info(`Deploying token contract at ${address.toString()}`);
201
- const sentTx = deploy.send(deployOpts);
202
- const txHash = await sentTx.getTxHash();
305
+ const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
203
306
  this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
204
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
307
+ await this.withNoMinTxsPerBlock(async () => {
308
+ await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
309
+ return token;
310
+ });
205
311
  }
312
+ return token;
206
313
  }
207
314
 
208
315
  /**
@@ -210,7 +317,7 @@ export class BotFactory {
210
317
  * @param wallet - Wallet to deploy the token contract from.
211
318
  * @returns The TokenContract instance.
212
319
  */
213
- private setupTokenContract(
320
+ private async setupTokenContract(
214
321
  deployer: AztecAddress,
215
322
  contractAddressSalt: Fr,
216
323
  name: string,
@@ -219,7 +326,8 @@ export class BotFactory {
219
326
  ): Promise<TokenContract> {
220
327
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
221
328
  const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
222
- return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
329
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
330
+ return TokenContract.at(instance.address, this.wallet);
223
331
  }
224
332
 
225
333
  private async setupAmmContract(
@@ -231,12 +339,14 @@ export class BotFactory {
231
339
  ): Promise<AMMContract> {
232
340
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
233
341
  const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
234
- const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
342
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
343
+ const amm = AMMContract.at(instance.address, this.wallet);
235
344
 
236
345
  this.log.info(`AMM deployed at ${amm.address}`);
237
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({ from: deployer });
238
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
239
- await minterTx.wait({ timeout: this.config.txMinedWaitSeconds });
346
+ const { receipt: minterReceipt } = await lpToken.methods
347
+ .set_minter(amm.address, true)
348
+ .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
349
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
240
350
  this.log.info(`Liquidity token initialized`);
241
351
 
242
352
  return amm;
@@ -252,9 +362,18 @@ export class BotFactory {
252
362
  ): Promise<void> {
253
363
  const getPrivateBalances = () =>
254
364
  Promise.all([
255
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
256
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
257
- lpToken.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
365
+ token0.methods
366
+ .balance_of_private(liquidityProvider)
367
+ .simulate({ from: liquidityProvider })
368
+ .then(r => r.result),
369
+ token1.methods
370
+ .balance_of_private(liquidityProvider)
371
+ .simulate({ from: liquidityProvider })
372
+ .then(r => r.result),
373
+ lpToken.methods
374
+ .balance_of_private(liquidityProvider)
375
+ .simulate({ from: liquidityProvider })
376
+ .then(r => r.result),
258
377
  ]);
259
378
 
260
379
  const authwitNonce = Fr.random();
@@ -295,23 +414,22 @@ export class BotFactory {
295
414
  .getFunctionCall(),
296
415
  });
297
416
 
298
- const mintTx = new BatchCall(this.wallet, [
417
+ const { receipt: mintReceipt } = await new BatchCall(this.wallet, [
299
418
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
300
419
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
301
- ]).send({ from: liquidityProvider });
420
+ ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
302
421
 
303
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
304
- await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
422
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
305
423
 
306
- const addLiquidityTx = amm.methods
424
+ const { receipt: addLiquidityReceipt } = await amm.methods
307
425
  .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
308
426
  .send({
309
427
  from: liquidityProvider,
310
428
  authWitnesses: [token0Authwit, token1Authwit],
429
+ wait: { timeout: this.config.txMinedWaitSeconds },
311
430
  });
312
431
 
313
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
314
- await addLiquidityTx.wait({ timeout: this.config.txMinedWaitSeconds });
432
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
315
433
  this.log.info(`Liquidity added`);
316
434
 
317
435
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -324,19 +442,22 @@ export class BotFactory {
324
442
  name: string,
325
443
  deploy: DeployMethod<T>,
326
444
  deployOpts: DeployOptions,
327
- ): Promise<T> {
328
- const address = (await deploy.getInstance(deployOpts)).address;
445
+ ): Promise<ContractInstanceWithAddress> {
446
+ const instance = await deploy.getInstance(deployOpts);
447
+ const address = instance.address;
329
448
  const metadata = await this.wallet.getContractMetadata(address);
330
449
  if (metadata.isContractPublished) {
331
450
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
332
- return deploy.register();
451
+ await deploy.register();
333
452
  } else {
334
453
  this.log.info(`Deploying contract ${name} at ${address.toString()}`);
335
- const sentTx = deploy.send(deployOpts);
336
- const txHash = await sentTx.getTxHash();
337
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
338
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
454
+ await this.withNoMinTxsPerBlock(async () => {
455
+ const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
456
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
457
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
458
+ });
339
459
  }
460
+ return instance;
340
461
  }
341
462
 
342
463
  /**
@@ -372,10 +493,18 @@ export class BotFactory {
372
493
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
373
494
  return;
374
495
  }
375
- const sentTx = new BatchCall(token.wallet, calls).send({ from: minter });
376
- const txHash = await sentTx.getTxHash();
377
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
378
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
496
+
497
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
498
+ const additionalScopes = isStandardToken ? undefined : [token.address];
499
+ await this.withNoMinTxsPerBlock(async () => {
500
+ const { txHash } = await new BatchCall(token.wallet, calls).send({
501
+ from: minter,
502
+ additionalScopes,
503
+ wait: NO_WAIT,
504
+ });
505
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
506
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
507
+ });
379
508
  }
380
509
 
381
510
  /**
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}`);
@@ -2,6 +2,7 @@ import { AztecAddress } from '@aztec/aztec.js/addresses';
2
2
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
3
3
  import { Fr } from '@aztec/foundation/curves/bn254';
4
4
  import { type Logger, createLogger } from '@aztec/foundation/log';
5
+ import { DateProvider } from '@aztec/foundation/timer';
5
6
  import type { AztecAsyncKVStore, AztecAsyncMap } from '@aztec/kv-store';
6
7
 
7
8
  export interface BridgeClaimData {
@@ -10,18 +11,38 @@ export interface BridgeClaimData {
10
11
  recipient: string;
11
12
  }
12
13
 
14
+ export interface PendingL1ToL2Message {
15
+ /** Random content field sent in the message. */
16
+ content: string;
17
+ /** Secret for consuming the message. */
18
+ secret: string;
19
+ /** Hash of the secret. */
20
+ secretHash: string;
21
+ /** Hash of the L1→L2 message. */
22
+ msgHash: string;
23
+ /** L1 sender address (hex). */
24
+ sender: string;
25
+ /** Global leaf index in the L1→L2 message tree. */
26
+ globalLeafIndex: string;
27
+ /** Timestamp when the message was seeded. */
28
+ timestamp: number;
29
+ }
30
+
13
31
  /**
14
32
  * Simple data store for the bot to persist L1 bridge claims.
15
33
  */
16
34
  export class BotStore {
17
35
  public static readonly SCHEMA_VERSION = 1;
18
36
  private readonly bridgeClaims: AztecAsyncMap<string, string>;
37
+ private readonly pendingL1ToL2: AztecAsyncMap<string, string>;
19
38
 
20
39
  constructor(
21
40
  private readonly store: AztecAsyncKVStore,
22
41
  private readonly log: Logger = createLogger('bot:store'),
42
+ private readonly dateProvider: DateProvider = new DateProvider(),
23
43
  ) {
24
44
  this.bridgeClaims = store.openMap<string, string>('bridge_claims');
45
+ this.pendingL1ToL2 = store.openMap<string, string>('pending_l1_to_l2');
25
46
  }
26
47
 
27
48
  /**
@@ -39,7 +60,7 @@ export class BotStore {
39
60
 
40
61
  const data = {
41
62
  claim: serializableClaim,
42
- timestamp: Date.now(),
63
+ timestamp: this.dateProvider.now(),
43
64
  recipient: recipient.toString(),
44
65
  };
45
66
 
@@ -115,7 +136,7 @@ export class BotStore {
115
136
  * Cleans up old bridge claims (older than 24 hours).
116
137
  */
117
138
  public async cleanupOldClaims(maxAgeMs: number = 24 * 60 * 60 * 1000): Promise<number> {
118
- const now = Date.now();
139
+ const now = this.dateProvider.now();
119
140
  let cleanedCount = 0;
120
141
  const entries = this.bridgeClaims.entriesAsync();
121
142
 
@@ -131,9 +152,43 @@ export class BotStore {
131
152
  return cleanedCount;
132
153
  }
133
154
 
134
- /**
135
- * Closes the store.
136
- */
155
+ /** Saves a pending L1→L2 message keyed by msgHash. */
156
+ public async savePendingL1ToL2Message(msg: PendingL1ToL2Message): Promise<void> {
157
+ await this.pendingL1ToL2.set(msg.msgHash, JSON.stringify(msg));
158
+ this.log.info(`Saved pending L1→L2 message ${msg.msgHash}`);
159
+ }
160
+
161
+ /** Returns all unconsumed pending L1→L2 messages. */
162
+ public async getUnconsumedL1ToL2Messages(): Promise<PendingL1ToL2Message[]> {
163
+ const messages: PendingL1ToL2Message[] = [];
164
+ for await (const [_, data] of this.pendingL1ToL2.entriesAsync()) {
165
+ messages.push(JSON.parse(data));
166
+ }
167
+ return messages;
168
+ }
169
+
170
+ /** Deletes a consumed L1→L2 message from the store. */
171
+ public async deleteL1ToL2Message(msgHash: string): Promise<void> {
172
+ await this.pendingL1ToL2.delete(msgHash);
173
+ this.log.info(`Deleted consumed L1→L2 message ${msgHash}`);
174
+ }
175
+
176
+ /** Cleans up pending L1→L2 messages older than maxAgeMs. */
177
+ public async cleanupOldPendingMessages(maxAgeMs: number = 24 * 60 * 60 * 1000): Promise<number> {
178
+ const now = this.dateProvider.now();
179
+ let cleanedCount = 0;
180
+ for await (const [key, data] of this.pendingL1ToL2.entriesAsync()) {
181
+ const parsed = JSON.parse(data);
182
+ if (now - parsed.timestamp > maxAgeMs) {
183
+ await this.pendingL1ToL2.delete(key);
184
+ cleanedCount++;
185
+ this.log.info(`Cleaned up old pending L1→L2 message ${key}`);
186
+ }
187
+ }
188
+ return cleanedCount;
189
+ }
190
+
191
+ /** Closes the store. */
137
192
  public async close(): Promise<void> {
138
193
  await this.store.close();
139
194
  this.log.info('Closed bot data store');
@@ -1 +1 @@
1
- export { BotStore, type BridgeClaimData } from './bot_store.js';
1
+ export { BotStore, type BridgeClaimData, type PendingL1ToL2Message } from './bot_store.js';