@aztec/bot 0.0.1-commit.43c09e3f → 0.0.1-commit.4d3c002

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 +24 -17
  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 +21 -32
  7. package/dest/bot.d.ts +4 -4
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +5 -8
  10. package/dest/config.d.ts +34 -18
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +39 -12
  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 +24 -5
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +282 -72
  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 +24 -19
  36. package/src/base_bot.ts +15 -33
  37. package/src/bot.ts +8 -10
  38. package/src/config.ts +44 -16
  39. package/src/cross_chain_bot.ts +203 -0
  40. package/src/factory.ts +337 -73
  41. package/src/index.ts +1 -0
  42. package/src/l1_to_l2_seeding.ts +79 -0
  43. package/src/runner.ts +18 -5
  44. package/src/store/bot_store.ts +60 -5
  45. package/src/store/index.ts +1 -1
  46. package/src/utils.ts +3 -3
package/src/factory.ts CHANGED
@@ -1,5 +1,5 @@
1
- import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
2
1
  import { getInitialTestAccountsData } from '@aztec/accounts/testing';
2
+ import { NO_FROM } from '@aztec/aztec.js/account';
3
3
  import { AztecAddress } from '@aztec/aztec.js/addresses';
4
4
  import {
5
5
  BatchCall,
@@ -9,64 +9,78 @@ import {
9
9
  type DeployOptions,
10
10
  NO_WAIT,
11
11
  } from '@aztec/aztec.js/contracts';
12
- import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
13
12
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
13
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
14
14
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
15
15
  import { deriveKeys } from '@aztec/aztec.js/keys';
16
16
  import { createLogger } from '@aztec/aztec.js/log';
17
17
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
18
18
  import { waitForTx } from '@aztec/aztec.js/node';
19
+ import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
20
+ import { ContractInitializationStatus } from '@aztec/aztec.js/wallet';
19
21
  import { createEthereumChain } from '@aztec/ethereum/chain';
20
22
  import { createExtendedL1Client } from '@aztec/ethereum/client';
23
+ import { RollupContract } from '@aztec/ethereum/contracts';
24
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
21
25
  import { Fr } from '@aztec/foundation/curves/bn254';
26
+ import { EthAddress } from '@aztec/foundation/eth-address';
22
27
  import { Timer } from '@aztec/foundation/timer';
23
28
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
24
29
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
25
30
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
31
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
26
32
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
27
- import { GasSettings } from '@aztec/stdlib/gas';
33
+ import { GasFees, GasSettings } from '@aztec/stdlib/gas';
28
34
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
29
35
  import { deriveSigningKey } from '@aztec/stdlib/keys';
30
- import { TestWallet } from '@aztec/test-wallet/server';
36
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
31
37
 
32
38
  import { type BotConfig, SupportedTokenContracts } from './config.js';
39
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
33
40
  import type { BotStore } from './store/index.js';
34
41
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
35
42
 
36
43
  const MINT_BALANCE = 1e12;
37
44
  const MIN_BALANCE = 1e3;
45
+ const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
46
+ const FEE_JUICE_TOP_UP_TARGET = 10_000n * 10n ** 18n;
38
47
 
39
48
  export class BotFactory {
40
49
  private log = createLogger('bot');
41
50
 
42
51
  constructor(
43
52
  private readonly config: BotConfig,
44
- private readonly wallet: TestWallet,
53
+ private readonly wallet: EmbeddedWallet,
45
54
  private readonly store: BotStore,
46
55
  private readonly aztecNode: AztecNode,
47
56
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
48
- ) {}
57
+ ) {
58
+ // Set fee padding on the wallet so that all transactions during setup
59
+ // (token deploy, minting, etc.) use the configured padding, not the default.
60
+ this.wallet.setMinFeePadding(config.minFeePadding);
61
+ }
49
62
 
50
63
  /**
51
64
  * Initializes a new bot by setting up the sender account, registering the recipient,
52
65
  * deploying the token contract, and minting tokens if necessary.
53
66
  */
54
67
  public async setup(): Promise<{
55
- wallet: TestWallet;
68
+ wallet: EmbeddedWallet;
56
69
  defaultAccountAddress: AztecAddress;
57
70
  token: TokenContract | PrivateTokenContract;
58
71
  node: AztecNode;
59
72
  recipient: AztecAddress;
60
73
  }> {
61
74
  const defaultAccountAddress = await this.setupAccount();
62
- const recipient = (await this.wallet.createAccount()).address;
63
- const token = await this.setupToken(defaultAccountAddress);
75
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
76
+ const token = await this.setupTokenWithOptionalEarlyRefuel(defaultAccountAddress);
77
+ await this.ensureFeeJuiceBalance(defaultAccountAddress, token);
64
78
  await this.mintTokens(token, defaultAccountAddress);
65
79
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
66
80
  }
67
81
 
68
82
  public async setupAmm(): Promise<{
69
- wallet: TestWallet;
83
+ wallet: EmbeddedWallet;
70
84
  defaultAccountAddress: AztecAddress;
71
85
  amm: AMMContract;
72
86
  token0: TokenContract;
@@ -74,7 +88,13 @@ export class BotFactory {
74
88
  node: AztecNode;
75
89
  }> {
76
90
  const defaultAccountAddress = await this.setupAccount();
77
- const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
91
+ const token0 = await this.setupTokenContractWithOptionalEarlyRefuel(
92
+ defaultAccountAddress,
93
+ this.config.tokenSalt,
94
+ 'BotToken0',
95
+ 'BOT0',
96
+ );
97
+ await this.ensureFeeJuiceBalance(defaultAccountAddress, token0);
78
98
  const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
79
99
  const liquidityToken = await this.setupTokenContract(
80
100
  defaultAccountAddress,
@@ -96,6 +116,89 @@ export class BotFactory {
96
116
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
97
117
  }
98
118
 
119
+ /**
120
+ * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
121
+ * seeding initial L1→L2 messages, and waiting for the first to be ready.
122
+ */
123
+ public async setupCrossChain(): Promise<{
124
+ wallet: EmbeddedWallet;
125
+ defaultAccountAddress: AztecAddress;
126
+ contract: TestContract;
127
+ node: AztecNode;
128
+ l1Client: ExtendedViemWalletClient;
129
+ rollupVersion: bigint;
130
+ }> {
131
+ const defaultAccountAddress = await this.setupAccount();
132
+
133
+ // Create L1 client (same pattern as bridgeL1FeeJuice)
134
+ const l1RpcUrls = this.config.l1RpcUrls;
135
+ if (!l1RpcUrls?.length) {
136
+ throw new Error('L1 RPC URLs required for cross-chain bot');
137
+ }
138
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
139
+ if (!mnemonicOrPrivateKey) {
140
+ throw new Error('L1 mnemonic or private key required for cross-chain bot');
141
+ }
142
+ const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
143
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
144
+ const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
145
+
146
+ // Fetch Rollup version (needed for Inbox L2Actor struct)
147
+ const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
148
+ const rollupVersion = await rollupContract.getVersion();
149
+
150
+ // Deploy TestContract
151
+ const contract = await this.setupTestContract(defaultAccountAddress);
152
+
153
+ // Recover any pending messages from store (clean up stale ones first)
154
+ await this.store.cleanupOldPendingMessages();
155
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
156
+
157
+ // Seed initial L1→L2 messages if pipeline is empty
158
+ const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
159
+ for (let i = 0; i < seedCount; i++) {
160
+ await seedL1ToL2Message(
161
+ l1Client,
162
+ EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()),
163
+ contract.address,
164
+ rollupVersion,
165
+ this.store,
166
+ this.log,
167
+ );
168
+ }
169
+
170
+ // Block until at least one message is ready
171
+ const allMessages = await this.store.getUnconsumedL1ToL2Messages();
172
+ if (allMessages.length > 0) {
173
+ this.log.info(`Waiting for first L1→L2 message to be ready...`);
174
+ const firstMsg = allMessages[0];
175
+ await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
176
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
177
+ });
178
+ this.log.info(`First L1→L2 message is ready`);
179
+ }
180
+
181
+ return {
182
+ wallet: this.wallet,
183
+ defaultAccountAddress,
184
+ contract,
185
+ node: this.aztecNode,
186
+ l1Client,
187
+ rollupVersion,
188
+ };
189
+ }
190
+
191
+ private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
192
+ const deployOpts: DeployOptions = {
193
+ from: deployer,
194
+ contractAddressSalt: this.config.tokenSalt,
195
+ universalDeploy: true,
196
+ };
197
+ const deploy = TestContract.deploy(this.wallet);
198
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
199
+ return TestContract.at(instance.address, this.wallet);
200
+ }
201
+
99
202
  /**
100
203
  * Checks if the sender account contract is initialized, and initializes it if necessary.
101
204
  * @returns The sender wallet.
@@ -114,14 +217,9 @@ export class BotFactory {
114
217
  private async setupAccountWithPrivateKey(secret: Fr) {
115
218
  const salt = this.config.senderSalt ?? Fr.ONE;
116
219
  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);
220
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
123
221
  const metadata = await this.wallet.getContractMetadata(accountManager.address);
124
- if (metadata.isContractInitialized) {
222
+ if (metadata.initializationStatus === ContractInitializationStatus.INITIALIZED) {
125
223
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
126
224
  const timer = new Timer();
127
225
  const address = accountManager.address;
@@ -136,13 +234,11 @@ export class BotFactory {
136
234
 
137
235
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
138
236
  const deployMethod = await accountManager.getDeployMethod();
139
- const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
140
- const gasSettings = GasSettings.default({ maxFeesPerGas });
141
237
 
142
238
  await this.withNoMinTxsPerBlock(async () => {
143
- const txHash = await deployMethod.send({
144
- from: AztecAddress.ZERO,
145
- fee: { gasSettings, paymentMethod },
239
+ const { txHash } = await deployMethod.send({
240
+ from: NO_FROM,
241
+ fee: { paymentMethod },
146
242
  wait: NO_WAIT,
147
243
  });
148
244
  this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
@@ -159,23 +255,87 @@ export class BotFactory {
159
255
 
160
256
  private async setupTestAccount() {
161
257
  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);
258
+ const accountManager = await this.wallet.createSchnorrAccount(
259
+ initialAccountData.secret,
260
+ initialAccountData.salt,
261
+ initialAccountData.signingKey,
262
+ );
168
263
  return accountManager.address;
169
264
  }
170
265
 
266
+ /**
267
+ * Setup token and refuel first: if the token already exists (restart scenario),
268
+ * run ensureFeeJuiceBalance before any step that might need fee juice. When deploying,
269
+ * use a bridge claim if balance is below threshold.
270
+ */
271
+ private async setupTokenWithOptionalEarlyRefuel(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
272
+ const token = await this.getTokenInstance(sender);
273
+ const address = token.address;
274
+ const metadata = await this.wallet.getContractMetadata(address);
275
+ if (metadata.isContractPublished) {
276
+ this.log.info(`Token at ${address.toString()} already deployed, refueling before setup`);
277
+ await this.ensureFeeJuiceBalance(sender, token);
278
+ }
279
+ return this.setupToken(sender);
280
+ }
281
+
282
+ /**
283
+ * Setup token0 for AMM with refuel-first behaviour when token already exists.
284
+ */
285
+ private async setupTokenContractWithOptionalEarlyRefuel(
286
+ deployer: AztecAddress,
287
+ contractAddressSalt: Fr,
288
+ name: string,
289
+ ticker: string,
290
+ decimals = 18,
291
+ ): Promise<TokenContract> {
292
+ const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
293
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
294
+ const instance = await deploy.getInstance(deployOpts);
295
+ const metadata = await this.wallet.getContractMetadata(instance.address);
296
+ if (metadata.isContractPublished) {
297
+ this.log.info(`Token ${name} at ${instance.address.toString()} already deployed, refueling before setup`);
298
+ const token = TokenContract.at(instance.address, this.wallet);
299
+ await this.ensureFeeJuiceBalance(deployer, token);
300
+ }
301
+ return this.setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals);
302
+ }
303
+
304
+ private async getTokenInstance(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
305
+ const deployOpts: DeployOptions = {
306
+ from: sender,
307
+ contractAddressSalt: this.config.tokenSalt,
308
+ universalDeploy: true,
309
+ };
310
+ if (this.config.contract === SupportedTokenContracts.TokenContract) {
311
+ const deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
312
+ const instance = await deploy.getInstance(deployOpts);
313
+ return TokenContract.at(instance.address, this.wallet);
314
+ }
315
+ if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
316
+ const tokenSecretKey = Fr.random();
317
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
318
+ const deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
319
+ const instance = await deploy.getInstance({
320
+ ...deployOpts,
321
+ skipInstancePublication: true,
322
+ skipClassPublication: true,
323
+ skipInitialization: false,
324
+ });
325
+ return PrivateTokenContract.at(instance.address, this.wallet);
326
+ }
327
+ throw new Error(`Unsupported token contract type: ${this.config.contract}`);
328
+ }
329
+
171
330
  /**
172
331
  * Checks if the token contract is deployed and deploys it if necessary.
173
- * @param wallet - Wallet to deploy the token contract from.
174
- * @returns The TokenContract instance.
332
+ * Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
333
+ * @param sender - Aztec address to deploy the token contract from.
334
+ * @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
335
+ * @returns The TokenContract or PrivateTokenContract instance.
175
336
  */
