@aztec-labs/bot 6.0.0-nightly.20260829

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 (58) hide show
  1. package/README.md +3 -0
  2. package/dest/amm_bot.d.ts +33 -0
  3. package/dest/amm_bot.d.ts.map +1 -0
  4. package/dest/amm_bot.js +108 -0
  5. package/dest/base_bot.d.ts +21 -0
  6. package/dest/base_bot.d.ts.map +1 -0
  7. package/dest/base_bot.js +69 -0
  8. package/dest/bot.d.ts +29 -0
  9. package/dest/bot.d.ts.map +1 -0
  10. package/dest/bot.js +60 -0
  11. package/dest/config.d.ts +175 -0
  12. package/dest/config.d.ts.map +1 -0
  13. package/dest/config.js +241 -0
  14. package/dest/cross_chain_bot.d.ts +56 -0
  15. package/dest/cross_chain_bot.d.ts.map +1 -0
  16. package/dest/cross_chain_bot.js +138 -0
  17. package/dest/factory.d.ts +74 -0
  18. package/dest/factory.d.ts.map +1 -0
  19. package/dest/factory.js +532 -0
  20. package/dest/index.d.ts +9 -0
  21. package/dest/index.d.ts.map +1 -0
  22. package/dest/index.js +8 -0
  23. package/dest/interface.d.ts +19 -0
  24. package/dest/interface.d.ts.map +1 -0
  25. package/dest/interface.js +38 -0
  26. package/dest/l1_to_l2_seeding.d.ts +8 -0
  27. package/dest/l1_to_l2_seeding.d.ts.map +1 -0
  28. package/dest/l1_to_l2_seeding.js +63 -0
  29. package/dest/rpc.d.ts +4 -0
  30. package/dest/rpc.d.ts.map +1 -0
  31. package/dest/rpc.js +8 -0
  32. package/dest/runner.d.ts +54 -0
  33. package/dest/runner.d.ts.map +1 -0
  34. package/dest/runner.js +566 -0
  35. package/dest/store/bot_store.d.ts +69 -0
  36. package/dest/store/bot_store.d.ts.map +1 -0
  37. package/dest/store/bot_store.js +138 -0
  38. package/dest/store/index.d.ts +2 -0
  39. package/dest/store/index.d.ts.map +1 -0
  40. package/dest/store/index.js +1 -0
  41. package/dest/utils.d.ts +19 -0
  42. package/dest/utils.d.ts.map +1 -0
  43. package/dest/utils.js +29 -0
  44. package/package.json +95 -0
  45. package/src/amm_bot.ts +132 -0
  46. package/src/base_bot.ts +75 -0
  47. package/src/bot.ts +102 -0
  48. package/src/config.ts +316 -0
  49. package/src/cross_chain_bot.ts +208 -0
  50. package/src/factory.ts +628 -0
  51. package/src/index.ts +14 -0
  52. package/src/interface.ts +31 -0
  53. package/src/l1_to_l2_seeding.ts +79 -0
  54. package/src/rpc.ts +8 -0
  55. package/src/runner.ts +220 -0
  56. package/src/store/bot_store.ts +196 -0
  57. package/src/store/index.ts +1 -0
  58. package/src/utils.ts +38 -0
