@aztec/bot 0.0.1-commit.4d79d1f2d → 0.0.1-commit.4d9804df

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 (50) hide show
  1. package/dest/amm_bot.d.ts +4 -3
  2. package/dest/amm_bot.d.ts.map +1 -1
  3. package/dest/amm_bot.js +26 -19
  4. package/dest/base_bot.d.ts +4 -4
  5. package/dest/base_bot.d.ts.map +1 -1
  6. package/dest/base_bot.js +21 -32
  7. package/dest/bot.d.ts +3 -2
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +5 -8
  10. package/dest/config.d.ts +47 -82
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +42 -15
  13. package/dest/cross_chain_bot.d.ts +56 -0
  14. package/dest/cross_chain_bot.d.ts.map +1 -0
  15. package/dest/cross_chain_bot.js +138 -0
  16. package/dest/factory.d.ts +31 -6
  17. package/dest/factory.d.ts.map +1 -1
  18. package/dest/factory.js +293 -158
  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/runner.d.ts +3 -2
  29. package/dest/runner.d.ts.map +1 -1
  30. package/dest/runner.js +20 -2
  31. package/dest/store/bot_store.d.ts +30 -5
  32. package/dest/store/bot_store.d.ts.map +1 -1
  33. package/dest/store/bot_store.js +37 -6
  34. package/dest/store/index.d.ts +2 -2
  35. package/dest/store/index.d.ts.map +1 -1
  36. package/dest/utils.js +3 -3
  37. package/package.json +17 -14
  38. package/src/amm_bot.ts +25 -17
  39. package/src/base_bot.ts +14 -31
  40. package/src/bot.ts +6 -5
  41. package/src/config.ts +47 -18
  42. package/src/cross_chain_bot.ts +208 -0
  43. package/src/factory.ts +314 -162
  44. package/src/index.ts +1 -0
  45. package/src/interface.ts +7 -7
  46. package/src/l1_to_l2_seeding.ts +79 -0
  47. package/src/runner.ts +39 -3
  48. package/src/store/bot_store.ts +60 -5
  49. package/src/store/index.ts +1 -1
  50. package/src/utils.ts +3 -3
package/src/factory.ts CHANGED
@@ -1,4 +1,5 @@
1
1
  import { getInitialTestAccountsData } from '@aztec/accounts/testing';
2
+ import { deriveSecretKeyFromSigningKey } from '@aztec/accounts/utils';
2
3
  import { AztecAddress } from '@aztec/aztec.js/addresses';