176
337
  private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
177
338
  let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
178
- let tokenInstance: ContractInstanceWithAddress | undefined;
179
339
  const deployOpts: DeployOptions = {
180
340
  from: sender,
181
341
  contractAddressSalt: this.config.tokenSalt,
@@ -184,8 +344,8 @@ export class BotFactory {
184
344
  let token: TokenContract | PrivateTokenContract;
185
345
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
186
346
  deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
187
- tokenInstance = await deploy.getInstance(deployOpts);
188
- token = TokenContract.at(tokenInstance.address, this.wallet);
347
+ const instance = await deploy.getInstance(deployOpts);
348
+ token = TokenContract.at(instance.address, this.wallet);
189
349
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
190
350
  // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
191
351
  const tokenSecretKey = Fr.random();
@@ -196,27 +356,16 @@ export class BotFactory {
196
356
  deployOpts.skipInitialization = false;
197
357
 
198
358
  // Register the contract with the secret key before deployment
199
- tokenInstance = await deploy.getInstance(deployOpts);
359
+ const tokenInstance = await deploy.getInstance(deployOpts);
200
360
  token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
201
361
  await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
362
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
363
+ deployOpts.additionalScopes = [tokenInstance.address];
202
364
  } else {
203
365
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
204
366
  }
205
367
 
206
- const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
207
- const metadata = await this.wallet.getContractMetadata(address);
208
- if (metadata.isContractPublished) {
209
- this.log.info(`Token at ${address.toString()} already deployed`);
210
- await deploy.register();
211
- } else {
212
- this.log.info(`Deploying token contract at ${address.toString()}`);
213
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
214
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
215
- await this.withNoMinTxsPerBlock(async () => {
216
- await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
217
- return token;
218
- });
219
- }
368
+ await this.registerOrDeployContract('token', deploy, deployOpts);
220
369
  return token;
221
370
  }
222
371
 
@@ -251,9 +400,11 @@ export class BotFactory {
251
400
  const amm = AMMContract.at(instance.address, this.wallet);
252
401
 
253
402
  this.log.info(`AMM deployed at ${amm.address}`);
254
- const minterReceipt = await lpToken.methods
255
- .set_minter(amm.address, true)
256
- .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
403
+ const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
404
+ const { receipt: minterReceipt } = await setMinterInteraction.send({
405
+ from: deployer,
406
+ wait: { timeout: this.config.txMinedWaitSeconds },
407
+ });
257
408
  this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
258
409
  this.log.info(`Liquidity token initialized`);
259
410
 
@@ -270,9 +421,18 @@ export class BotFactory {
270
421
  ): Promise<void> {
271
422
  const getPrivateBalances = () =>
272
423
  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 }),
424
+ token0.methods
425
+ .balance_of_private(liquidityProvider)
426
+ .simulate({ from: liquidityProvider })
427
+ .then(r => r.result),
428
+ token1.methods
429
+ .balance_of_private(liquidityProvider)
430
+ .simulate({ from: liquidityProvider })
431
+ .then(r => r.result),
432
+ lpToken.methods
433
+ .balance_of_private(liquidityProvider)
434
+ .simulate({ from: liquidityProvider })
435
+ .then(r => r.result),
276
436
  ]);
