@aztec/bot 0.0.1-commit.87a0206 → 0.0.1-commit.88e6f9396

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 +4 -4
  2. package/dest/amm_bot.d.ts.map +1 -1
  3. package/dest/amm_bot.js +23 -16
  4. package/dest/base_bot.d.ts +4 -4
  5. package/dest/base_bot.d.ts.map +1 -1
  6. package/dest/base_bot.js +9 -10
  7. package/dest/bot.d.ts +4 -4
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +4 -3
  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 -5
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +187 -39
  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 +23 -18
  36. package/src/base_bot.ts +8 -18
  37. package/src/bot.ts +7 -6
  38. package/src/config.ts +40 -12
  39. package/src/cross_chain_bot.ts +203 -0
  40. package/src/factory.ts +191 -50
  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 {
@@ -9,8 +8,8 @@ import {
9
8
  type DeployOptions,
10
9
  NO_WAIT,
11
10
  } from '@aztec/aztec.js/contracts';
12
- import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
13
11
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
12
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
14
13
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
15
14
  import { deriveKeys } from '@aztec/aztec.js/keys';
16
15
  import { createLogger } from '@aztec/aztec.js/log';
@@ -18,18 +17,23 @@ import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
18
17
  import { waitForTx } from '@aztec/aztec.js/node';
19
18
  import { createEthereumChain } from '@aztec/ethereum/chain';
20
19
  import { createExtendedL1Client } from '@aztec/ethereum/client';
20
+ import { RollupContract } from '@aztec/ethereum/contracts';
21
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
21
22
  import { Fr } from '@aztec/foundation/curves/bn254';
23
+ import { EthAddress } from '@aztec/foundation/eth-address';
22
24
  import { Timer } from '@aztec/foundation/timer';
23
25
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
24
26
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
25
27
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
28
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
26
29
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
27
- import { GasSettings } from '@aztec/stdlib/gas';
30
+ import { GasFees, GasSettings } from '@aztec/stdlib/gas';
28
31
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
29
32
  import { deriveSigningKey } from '@aztec/stdlib/keys';
30
- import { TestWallet } from '@aztec/test-wallet/server';
33
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
31
34
 
32
35
  import { type BotConfig, SupportedTokenContracts } from './config.js';
36
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
33
37
  import type { BotStore } from './store/index.js';
34
38
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
35
39
 
@@ -41,32 +45,36 @@ export class BotFactory {
41
45
 
42
46
  constructor(
43
47
  private readonly config: BotConfig,
44
- private readonly wallet: TestWallet,
48
+ private readonly wallet: EmbeddedWallet,
45
49
  private readonly store: BotStore,
46
50
  private readonly aztecNode: AztecNode,
47
51
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
48
- ) {}
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
+ }
49
57
 
50
58
  /**
51
59
  * Initializes a new bot by setting up the sender account, registering the recipient,
52
60
  * deploying the token contract, and minting tokens if necessary.
53
61
  */
54
62
  public async setup(): Promise<{
55
- wallet: TestWallet;
63
+ wallet: EmbeddedWallet;
56
64
  defaultAccountAddress: AztecAddress;
57
65
  token: TokenContract | PrivateTokenContract;
58
66
  node: AztecNode;
59
67
  recipient: AztecAddress;
60
68
  }> {
61
69
  const defaultAccountAddress = await this.setupAccount();
62
- const recipient = (await this.wallet.createAccount()).address;
70
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
63
71
  const token = await this.setupToken(defaultAccountAddress);
64
72
  await this.mintTokens(token, defaultAccountAddress);
65
73
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
66
74
  }
67
75
 
68
76
  public async setupAmm(): Promise<{
69
- wallet: TestWallet;
77
+ wallet: EmbeddedWallet;
70
78
  defaultAccountAddress: AztecAddress;
71
79
  amm: AMMContract;
72
80
  token0: TokenContract;
@@ -96,6 +104,89 @@ export class BotFactory {
96
104
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
97
105
  }
98
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
+
99
190
  /**
100
191
  * Checks if the sender account contract is initialized, and initializes it if necessary.
101
192
  * @returns The sender wallet.
@@ -114,12 +205,7 @@ export class BotFactory {
114
205
  private async setupAccountWithPrivateKey(secret: Fr) {
115
206
  const salt = this.config.senderSalt ?? Fr.ONE;
116
207
  const signingKey = deriveSigningKey(secret);
117
- const accountData = {
118
- secret,
119
- salt,
120
- contract: new SchnorrAccountContract(signingKey!),
121
- };
122
- const accountManager = await this.wallet.createAccount(accountData);
208
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
123
209
  const metadata = await this.wallet.getContractMetadata(accountManager.address);
124
210
  if (metadata.isContractInitialized) {
125
211
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
@@ -137,15 +223,20 @@ export class BotFactory {
137
223
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
138
224
  const deployMethod = await accountManager.getDeployMethod();
139
225
  const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
140
- const gasSettings = GasSettings.default({ maxFeesPerGas });
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() });
141
232
 
142
233
  await this.withNoMinTxsPerBlock(async () => {
143
- const txHash = await deployMethod.send({
234
+ const { txHash } = await deployMethod.send({
144
235
  from: AztecAddress.ZERO,
145
236
  fee: { gasSettings, paymentMethod },
146
237
  wait: NO_WAIT,
147
238
  });
148
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
239
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`, { gasSettings });
149
240
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
150
241
  });
151
242
  this.log.info(`Account deployed at ${address}`);
@@ -159,12 +250,11 @@ export class BotFactory {
159
250
 
160
251
  private async setupTestAccount() {
161
252
  const [initialAccountData] = await getInitialTestAccountsData();
162
- const accountData = {
163
- secret: initialAccountData.secret,
164
- salt: initialAccountData.salt,
165
- contract: new SchnorrAccountContract(initialAccountData.signingKey),
166
- };
167
- const accountManager = await this.wallet.createAccount(accountData);
253
+ const accountManager = await this.wallet.createSchnorrAccount(
254
+ initialAccountData.secret,
255
+ initialAccountData.salt,
256
+ initialAccountData.signingKey,
257
+ );
168
258
  return accountManager.address;
169
259
  }
170
260
 
@@ -199,6 +289,8 @@ export class BotFactory {
199
289
  tokenInstance = await deploy.getInstance(deployOpts);
200
290
  token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
201
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];
202
294
  } else {
203
295
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
204
296
  }
@@ -210,8 +302,9 @@ export class BotFactory {
210
302
  await deploy.register();
211
303
  } else {
212
304
  this.log.info(`Deploying token contract at ${address.toString()}`);
213
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
214
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
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 });
215
308
  await this.withNoMinTxsPerBlock(async () => {
216
309
  await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
217
310
  return token;
@@ -251,10 +344,19 @@ export class BotFactory {
251
344
  const amm = AMMContract.at(instance.address, this.wallet);
252
345
 
253
346
  this.log.info(`AMM deployed at ${amm.address}`);
254
- const minterReceipt = await lpToken.methods
255
- .set_minter(amm.address, true)
256
- .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
257
- this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
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
+ });
258
360
  this.log.info(`Liquidity token initialized`);
259
361
 
260
362
  return amm;
@@ -270,9 +372,18 @@ export class BotFactory {
270
372
  ): Promise<void> {
271
373
  const getPrivateBalances = () =>
272
374
  Promise.all([
273
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
274
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
275
- 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),
276
387
  ]);
277
388
 
278
389
  const authwitNonce = Fr.random();
@@ -313,22 +424,44 @@ export class BotFactory {
313
424
  .getFunctionCall(),
314
425
  });
315
426
 
316
- const mintReceipt = await new BatchCall(this.wallet, [
427
+ const mintBatch = new BatchCall(this.wallet, [
317
428
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
318
429
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
319
- ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
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
+ });
320
440
 
321
- this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
441
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`, { estimatedGas: mintGas });
322
442
 
323
- const addLiquidityReceipt = await amm.methods
324
- .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
325
- .send({
326
- from: liquidityProvider,
327
- authWitnesses: [token0Authwit, token1Authwit],
328
- wait: { timeout: this.config.txMinedWaitSeconds },
329
- });
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
+ });
330
461
 
331
- this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
462
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`, {
463
+ estimatedGas: addLiquidityGas,
464
+ });
332
465
  this.log.info(`Liquidity added`);
333
466
 
334
467
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -349,9 +482,10 @@ export class BotFactory {
349
482
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
350
483
  await deploy.register();
351
484
  } else {
352
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
485
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
486
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
353
487
  await this.withNoMinTxsPerBlock(async () => {
354
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
488
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
355
489
  this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
356
490
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
357
491
  });
@@ -393,9 +527,18 @@ export class BotFactory {
393
527
  return;
394
528
  }
395
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 });
396
534
  await this.withNoMinTxsPerBlock(async () => {
397
- const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
398
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
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 });
399
542
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
400
543
  });
401
544
  }
@@ -417,7 +560,6 @@ export class BotFactory {
417
560
  await this.withNoMinTxsPerBlock(() =>
418
561
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
419
562
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
420
- forPublicConsumption: false,
421
563
  }),
422
564
  );
423
565
  return existingClaim.claim;
@@ -456,7 +598,6 @@ export class BotFactory {
456
598
  await this.withNoMinTxsPerBlock(() =>
457
599
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
458
600
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
459
- forPublicConsumption: false,
460
601
  }),
461
602
  );
462
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}`);
@@ -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';
package/src/utils.ts CHANGED
@@ -15,8 +15,8 @@ export async function getBalances(
15
15
  who: AztecAddress,
16
16
  from?: AztecAddress,
17
17
  ): Promise<{ privateBalance: bigint; publicBalance: bigint }> {
18
- const privateBalance = await token.methods.balance_of_private(who).simulate({ from: from ?? who });
19
- const publicBalance = await token.methods.balance_of_public(who).simulate({ from: from ?? who });
18
+ const { result: privateBalance } = await token.methods.balance_of_private(who).simulate({ from: from ?? who });
19
+ const { result: publicBalance } = await token.methods.balance_of_public(who).simulate({ from: from ?? who });
20
20
  return { privateBalance, publicBalance };
21
21
  }
22
22
 
@@ -25,7 +25,7 @@ export async function getPrivateBalance(
25
25
  who: AztecAddress,
26
26
  from?: AztecAddress,
27
27
  ): Promise<bigint> {
28
- const privateBalance = await token.methods.get_balance(who).simulate({ from: from ?? who });
28
+ const { result: privateBalance } = await token.methods.get_balance(who).simulate({ from: from ?? who });
29
29
  return privateBalance;
30
30
  }
31
31