@aztec/bot 0.0.1-commit.e6bd8901 → 0.0.1-commit.ec5f612

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 (43) hide show
  1. package/dest/amm_bot.d.ts +4 -4
  2. package/dest/amm_bot.d.ts.map +1 -1
  3. package/dest/base_bot.d.ts +4 -4
  4. package/dest/base_bot.d.ts.map +1 -1
  5. package/dest/base_bot.js +9 -10
  6. package/dest/bot.d.ts +4 -4
  7. package/dest/bot.d.ts.map +1 -1
  8. package/dest/bot.js +2 -2
  9. package/dest/config.d.ts +30 -14
  10. package/dest/config.d.ts.map +1 -1
  11. package/dest/config.js +36 -8
  12. package/dest/cross_chain_bot.d.ts +54 -0
  13. package/dest/cross_chain_bot.d.ts.map +1 -0
  14. package/dest/cross_chain_bot.js +140 -0
  15. package/dest/factory.d.ts +20 -5
  16. package/dest/factory.d.ts.map +1 -1
  17. package/dest/factory.js +80 -14
  18. package/dest/index.d.ts +2 -1
  19. package/dest/index.d.ts.map +1 -1
  20. package/dest/index.js +1 -0
  21. package/dest/l1_to_l2_seeding.d.ts +8 -0
  22. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  23. package/dest/l1_to_l2_seeding.js +63 -0
  24. package/dest/runner.d.ts +3 -3
  25. package/dest/runner.d.ts.map +1 -1
  26. package/dest/runner.js +17 -1
  27. package/dest/store/bot_store.d.ts +30 -5
  28. package/dest/store/bot_store.d.ts.map +1 -1
  29. package/dest/store/bot_store.js +37 -6
  30. package/dest/store/index.d.ts +2 -2
  31. package/dest/store/index.d.ts.map +1 -1
  32. package/package.json +16 -13
  33. package/src/amm_bot.ts +3 -3
  34. package/src/base_bot.ts +8 -18
  35. package/src/bot.ts +5 -5
  36. package/src/config.ts +39 -10
  37. package/src/cross_chain_bot.ts +209 -0
  38. package/src/factory.ts +110 -20
  39. package/src/index.ts +1 -0
  40. package/src/l1_to_l2_seeding.ts +79 -0
  41. package/src/runner.ts +18 -5
  42. package/src/store/bot_store.ts +60 -5
  43. package/src/store/index.ts +1 -1
package/src/amm_bot.ts CHANGED
@@ -6,7 +6,7 @@ import { jsonStringify } from '@aztec/foundation/json-rpc';
6
6
  import type { AMMContract } from '@aztec/noir-contracts.js/AMM';
7
7
  import type { TokenContract } from '@aztec/noir-contracts.js/Token';
8
8
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
9
- import type { TestWallet } from '@aztec/test-wallet/server';
9
+ import type { EmbeddedWallet } from '@aztec/wallets/embedded';
10
10
 
11
11
  import { BaseBot } from './base_bot.js';
12
12
  import type { BotConfig } from './config.js';
