@aztec/bot 0.0.1-commit.6d3c34e → 0.0.1-commit.7cf39cb55

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 (44) hide show
  1. package/dest/amm_bot.d.ts +6 -7
  2. package/dest/amm_bot.d.ts.map +1 -1
  3. package/dest/amm_bot.js +5 -1
  4. package/dest/base_bot.d.ts +6 -6
  5. package/dest/base_bot.d.ts.map +1 -1
  6. package/dest/base_bot.js +12 -13
  7. package/dest/bot.d.ts +6 -6
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +7 -4
  10. package/dest/config.d.ts +30 -14
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +36 -8
  13. package/dest/cross_chain_bot.d.ts +54 -0
  14. package/dest/cross_chain_bot.d.ts.map +1 -0
  15. package/dest/cross_chain_bot.js +140 -0
  16. package/dest/factory.d.ts +20 -10
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +152 -70
  19. package/dest/index.d.ts +2 -1
  20. package/dest/index.d.ts.map +1 -1
  21. package/dest/index.js +1 -0
  22. package/dest/l1_to_l2_seeding.d.ts +8 -0
  23. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  24. package/dest/l1_to_l2_seeding.js +63 -0
  25. package/dest/runner.d.ts +3 -3
  26. package/dest/runner.d.ts.map +1 -1
  27. package/dest/runner.js +17 -1
  28. package/dest/store/bot_store.d.ts +31 -6
  29. package/dest/store/bot_store.d.ts.map +1 -1
  30. package/dest/store/bot_store.js +37 -6
  31. package/dest/store/index.d.ts +2 -2
  32. package/dest/store/index.d.ts.map +1 -1
  33. package/package.json +17 -14
  34. package/src/amm_bot.ts +7 -7
  35. package/src/base_bot.ts +11 -25
  36. package/src/bot.ts +9 -8
  37. package/src/config.ts +39 -10
  38. package/src/cross_chain_bot.ts +209 -0
  39. package/src/factory.ts +165 -55
  40. package/src/index.ts +1 -0
  41. package/src/l1_to_l2_seeding.ts +79 -0
  42. package/src/runner.ts +18 -5
  43. package/src/store/bot_store.ts +60 -5
  44. package/src/store/index.ts +1 -1
