@aztec/bot 0.0.1-commit.b655e406 → 0.0.1-commit.b9865e97

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 (52) 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 +28 -17
  4. package/dest/base_bot.d.ts +8 -8
  5. package/dest/base_bot.d.ts.map +1 -1
  6. package/dest/base_bot.js +26 -37
  7. package/dest/bot.d.ts +6 -6
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +9 -9
  10. package/dest/config.d.ts +63 -99
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +46 -19
  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 +137 -0
  16. package/dest/factory.d.ts +23 -26
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +374 -114
  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/interface.d.ts +2 -6
  23. package/dest/interface.d.ts.map +1 -1
  24. package/dest/interface.js +30 -7
  25. package/dest/l1_to_l2_seeding.d.ts +8 -0
  26. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  27. package/dest/l1_to_l2_seeding.js +63 -0
  28. package/dest/rpc.d.ts +1 -1
  29. package/dest/runner.d.ts +3 -3
  30. package/dest/runner.d.ts.map +1 -1
  31. package/dest/runner.js +429 -31
  32. package/dest/store/bot_store.d.ts +30 -5
  33. package/dest/store/bot_store.d.ts.map +1 -1
  34. package/dest/store/bot_store.js +38 -7
  35. package/dest/store/index.d.ts +2 -2
  36. package/dest/store/index.d.ts.map +1 -1
  37. package/dest/utils.d.ts +1 -1
  38. package/dest/utils.js +3 -3
  39. package/package.json +20 -16
  40. package/src/amm_bot.ts +27 -22
  41. package/src/base_bot.ts +23 -44
  42. package/src/bot.ts +11 -12
  43. package/src/config.ts +94 -64
  44. package/src/cross_chain_bot.ts +204 -0
  45. package/src/factory.ts +414 -95
  46. package/src/index.ts +1 -0
  47. package/src/interface.ts +7 -7
  48. package/src/l1_to_l2_seeding.ts +79 -0
  49. package/src/runner.ts +18 -5
  50. package/src/store/bot_store.ts +61 -6
  51. package/src/store/index.ts +1 -1
  52. 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,
@@ -7,56 +7,94 @@ import {
7
7
  ContractFunctionInteraction,
8
8
  type DeployMethod,
9
9
  type DeployOptions,
10
+ NO_WAIT,
10
11
  } from '@aztec/aztec.js/contracts';
11
- import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
12
12
  import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
13
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
13
14
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
15
+ import { deriveKeys } from '@aztec/aztec.js/keys';
14
16
  import { createLogger } from '@aztec/aztec.js/log';
15
17
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
16
- import { createEthereumChain, createExtendedL1Client } from '@aztec/ethereum';
17
- import { Fr } from '@aztec/foundation/fields';
18
+ import { waitForTx } from '@aztec/aztec.js/node';
19
+ import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
20
+ import { ContractInitializationStatus } from '@aztec/aztec.js/wallet';
21
+ import { createEthereumChain } from '@aztec/ethereum/chain';
22
+ import { createExtendedL1Client } from '@aztec/ethereum/client';
23
+ import { RollupContract } from '@aztec/ethereum/contracts';
24
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
25
+ import { Fr } from '@aztec/foundation/curves/bn254';
26
+ import { EthAddress } from '@aztec/foundation/eth-address';
18
27
  import { Timer } from '@aztec/foundation/timer';