@@ -21,7 +21,7 @@ type Balances = { token0: bigint; token1: bigint };
21
21
  export class AmmBot extends BaseBot {
22
22
  protected constructor(
23
23
  node: AztecNode,
24
- wallet: TestWallet,
24
+ wallet: EmbeddedWallet,
25
25
  defaultAccountAddress: AztecAddress,
26
26
  public readonly amm: AMMContract,
27
27
  public readonly token0: TokenContract,
@@ -33,7 +33,7 @@ export class AmmBot extends BaseBot {
33
33
 
34
34
  static async create(
35
35
  config: BotConfig,
36
- wallet: TestWallet,
36
+ wallet: EmbeddedWallet,
37
37
  aztecNode: AztecNode,
38
38
  aztecNodeAdmin: AztecNodeAdmin | undefined,
39
39
  store: BotStore,
package/src/base_bot.ts CHANGED
@@ -1,16 +1,11 @@
1
1
  import { AztecAddress } from '@aztec/aztec.js/addresses';
2
- import {
3
- BatchCall,
4
- ContractFunctionInteraction,
5
- type SendInteractionOptions,
6
- waitForProven,
7
- } from '@aztec/aztec.js/contracts';
2
+ import { BatchCall, ContractFunctionInteraction, type SendInteractionOptions } from '@aztec/aztec.js/contracts';
8
3
  import { createLogger } from '@aztec/aztec.js/log';
9
4
  import { waitForTx } from '@aztec/aztec.js/node';
10
- import { TxHash, TxReceipt } from '@aztec/aztec.js/tx';
5
+ import { TxHash, TxReceipt, TxStatus } from '@aztec/aztec.js/tx';
11
6
  import { Gas } from '@aztec/stdlib/gas';
12
7
  import type { AztecNode } from '@aztec/stdlib/interfaces/client';
13
- import type { TestWallet } from '@aztec/test-wallet/server';
8
+ import type { EmbeddedWallet } from '@aztec/wallets/embedded';
14
9
 
15
10
  import type { BotConfig } from './config.js';
16
11
 
@@ -22,15 +17,15 @@ export abstract class BaseBot {
22
17
 
23
18
  protected constructor(
24
19
  public readonly node: AztecNode,
25
- public readonly wallet: TestWallet,
20
+ public readonly wallet: EmbeddedWallet,
26
21
  public readonly defaultAccountAddress: AztecAddress,
27
22
  public config: BotConfig,
28
23
  ) {}
29
24
 
30
25
  public async run(): Promise<TxReceipt | TxHash> {
31
26
  this.attempts++;
32
- const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000) };
33
27
  const { followChain, txMinedWaitSeconds } = this.config;
28
+ const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000), followChain, txMinedWaitSeconds };
34
29
 
35
30
  this.log.verbose(`Creating tx`, logCtx);
36
31
  const txHash = await this.createAndSendTx(logCtx);
@@ -40,14 +35,9 @@ export abstract class BaseBot {
40
35
  return txHash;
41
36
  }
42
37
 
43
- this.log.verbose(
44
- `Awaiting tx ${txHash.toString()} to be on the ${followChain} chain (timeout ${txMinedWaitSeconds}s)`,
45
- logCtx,
46
- );
47
- const receipt = await waitForTx(this.node, txHash, { timeout: txMinedWaitSeconds });
48
- if (followChain === 'PROVEN') {
49
- await waitForProven(this.node, receipt, { provenTimeout: txMinedWaitSeconds });
50
- }
38
+ const waitForStatus = TxStatus[followChain];
39
+ this.log.verbose(`Awaiting tx ${txHash.toString()} to be on the ${followChain} chain`, logCtx);
40
+ const receipt = await waitForTx(this.node, txHash, { timeout: txMinedWaitSeconds, waitForStatus });
51
41
  this.successes++;
52
42
  this.log.info(
53
43
  `Tx #${this.attempts} ${receipt.txHash} successfully mined in block ${receipt.blockNumber} (stats: ${this.successes}/${this.attempts} success)`,
package/src/bot.ts CHANGED
@@ -5,7 +5,7 @@ import { times } from '@aztec/foundation/collection';
5
5
  import type { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
6
6
  import type { TokenContract } from '@aztec/noir-contracts.js/Token';
7
7
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
8
- import type { TestWallet } from '@aztec/test-wallet/server';
8
+ import type { EmbeddedWallet } from '@aztec/wallets/embedded';
9
9
 
10
10
  import { BaseBot } from './base_bot.js';
11
11
  import type { BotConfig } from './config.js';
@@ -18,7 +18,7 @@ const TRANSFER_AMOUNT = 1;
18
18
  export class Bot extends BaseBot {
19
19
  protected constructor(
20
20
  node: AztecNode,
21
- wallet: TestWallet,
21
+ wallet: EmbeddedWallet,
22
22
  defaultAccountAddress: AztecAddress,
23
23
  public readonly token: TokenContract | PrivateTokenContract,
24
24
  public readonly recipient: AztecAddress,
@@ -29,7 +29,7 @@ export class Bot extends BaseBot {
29
29
 
30
30
  static async create(
31
31
  config: BotConfig,
32
- wallet: TestWallet,
32
+ wallet: EmbeddedWallet,
33
33
  aztecNode: AztecNode,
34
34
  aztecNodeAdmin: AztecNodeAdmin | undefined,
35
35
  store: BotStore,
@@ -83,7 +83,7 @@ export class Bot extends BaseBot {
83
83
  if (isStandardTokenContract(this.token)) {
84
84
  return {
85
85
  sender: await getBalances(this.token, this.defaultAccountAddress),
86
- recipient: await getBalances(this.token, this.recipient, this.defaultAccountAddress),
86
+ recipient: await getBalances(this.token, this.recipient),
87
87
  };
88
88
  } else {
89
89
  return {
@@ -92,7 +92,7 @@ export class Bot extends BaseBot {
92
92
  publicBalance: 0n,
93
93
  },
94
94
  recipient: {
95
- privateBalance: await getPrivateBalance(this.token, this.recipient, this.defaultAccountAddress),
95
+ privateBalance: await getPrivateBalance(this.token, this.recipient),
96
96
  publicBalance: 0n,
97
97
  },
98
98
  };
package/src/config.ts CHANGED
@@ -19,9 +19,12 @@ import type { ComponentsVersions } from '@aztec/stdlib/versioning';
19
19
 
20
20
  import { z } from 'zod';
21
21
 
22
- const BotFollowChain = ['NONE', 'PENDING', 'PROVEN'] as const;
22
+ const BotFollowChain = ['NONE', 'PROPOSED', 'CHECKPOINTED', 'PROVEN'] as const;
23
23
  type BotFollowChain = (typeof BotFollowChain)[number];
24
24
 
25
+ const BotMode = ['transfer', 'amm', 'crosschain'] as const;
26
+ type BotMode = (typeof BotMode)[number];
27
+
25
28
  export enum SupportedTokenContracts {
26
29
  TokenContract = 'TokenContract',
27
30
  PrivateTokenContract = 'PrivateTokenContract',
@@ -76,8 +79,12 @@ export type BotConfig = {
76
79
  maxConsecutiveErrors: number;
77
80
  /** Stops the bot if service becomes unhealthy */
78
81
  stopWhenUnhealthy: boolean;
79
- /** Deploy an AMM contract and do swaps instead of transfers */
80
- ammTxs: boolean;
82
+ /** Bot mode: transfer, amm, or crosschain. */
83
+ botMode: BotMode;
84
+ /** Number of L2→L1 messages per tx (crosschain mode). */
85
+ l2ToL1MessagesPerTx: number;
86
+ /** Max L1→L2 messages to keep in-flight (crosschain mode). */
87
+ l1ToL2SeedCount: number;
81
88
  } & Pick<DataStoreConfig, 'dataDirectory' | 'dataStoreMapSizeKb'>;
82
89
 
83
90
  export const BotConfigSchema = zodFor<BotConfig>()(
@@ -107,7 +114,9 @@ export const BotConfigSchema = zodFor<BotConfig>()(
107
114
  contract: z.nativeEnum(SupportedTokenContracts),
108
115
  maxConsecutiveErrors: z.number().int().nonnegative(),
109
116
  stopWhenUnhealthy: z.boolean(),
110
- ammTxs: z.boolean().default(false),
117
+ botMode: z.enum(BotMode).default('transfer'),
118
+ l2ToL1MessagesPerTx: z.number().int().nonnegative().default(1),
119
+ l1ToL2SeedCount: z.number().int().nonnegative().default(1),
111
120
  dataDirectory: z.string().optional(),
112
121
  dataStoreMapSizeKb: z.number().optional(),
113
122
  })
@@ -213,10 +222,14 @@ export const botConfigMappings: ConfigMappingsType<BotConfig> = {
213
222
  description: 'Which chain the bot follows',
214
223
  defaultValue: 'NONE',
215
224
  parseEnv(val) {
216
- if (!(BotFollowChain as readonly string[]).includes(val.toUpperCase())) {
225
+ const upper = val.toUpperCase();
226
+ if (upper === 'PENDING') {
227
+ return 'CHECKPOINTED';
228
+ }
229
+ if (!(BotFollowChain as readonly string[]).includes(upper)) {
217
230
  throw new Error(`Invalid value for BOT_FOLLOW_CHAIN: ${val}`);
218
231
  }
219
- return val as BotFollowChain;
232
+ return upper as BotFollowChain;
220
233
  },
221
234
  },
222
235
  maxPendingTxs: {
@@ -264,10 +277,26 @@ export const botConfigMappings: ConfigMappingsType<BotConfig> = {
264
277
  description: 'Stops the bot if service becomes unhealthy',
265
278
  ...booleanConfigHelper(false),
266
279
  },
267
- ammTxs: {
268
- env: 'BOT_AMM_TXS',
269
- description: 'Deploy an AMM and send swaps to it',
270
- ...booleanConfigHelper(false),
280
+ botMode: {
281
+ env: 'BOT_MODE',
282
+ description: 'Bot mode: transfer, amm, or crosschain',
283
+ defaultValue: 'transfer' as BotMode,
284
+ parseEnv(val: string) {
285
+ if (!(BotMode as readonly string[]).includes(val)) {
286
+ throw new Error(`Invalid value for BOT_MODE: ${val}`);
287
+ }
288
+ return val as BotMode;
289
+ },
290
+ },
291
+ l2ToL1MessagesPerTx: {
292
+ env: 'BOT_L2_TO_L1_MESSAGES_PER_TX',
293
+ description: 'Number of L2→L1 messages per tx (crosschain mode)',
294
+ ...numberConfigHelper(1),
295
+ },
296
+ l1ToL2SeedCount: {
297
+ env: 'BOT_L1_TO_L2_SEED_COUNT',
298
+ description: 'Max L1→L2 messages to keep in-flight (crosschain mode)',
299
+ ...numberConfigHelper(1),
271
300
  },
272
301
  ...pickConfigMappings(dataConfigMappings, ['dataStoreMapSizeKb', 'dataDirectory']),
273
302
  };
@@ -0,0 +1,209 @@
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
+ return batch.send({ ...opts, wait: NO_WAIT });
144
+ }
145
+
146
+ protected override async onTxMined(receipt: TxReceipt, logCtx: object): Promise<void> {
147
+ // Verify L2→L1 messages appeared in this tx's effects
148
+ const indexed = await this.node.getTxEffect(receipt.txHash);
149
+ if (indexed) {
150
+ const l2ToL1Msgs = indexed.data.l2ToL1Msgs.filter(m => !m.isZero());
151
+ if (l2ToL1Msgs.length >= this.config.l2ToL1MessagesPerTx) {
152
+ this.l2ToL1Sent += l2ToL1Msgs.length;
153
+ } else {
154
+ this.log.error(`Expected ${this.config.l2ToL1MessagesPerTx} L2→L1 messages but found ${l2ToL1Msgs.length}`, {
155
+ ...logCtx,
156
+ blockNumber: receipt.blockNumber,
157
+ txHash: receipt.txHash.toString(),
158
+ });
159
+ }
160
+ }
161
+
162
+ const pendingCount = (await this.store.getUnconsumedL1ToL2Messages()).length;
163
+ this.log.info(`CrossChainBot txs mined`, {
164
+ ...logCtx,
165
+ l2ToL1Sent: this.l2ToL1Sent,
166
+ l1ToL2Consumed: this.l1ToL2Consumed,
167
+ l1ToL2Pending: pendingCount,
168
+ });
169
+ }
170
+
171
+ /** Finds the oldest pending message that is ready for consumption. */
172
+ private async getReadyL1ToL2Message(
173
+ pendingMessages: PendingL1ToL2Message[],
174
+ ): Promise<PendingL1ToL2Message | undefined> {
175
+ const now = Date.now();
176
+ for (const msg of pendingMessages) {
177
+ const ready = await isL1ToL2MessageReady(this.node, Fr.fromHexString(msg.msgHash), {
178
+ // Use forPublicConsumption: false so we wait until blockNumber >= messageBlockNumber.
179
+ // With forPublicConsumption: true, the check returns true one block early (the sequencer
180
+ // includes L1→L2 messages before executing the block's txs), but gas estimation simulates
181
+ // against the current world state which doesn't yet have the message.
182
+ // See https://linear.app/aztec-labs/issue/A-548 for details.
183
+ forPublicConsumption: false,
184
+ });
185
+ if (ready) {
186
+ return msg;
187
+ }
188
+
189
+ // Time-based stale detection: if the message is old and still not ready, remove it
190
+ if (now - msg.timestamp > STALE_MESSAGE_THRESHOLD_MS) {
191
+ await this.store.deleteL1ToL2Message(msg.msgHash);
192
+ this.log.warn(`Removed stale L1→L2 message ${msg.msgHash}`);
193
+ }
194
+ }
195
+ return undefined;
196
+ }
197
+
198
+ /** Seeds a new L1→L2 message on L1 and stores it. */
199
+ private async seedNewL1ToL2Message(): Promise<void> {
200
+ await seedL1ToL2Message(
201
+ this.l1Client,
202
+ this.inboxAddress,
203
+ this.contract.address,
204
+ this.rollupVersion,
205
+ this.store,
206
+ this.log,
207
+ );
208
+ }
209
+ }
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,94 @@ 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
+ // Use forPublicConsumption: false so we wait until the message is in the current world
162
+ // state. With true, it returns one block early which causes gas estimation simulation to
163
+ // fail since it runs against the current state.
164
+ // See https://linear.app/aztec-labs/issue/A-548 for details.
165
+ forPublicConsumption: false,
166
+ });
167
+ this.log.info(`First L1→L2 message is ready`);
168
+ }
169
+
170
+ return {
171
+ wallet: this.wallet,
172
+ defaultAccountAddress,
173
+ contract,
174
+ node: this.aztecNode,
175
+ l1Client,
176
+ rollupVersion,
177
+ };
178
+ }
179
+
180
+ private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
181
+ const deployOpts: DeployOptions = {
182
+ from: deployer,
183
+ contractAddressSalt: this.config.tokenSalt,
184
+ universalDeploy: true,
185
+ };
186
+ const deploy = TestContract.deploy(this.wallet);
187
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
188
+ return TestContract.at(instance.address, this.wallet);
189
+ }
190
+
99
191
  /**
100
192
  * Checks if the sender account contract is initialized, and initializes it if necessary.
101
193
  * @returns The sender wallet.
@@ -114,12 +206,7 @@ export class BotFactory {
114
206
  private async setupAccountWithPrivateKey(secret: Fr) {
115
207
  const salt = this.config.senderSalt ?? Fr.ONE;
116
208
  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);
209
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
123
210
  const metadata = await this.wallet.getContractMetadata(accountManager.address);
124
211
  if (metadata.isContractInitialized) {
125
212
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
@@ -159,12 +246,11 @@ export class BotFactory {
159
246
 
160
247
  private async setupTestAccount() {
161
248
  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);
249
+ const accountManager = await this.wallet.createSchnorrAccount(
250
+ initialAccountData.secret,
251
+ initialAccountData.salt,
252
+ initialAccountData.signingKey,
253
+ );
168
254
  return accountManager.address;
169
255
  }
170
256
 
@@ -199,6 +285,8 @@ export class BotFactory {
199
285
  tokenInstance = await deploy.getInstance(deployOpts);
200
286
  token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
201
287
  await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
288
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
289
+ deployOpts.additionalScopes = [tokenInstance.address];
202
290
  } else {
203
291
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
204
292
  }
@@ -393,8 +481,10 @@ export class BotFactory {
393
481
  return;
394
482
  }
395
483
 
484
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
485
+ const additionalScopes = isStandardToken ? undefined : [token.address];
396
486
  await this.withNoMinTxsPerBlock(async () => {
397
- const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
487
+ const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, additionalScopes, wait: NO_WAIT });
398
488
  this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
399
489
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
400
490
  });
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 {