package/src/factory.ts ADDED
@@ -0,0 +1,628 @@
1
+ import { getInitialTestAccountsData } from '@aztec-labs/accounts/testing';
2
+ import { deriveSecretKeyFromSigningKey } from '@aztec-labs/accounts/utils';
3
+ import { AztecAddress } from '@aztec-labs/aztec.js/addresses';
4
+ import {
5
+ BatchCall,
6
+ ContractBase,
7
+ ContractFunctionInteraction,
8
+ type DeployMethod,
9
+ type DeployOptions,
10
+ NO_WAIT,
11
+ } from '@aztec-labs/aztec.js/contracts';
12
+ import type { L2AmountClaim } from '@aztec-labs/aztec.js/ethereum';
13
+ import { L1FeeJuicePortalManager } from '@aztec-labs/aztec.js/ethereum';
14
+ import { FeeJuicePaymentMethodWithClaim } from '@aztec-labs/aztec.js/fee';
15
+ import { deriveKeys } from '@aztec-labs/aztec.js/keys';
16
+ import { createLogger } from '@aztec-labs/aztec.js/log';
17
+ import { waitForL1ToL2MessageReady } from '@aztec-labs/aztec.js/messaging';
18
+ import { waitForTx } from '@aztec-labs/aztec.js/node';
19
+ import { getFeeJuiceBalance } from '@aztec-labs/aztec.js/utils';
20
+ import { createEthereumChain } from '@aztec-labs/ethereum/chain';
21
+ import { createExtendedL1Client } from '@aztec-labs/ethereum/client';
22
+ import { RollupContract } from '@aztec-labs/ethereum/contracts';
23
+ import type { ExtendedViemWalletClient } from '@aztec-labs/ethereum/types';
24
+ import { Fr } from '@aztec-labs/foundation/curves/bn254';
25
+ import { GrumpkinScalar } from '@aztec-labs/foundation/curves/grumpkin';
26
+ import { EthAddress } from '@aztec-labs/foundation/eth-address';
27
+ import { AMMContract } from '@aztec-labs/noir-contracts.js/AMM';
28
+ import { PrivateTokenContract } from '@aztec-labs/noir-contracts.js/PrivateToken';
29
+ import { TokenContract } from '@aztec-labs/noir-contracts.js/Token';
30
+ import { TestContract } from '@aztec-labs/noir-test-contracts.js/Test';
31
+ import type { BlockTag } from '@aztec-labs/stdlib/block';
32
+ import type { ContractInstanceWithAddress } from '@aztec-labs/stdlib/contract';
33
+ import type { AztecNode, AztecNodeAdmin } from '@aztec-labs/stdlib/interfaces/client';
34
+ import { EmbeddedWallet } from '@aztec-labs/wallets/embedded';
35
+
36
+ import { type BotConfig, SupportedTokenContracts } from './config.js';
37
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
38
+ import type { BotStore } from './store/index.js';
39
+ import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
40
+
41
+ const MINT_BALANCE = 1e12;
42
+ const MIN_BALANCE = 1e3;
43
+ const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
44
+
45
+ export class BotFactory {
46
+ private log = createLogger('bot');
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
+
53
+ constructor(
54
+ private readonly config: BotConfig,
55
+ private readonly wallet: EmbeddedWallet,
56
+ private readonly store: BotStore,
57
+ private readonly aztecNode: AztecNode,
58
+ private readonly aztecNodeAdmin?: AztecNodeAdmin,
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
+ }
65
+
66
+ /**
67
+ * Initializes a new bot by setting up the sender account, registering the recipient,
68
+ * deploying the token contract, and minting tokens if necessary.
69
+ */
70
+ public async setup(): Promise<{
71
+ wallet: EmbeddedWallet;
72
+ defaultAccountAddress: AztecAddress;
73
+ token: TokenContract | PrivateTokenContract;
74
+ node: AztecNode;
75
+ recipient: AztecAddress;
76
+ }> {
77
+ const defaultAccountAddress = await this.setupAccount();
78
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random(), GrumpkinScalar.random()))
79
+ .address;
80
+ await this.ensureFeeJuiceBalance(defaultAccountAddress);
81
+ const token = await this.setupToken(defaultAccountAddress);
82
+ await this.mintTokens(token, defaultAccountAddress);
83
+ return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
84
+ }
85
+
86
+ public async setupAmm(): Promise<{
87
+ wallet: EmbeddedWallet;
88
+ defaultAccountAddress: AztecAddress;
89
+ amm: AMMContract;
90
+ token0: TokenContract;
91
+ token1: TokenContract;
92
+ node: AztecNode;
93
+ }> {
94
+ const defaultAccountAddress = await this.setupAccount();
95
+ await this.ensureFeeJuiceBalance(defaultAccountAddress);
96
+
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);
125
+ this.log.info(`AMM initialized and funded`);
126
+
127
+ return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
128
+ }
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
+
216
+ /**
217
+ * Checks if the sender account contract is initialized, and initializes it if necessary.
218
+ * @returns The sender wallet.
219
+ */
220
+ private async setupAccount() {
221
+ const privateKey = this.config.senderPrivateKey?.getValue();
222
+ if (privateKey) {
223
+ this.log.info(`Setting up account with provided private key`);
224
+ return await this.setupAccountWithPrivateKey(privateKey);
225
+ } else {
226
+ this.log.info(`Setting up test account`);
227
+ return await this.setupTestAccount();
228
+ }
229
+ }
230
+
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
+ */
238
+ private async setupTestAccount() {
239
+ const [initialAccountData] = await getInitialTestAccountsData();
240
+ const accountManager = await this.wallet.createSchnorrInitializerlessAccount(
241
+ initialAccountData.secret,
242
+ initialAccountData.salt,
243
+ initialAccountData.signingKey,
244
+ );
245
+ return accountManager.address;
246
+ }
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
+
256
+ /**
257
+ * Checks if the token contract is deployed and deploys it if necessary.
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.
262
+ */
263
+ private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
264
+ let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
265
+ const salt = this.config.tokenSalt;
266
+ const deployOpts: DeployOptions = { from: sender };
267
+ let token: TokenContract | PrivateTokenContract;
268
+ if (this.config.contract === SupportedTokenContracts.TokenContract) {
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);
272
+ } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
273
+ // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
274
+ const tokenSecretKey = Fr.random();
275
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
276
+ deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
277
+ salt,
278
+ universalDeploy: true,
279
+ publicKeys: tokenPublicKeys,
280
+ });
281
+ deployOpts.skipInstancePublication = true;
282
+ deployOpts.skipClassPublication = true;
283
+ deployOpts.skipInitialization = false;
284
+
285
+ // Register the contract with the secret key before deployment
286
+ const tokenInstance = await deploy.getInstance();
287
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
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];
291
+ } else {
292
+ throw new Error(`Unsupported token contract type: ${this.config.contract}`);
293
+ }
294
+
295
+ await this.registerOrDeployContract('token', deploy, deployOpts);
296
+ return token;
297
+ }
298
+
299
+ /**
300
+ * Checks if the token contract is deployed and deploys it if necessary.
301
+ * @param wallet - Wallet to deploy the token contract from.
302
+ * @returns The TokenContract instance.
303
+ */
304
+ private async setupTokenContract(
305
+ deployer: AztecAddress,
306
+ salt: Fr,
307
+ name: string,
308
+ ticker: string,
309
+ decimals = 18,
310
+ ): Promise<TokenContract> {
311
+ const deployOpts: DeployOptions = { from: deployer };
312
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
313
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
314
+ return TokenContract.at(instance.address, this.wallet);
315
+ }
316
+
317
+ private async deployAmmContract(deployer: AztecAddress, deploy: DeployMethod<AMMContract>): Promise<AMMContract> {
318
+ const instance = await this.registerOrDeployContract('AMM', deploy, { from: deployer });
319
+ const amm = AMMContract.at(instance.address, this.wallet);
320
+ this.log.info(`AMM deployed at ${amm.address}`);
321
+ return amm;
322
+ }
323
+
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(
348
+ defaultAccountAddress: AztecAddress,
349
+ liquidityProvider: AztecAddress,
350
+ amm: AMMContract,
351
+ token0: TokenContract,
352
+ token1: TokenContract,
353
+ lpToken: TokenContract,
354
+ ): Promise<void> {
355
+ const authwitNonce = Fr.random();
356
+
357
+ // keep some tokens for swapping
358
+ const amount0Max = MINT_BALANCE / 2;
359
+ const amount0Min = MINT_BALANCE / 4;
360
+ const amount1Max = MINT_BALANCE / 2;
361
+ const amount1Min = MINT_BALANCE / 4;
362
+
363
+ // Add authwitnesses for the transfers in AMM::add_liquidity function
364
+ const token0Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
365
+ caller: amm.address,
366
+ call: await token0.methods
367
+ .transfer_to_public_and_prepare_private_balance_increase(
368
+ liquidityProvider,
369
+ amm.address,
370
+ amount0Max,
371
+ authwitNonce,
372
+ )
373
+ .getFunctionCall(),
374
+ });
375
+ const token1Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
376
+ caller: amm.address,
377
+ call: await token1.methods
378
+ .transfer_to_public_and_prepare_private_balance_increase(
379
+ liquidityProvider,
380
+ amm.address,
381
+ amount1Max,
382
+ authwitNonce,
383
+ )
384
+ .getFunctionCall(),
385
+ });
386
+
387
+ const { receipt } = await amm.methods
388
+ .add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
389
+ .send({
390
+ from: liquidityProvider,
391
+ authWitnesses: [token0Authwit, token1Authwit],
392
+ wait: { timeout: this.config.txMinedWaitSeconds },
393
+ });
394
+
395
+ this.log.info(`Sent tx to add liquidity to the AMM: ${receipt.txHash.toString()}`);
396
+ this.log.info(`Liquidity added`);
397
+
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
+ ]);
412
+ this.log.info(
413
+ `Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`,
414
+ );
415
+ }
416
+
417
+ private async registerOrDeployContract<T extends ContractBase>(
418
+ name: string,
419
+ deploy: DeployMethod<T>,
420
+ deployOpts: DeployOptions,
421
+ ): Promise<ContractInstanceWithAddress> {
422
+ const instance = await deploy.getInstance();
423
+ const address = instance.address;
424
+ const metadata = await this.wallet.getContractMetadata(address);
425
+ if (metadata.isContractPublished) {
426
+ this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
427
+ await deploy.register();
428
+ return instance;
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
+
441
+ return instance;
442
+ }
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
+
450
+ /**
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.
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
+
491
+ private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
492
+ const isStandardToken = isStandardTokenContract(token);
493
+ let privateBalance = 0n;
494
+ let publicBalance = 0n;
495
+
496
+ if (isStandardToken) {
497
+ ({ privateBalance, publicBalance } = await getBalances(token, minter));
498
+ } else {
499
+ privateBalance = await getPrivateBalance(token, minter);
500
+ }
501
+
502
+ const calls: ContractFunctionInteraction[] = [];
503
+ if (privateBalance < MIN_BALANCE) {
504
+ this.log.info(`Minting private tokens for ${minter.toString()}`);
505
+
506
+ calls.push(
507
+ isStandardToken
508
+ ? token.methods.mint_to_private(minter, MINT_BALANCE)
509
+ : token.methods.mint(MINT_BALANCE, minter),
510
+ );
511
+ }
512
+ if (isStandardToken && publicBalance < MIN_BALANCE) {
513
+ this.log.info(`Minting public tokens for ${minter.toString()}`);
514
+ calls.push(token.methods.mint_to_public(minter, MINT_BALANCE));
515
+ }
516
+ if (calls.length === 0) {
517
+ this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
518
+ return;
519
+ }
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);
524
+ await this.withNoMinTxsPerBlock(async () => {
525
+ const { txHash } = await mintBatch.send({
526
+ from: minter,
527
+ additionalScopes,
528
+ wait: NO_WAIT,
529
+ });
530
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
531
+ return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
532
+ });
533
+ }
534
+
535
+ /**
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.
539
+ */
540
+ private async getOrCreateBridgeClaim(recipient: AztecAddress): Promise<L2AmountClaim> {
541
+ const existingClaim = await this.store.getBridgeClaim(recipient);
542
+ if (existingClaim) {
543
+ this.log.info(`Found existing bridge claim for ${recipient.toString()}, checking validity...`);
544
+ try {
545
+ const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
546
+ await this.withNoMinTxsPerBlock(() =>
547
+ waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
548
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
549
+ chainTip: this.syncChainTip,
550
+ }),
551
+ );
552
+ return existingClaim.claim;
553
+ } catch (err) {
554
+ this.log.warn(`Failed to verify existing claim, creating new one: ${err}`);
555
+ await this.store.deleteBridgeClaim(recipient);
556
+ }
557
+ }
558
+
559
+ const claim = await this.bridgeL1FeeJuice(recipient);
560
+ await this.store.saveBridgeClaim(recipient, claim);
561
+ return claim;
562
+ }
563
+
564
+ private async bridgeL1FeeJuice(recipient: AztecAddress): Promise<L2AmountClaim> {
565
+ const l1RpcUrls = this.config.l1RpcUrls;
566
+ if (!l1RpcUrls?.length) {
567
+ throw new Error('L1 Rpc url is required to bridge the fee juice to fund the deployment of the account.');
568
+ }
569
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
570
+ if (!mnemonicOrPrivateKey) {
571
+ throw new Error(
572
+ 'Either a mnemonic or private key of an L1 account is required to bridge the fee juice to fund the deployment of the account.',
573
+ );
574
+ }
575
+
576
+ const { l1ChainId } = await this.aztecNode.getNodeInfo();
577
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
578
+ const extendedClient = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
579
+
580
+ const portal = await L1FeeJuicePortalManager.new(this.aztecNode, extendedClient, this.log);
581
+ const mintAmount = await portal.getTokenManager().getMintAmount();
582
+ const claim = await portal.bridgeTokensPublic(recipient, mintAmount, true /* mint */);
583
+
584
+ await this.withNoMinTxsPerBlock(() =>
585
+ waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
586
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
587
+ chainTip: this.syncChainTip,
588
+ }),
589
+ );
590
+
591
+ this.log.info(`Created a claim for ${mintAmount} L1 fee juice to ${recipient}.`, claim);
592
+
593
+ return claim as L2AmountClaim;
594
+ }
595
+
596
+ protected async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
597
+ if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
598
+ this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
599
+ return fn();
600
+ }
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
+ }
614
+ try {
615
+ await this.savedMinTxsPerBlock;
616
+ return await fn();
617
+ } finally {
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
+ }
626
+ }
627
+ }
628
+ }
package/src/index.ts ADDED
@@ -0,0 +1,14 @@
1
+ export { Bot } from './bot.js';
2
+ export { AmmBot } from './amm_bot.js';
3
+ export { CrossChainBot } from './cross_chain_bot.js';
4
+ export { BotRunner } from './runner.js';
5
+ export { BotStore } from './store/bot_store.js';
6
+ export {
7
+ type BotConfig,
8
+ getBotConfigFromEnv,
9
+ getBotDefaultConfig,
10
+ botConfigMappings,
11
+ SupportedTokenContracts,
12
+ } from './config.js';
13
+ export { getBotRunnerApiHandler } from './rpc.js';
14
+ export * from './interface.js';
@@ -0,0 +1,31 @@
1
+ import { AztecAddress } from '@aztec-labs/aztec.js/addresses';
2
+ import type { ApiSchemaFor } from '@aztec-labs/stdlib/schemas';
3
+ import { z } from 'zod';
4
+
5
+ import { type BotConfig, BotConfigSchema } from './config.js';
6
+
7
+ export const BotInfoSchema = z.object({
8
+ botAddress: AztecAddress.schema,
9
+ });
10
+
11
+ export type BotInfo = z.infer<typeof BotInfoSchema>;
12
+
13
+ export interface BotRunnerApi {
14
+ start(): Promise<void>;
15
+ stop(): Promise<void>;
16
+ run(): Promise<void>;
17
+ setup(): Promise<void>;
18
+ getConfig(): Promise<BotConfig>;
19
+ getInfo(): Promise<BotInfo>;
20
+ update(config: BotConfig): Promise<void>;
21
+ }
22
+
23
+ export const BotRunnerApiSchema: ApiSchemaFor<BotRunnerApi> = {
24
+ start: z.function({ input: z.tuple([]), output: z.void() }),
25
+ stop: z.function({ input: z.tuple([]), output: z.void() }),
26
+ run: z.function({ input: z.tuple([]), output: z.void() }),
27
+ setup: z.function({ input: z.tuple([]), output: z.void() }),
28
+ getInfo: z.function({ input: z.tuple([]), output: BotInfoSchema }),
29
+ getConfig: z.function({ input: z.tuple([]), output: BotConfigSchema }),
30
+ update: z.function({ input: z.tuple([BotConfigSchema]), output: z.void() }),
31
+ };