19
28
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
20
29
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
21
30
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
22
- import { GasSettings } from '@aztec/stdlib/gas';
31
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
32
+ import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
33
+ import { GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
23
34
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
24
35
  import { deriveSigningKey } from '@aztec/stdlib/keys';
25
- import { TestWallet } from '@aztec/test-wallet/server';
36
+ import { EmbeddedWallet } from '@aztec/wallets/embedded';
26
37
 
27
38
  import { type BotConfig, SupportedTokenContracts } from './config.js';
39
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
28
40
  import type { BotStore } from './store/index.js';
29
41
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
30
42
 
31
43
  const MINT_BALANCE = 1e12;
32
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;
33
47
 
34
48
  export class BotFactory {
35
49
  private log = createLogger('bot');
36
50
 
37
51
  constructor(
38
52
  private readonly config: BotConfig,
39
- private readonly wallet: TestWallet,
53
+ private readonly wallet: EmbeddedWallet,
40
54
  private readonly store: BotStore,
41
55
  private readonly aztecNode: AztecNode,
42
56
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
43
- ) {}
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
+ }
44
62
 
45
63
  /**
46
64
  * Initializes a new bot by setting up the sender account, registering the recipient,
47
65
  * deploying the token contract, and minting tokens if necessary.
48
66
  */
49
- public async setup() {
50
- const recipient = (await this.wallet.createAccount()).address;
67
+ public async setup(): Promise<{
68
+ wallet: EmbeddedWallet;
69
+ defaultAccountAddress: AztecAddress;
70
+ token: TokenContract | PrivateTokenContract;
71
+ node: AztecNode;
72
+ recipient: AztecAddress;
73
+ }> {
51
74
  const defaultAccountAddress = await this.setupAccount();
52
- 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);
53
78
  await this.mintTokens(token, defaultAccountAddress);
54
79
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
55
80
  }
56
81
 
57
- public async setupAmm() {
82
+ public async setupAmm(): Promise<{
83
+ wallet: EmbeddedWallet;
84
+ defaultAccountAddress: AztecAddress;
85
+ amm: AMMContract;
86
+ token0: TokenContract;
87
+ token1: TokenContract;
88
+ node: AztecNode;
89
+ }> {
58
90
  const defaultAccountAddress = await this.setupAccount();
59
- 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);
60
98
  const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
61
99
  const liquidityToken = await this.setupTokenContract(
62
100
  defaultAccountAddress,
@@ -78,6 +116,85 @@ export class BotFactory {
78
116
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
79
117
  }
80
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 = { from: deployer };
193
+ const deploy = TestContract.deploy(this.wallet, { salt: this.config.tokenSalt, universalDeploy: true });
194
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
195
+ return TestContract.at(instance.address, this.wallet);
196
+ }
197
+
81
198
  /**
82
199
  * Checks if the sender account contract is initialized, and initializes it if necessary.
83
200
  * @returns The sender wallet.
@@ -96,14 +213,9 @@ export class BotFactory {
96
213
  private async setupAccountWithPrivateKey(secret: Fr) {
97
214
  const salt = this.config.senderSalt ?? Fr.ONE;
98
215
  const signingKey = deriveSigningKey(secret);
99
- const accountData = {
100
- secret,
101
- salt,
102
- contract: new SchnorrAccountContract(signingKey!),
103
- };
104
- const accountManager = await this.wallet.createAccount(accountData);
105
- const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
106
- if (isInit) {
216
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
217
+ const metadata = await this.wallet.getContractMetadata(accountManager.address);
218
+ if (metadata.initializationStatus === ContractInitializationStatus.INITIALIZED) {
107
219
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
108
220
  const timer = new Timer();
109
221
  const address = accountManager.address;
@@ -118,12 +230,16 @@ export class BotFactory {
118
230
 
119
231
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
120
232
  const deployMethod = await accountManager.getDeployMethod();
121
- const maxFeesPerGas = (await this.aztecNode.getCurrentBaseFees()).mul(1 + this.config.baseFeePadding);
122
- const gasSettings = GasSettings.default({ maxFeesPerGas });
123
- const sentTx = deployMethod.send({ from: AztecAddress.ZERO, fee: { gasSettings, paymentMethod } });
124
- const txHash = await sentTx.getTxHash();
125
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
126
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
233
+
234
+ await this.withNoMinTxsPerBlock(async () => {
235
+ const { txHash } = await deployMethod.send({
236
+ from: NO_FROM,
237
+ fee: { paymentMethod },
238
+ wait: NO_WAIT,
239
+ });
240
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
241
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
242
+ });
127
243
  this.log.info(`Account deployed at ${address}`);
128
244
 
129
245
  // Clean up the consumed bridge claim
@@ -135,49 +251,113 @@ export class BotFactory {
135
251
 
136
252
  private async setupTestAccount() {
137
253
  const [initialAccountData] = await getInitialTestAccountsData();
138
- const accountData = {
139
- secret: initialAccountData.secret,
140
- salt: initialAccountData.salt,
141
- contract: new SchnorrAccountContract(initialAccountData.signingKey),
142
- };
143
- const accountManager = await this.wallet.createAccount(accountData);
254
+ const accountManager = await this.wallet.createSchnorrAccount(
255
+ initialAccountData.secret,
256
+ initialAccountData.salt,
257
+ initialAccountData.signingKey,
258
+ );
144
259
  return accountManager.address;
145
260
  }
146
261
 
262
+ /**
263
+ * Setup token and refuel first: if the token already exists (restart scenario),
264
+ * run ensureFeeJuiceBalance before any step that might need fee juice. When deploying,
265
+ * use a bridge claim if balance is below threshold.
266
+ */
267
+ private async setupTokenWithOptionalEarlyRefuel(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
268
+ const token = await this.getTokenInstance(sender);
269
+ const address = token.address;
270
+ const metadata = await this.wallet.getContractMetadata(address);
271
+ if (metadata.isContractPublished) {
272
+ this.log.info(`Token at ${address.toString()} already deployed, refueling before setup`);
273
+ await this.ensureFeeJuiceBalance(sender, token);
274
+ }
275
+ return this.setupToken(sender);
276
+ }
277
+
278
+ /**
279
+ * Setup token0 for AMM with refuel-first behaviour when token already exists.
280
+ */
281
+ private async setupTokenContractWithOptionalEarlyRefuel(
282
+ deployer: AztecAddress,
283
+ salt: Fr,
284
+ name: string,
285
+ ticker: string,
286
+ decimals = 18,
287
+ ): Promise<TokenContract> {
288
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
289
+ const instance = await deploy.getInstance();
290
+ const metadata = await this.wallet.getContractMetadata(instance.address);
291
+ if (metadata.isContractPublished) {
292
+ this.log.info(`Token ${name} at ${instance.address.toString()} already deployed, refueling before setup`);
293
+ const token = TokenContract.at(instance.address, this.wallet);
294
+ await this.ensureFeeJuiceBalance(deployer, token);
295
+ }
296
+ return this.setupTokenContract(deployer, salt, name, ticker, decimals);
297
+ }
298
+
299
+ private async getTokenInstance(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
300
+ const salt = this.config.tokenSalt;
301
+ if (this.config.contract === SupportedTokenContracts.TokenContract) {
302
+ const deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
303
+ const instance = await deploy.getInstance();
304
+ return TokenContract.at(instance.address, this.wallet);
305
+ }
306
+ if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
307
+ const tokenSecretKey = Fr.random();
308
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
309
+ const deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
310
+ salt,
311
+ universalDeploy: true,
312
+ publicKeys: tokenPublicKeys,
313
+ });
314
+ const instance = await deploy.getInstance();
315
+ return PrivateTokenContract.at(instance.address, this.wallet);
316
+ }
317
+ throw new Error(`Unsupported token contract type: ${this.config.contract}`);
318
+ }
319
+
147
320
  /**
148
321
  * Checks if the token contract is deployed and deploys it if necessary.
149
- * @param wallet - Wallet to deploy the token contract from.
150
- * @returns The TokenContract instance.
322
+ * Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
323
+ * @param sender - Aztec address to deploy the token contract from.
324
+ * @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
325
+ * @returns The TokenContract or PrivateTokenContract instance.
151
326
  */