277
437
 
278
438
  const authwitNonce = Fr.random();
@@ -313,20 +473,29 @@ export class BotFactory {
313
473
  .getFunctionCall(),
314
474
  });
315
475
 
316
- const mintReceipt = await new BatchCall(this.wallet, [
476
+ const mintBatch = new BatchCall(this.wallet, [
317
477
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
318
478
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
319
- ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
479
+ ]);
480
+ const { receipt: mintReceipt } = await mintBatch.send({
481
+ from: liquidityProvider,
482
+ wait: { timeout: this.config.txMinedWaitSeconds },
483
+ });
320
484
 
321
485
  this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
322
486
 
323
- const addLiquidityReceipt = await amm.methods
324
- .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
325
- .send({
326
- from: liquidityProvider,
327
- authWitnesses: [token0Authwit, token1Authwit],
328
- wait: { timeout: this.config.txMinedWaitSeconds },
329
- });
487
+ const addLiquidityInteraction = amm.methods.add_liquidity(
488
+ amount0Max,
489
+ amount1Max,
490
+ amount0Min,
491
+ amount1Min,
492
+ authwitNonce,
493
+ );
494
+ const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
495
+ from: liquidityProvider,
496
+ authWitnesses: [token0Authwit, token1Authwit],
497
+ wait: { timeout: this.config.txMinedWaitSeconds },
498
+ });
330
499
 