@@ -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 {
@@ -7,27 +6,34 @@ import {
7
6
  ContractFunctionInteraction,
8
7
  type DeployMethod,
9
8
  type DeployOptions,
9
+ NO_WAIT,
10
10
  } from '@aztec/aztec.js/contracts';
11
- import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
12
11
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
12
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
13
13
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
14
14
  import { deriveKeys } from '@aztec/aztec.js/keys';
15
15
  import { createLogger } from '@aztec/aztec.js/log';
16
16
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
17
+ import { waitForTx } from '@aztec/aztec.js/node';
17
18
  import { createEthereumChain } from '@aztec/ethereum/chain';
18
19
  import { createExtendedL1Client } from '@aztec/ethereum/client';
20
+ import { RollupContract } from '@aztec/ethereum/contracts';
21
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
19
22
  import { Fr } from '@aztec/foundation/curves/bn254';
23
+ import { EthAddress } from '@aztec/foundation/eth-address';
20
24
  import { Timer } from '@aztec/foundation/timer';
21
25
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
22
26
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
23
27
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
28
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
24
29
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
25
30
  import { GasSettings } from '@aztec/stdlib/gas';
26
31
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
27
32
  import { deriveSigningKey } from '@aztec/stdlib/keys';
28
- import { TestWallet } from '@aztec/test-wallet/server';
33
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
29
34
 
30
35
  import { type BotConfig, SupportedTokenContracts } from './config.js';
36
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
31
37
  import type { BotStore } from './store/index.js';
32
38
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
33
39
 
@@ -39,7 +45,7 @@ export class BotFactory {
39
45
 
40
46
  constructor(
41
47
  private readonly config: BotConfig,
42
- private readonly wallet: TestWallet,
48
+ private readonly wallet: EmbeddedWallet,
43
49
  private readonly store: BotStore,
44
50
  private readonly aztecNode: AztecNode,
45
51
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
@@ -50,21 +56,21 @@ export class BotFactory {
50
56
  * deploying the token contract, and minting tokens if necessary.
51
57
  */
52
58
  public async setup(): Promise<{
53
- wallet: TestWallet;
59
+ wallet: EmbeddedWallet;
54
60
  defaultAccountAddress: AztecAddress;
55
61
  token: TokenContract | PrivateTokenContract;
56
62
  node: AztecNode;
57
63
  recipient: AztecAddress;
58
64
  }> {
59
- const recipient = (await this.wallet.createAccount()).address;
60
65
  const defaultAccountAddress = await this.setupAccount();
66
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
61
67
  const token = await this.setupToken(defaultAccountAddress);
62
68
  await this.mintTokens(token, defaultAccountAddress);
63
69
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
64
70
  }
65
71
 
66
72
  public async setupAmm(): Promise<{
67
- wallet: TestWallet;
73
+ wallet: EmbeddedWallet;
68
74
  defaultAccountAddress: AztecAddress;
69
75
  amm: AMMContract;
70
76
  token0: TokenContract;
@@ -94,6 +100,94 @@ export class BotFactory {
94
100
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
95
101
  }
96
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
+
97
191
  /**
98
192
  * Checks if the sender account contract is initialized, and initializes it if necessary.
99
193
  * @returns The sender wallet.
@@ -112,14 +206,9 @@ export class BotFactory {
112
206
  private async setupAccountWithPrivateKey(secret: Fr) {
113
207
  const salt = this.config.senderSalt ?? Fr.ONE;
114
208
  const signingKey = deriveSigningKey(secret);
115
- const accountData = {
116
- secret,
117
- salt,
118
- contract: new SchnorrAccountContract(signingKey!),
119
- };
120
- const accountManager = await this.wallet.createAccount(accountData);
121
- const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
122
- if (isInit) {
209
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
210
+ const metadata = await this.wallet.getContractMetadata(accountManager.address);
211
+ if (metadata.isContractInitialized) {
123
212
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
124
213
  const timer = new Timer();
125
214
  const address = accountManager.address;
@@ -136,10 +225,16 @@ export class BotFactory {
136
225
  const deployMethod = await accountManager.getDeployMethod();
137
226
  const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
138
227
  const gasSettings = GasSettings.default({ maxFeesPerGas });
139
- const sentTx = deployMethod.send({ from: AztecAddress.ZERO, fee: { gasSettings, paymentMethod } });
140
- const txHash = await sentTx.getTxHash();
141
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
142
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
228
+
229
+ await this.withNoMinTxsPerBlock(async () => {
230
+ const txHash = await deployMethod.send({
231
+ from: AztecAddress.ZERO,
232
+ fee: { gasSettings, paymentMethod },
233
+ wait: NO_WAIT,
234
+ });
235
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
236
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
237
+ });
143
238
  this.log.info(`Account deployed at ${address}`);
144
239
 
145
240
  // Clean up the consumed bridge claim
@@ -151,12 +246,11 @@ export class BotFactory {
151
246
 
152
247
  private async setupTestAccount() {
153
248
  const [initialAccountData] = await getInitialTestAccountsData();
154
- const accountData = {
155
- secret: initialAccountData.secret,
156
- salt: initialAccountData.salt,
157
- contract: new SchnorrAccountContract(initialAccountData.signingKey),
158
- };
159
- const accountManager = await this.wallet.createAccount(accountData);
249
+ const accountManager = await this.wallet.createSchnorrAccount(
250
+ initialAccountData.secret,
251
+ initialAccountData.salt,
252
+ initialAccountData.signingKey,
253
+ );
160
254
  return accountManager.address;
161
255
  }
162
256
 
@@ -173,8 +267,11 @@ export class BotFactory {
173
267
  contractAddressSalt: this.config.tokenSalt,
174
268
  universalDeploy: true,
175
269
  };
270
+ let token: TokenContract | PrivateTokenContract;
176
271
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
177
272
  deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
273
+ tokenInstance = await deploy.getInstance(deployOpts);
274
+ token = TokenContract.at(tokenInstance.address, this.wallet);
178
275
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
179
276
  // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
180
277
  const tokenSecretKey = Fr.random();
@@ -186,22 +283,27 @@ export class BotFactory {
186
283
 
187
284
  // Register the contract with the secret key before deployment
188
285
  tokenInstance = await deploy.getInstance(deployOpts);
286
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
189
287
  await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
190
288
  } else {
191
289
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
192
290
  }
193
291
 
194
292
  const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
195
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
293
+ const metadata = await this.wallet.getContractMetadata(address);
294
+ if (metadata.isContractPublished) {
196
295
  this.log.info(`Token at ${address.toString()} already deployed`);
197
- return deploy.register();
296
+ await deploy.register();
198
297
  } else {
199
298
  this.log.info(`Deploying token contract at ${address.toString()}`);
200
- const sentTx = deploy.send(deployOpts);
201
- const txHash = await sentTx.getTxHash();
299
+ const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
202
300
  this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
203
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
301
+ await this.withNoMinTxsPerBlock(async () => {
302
+ await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
303
+ return token;
304
+ });
204
305
  }
306
+ return token;
205
307
  }
206
308
 
207
309
  /**
@@ -209,7 +311,7 @@ export class BotFactory {
209
311
  * @param wallet - Wallet to deploy the token contract from.
210
312
  * @returns The TokenContract instance.
211
313
  */
212
- private setupTokenContract(
314
+ private async setupTokenContract(
213
315
  deployer: AztecAddress,
214
316
  contractAddressSalt: Fr,
215
317
  name: string,
@@ -218,7 +320,8 @@ export class BotFactory {
218
320
  ): Promise<TokenContract> {
219
321
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
220
322
  const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
221
- return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
323
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
324
+ return TokenContract.at(instance.address, this.wallet);
222
325
  }
223
326
 
224
327
  private async setupAmmContract(
@@ -230,12 +333,14 @@ export class BotFactory {
230
333
  ): Promise<AMMContract> {
231
334
  const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
232
335
  const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
233
- const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
336
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
337
+ const amm = AMMContract.at(instance.address, this.wallet);
234
338
 
235
339
  this.log.info(`AMM deployed at ${amm.address}`);
236
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({ from: deployer });
237
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
238
- await minterTx.wait({ timeout: this.config.txMinedWaitSeconds });
340
+ const minterReceipt = await lpToken.methods
341
+ .set_minter(amm.address, true)
342
+ .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
343
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
239
344
  this.log.info(`Liquidity token initialized`);
240
345
 
241
346
  return amm;
@@ -294,23 +399,22 @@ export class BotFactory {
294
399
  .getFunctionCall(),
295
400
  });
296
401
 
297
- const mintTx = new BatchCall(this.wallet, [
402
+ const mintReceipt = await new BatchCall(this.wallet, [
298
403
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
299
404
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
300
- ]).send({ from: liquidityProvider });
405
+ ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
301
406
 
302
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
303
- await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
407
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
304
408
 
305
- const addLiquidityTx = amm.methods
409
+ const addLiquidityReceipt = await amm.methods
306
410
  .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
307
411
  .send({
308
412
  from: liquidityProvider,
309
413
  authWitnesses: [token0Authwit, token1Authwit],
414
+ wait: { timeout: this.config.txMinedWaitSeconds },
310
415
  });
311
416
 
312
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
313
- await addLiquidityTx.wait({ timeout: this.config.txMinedWaitSeconds });
417
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
314
418
  this.log.info(`Liquidity added`);
315
419
 
316
420
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -323,18 +427,22 @@ export class BotFactory {
323
427
  name: string,
324
428
  deploy: DeployMethod<T>,
325
429
  deployOpts: DeployOptions,
326
- ): Promise<T> {
327
- const address = (await deploy.getInstance(deployOpts)).address;
328
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
430
+ ): Promise<ContractInstanceWithAddress> {
431
+ const instance = await deploy.getInstance(deployOpts);
432
+ const address = instance.address;
433
+ const metadata = await this.wallet.getContractMetadata(address);
434
+ if (metadata.isContractPublished) {
329
435
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
330
- return deploy.register();
436
+ await deploy.register();
331
437
  } else {
332
438
  this.log.info(`Deploying contract ${name} at ${address.toString()}`);
333
- const sentTx = deploy.send(deployOpts);
334
- const txHash = await sentTx.getTxHash();
335
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
336
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
439
+ await this.withNoMinTxsPerBlock(async () => {
440
+ const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
441
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
442
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
443
+ });
337
444
  }
445
+ return instance;
338
446
  }
339
447
 
340
448
  /**
@@ -370,10 +478,12 @@ export class BotFactory {
370
478
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
371
479
  return;
372
480
  }
373
- const sentTx = new BatchCall(token.wallet, calls).send({ from: minter });
374
- const txHash = await sentTx.getTxHash();
375
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
376
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
481
+
482
+ await this.withNoMinTxsPerBlock(async () => {
483
+ const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
484
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
485
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
486
+ });
377
487
  }
378
488
 
379
489
  /**
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
+ }