152
327
  private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
153
328
  let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
154
- const deployOpts: DeployOptions = {
155
- from: sender,
156
- contractAddressSalt: this.config.tokenSalt,
157
- universalDeploy: true,
158
- };
329
+ const salt = this.config.tokenSalt;
330
+ const deployOpts: DeployOptions = { from: sender };
331
+ let token: TokenContract | PrivateTokenContract;
159
332
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
160
- deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
333
+ deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
334
+ const instance = await deploy.getInstance();
335
+ token = TokenContract.at(instance.address, this.wallet);
161
336
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
162
- deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender);
337
+ // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
338
+ const tokenSecretKey = Fr.random();
339
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
340
+ deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
341
+ salt,
342
+ universalDeploy: true,
343
+ publicKeys: tokenPublicKeys,
344
+ });
163
345
  deployOpts.skipInstancePublication = true;
164
346
  deployOpts.skipClassPublication = true;
165
347
  deployOpts.skipInitialization = false;
348
+
349
+ // Register the contract with the secret key before deployment
350
+ const tokenInstance = await deploy.getInstance();
351
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
352
+ await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
353
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
354
+ deployOpts.additionalScopes = [tokenInstance.address];
166
355
  } else {
167
356
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
168
357
  }
169
358
 
170
- const address = (await deploy.getInstance(deployOpts)).address;
171
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
172
- this.log.info(`Token at ${address.toString()} already deployed`);
173
- return deploy.register();
174
- } else {
175
- this.log.info(`Deploying token contract at ${address.toString()}`);
176
- const sentTx = deploy.send(deployOpts);
177
- const txHash = await sentTx.getTxHash();
178
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
179
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
180
- }
359
+ await this.registerOrDeployContract('token', deploy, deployOpts);
360
+ return token;
181
361
  }