331
500
  this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
332
501
  this.log.info(`Liquidity added`);
@@ -349,12 +518,42 @@ export class BotFactory {
349
518
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
350
519
  await deploy.register();
351
520
  } else {
352
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
353
- await this.withNoMinTxsPerBlock(async () => {
354
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
355
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
356
- return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
357
- });
521
+ const sender = deployOpts.from === NO_FROM ? undefined : deployOpts.from;
522
+ const balance = sender ? await getFeeJuiceBalance(sender, this.aztecNode) : 0n;
523
+ const useClaim =
524
+ sender &&
525
+ balance < FEE_JUICE_TOP_UP_THRESHOLD &&
526
+ this.config.feePaymentMethod === 'fee_juice' &&
527
+ !!this.config.l1RpcUrls?.length;
528
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
529
+
530
+ if (useClaim && mnemonicOrPrivateKey) {
531
+ const claim = await this.getOrCreateBridgeClaim(sender!);
532
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(sender!, claim);
533
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true, paymentMethod } });
534
+ const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
535
+ const gasSettings = GasSettings.from({
536
+ ...estimatedGas!,
537
+ maxFeesPerGas,
538
+ maxPriorityFeesPerGas: GasFees.empty(),
539
+ });
540
+ await this.withNoMinTxsPerBlock(async () => {
541
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings, paymentMethod }, wait: NO_WAIT });
542
+ this.log.info(
543
+ `Sent contract ${name} deploy tx ${txHash.toString()} (using bridge claim, balance was ${balance})`,
544
+ );
545
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
546
+ });
547
+ await this.store.deleteBridgeClaim(sender!);
548
+ } else {
549
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
550
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
551
+ await this.withNoMinTxsPerBlock(async () => {
552
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
553
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
554
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
555
+ });
556
+ }
358
557
  }
