@aztec/bot 0.0.1-commit.f2ce05ee → 0.0.1-commit.f81dbcf

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 +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 +134 -0
  16. package/dest/factory.d.ts +20 -5
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +87 -28
  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 +39 -10
  39. package/src/cross_chain_bot.ts +203 -0
  40. package/src/factory.ts +127 -31
  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
@@ -0,0 +1,203 @@
1
+ /**
2
+ * CrossChainBot exercises L2->L1 and L1->L2 messaging.
3
+ *
4
+ * createAndSendTx onTxMined
5
+ * ────────────────────────────────────── ──────────────────────────────
6
+ *
7
+ * 1. SEED (fire-and-forget) 3. VERIFY L2->L1
8
+ * if store has fewer pending messages Query getTxEffect, confirm
9
+ * than seedCount and no seed is the expected L2->L1 messages
10
+ * in-flight: appeared in tx effects.
11
+ * * kick off L1 inbox tx
12
+ * * store msg on completion
13
+ *
14
+ * 2. BUILD & SEND BATCH
15
+ * Always:
16
+ * N x create_l2_to_l1_message
17
+ * (random content, fixed
18
+ * L1 recipient)
19
+ * If a ready L1->L2 msg exists:
20
+ * 1 x consume_message_from_
21
+ * arbitrary_sender_public
22
+ * delete consumed msg from store
23
+ * Send batch tx (no wait)
24
+ *
25
+ */
26
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
27
+ import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
28
+ import { isL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
29
+ import { TxHash, TxReceipt } from '@aztec/aztec.js/tx';
30
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
31
+ import { Fr } from '@aztec/foundation/curves/bn254';
32
+ import { EthAddress } from '@aztec/foundation/eth-address';
33
+ import type { TestContract } from '@aztec/noir-test-contracts.js/Test';
34
+ import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
35
+ import type { EmbeddedWallet } from '@aztec/wallets/embedded';
36
+
37
+ import { BaseBot } from './base_bot.js';
38
+ import type { BotConfig } from './config.js';
39
+ import { BotFactory } from './factory.js';
40
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
41
+ import type { BotStore, PendingL1ToL2Message } from './store/index.js';
42
+
43
+ /** Stale message threshold: messages older than this are removed. */
44
+ const STALE_MESSAGE_THRESHOLD_MS = 2 * 60 * 60 * 1000; // 2 hours
45
+
46
+ /** Bot that exercises both L2→L1 and L1→L2 cross-chain messaging. */
47
+ export class CrossChainBot extends BaseBot {
48
+ private l2ToL1Sent = 0;
49
+ private l1ToL2Consumed = 0;
50
+ private pendingSeedPromise: Promise<void> | undefined;
51
+
52
+ protected constructor(
53
+ node: AztecNode,
54
+ wallet: EmbeddedWallet,
55
+ defaultAccountAddress: AztecAddress,
56
+ private readonly contract: TestContract,
57
+ private readonly l1Client: ExtendedViemWalletClient,
58
+ private readonly l1Recipient: EthAddress,
59
+ private readonly inboxAddress: EthAddress,
60
+ private readonly rollupVersion: bigint,
61
+ private readonly store: BotStore,
62
+ config: BotConfig,
63
+ ) {
64
+ super(node, wallet, defaultAccountAddress, config);
65
+ }
66
+
67
+ static async create(
68
+ config: BotConfig,
69
+ wallet: EmbeddedWallet,
70
+ aztecNode: AztecNode,
71
+ aztecNodeAdmin: AztecNodeAdmin | undefined,
72
+ store: BotStore,
73
+ ): Promise<CrossChainBot> {
74
+ if (config.followChain === 'NONE') {
75
+ throw new Error(`CrossChainBot requires followChain to be set (got NONE)`);
76
+ }
77
+ const factory = new BotFactory(config, wallet, store, aztecNode, aztecNodeAdmin);
78
+ const { defaultAccountAddress, contract, l1Client, rollupVersion } = await factory.setupCrossChain();
79
+ const l1Recipient = EthAddress.fromString(l1Client.account!.address);
80
+ const { l1ContractAddresses } = await aztecNode.getNodeInfo();
81
+ const inboxAddress = EthAddress.fromString(l1ContractAddresses.inboxAddress.toString());
82
+ return new CrossChainBot(
83
+ aztecNode,
84
+ wallet,
85
+ defaultAccountAddress,
86
+ contract,
87
+ l1Client,
88
+ l1Recipient,
89
+ inboxAddress,
90
+ rollupVersion,
91
+ store,
92
+ config,
93
+ );
94
+ }
95
+
96
+ protected async createAndSendTx(logCtx: object): Promise<TxHash> {
97
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
98
+
99
+ // Send an L1→L2 message if we're below the threshold and not already seeding one
100
+ if (pendingMessages.length < this.config.l1ToL2SeedCount && !this.pendingSeedPromise) {
101
+ this.pendingSeedPromise = this.seedNewL1ToL2Message()
102
+ .catch(err => this.log.warn(`Failed to seed L1→L2 message: ${err}`, logCtx))
103
+ .finally(() => {
104
+ this.pendingSeedPromise = undefined;
105
+ });
106
+ }
107
+
108
+ // Build batch: always L2→L1, optionally consume L1→L2
109
+ const calls = [];
110
+
111
+ // L2→L1: create messages with random content
112
+ for (let i = 0; i < this.config.l2ToL1MessagesPerTx; i++) {
113
+ calls.push(
114
+ this.contract.methods.create_l2_to_l1_message_arbitrary_recipient_public(Fr.random(), this.l1Recipient),
115
+ );
116
+ }
117
+
118
+ // L1→L2: consume oldest ready message if available
119
+ const readyMsg = await this.getReadyL1ToL2Message(pendingMessages);
120
+ if (readyMsg) {
121
+ calls.push(
122
+ this.contract.methods.consume_message_from_arbitrary_sender_public(
123
+ Fr.fromHexString(readyMsg.content),
124
+ Fr.fromHexString(readyMsg.secret),
125
+ EthAddress.fromString(readyMsg.sender),
126
+ new Fr(BigInt(readyMsg.globalLeafIndex)),
127
+ ),
128
+ );
129
+ // Delete consumed message immediately so it works with FOLLOW_CHAIN=NONE
130
+ await this.store.deleteL1ToL2Message(readyMsg.msgHash);
131
+ this.l1ToL2Consumed++;
132
+ } else {
133
+ this.log.warn(`No ready L1→L2 message to consume`, {
134
+ ...logCtx,
135
+ pendingCount: pendingMessages.length,
136
+ });
137
+ }
138
+
139
+ const batch = new BatchCall(this.wallet, calls);
140
+ const opts = await this.getSendMethodOpts(batch);
141
+
142
+ this.log.verbose(`Sending cross-chain batch with ${calls.length} calls`, logCtx);
143
+ const { txHash } = await batch.send({ ...opts, wait: NO_WAIT });
144
+ return txHash;
145
+ }
146
+
147
+ protected override async onTxMined(receipt: TxReceipt, logCtx: object): Promise<void> {
148
+ // Verify L2→L1 messages appeared in this tx's effects
149
+ const indexed = await this.node.getTxEffect(receipt.txHash);
150
+ if (indexed) {
151
+ const l2ToL1Msgs = indexed.data.l2ToL1Msgs.filter(m => !m.isZero());
152
+ if (l2ToL1Msgs.length >= this.config.l2ToL1MessagesPerTx) {
153
+ this.l2ToL1Sent += l2ToL1Msgs.length;
154
+ } else {
155
+ this.log.error(`Expected ${this.config.l2ToL1MessagesPerTx} L2→L1 messages but found ${l2ToL1Msgs.length}`, {
156
+ ...logCtx,
157
+ blockNumber: receipt.blockNumber,
158
+ txHash: receipt.txHash.toString(),
159
+ });
160
+ }
161
+ }
162
+
163
+ const pendingCount = (await this.store.getUnconsumedL1ToL2Messages()).length;
164
+ this.log.info(`CrossChainBot txs mined`, {
165
+ ...logCtx,
166
+ l2ToL1Sent: this.l2ToL1Sent,
167
+ l1ToL2Consumed: this.l1ToL2Consumed,
168
+ l1ToL2Pending: pendingCount,
169
+ });
170
+ }
171
+
172
+ /** Finds the oldest pending message that is ready for consumption. */
173
+ private async getReadyL1ToL2Message(
174
+ pendingMessages: PendingL1ToL2Message[],
175
+ ): Promise<PendingL1ToL2Message | undefined> {
176
+ const now = Date.now();
177
+ for (const msg of pendingMessages) {
178
+ const ready = await isL1ToL2MessageReady(this.node, Fr.fromHexString(msg.msgHash));
179
+ if (ready) {
180
+ return msg;
181
+ }
182
+
183
+ // Time-based stale detection: if the message is old and still not ready, remove it
184
+ if (now - msg.timestamp > STALE_MESSAGE_THRESHOLD_MS) {
185
+ await this.store.deleteL1ToL2Message(msg.msgHash);
186
+ this.log.warn(`Removed stale L1→L2 message ${msg.msgHash}`);
187
+ }
188
+ }
189
+ return undefined;
190
+ }
191
+
192
+ /** Seeds a new L1→L2 message on L1 and stores it. */
193
+ private async seedNewL1ToL2Message(): Promise<void> {
194
+ await seedL1ToL2Message(
195
+ this.l1Client,
196
+ this.inboxAddress,
197
+ this.contract.address,
198
+ this.rollupVersion,
199
+ this.store,
200
+ this.log,
201
+ );
202
+ }
203
+ }
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
30
  import { 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,7 +45,7 @@ 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,
@@ -52,21 +56,21 @@ export class BotFactory {
52
56
  * deploying the token contract, and minting tokens if necessary.
53
57
  */
54
58
  public async setup(): Promise<{
55
- wallet: TestWallet;
59
+ wallet: EmbeddedWallet;
56
60
  defaultAccountAddress: AztecAddress;
57
61
  token: TokenContract | PrivateTokenContract;
58
62
  node: AztecNode;
59
63
  recipient: AztecAddress;
60
64
  }> {
61
65
  const defaultAccountAddress = await this.setupAccount();
62
- const recipient = (await this.wallet.createAccount()).address;
66
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
63
67
  const token = await this.setupToken(defaultAccountAddress);
64
68
  await this.mintTokens(token, defaultAccountAddress);
65
69
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
66
70
  }
67
71
 
68
72
  public async setupAmm(): Promise<{
69
- wallet: TestWallet;
73
+ wallet: EmbeddedWallet;
70
74
  defaultAccountAddress: AztecAddress;
71
75
  amm: AMMContract;
72
76
  token0: TokenContract;
@@ -96,6 +100,89 @@ export class BotFactory {
96
100
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
97
101
  }
98
102
 
103
+ /**
104
+ * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
105
+ * seeding initial L1→L2 messages, and waiting for the first to be ready.
106
+ */
107
+ public async setupCrossChain(): Promise<{
108
+ wallet: EmbeddedWallet;
109
+ defaultAccountAddress: AztecAddress;
110
+ contract: TestContract;
111
+ node: AztecNode;
112
+ l1Client: ExtendedViemWalletClient;
113
+ rollupVersion: bigint;
114
+ }> {
115
+ const defaultAccountAddress = await this.setupAccount();
116
+
117
+ // Create L1 client (same pattern as bridgeL1FeeJuice)
118
+ const l1RpcUrls = this.config.l1RpcUrls;
119
+ if (!l1RpcUrls?.length) {
120
+ throw new Error('L1 RPC URLs required for cross-chain bot');
121
+ }
122
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
123
+ if (!mnemonicOrPrivateKey) {
124
+ throw new Error('L1 mnemonic or private key required for cross-chain bot');
125
+ }
126
+ const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
127
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
128
+ const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
129
+
130
+ // Fetch Rollup version (needed for Inbox L2Actor struct)
131
+ const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
132
+ const rollupVersion = await rollupContract.getVersion();
133
+
134
+ // Deploy TestContract
135
+ const contract = await this.setupTestContract(defaultAccountAddress);
136
+
137
+ // Recover any pending messages from store (clean up stale ones first)
138
+ await this.store.cleanupOldPendingMessages();
139
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
140
+
141
+ // Seed initial L1→L2 messages if pipeline is empty
142
+ const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
143
+ for (let i = 0; i < seedCount; i++) {
144
+ await seedL1ToL2Message(
145
+ l1Client,
146
+ EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()),
147
+ contract.address,
148
+ rollupVersion,
149
+ this.store,
150
+ this.log,
151
+ );
152
+ }
153
+
154
+ // Block until at least one message is ready
155
+ const allMessages = await this.store.getUnconsumedL1ToL2Messages();
156
+ if (allMessages.length > 0) {
157
+ this.log.info(`Waiting for first L1→L2 message to be ready...`);
158
+ const firstMsg = allMessages[0];
159
+ await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
160
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
161
+ });
162
+ this.log.info(`First L1→L2 message is ready`);
163
+ }
164
+
165
+ return {
166
+ wallet: this.wallet,
167
+ defaultAccountAddress,
168
+ contract,
169
+ node: this.aztecNode,
170
+ l1Client,
171
+ rollupVersion,
172
+ };
173
+ }
174
+
175
+ private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
176
+ const deployOpts: DeployOptions = {
177
+ from: deployer,
178
+ contractAddressSalt: this.config.tokenSalt,
179
+ universalDeploy: true,
180
+ };
181
+ const deploy = TestContract.deploy(this.wallet);
182
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
183
+ return TestContract.at(instance.address, this.wallet);
184
+ }
185
+
99
186
  /**
100
187
  * Checks if the sender account contract is initialized, and initializes it if necessary.
101
188
  * @returns The sender wallet.
@@ -114,12 +201,7 @@ export class BotFactory {
114
201
  private async setupAccountWithPrivateKey(secret: Fr) {
115
202
  const salt = this.config.senderSalt ?? Fr.ONE;
116
203
  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);
204
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
123
205
  const metadata = await this.wallet.getContractMetadata(accountManager.address);
124
206
  if (metadata.isContractInitialized) {
125
207
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
@@ -140,7 +222,7 @@ export class BotFactory {
140
222
  const gasSettings = GasSettings.default({ maxFeesPerGas });
141
223
 
142
224
  await this.withNoMinTxsPerBlock(async () => {
143
- const txHash = await deployMethod.send({
225
+ const { txHash } = await deployMethod.send({
144
226
  from: AztecAddress.ZERO,
145
227
  fee: { gasSettings, paymentMethod },
146
228
  wait: NO_WAIT,
@@ -159,12 +241,11 @@ export class BotFactory {
159
241
 
160
242
  private async setupTestAccount() {
161
243
  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);
244
+ const accountManager = await this.wallet.createSchnorrAccount(
245
+ initialAccountData.secret,
246
+ initialAccountData.salt,
247
+ initialAccountData.signingKey,
248
+ );
168
249
  return accountManager.address;
169
250
  }
170
251
 
@@ -199,6 +280,8 @@ export class BotFactory {
199
280
  tokenInstance = await deploy.getInstance(deployOpts);
200
281
  token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
201
282
  await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
283
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
284
+ deployOpts.additionalScopes = [tokenInstance.address];
202
285
  } else {
203
286
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
204
287
  }
@@ -210,7 +293,7 @@ export class BotFactory {
210
293
  await deploy.register();
211
294
  } else {
212
295
  this.log.info(`Deploying token contract at ${address.toString()}`);
213
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
296
+ const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
214
297
  this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
215
298
  await this.withNoMinTxsPerBlock(async () => {
216
299
  await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
@@ -251,7 +334,7 @@ export class BotFactory {
251
334
  const amm = AMMContract.at(instance.address, this.wallet);
252
335
 
253
336
  this.log.info(`AMM deployed at ${amm.address}`);
254
- const minterReceipt = await lpToken.methods
337
+ const { receipt: minterReceipt } = await lpToken.methods
255
338
  .set_minter(amm.address, true)
256
339
  .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
257
340
  this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
@@ -270,9 +353,18 @@ export class BotFactory {
270
353
  ): Promise<void> {
271
354
  const getPrivateBalances = () =>
272
355
  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 }),
356
+ token0.methods
357
+ .balance_of_private(liquidityProvider)
358
+ .simulate({ from: liquidityProvider })
359
+ .then(r => r.result),
360
+ token1.methods
361
+ .balance_of_private(liquidityProvider)
362
+ .simulate({ from: liquidityProvider })
363
+ .then(r => r.result),
364
+ lpToken.methods
365
+ .balance_of_private(liquidityProvider)
366
+ .simulate({ from: liquidityProvider })
367
+ .then(r => r.result),
276
368
  ]);
277
369
 
278
370
  const authwitNonce = Fr.random();
@@ -313,14 +405,14 @@ export class BotFactory {
313
405
  .getFunctionCall(),
314
406
  });
315
407
 
316
- const mintReceipt = await new BatchCall(this.wallet, [
408
+ const { receipt: mintReceipt } = await new BatchCall(this.wallet, [
317
409
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
318
410
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
319
411
  ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
320
412
 
321
413
  this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
322
414
 
323
- const addLiquidityReceipt = await amm.methods
415
+ const { receipt: addLiquidityReceipt } = await amm.methods
324
416
  .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
325
417
  .send({
326
418
  from: liquidityProvider,
@@ -351,7 +443,7 @@ export class BotFactory {
351
443
  } else {
352
444
  this.log.info(`Deploying contract ${name} at ${address.toString()}`);
353
445
  await this.withNoMinTxsPerBlock(async () => {
354
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
446
+ const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
355
447
  this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
356
448
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
357
449
  });
@@ -393,8 +485,14 @@ export class BotFactory {
393
485
  return;
394
486
  }
395
487
 
488
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
489
+ const additionalScopes = isStandardToken ? undefined : [token.address];
396
490
  await this.withNoMinTxsPerBlock(async () => {
397
- const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
491
+ const { txHash } = await new BatchCall(token.wallet, calls).send({
492
+ from: minter,
493
+ additionalScopes,
494
+ wait: NO_WAIT,
495
+ });
398
496
  this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
399
497
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
400
498
  });
@@ -417,7 +515,6 @@ export class BotFactory {
417
515
  await this.withNoMinTxsPerBlock(() =>
418
516
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
419
517
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
420
- forPublicConsumption: false,
421
518
  }),
422
519
  );
423
520
  return existingClaim.claim;
@@ -456,7 +553,6 @@ export class BotFactory {
456
553
  await this.withNoMinTxsPerBlock(() =>
457
554
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
458
555
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
459
- forPublicConsumption: false,
460
556
  }),
461
557
  );
462
558
 
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}`);