182
362
 
183
363
  /**
@@ -185,33 +365,41 @@ export class BotFactory {
185
365
  * @param wallet - Wallet to deploy the token contract from.
186
366
  * @returns The TokenContract instance.
187
367
  */
188
- private setupTokenContract(
368
+ private async setupTokenContract(
189
369
  deployer: AztecAddress,
190
- contractAddressSalt: Fr,
370
+ salt: Fr,
191
371
  name: string,
192
372
  ticker: string,
193
373
  decimals = 18,
194
374
  ): Promise<TokenContract> {
195
- const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
196
- const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
197
- return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
375
+ const deployOpts: DeployOptions = { from: deployer };
376
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
377
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
378
+ return TokenContract.at(instance.address, this.wallet);
198
379
  }
199
380
 
200
381
  private async setupAmmContract(
201
382
  deployer: AztecAddress,
202
- contractAddressSalt: Fr,
383
+ salt: Fr,
203
384
  token0: TokenContract,
204
385
  token1: TokenContract,
205
386
  lpToken: TokenContract,
206
387
  ): Promise<AMMContract> {
207
- const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
208
- const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
209
- const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
388
+ const deployOpts: DeployOptions = { from: deployer };
389
+ const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address, {
390
+ salt,
391
+ universalDeploy: true,
392
+ });
393
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
394
+ const amm = AMMContract.at(instance.address, this.wallet);
210
395
 
211
396
  this.log.info(`AMM deployed at ${amm.address}`);
212
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({ from: deployer });
213
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
214
- await minterTx.wait({ timeout: this.config.txMinedWaitSeconds });
397
+ const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
398
+ const { receipt: minterReceipt } = await setMinterInteraction.send({
399
+ from: deployer,
400
+ wait: { timeout: this.config.txMinedWaitSeconds },
401
+ });
402
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
215
403
  this.log.info(`Liquidity token initialized`);
216
404
 
217
405
  return amm;
@@ -227,9 +415,18 @@ export class BotFactory {
227
415
  ): Promise<void> {
228
416
  const getPrivateBalances = () =>
229
417
  Promise.all([
230
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
231
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
232
- lpToken.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
418
+ token0.methods
419
+ .balance_of_private(liquidityProvider)
420
+ .simulate({ from: liquidityProvider })
421
+ .then(r => r.result),
422
+ token1.methods
423
+ .balance_of_private(liquidityProvider)
424
+ .simulate({ from: liquidityProvider })
425
+ .then(r => r.result),
426
+ lpToken.methods
427
+ .balance_of_private(liquidityProvider)
428
+ .simulate({ from: liquidityProvider })
429
+ .then(r => r.result),
233
430
  ]);
234
431
 
235
432
  const authwitNonce = Fr.random();
@@ -270,23 +467,31 @@ export class BotFactory {
270
467
  .getFunctionCall(),
271
468
  });
272
469
 
273
- const mintTx = new BatchCall(this.wallet, [
470
+ const mintBatch = new BatchCall(this.wallet, [
274
471
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
275
472
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
276
- ]).send({ from: liquidityProvider });
473
+ ]);
474
+ const { receipt: mintReceipt } = await mintBatch.send({
475
+ from: liquidityProvider,
476
+ wait: { timeout: this.config.txMinedWaitSeconds },
477
+ });
277
478
 
278
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
279
- await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
479
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
280
480
 
281
- const addLiquidityTx = amm.methods
282
- .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
283
- .send({
284
- from: liquidityProvider,
285
- authWitnesses: [token0Authwit, token1Authwit],
286
- });
481
+ const addLiquidityInteraction = amm.methods.add_liquidity(
482
+ amount0Max,
483
+ amount1Max,
484
+ amount0Min,
485
+ amount1Min,
486
+ authwitNonce,
487
+ );
488
+ const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
489
+ from: liquidityProvider,
490
+ authWitnesses: [token0Authwit, token1Authwit],
491
+ wait: { timeout: this.config.txMinedWaitSeconds },
492
+ });
287
493
 