359
558
  return instance;
360
559
  }
@@ -363,6 +562,66 @@ export class BotFactory {
363
562
  * Mints private and public tokens for the sender if their balance is below the minimum.
364
563
  * @param token - Token contract.
365
564
  */
565
+ /**
566
+ * Ensures the account has sufficient fee juice by bridging from L1 if balance is below threshold.
567
+ * Bridges repeatedly until balance reaches the target (10k FJ).
568
+ * Used on startup/restart to top up when the account has run out after previous runs.
569
+ */
570
+ private async ensureFeeJuiceBalance(
571
+ account: AztecAddress,
572
+ token: TokenContract | PrivateTokenContract,
573
+ ): Promise<void> {
574
+ const { feePaymentMethod, l1RpcUrls } = this.config;
575
+ if (feePaymentMethod !== 'fee_juice' || !l1RpcUrls?.length) {
576
+ return;
577
+ }
578
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
579
+ if (!mnemonicOrPrivateKey) {
580
+ return;
581
+ }
582
+
583
+ let balance = await getFeeJuiceBalance(account, this.aztecNode);
584
+ if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
585
+ this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
586
+ return;
587
+ }
588
+
589
+ this.log.info(
590
+ `Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1 until ${FEE_JUICE_TOP_UP_TARGET}`,
591
+ );
592
+ const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
593
+ const minimalInteraction = isStandardTokenContract(token)
594
+ ? token.methods.transfer_in_public(account, account, 0n, 0)
595
+ : token.methods.transfer(0n, account, account);
596
+
597
+ while (balance < FEE_JUICE_TOP_UP_TARGET) {
598
+ const claim = await this.bridgeL1FeeJuice(account);
599
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
600
+ const { estimatedGas } = await minimalInteraction.simulate({
601
+ from: account,
602
+ fee: { estimateGas: true, paymentMethod },
603
+ });
604
+ const gasSettings = GasSettings.from({
605
+ ...estimatedGas!,
606
+ maxFeesPerGas,
607
+ maxPriorityFeesPerGas: GasFees.empty(),
608
+ });
609
+
610
+ await this.withNoMinTxsPerBlock(async () => {
611
+ const { txHash } = await minimalInteraction.send({
612
+ from: account,
613
+ fee: { gasSettings, paymentMethod },
614
+ wait: NO_WAIT,
615
+ });
616
+ this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
617
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
618
+ });
619
+ balance = await getFeeJuiceBalance(account, this.aztecNode);
620
+ this.log.info(`Fee juice balance after top-up: ${balance}`);
621
+ }
622
+ this.log.info(`Fee juice top-up complete for ${account.toString()}`);
623
+ }
624
+
366
625
  private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
367
626
  const isStandardToken = isStandardTokenContract(token);
368
627
  let privateBalance = 0n;
@@ -393,8 +652,15 @@ export class BotFactory {
393
652
  return;
394
653
  }
395
654
 
655
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
656
+ const additionalScopes = isStandardToken ? undefined : [token.address];
657
+ const mintBatch = new BatchCall(token.wallet, calls);
396
658
  await this.withNoMinTxsPerBlock(async () => {
397
- const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
659
+ const { txHash } = await mintBatch.send({
660
+ from: minter,
661
+ additionalScopes,
662
+ wait: NO_WAIT,
663
+ });
398
664
  this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
399
665
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
400
666
  });
@@ -417,7 +683,6 @@ export class BotFactory {
417
683
  await this.withNoMinTxsPerBlock(() =>
418
684
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
419
685
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
420
- forPublicConsumption: false,
421
686
  }),
422
687
  );
423
688
  return existingClaim.claim;
@@ -456,7 +721,6 @@ export class BotFactory {
456
721
  await this.withNoMinTxsPerBlock(() =>
457
722
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
458
723
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
459
- forPublicConsumption: false,
460
724
  }),
461
725
  );
462
726
 
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
+ }