3
4
  import {
4
5
  BatchCall,
@@ -8,43 +9,59 @@ import {
8
9
  type DeployOptions,
9
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';
14
15
  import { deriveKeys } from '@aztec/aztec.js/keys';
15
16
  import { createLogger } from '@aztec/aztec.js/log';
16
17
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
17
18
  import { waitForTx } from '@aztec/aztec.js/node';
19
+ import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
18
20
  import { createEthereumChain } from '@aztec/ethereum/chain';
19
21
  import { createExtendedL1Client } from '@aztec/ethereum/client';
22
+ import { RollupContract } from '@aztec/ethereum/contracts';
23
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
20
24
  import { Fr } from '@aztec/foundation/curves/bn254';
21
- import { Timer } from '@aztec/foundation/timer';
25
+ import { GrumpkinScalar } from '@aztec/foundation/curves/grumpkin';
26
+ import { EthAddress } from '@aztec/foundation/eth-address';
22
27
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
23
28
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
24
29
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
30
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
31
+ import type { BlockTag } from '@aztec/stdlib/block';
25
32
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
26
- import { GasSettings } from '@aztec/stdlib/gas';
27
33
  import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
28
- import { deriveSigningKey } from '@aztec/stdlib/keys';
29
34
  import { EmbeddedWallet } from '@aztec/wallets/embedded';
30
35
 
31
36
  import { type BotConfig, SupportedTokenContracts } from './config.js';
37
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
32
38
  import type { BotStore } from './store/index.js';
33
39
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
34
40
 
35
41
  const MINT_BALANCE = 1e12;
36
42
  const MIN_BALANCE = 1e3;
43
+ const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
37
44
 
38
45
  export class BotFactory {
39
46
  private log = createLogger('bot');
40
47
 
48
+ /** Number of in-flight withNoMinTxsPerBlock calls; see that method for why they are counted. */
49
+ private noMinTxsPerBlockDepth = 0;
50
+ /** Set by the first withNoMinTxsPerBlock entrant; resolves to the minTxsPerBlock value to restore. */
51
+ private savedMinTxsPerBlock?: Promise<{ minTxsPerBlock?: number }>;
52
+
41
53
  constructor(
42
54
  private readonly config: BotConfig,
43
55
  private readonly wallet: EmbeddedWallet,
44
56
  private readonly store: BotStore,
45
57
  private readonly aztecNode: AztecNode,
46
58
  private readonly aztecNodeAdmin?: AztecNodeAdmin,
47
- ) {}
59
+ private readonly syncChainTip?: BlockTag,
60
+ ) {
61
+ // Set fee padding on the wallet so that all transactions during setup
62
+ // (token deploy, minting, etc.) use the configured padding, not the default.
63
+ this.wallet.setMinFeePadding(config.minFeePadding);
64
+ }
48
65
 
49
66
  /**
50
67
  * Initializes a new bot by setting up the sender account, registering the recipient,
@@ -58,7 +75,9 @@ export class BotFactory {
58
75
  recipient: AztecAddress;
59
76
  }> {
60
77
  const defaultAccountAddress = await this.setupAccount();
61
- const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
78
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random(), GrumpkinScalar.random()))
79
+ .address;
80
+ await this.ensureFeeJuiceBalance(defaultAccountAddress);
62
81
  const token = await this.setupToken(defaultAccountAddress);
63
82
  await this.mintTokens(token, defaultAccountAddress);
64
83
  return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
@@ -73,28 +92,127 @@ export class BotFactory {
73
92
  node: AztecNode;
74
93
  }> {
75
94
  const defaultAccountAddress = await this.setupAccount();
76
- const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
77
- const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
78
- const liquidityToken = await this.setupTokenContract(
79
- defaultAccountAddress,
80
- this.config.tokenSalt,
81
- 'BotLPToken',
82
- 'BOTLP',
83
- );
84
- const amm = await this.setupAmmContract(
85
- defaultAccountAddress,
86
- this.config.tokenSalt,
87
- token0,
88
- token1,
89
- liquidityToken,
90
- );
95
+ await this.ensureFeeJuiceBalance(defaultAccountAddress);
91
96
 
92
- await this.fundAmm(defaultAccountAddress, defaultAccountAddress, amm, token0, token1, liquidityToken);
97
+ const salt = this.config.tokenSalt;
98
+
99
+ // token0, token1 and the LP token are independent contracts with no shared state, so deploy them
100
+ // concurrently rather than one slot at a time.
101
+ const [token0, token1, liquidityToken] = await Promise.all([
102
+ this.setupTokenContract(defaultAccountAddress, salt, 'BotToken0', 'BOT0'),
103
+ this.setupTokenContract(defaultAccountAddress, salt, 'BotToken1', 'BOT1'),
104
+ this.setupTokenContract(defaultAccountAddress, salt, 'BotLPToken', 'BOTLP'),
105
+ ]);
106
+
107
+ const ammDeploy = AMMContract.deploy(this.wallet, token0.address, token1.address, liquidityToken.address, {
108
+ salt,
109
+ universalDeploy: true,
110
+ });
111
+ const ammAddress = (await ammDeploy.getInstance()).address;
112
+
113
+ // The AMM constructor only stores the (already-derived) token addresses, and set_minter only records
114
+ // the AMM address on the LP token: neither reads the other's on-chain state, so the AMM deploy, the
115
+ // LP-minter grant, and the token0/token1 mints are mutually independent and run concurrently.
116
+ const [amm] = await Promise.all([
117
+ this.deployAmmContract(defaultAccountAddress, ammDeploy),
118
+ this.grantLpTokenMinter(defaultAccountAddress, liquidityToken, ammAddress),
119
+ this.mintAmmLiquidity(defaultAccountAddress, token0, token1),
120
+ ]);
121
+
122
+ // add_liquidity spends the minted token0/token1 balances and mints LP tokens, so it must follow both
123
+ // the mints and the minter grant, and target the deployed AMM.
124
+ await this.addAmmLiquidity(defaultAccountAddress, defaultAccountAddress, amm, token0, token1, liquidityToken);
93
125
  this.log.info(`AMM initialized and funded`);
94
126
 
95
127
  return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
96
128
  }
97
129
 
130
+ /**
131
+ * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
132
+ * seeding initial L1→L2 messages, and waiting for the first to be ready.
133
+ */
134
+ public async setupCrossChain(): Promise<{
135
+ wallet: EmbeddedWallet;
136
+ defaultAccountAddress: AztecAddress;
137
+ contract: TestContract;
138
+ node: AztecNode;
139
+ l1Client: ExtendedViemWalletClient;
140
+ rollupVersion: bigint;
141
+ }> {
142
+ const defaultAccountAddress = await this.setupAccount();
143
+ await this.ensureFeeJuiceBalance(defaultAccountAddress);
144
+
145
+ // Create L1 client (same pattern as bridgeL1FeeJuice)
146
+ const l1RpcUrls = this.config.l1RpcUrls;
147
+ if (!l1RpcUrls?.length) {
148
+ throw new Error('L1 RPC URLs required for cross-chain bot');
149
+ }
150
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
151
+ if (!mnemonicOrPrivateKey) {
152
+ throw new Error('L1 mnemonic or private key required for cross-chain bot');
153
+ }
154
+ const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
155
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
156
+ const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
157
+
158
+ // Fetch Rollup version (needed for Inbox L2Actor struct)
159
+ const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
160
+ const rollupVersion = await rollupContract.getVersion();
161
+
162
+ // Derive the TestContract address up front (deterministic from the salt). Seeding L1→L2 messages only
163
+ // needs the L2 recipient address — the messages are queued on L1 and don't require the L2 contract to
164
+ // exist yet (they're consumed later, after setup completes) — so the deploy (an L2 tx paying from the
165
+ // standing balance funded above) and the L1 seeding run concurrently.
166
+ const testContractDeploy = TestContract.deploy(this.wallet, {
167
+ salt: this.config.tokenSalt,
168
+ universalDeploy: true,
169
+ });
170
+ const contractAddress = (await testContractDeploy.getInstance()).address;
171
+
172
+ // Recover any pending messages from store (clean up stale ones first)
173
+ await this.store.cleanupOldPendingMessages();
174
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
175
+
176
+ // Seed initial L1→L2 messages if pipeline is empty. The seeds are sent one at a time: they share the
177
+ // bot's L1 account, so concurrent sends would race on the L1 nonce.
178
+ const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
179
+ const inboxAddress = EthAddress.fromString(l1ContractAddresses.inboxAddress.toString());
180
+ const [contract] = await Promise.all([
181
+ this.deployTestContract(defaultAccountAddress, testContractDeploy),
182
+ (async () => {
183
+ for (let i = 0; i < seedCount; i++) {
184
+ await seedL1ToL2Message(l1Client, inboxAddress, contractAddress, rollupVersion, this.store, this.log);
185
+ }
186
+ })(),
187
+ ]);
188
+
189
+ // Block until at least one message is ready
190
+ const allMessages = await this.store.getUnconsumedL1ToL2Messages();
191
+ if (allMessages.length > 0) {
192
+ this.log.info(`Waiting for first L1→L2 message to be ready...`);
193
+ const firstMsg = allMessages[0];
194
+ await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
195
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
196
+ chainTip: this.syncChainTip,
197
+ });
198
+ this.log.info(`First L1→L2 message is ready`);
199
+ }
200
+
201
+ return {
202
+ wallet: this.wallet,
203
+ defaultAccountAddress,
204
+ contract,
205
+ node: this.aztecNode,
206
+ l1Client,
207
+ rollupVersion,
208
+ };
209
+ }
210
+
211
+ private async deployTestContract(deployer: AztecAddress, deploy: DeployMethod<TestContract>): Promise<TestContract> {
212
+ const instance = await this.registerOrDeployContract('TestContract', deploy, { from: deployer });
213
+ return TestContract.at(instance.address, this.wallet);
214
+ }
215
+
98
216
  /**
99
217
  * Checks if the sender account contract is initialized, and initializes it if necessary.
100
218
  * @returns The sender wallet.
@@ -110,50 +228,16 @@ export class BotFactory {
110
228
  }
111
229
  }
112
230
 
113
- private async setupAccountWithPrivateKey(secret: Fr) {
114
- const salt = this.config.senderSalt ?? Fr.ONE;
115
- const signingKey = deriveSigningKey(secret);
116
- const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
117
- const metadata = await this.wallet.getContractMetadata(accountManager.address);
118
- if (metadata.isContractInitialized) {
119
- this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
120
- const timer = new Timer();
121
- const address = accountManager.address;
122
- this.log.info(`Account at ${address} registered. duration=${timer.ms()}`);
123
- await this.store.deleteBridgeClaim(address);
124
- return address;
125
- } else {
126
- const address = accountManager.address;
127
- this.log.info(`Deploying account at ${address}`);
128
-
129
- const claim = await this.getOrCreateBridgeClaim(address);
130
-
131
- const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
132
- const deployMethod = await accountManager.getDeployMethod();
133
- const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
134
- const gasSettings = GasSettings.default({ maxFeesPerGas });
135
-
136
- await this.withNoMinTxsPerBlock(async () => {
137
- const txHash = await deployMethod.send({
138
- from: AztecAddress.ZERO,
139
- fee: { gasSettings, paymentMethod },
140
- wait: NO_WAIT,
141
- });
142
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
143
- return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
144
- });
145
- this.log.info(`Account deployed at ${address}`);
146
-
147
- // Clean up the consumed bridge claim
148
- await this.store.deleteBridgeClaim(address);
149
-
150
- return accountManager.address;
151
- }
152
- }
153
-
231
+ /**
232
+ * Keyless fallback for tests and local dev: reuses the first genesis test account, whose address is
233
+ * pre-funded with fee juice via `initialFundedAccounts`. The test accounts are initializerless, so this
234
+ * must create an initializerless account for the address to match the funded one. Production bots set a
235
+ * sender private key and fund the resulting initializerless account from L1 instead; see
236
+ * setupAccountWithPrivateKey.
237
+ */
154
238
  private async setupTestAccount() {
155
239
  const [initialAccountData] = await getInitialTestAccountsData();
156
- const accountManager = await this.wallet.createSchnorrAccount(
240
+ const accountManager = await this.wallet.createSchnorrInitializerlessAccount(
157
241
  initialAccountData.secret,
158
242
  initialAccountData.salt,
159
243
  initialAccountData.signingKey,
@@ -161,55 +245,54 @@ export class BotFactory {
161
245
  return accountManager.address;
162
246
  }
163
247
 
248
+ private async setupAccountWithPrivateKey(privateKey: Fr) {
249
+ const salt = this.config.senderSalt ?? Fr.ONE;
250
+ const signingKey = GrumpkinScalar.fromBuffer(privateKey.toBuffer());
251
+ const secret = await deriveSecretKeyFromSigningKey(signingKey);
252
+ const accountManager = await this.wallet.createSchnorrInitializerlessAccount(secret, salt, signingKey);
253
+ return accountManager.address;
254
+ }
255
+
164
256
  /**
165
257
  * Checks if the token contract is deployed and deploys it if necessary.
166
- * @param wallet - Wallet to deploy the token contract from.
167
- * @returns The TokenContract instance.
258
+ * Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
259
+ * @param sender - Aztec address to deploy the token contract from.
260
+ * @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
261
+ * @returns The TokenContract or PrivateTokenContract instance.
168
262
  */
169
263
  private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
170
264
  let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
171
- let tokenInstance: ContractInstanceWithAddress | undefined;
172
- const deployOpts: DeployOptions = {
173
- from: sender,
174
- contractAddressSalt: this.config.tokenSalt,
175
- universalDeploy: true,
176
- };
265
+ const salt = this.config.tokenSalt;
266
+ const deployOpts: DeployOptions = { from: sender };
177
267
  let token: TokenContract | PrivateTokenContract;
178
268
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
179
- deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
180
- tokenInstance = await deploy.getInstance(deployOpts);
181
- token = TokenContract.at(tokenInstance.address, this.wallet);
269
+ deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
270
+ const instance = await deploy.getInstance();
271
+ token = TokenContract.at(instance.address, this.wallet);
182
272
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
183
273
  // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
184
274
  const tokenSecretKey = Fr.random();
185
275
  const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
186
- deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
276
+ deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
277
+ salt,
278
+ universalDeploy: true,
279
+ publicKeys: tokenPublicKeys,
280
+ });
187
281
  deployOpts.skipInstancePublication = true;
188
282
  deployOpts.skipClassPublication = true;
189
283
  deployOpts.skipInitialization = false;
190
284
 
191
285
  // Register the contract with the secret key before deployment
192
- tokenInstance = await deploy.getInstance(deployOpts);
286
+ const tokenInstance = await deploy.getInstance();
193
287
  token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
194
288
  await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
289
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
290
+ deployOpts.additionalScopes = [tokenInstance.address];
195
291
  } else {
196
292
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
197
293
  }