288
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
289
- await addLiquidityTx.wait({ timeout: this.config.txMinedWaitSeconds });
494
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
290
495
  this.log.info(`Liquidity added`);
291
496
 
292
497
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
@@ -299,24 +504,118 @@ export class BotFactory {
299
504
  name: string,
300
505
  deploy: DeployMethod<T>,
301
506
  deployOpts: DeployOptions,
302
- ): Promise<T> {
303
- const address = (await deploy.getInstance(deployOpts)).address;
304
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
507
+ ): Promise<ContractInstanceWithAddress> {
508
+ const instance = await deploy.getInstance();
509
+ const address = instance.address;
510
+ const metadata = await this.wallet.getContractMetadata(address);
511
+ if (metadata.isContractPublished) {
305
512
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
306
- return deploy.register();
513
+ await deploy.register();
307
514
  } else {
308
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
309
- const sentTx = deploy.send(deployOpts);
310
- const txHash = await sentTx.getTxHash();
311
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
312
- return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
515
+ const sender = deployOpts.from === NO_FROM ? undefined : deployOpts.from;
516
+ const balance = sender ? await getFeeJuiceBalance(sender, this.aztecNode) : 0n;
517
+ const useClaim =
518
+ sender &&
519
+ balance < FEE_JUICE_TOP_UP_THRESHOLD &&
520
+ this.config.feePaymentMethod === 'fee_juice' &&
521
+ !!this.config.l1RpcUrls?.length;
522
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
523
+
524
+ if (useClaim && mnemonicOrPrivateKey) {
525
+ const claim = await this.getOrCreateBridgeClaim(sender!);
526
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(sender!, claim);
527
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true, paymentMethod } });
528
+ const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
529
+ const gasSettings = GasSettings.from({
530
+ ...estimatedGas!,
531
+ maxFeesPerGas,
532
+ maxPriorityFeesPerGas: GasFees.empty(),
533
+ });
534
+ await this.withNoMinTxsPerBlock(async () => {
535
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings, paymentMethod }, wait: NO_WAIT });
536
+ this.log.info(
537
+ `Sent contract ${name} deploy tx ${txHash.toString()} (using bridge claim, balance was ${balance})`,
538
+ );
539
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
540
+ });
541
+ await this.store.deleteBridgeClaim(sender!);
542
+ } else {
543
+ const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
544
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
545
+ await this.withNoMinTxsPerBlock(async () => {
546
+ const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
547
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
548
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
549
+ });
550
+ }
313
551
  }
552
+ return instance;
314
553
  }
315
554
 
316
555
  /**
317
556
  * Mints private and public tokens for the sender if their balance is below the minimum.
318
557
  * @param token - Token contract.
319
558
  */