198
294
 
199
- const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
200
- const metadata = await this.wallet.getContractMetadata(address);
201
- if (metadata.isContractPublished) {
202
- this.log.info(`Token at ${address.toString()} already deployed`);
203
- await deploy.register();
204
- } else {
205
- this.log.info(`Deploying token contract at ${address.toString()}`);
206
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
207
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
208
- await this.withNoMinTxsPerBlock(async () => {
209
- await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
210
- return token;
211
- });
212
- }
295
+ await this.registerOrDeployContract('token', deploy, deployOpts);
213
296
  return token;
214
297
  }
215
298
 
@@ -220,40 +303,48 @@ export class BotFactory {
220
303
  */
221
304
  private async setupTokenContract(
222
305
  deployer: AztecAddress,
223
- contractAddressSalt: Fr,
306
+ salt: Fr,
224
307
  name: string,
225
308
  ticker: string,
226
309
  decimals = 18,
227
310
  ): Promise<TokenContract> {
228
- const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
229
- const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
311
+ const deployOpts: DeployOptions = { from: deployer };
312
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
230
313
  const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
231
314
  return TokenContract.at(instance.address, this.wallet);
232
315
  }
233
316
 
234
- private async setupAmmContract(
235
- deployer: AztecAddress,
236
- contractAddressSalt: Fr,
237
- token0: TokenContract,
238
- token1: TokenContract,
239
- lpToken: TokenContract,
240
- ): Promise<AMMContract> {
241
- const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
242
- const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
243
- const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
317
+ private async deployAmmContract(deployer: AztecAddress, deploy: DeployMethod<AMMContract>): Promise<AMMContract> {
318
+ const instance = await this.registerOrDeployContract('AMM', deploy, { from: deployer });
244
319
  const amm = AMMContract.at(instance.address, this.wallet);
245
-
246
320
  this.log.info(`AMM deployed at ${amm.address}`);
247
- const minterReceipt = await lpToken.methods
248
- .set_minter(amm.address, true)
249
- .send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
250
- this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
251
- this.log.info(`Liquidity token initialized`);
252
-
253
321
  return amm;
254
322
  }
255
323
 
256
- private async fundAmm(
324
+ /** Grants the AMM minting rights over the LP token. set_minter only records the address, so it does not
325
+ * require the AMM contract to be deployed first. */
326
+ private async grantLpTokenMinter(deployer: AztecAddress, lpToken: TokenContract, amm: AztecAddress): Promise<void> {
327
+ const { receipt } = await lpToken.methods.set_minter(amm, true).send({
328
+ from: deployer,
329
+ wait: { timeout: this.config.txMinedWaitSeconds },
330
+ });
331
+ this.log.info(`Set LP token minter to AMM txHash=${receipt.txHash.toString()}`);
332
+ }
333
+
334
+ private async mintAmmLiquidity(minter: AztecAddress, token0: TokenContract, token1: TokenContract): Promise<void> {
335
+ this.log.info(`Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1 for ${minter}`);
336
+ const mintBatch = new BatchCall(this.wallet, [
337
+ token0.methods.mint_to_private(minter, MINT_BALANCE),
338
+ token1.methods.mint_to_private(minter, MINT_BALANCE),
339
+ ]);
340
+ const { receipt } = await mintBatch.send({
341
+ from: minter,
342
+ wait: { timeout: this.config.txMinedWaitSeconds },
343
+ });
344
+ this.log.info(`Sent mint tx: ${receipt.txHash.toString()}`);
345
+ }
346
+
347
+ private async addAmmLiquidity(
257
348
  defaultAccountAddress: AztecAddress,
258
349
  liquidityProvider: AztecAddress,
259
350
  amm: AMMContract,
@@ -261,13 +352,6 @@ export class BotFactory {
261
352
  token1: TokenContract,
262
353
  lpToken: TokenContract,
263
354
  ): Promise<void> {
264
- const getPrivateBalances = () =>
265
- Promise.all([
266
- token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
267
- token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
268
- lpToken.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
269
- ]);
270
-
271
355
  const authwitNonce = Fr.random();
272
356
 
273
357
  // keep some tokens for swapping
@@ -276,12 +360,6 @@ export class BotFactory {
276
360
  const amount1Max = MINT_BALANCE / 2;
277
361
  const amount1Min = MINT_BALANCE / 4;
278
362
 
279
- const [t0Bal, t1Bal, lpBal] = await getPrivateBalances();
280
-
281
- this.log.info(
282
- `Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1. Current private balances of ${liquidityProvider}: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`,
283
- );
284
-
285
363
  // Add authwitnesses for the transfers in AMM::add_liquidity function
286
364
  const token0Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
287
365
  caller: amm.address,
@@ -306,14 +384,7 @@ export class BotFactory {
306
384
  .getFunctionCall(),
307
385
  });
308
386
 
309
- const mintReceipt = await new BatchCall(this.wallet, [
310
- token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
311
- token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
312
- ]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
313
-
314
- this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
315
-
316
- const addLiquidityReceipt = await amm.methods
387
+ const { receipt } = await amm.methods
317
388
  .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
318
389
  .send({
319
390
  from: liquidityProvider,
@@ -321,12 +392,25 @@ export class BotFactory {
321
392
  wait: { timeout: this.config.txMinedWaitSeconds },
322
393
  });
323
394
 
324
- this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
395
+ this.log.info(`Sent tx to add liquidity to the AMM: ${receipt.txHash.toString()}`);
325
396
  this.log.info(`Liquidity added`);
326
397
 
327
- const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
398
+ const [t0Bal, t1Bal, lpBal] = await Promise.all([
399
+ token0.methods
400
+ .balance_of_private(liquidityProvider)
401
+ .simulate({ from: liquidityProvider })
402
+ .then(r => r.result),
403
+ token1.methods
404
+ .balance_of_private(liquidityProvider)
405
+ .simulate({ from: liquidityProvider })
406
+ .then(r => r.result),
407
+ lpToken.methods
408
+ .balance_of_private(liquidityProvider)
409
+ .simulate({ from: liquidityProvider })
410
+ .then(r => r.result),
411
+ ]);
328
412
  this.log.info(
329
- `Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${newT0Bal}, token1=${newT1Bal}, lp=${newLPBal}`,
413
+ `Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`,
330
414
  );
331
415
  }
332
416
 
@@ -335,27 +419,75 @@ export class BotFactory {
335
419
  deploy: DeployMethod<T>,
336
420
  deployOpts: DeployOptions,
337
421
  ): Promise<ContractInstanceWithAddress> {
338
- const instance = await deploy.getInstance(deployOpts);
422
+ const instance = await deploy.getInstance();
339
423
  const address = instance.address;
340
424
  const metadata = await this.wallet.getContractMetadata(address);
341
425
  if (metadata.isContractPublished) {
342
426
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
343
427
  await deploy.register();
344
- } else {
345
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
346
- await this.withNoMinTxsPerBlock(async () => {
347
- const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
348
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
349
- return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
350
- });
428
+ return instance;
351
429
  }
430
+
431
+ // Setup always runs ensureFeeJuiceBalance before any deploy, so the account pays from its standing
432
+ // balance here. No manual gas estimation: the embedded wallet simulates before sending and derives
433
+ // the gas limits and padded maxFeesPerGas itself.
434
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`);
435
+ await this.withNoMinTxsPerBlock(async () => {
436
+ const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
437
+ this.log.info(`Sent contract ${name} deploy tx ${txHash.toString()}`);
438
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
439
+ });
440
+
352
441
  return instance;
353
442
  }
354
443
 
444
+ /** True when the config allows bridging fee juice from L1 (fee_juice mode, an L1 RPC, and an L1 key). */
445
+ private isL1BridgingConfigured(): boolean {
446
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
447
+ return this.config.feePaymentMethod === 'fee_juice' && !!this.config.l1RpcUrls?.length && !!mnemonicOrPrivateKey;
448
+ }
449
+
355
450
  /**
356
- * Mints private and public tokens for the sender if their balance is below the minimum.
357
- * @param token - Token contract.
451
+ * Ensures the account holds enough fee juice before any other setup step. The account starts empty
452
+ * (initializerless accounts have no deployment tx) and the runtime loop pays fees from this balance and
453
+ * never refuels itself, so every flow funds the account up front. Bridges claims from L1 and consumes
454
+ * each with a claim-only tx until the balance clears the threshold, working from a zero (fresh run) or
455
+ * drained (restart) balance. Each bridge mints a fixed amount well above the threshold, so this is a
456
+ * single bridge in practice. No-op when L1 bridging is not configured or the balance is already above
457
+ * the threshold.
358
458
  */
459
+ private async ensureFeeJuiceBalance(account: AztecAddress): Promise<void> {
460
+ if (!this.isL1BridgingConfigured()) {
461
+ return;
462
+ }
463
+
464
+ let balance = await getFeeJuiceBalance(account, this.aztecNode);
465
+ if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
466
+ this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
467
+ return;
468
+ }
469
+
470
+ this.log.info(`Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1`);
471
+
472
+ while (balance < FEE_JUICE_TOP_UP_THRESHOLD) {
473
+ // Persist the claim before consuming it: if the top-up tx fails or the bot crashes mid-loop, the
474
+ // next run reuses the pending claim instead of bridging again (and wasting the bridged funds).
475
+ const claim = await this.getOrCreateBridgeClaim(account);
476
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
477
+
478
+ await this.withNoMinTxsPerBlock(async () => {
479
+ const executionPayload = await paymentMethod.getExecutionPayload();
480
+ const { txHash } = await this.wallet.sendTx(executionPayload, { from: account, wait: NO_WAIT });
481
+ this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
482
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
483
+ });
484
+ await this.store.deleteBridgeClaim(account);
485
+ balance = await getFeeJuiceBalance(account, this.aztecNode);
486
+ this.log.info(`Fee juice balance after top-up: ${balance}`);
487
+ }
488
+ this.log.info(`Fee juice top-up complete for ${account.toString()}`);
489
+ }
490
+
359
491
  private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
360
492
  const isStandardToken = isStandardTokenContract(token);
361
493
  let privateBalance = 0n;
@@ -386,31 +518,35 @@ export class BotFactory {
386
518
  return;
387
519
  }
388
520
 
521
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
522
+ const additionalScopes = isStandardToken ? undefined : [token.address];
523
+ const mintBatch = new BatchCall(token.wallet, calls);
389
524
  await this.withNoMinTxsPerBlock(async () => {
390
- const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
525
+ const { txHash } = await mintBatch.send({
526
+ from: minter,
527
+ additionalScopes,
528
+ wait: NO_WAIT,
529
+ });
391
530
  this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
392
531
  return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
393
532
  });
394
533
  }
395
534
 
396
535
  /**
397
- * Gets or creates a bridge claim for the recipient.
398
- * Checks if a claim already exists in the store and reuses it if valid.
399
- * Only creates a new bridge if fee juice balance is below threshold.
536
+ * Returns a usable bridge claim for the recipient, reusing a persisted one when its L1→L2 message is
537
+ * still available (resuming a top-up that failed or crashed before the claim was consumed) and bridging
538
+ * a fresh claim otherwise. The caller deletes the claim from the store once it has been consumed.
400
539
  */
401
540
  private async getOrCreateBridgeClaim(recipient: AztecAddress): Promise<L2AmountClaim> {
402
- // Check if we have an existing claim in the store
403
541
  const existingClaim = await this.store.getBridgeClaim(recipient);
404
542
  if (existingClaim) {
405
543
  this.log.info(`Found existing bridge claim for ${recipient.toString()}, checking validity...`);
406
-
407
- // Check if the message is ready on L2
408
544
  try {
409
545
  const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
410
546
  await this.withNoMinTxsPerBlock(() =>
411
547
  waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
412
548
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
413
- forPublicConsumption: false,
549
+ chainTip: this.syncChainTip,
414
550
  }),
415
551
  );
416
552
  return existingClaim.claim;
@@ -422,7 +558,6 @@ export class BotFactory {
422
558
 
423
559
  const claim = await this.bridgeL1FeeJuice(recipient);
424
560
  await this.store.saveBridgeClaim(recipient, claim);
425
-
426
561
  return claim;
427
562
  }
428
563
 
@@ -449,7 +584,7 @@ export class BotFactory {
449
584
  await this.withNoMinTxsPerBlock(() =>
450
585
  waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
451
586
  timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
452
- forPublicConsumption: false,
587
+ chainTip: this.syncChainTip,
453
588
  }),
454
589
  );
455
590
 
@@ -458,19 +593,36 @@ export class BotFactory {
458
593
  return claim as L2AmountClaim;
459
594
  }
460
595
 
461
- private async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
596
+ protected async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
462
597
  if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
463
598
  this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
464
599
  return fn();
465
600
  }
466
- const { minTxsPerBlock } = await this.aztecNodeAdmin.getConfig();
467
- this.log.warn(`Setting sequencer minTxsPerBlock to 0 from ${minTxsPerBlock} to flush setup transactions`);
468
- await this.aztecNodeAdmin.setConfig({ minTxsPerBlock: 0 });
601
+ const aztecNodeAdmin = this.aztecNodeAdmin;
602
+ // Setup steps run concurrently, so this wrapper can be re-entered while another call is in flight.
603
+ // Reference-count the entrants: the first saves the current value and zeroes it, the last restores it.
604
+ // A naive save/zero/restore per call could interleave, with a late entrant reading the already-zeroed
605
+ // value and "restoring" 0 at the end.
606
+ if (this.noMinTxsPerBlockDepth++ === 0) {
607
+ this.savedMinTxsPerBlock = (async () => {
608
+ const { minTxsPerBlock } = await aztecNodeAdmin.getConfig();
609
+ this.log.warn(`Setting sequencer minTxsPerBlock to 0 from ${minTxsPerBlock} to flush setup transactions`);
610
+ await aztecNodeAdmin.setConfig({ minTxsPerBlock: 0 });
611
+ return { minTxsPerBlock };
612
+ })();
613
+ }
469
614
  try {
615
+ await this.savedMinTxsPerBlock;
470
616
  return await fn();
471
617
  } finally {
472
- this.log.warn(`Restoring sequencer minTxsPerBlock to ${minTxsPerBlock}`);
473
- await this.aztecNodeAdmin.setConfig({ minTxsPerBlock });
618
+ if (--this.noMinTxsPerBlockDepth === 0) {
619
+ // If saving/zeroing itself failed there is nothing to restore.
620
+ const saved = await this.savedMinTxsPerBlock!.catch(() => undefined);
621
+ if (saved) {
622
+ this.log.warn(`Restoring sequencer minTxsPerBlock to ${saved.minTxsPerBlock}`);
623
+ await aztecNodeAdmin.setConfig({ minTxsPerBlock: saved.minTxsPerBlock });
624
+ }
625
+ }
474
626
  }
475
627
  }
476
628
  }