559
+ /**
560
+ * Ensures the account has sufficient fee juice by bridging from L1 if balance is below threshold.
561
+ * Bridges repeatedly until balance reaches the target (10k FJ).
562
+ * Used on startup/restart to top up when the account has run out after previous runs.
563
+ */
564
+ private async ensureFeeJuiceBalance(
565
+ account: AztecAddress,
566
+ token: TokenContract | PrivateTokenContract,
567
+ ): Promise<void> {
568
+ const { feePaymentMethod, l1RpcUrls } = this.config;
569
+ if (feePaymentMethod !== 'fee_juice' || !l1RpcUrls?.length) {
570
+ return;
571
+ }
572
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
573
+ if (!mnemonicOrPrivateKey) {
574
+ return;
575
+ }
576
+
577
+ let balance = await getFeeJuiceBalance(account, this.aztecNode);
578
+ if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
579
+ this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
580
+ return;
581
+ }
582
+
583
+ this.log.info(
584
+ `Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1 until ${FEE_JUICE_TOP_UP_TARGET}`,
585
+ );
586
+ const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
587
+ const minimalInteraction = isStandardTokenContract(token)
588
+ ? token.methods.transfer_in_public(account, account, 0n, 0)
589
+ : token.methods.transfer(0n, account, account);
590
+
591
+ while (balance < FEE_JUICE_TOP_UP_TARGET) {
592
+ const claim = await this.bridgeL1FeeJuice(account);
593
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
594
+ const { estimatedGas } = await minimalInteraction.simulate({
595
+ from: account,
596
+ fee: { estimateGas: true, paymentMethod },
597
+ });
598
+ const gasSettings = GasSettings.from({
599
+ ...estimatedGas!,
600
+ maxFeesPerGas,
601
+ maxPriorityFeesPerGas: GasFees.empty(),
602
+ });
603
+
604
+ await this.withNoMinTxsPerBlock(async () => {
605
+ const { txHash } = await minimalInteraction.send({
606
+ from: account,
607
+ fee: { gasSettings, paymentMethod },
608
+ wait: NO_WAIT,
609
+ });
610
+ this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
611
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
612
+ });
613
+ balance = await getFeeJuiceBalance(account, this.aztecNode);
614
+ this.log.info(`Fee juice balance after top-up: ${balance}`);
615
+ }
616
+ this.log.info(`Fee juice top-up complete for ${account.toString()}`);
617
+ }
618
+
320
619
  private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
321
620
  const isStandardToken = isStandardTokenContract(token);
322
621
  let privateBalance = 0n;
@@ -346,10 +645,19 @@ export class BotFactory {
346
645
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
347
646
  return;
348
647
  }
349
- const sentTx = new BatchCall(token.wallet, calls).send({ from: minter });
350
- const txHash = await sentTx.getTxHash();
351
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
352
- await this.withNoMinTxsPerBlock(() => sentTx.wait({ timeout: this.config.txMinedWaitSeconds }));
648
+
649
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
650
+ const additionalScopes = isStandardToken ? undefined : [token.address];
651
+ const mintBatch = new BatchCall(token.wallet, calls);
652
+ await this.withNoMinTxsPerBlock(async () => {
653
+ const { txHash } = await mintBatch.send({
654
+ from: minter,
655
+ additionalScopes,
656
+ wait: NO_WAIT,
657
+ });
658
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
659
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
660
+ });
353
661
  }
354
662
 
355
663
  /**
@@ -369,7 +677,6 @@ export class BotFactory {
369
677
  await this.withNoMinTxsPerBlock(() =>
370
678
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
371
679
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
372
- forPublicConsumption: false,
373
680
  }),
374
681
  );
375
682
  return existingClaim.claim;
@@ -408,7 +715,6 @@ export class BotFactory {
408
715
  await this.withNoMinTxsPerBlock(() =>
409
716
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
410
717
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
411
- forPublicConsumption: false,
412
718
  }),
413
719
  );
414
720
 
@@ -417,6 +723,19 @@ export class BotFactory {
417
723
  return claim as L2AmountClaim;
418
724
  }
419
725
 
726
+ /** Returns worst-case min fees across predicted slots, with fallback to current min fees. */
727
+ private async getMinFees(): Promise<GasFees> {
728
+ try {
729
+ const predicted = await this.aztecNode.getPredictedMinFees(ManaUsageEstimate.Limit);
730
+ if (predicted.length === 0) {
731
+ return this.aztecNode.getCurrentMinFees();
732
+ }
733
+ return predicted.reduce((worst, fees) => (fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst));
734
+ } catch {
735
+ return this.aztecNode.getCurrentMinFees();
736
+ }
737
+ }
738
+
420
739
  private async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
421
740
  if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
422
741
  this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);