@aztec/bot 0.0.1-commit.24de95ac → 0.0.1-commit.2606882

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/dest/factory.js CHANGED
@@ -1,23 +1,33 @@
1
- import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
2
1
  import { getInitialTestAccountsData } from '@aztec/accounts/testing';
3
- import { AztecAddress } from '@aztec/aztec.js/addresses';
4
- import { BatchCall } from '@aztec/aztec.js/contracts';
2
+ import { NO_FROM } from '@aztec/aztec.js/account';
3
+ import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
5
4
  import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
6
5
  import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
6
+ import { deriveKeys } from '@aztec/aztec.js/keys';
7
7
  import { createLogger } from '@aztec/aztec.js/log';
8
8
  import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
9
- import { createEthereumChain, createExtendedL1Client } from '@aztec/ethereum';
10
- import { Fr } from '@aztec/foundation/fields';
9
+ import { waitForTx } from '@aztec/aztec.js/node';
10
+ import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
11
+ import { ContractInitializationStatus } from '@aztec/aztec.js/wallet';
12
+ import { createEthereumChain } from '@aztec/ethereum/chain';
13
+ import { createExtendedL1Client } from '@aztec/ethereum/client';
14
+ import { RollupContract } from '@aztec/ethereum/contracts';
15
+ import { Fr } from '@aztec/foundation/curves/bn254';
16
+ import { EthAddress } from '@aztec/foundation/eth-address';
11
17
  import { Timer } from '@aztec/foundation/timer';
12
18
  import { AMMContract } from '@aztec/noir-contracts.js/AMM';
13
19
  import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
14
20
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
15
- import { GasSettings } from '@aztec/stdlib/gas';
21
+ import { TestContract } from '@aztec/noir-test-contracts.js/Test';
22
+ import { GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
16
23
  import { deriveSigningKey } from '@aztec/stdlib/keys';
17
24
  import { SupportedTokenContracts } from './config.js';
25
+ import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
18
26
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
19
27
  const MINT_BALANCE = 1e12;
20
28
  const MIN_BALANCE = 1e3;
29
+ const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
30
+ const FEE_JUICE_TOP_UP_TARGET = 10_000n * 10n ** 18n;
21
31
  export class BotFactory {
22
32
  config;
23
33
  wallet;
@@ -32,14 +42,18 @@ export class BotFactory {
32
42
  this.aztecNode = aztecNode;
33
43
  this.aztecNodeAdmin = aztecNodeAdmin;
34
44
  this.log = createLogger('bot');
45
+ // Set fee padding on the wallet so that all transactions during setup
46
+ // (token deploy, minting, etc.) use the configured padding, not the default.
47
+ this.wallet.setMinFeePadding(config.minFeePadding);
35
48
  }
36
49
  /**
37
50
  * Initializes a new bot by setting up the sender account, registering the recipient,
38
51
  * deploying the token contract, and minting tokens if necessary.
39
52
  */ async setup() {
40
- const recipient = (await this.wallet.createAccount()).address;
41
53
  const defaultAccountAddress = await this.setupAccount();
42
- const token = await this.setupToken(defaultAccountAddress);
54
+ const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
55
+ const token = await this.setupTokenWithOptionalEarlyRefuel(defaultAccountAddress);
56
+ await this.ensureFeeJuiceBalance(defaultAccountAddress, token);
43
57
  await this.mintTokens(token, defaultAccountAddress);
44
58
  return {
45
59
  wallet: this.wallet,
@@ -51,7 +65,8 @@ export class BotFactory {
51
65
  }
52
66
  async setupAmm() {
53
67
  const defaultAccountAddress = await this.setupAccount();
54
- const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
68
+ const token0 = await this.setupTokenContractWithOptionalEarlyRefuel(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
69
+ await this.ensureFeeJuiceBalance(defaultAccountAddress, token0);
55
70
  const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
56
71
  const liquidityToken = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotLPToken', 'BOTLP');
57
72
  const amm = await this.setupAmmContract(defaultAccountAddress, this.config.tokenSalt, token0, token1, liquidityToken);
@@ -67,6 +82,66 @@ export class BotFactory {
67
82
  };
68
83
  }
69
84
  /**
85
+ * Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
86
+ * seeding initial L1→L2 messages, and waiting for the first to be ready.
87
+ */ async setupCrossChain() {
88
+ const defaultAccountAddress = await this.setupAccount();
89
+ // Create L1 client (same pattern as bridgeL1FeeJuice)
90
+ const l1RpcUrls = this.config.l1RpcUrls;
91
+ if (!l1RpcUrls?.length) {
92
+ throw new Error('L1 RPC URLs required for cross-chain bot');
93
+ }
94
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
95
+ if (!mnemonicOrPrivateKey) {
96
+ throw new Error('L1 mnemonic or private key required for cross-chain bot');
97
+ }
98
+ const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
99
+ const chain = createEthereumChain(l1RpcUrls, l1ChainId);
100
+ const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
101
+ // Fetch Rollup version (needed for Inbox L2Actor struct)
102
+ const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
103
+ const rollupVersion = await rollupContract.getVersion();
104
+ // Deploy TestContract
105
+ const contract = await this.setupTestContract(defaultAccountAddress);
106
+ // Recover any pending messages from store (clean up stale ones first)
107
+ await this.store.cleanupOldPendingMessages();
108
+ const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
109
+ // Seed initial L1→L2 messages if pipeline is empty
110
+ const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
111
+ for(let i = 0; i < seedCount; i++){
112
+ await seedL1ToL2Message(l1Client, EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()), contract.address, rollupVersion, this.store, this.log);
113
+ }
114
+ // Block until at least one message is ready
115
+ const allMessages = await this.store.getUnconsumedL1ToL2Messages();
116
+ if (allMessages.length > 0) {
117
+ this.log.info(`Waiting for first L1→L2 message to be ready...`);
118
+ const firstMsg = allMessages[0];
119
+ await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
120
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
121
+ });
122
+ this.log.info(`First L1→L2 message is ready`);
123
+ }
124
+ return {
125
+ wallet: this.wallet,
126
+ defaultAccountAddress,
127
+ contract,
128
+ node: this.aztecNode,
129
+ l1Client,
130
+ rollupVersion
131
+ };
132
+ }
133
+ async setupTestContract(deployer) {
134
+ const deployOpts = {
135
+ from: deployer
136
+ };
137
+ const deploy = TestContract.deploy(this.wallet, {
138
+ salt: this.config.tokenSalt,
139
+ universalDeploy: true
140
+ });
141
+ const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
142
+ return TestContract.at(instance.address, this.wallet);
143
+ }
144
+ /**
70
145
  * Checks if the sender account contract is initialized, and initializes it if necessary.
71
146
  * @returns The sender wallet.
72
147
  */ async setupAccount() {
@@ -82,14 +157,9 @@ export class BotFactory {
82
157
  async setupAccountWithPrivateKey(secret) {
83
158
  const salt = this.config.senderSalt ?? Fr.ONE;
84
159
  const signingKey = deriveSigningKey(secret);
85
- const accountData = {
86
- secret,
87
- salt,
88
- contract: new SchnorrAccountContract(signingKey)
89
- };
90
- const accountManager = await this.wallet.createAccount(accountData);
91
- const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
92
- if (isInit) {
160
+ const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
161
+ const metadata = await this.wallet.getContractMetadata(accountManager.address);
162
+ if (metadata.initializationStatus === ContractInitializationStatus.INITIALIZED) {
93
163
  this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
94
164
  const timer = new Timer();
95
165
  const address = accountManager.address;
@@ -102,22 +172,19 @@ export class BotFactory {
102
172
  const claim = await this.getOrCreateBridgeClaim(address);
103
173
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
104
174
  const deployMethod = await accountManager.getDeployMethod();
105
- const maxFeesPerGas = (await this.aztecNode.getCurrentBaseFees()).mul(1 + this.config.baseFeePadding);
106
- const gasSettings = GasSettings.default({
107
- maxFeesPerGas
108
- });
109
- const sentTx = deployMethod.send({
110
- from: AztecAddress.ZERO,
111
- fee: {
112
- gasSettings,
113
- paymentMethod
114
- }
115
- });
116
- const txHash = await sentTx.getTxHash();
117
- this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
118
- await this.withNoMinTxsPerBlock(()=>sentTx.wait({
175
+ await this.withNoMinTxsPerBlock(async ()=>{
176
+ const { txHash } = await deployMethod.send({
177
+ from: NO_FROM,
178
+ fee: {
179
+ paymentMethod
180
+ },
181
+ wait: NO_WAIT
182
+ });
183
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
184
+ return waitForTx(this.aztecNode, txHash, {
119
185
  timeout: this.config.txMinedWaitSeconds
120
- }));
186
+ });
187
+ });
121
188
  this.log.info(`Account deployed at ${address}`);
122
189
  // Clean up the consumed bridge claim
123
190
  await this.store.deleteBridgeClaim(address);
@@ -126,78 +193,142 @@ export class BotFactory {
126
193
  }
127
194
  async setupTestAccount() {
128
195
  const [initialAccountData] = await getInitialTestAccountsData();
129
- const accountData = {
130
- secret: initialAccountData.secret,
131
- salt: initialAccountData.salt,
132
- contract: new SchnorrAccountContract(initialAccountData.signingKey)
133
- };
134
- const accountManager = await this.wallet.createAccount(accountData);
196
+ const accountManager = await this.wallet.createSchnorrAccount(initialAccountData.secret, initialAccountData.salt, initialAccountData.signingKey);
135
197
  return accountManager.address;
136
198
  }
137
199
  /**
200
+ * Setup token and refuel first: if the token already exists (restart scenario),
201
+ * run ensureFeeJuiceBalance before any step that might need fee juice. When deploying,
202
+ * use a bridge claim if balance is below threshold.
203
+ */ async setupTokenWithOptionalEarlyRefuel(sender) {
204
+ const token = await this.getTokenInstance(sender);
205
+ const address = token.address;
206
+ const metadata = await this.wallet.getContractMetadata(address);
207
+ if (metadata.isContractPublished) {
208
+ this.log.info(`Token at ${address.toString()} already deployed, refueling before setup`);
209
+ await this.ensureFeeJuiceBalance(sender, token);
210
+ }
211
+ return this.setupToken(sender);
212
+ }
213
+ /**
214
+ * Setup token0 for AMM with refuel-first behaviour when token already exists.
215
+ */ async setupTokenContractWithOptionalEarlyRefuel(deployer, salt, name, ticker, decimals = 18) {
216
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, {
217
+ salt,
218
+ universalDeploy: true
219
+ });
220
+ const instance = await deploy.getInstance();
221
+ const metadata = await this.wallet.getContractMetadata(instance.address);
222
+ if (metadata.isContractPublished) {
223
+ this.log.info(`Token ${name} at ${instance.address.toString()} already deployed, refueling before setup`);
224
+ const token = TokenContract.at(instance.address, this.wallet);
225
+ await this.ensureFeeJuiceBalance(deployer, token);
226
+ }
227
+ return this.setupTokenContract(deployer, salt, name, ticker, decimals);
228
+ }
229
+ async getTokenInstance(sender) {
230
+ const salt = this.config.tokenSalt;
231
+ if (this.config.contract === SupportedTokenContracts.TokenContract) {
232
+ const deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, {
233
+ salt,
234
+ universalDeploy: true
235
+ });
236
+ const instance = await deploy.getInstance();
237
+ return TokenContract.at(instance.address, this.wallet);
238
+ }
239
+ if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
240
+ const tokenSecretKey = Fr.random();
241
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
242
+ const deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
243
+ salt,
244
+ universalDeploy: true,
245
+ publicKeys: tokenPublicKeys
246
+ });
247
+ const instance = await deploy.getInstance();
248
+ return PrivateTokenContract.at(instance.address, this.wallet);
249
+ }
250
+ throw new Error(`Unsupported token contract type: ${this.config.contract}`);
251
+ }
252
+ /**
138
253
  * Checks if the token contract is deployed and deploys it if necessary.
139
- * @param wallet - Wallet to deploy the token contract from.
140
- * @returns The TokenContract instance.
254
+ * Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
255
+ * @param sender - Aztec address to deploy the token contract from.
256
+ * @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
257
+ * @returns The TokenContract or PrivateTokenContract instance.
141
258
  */ async setupToken(sender) {
142
259
  let deploy;
260
+ const salt = this.config.tokenSalt;
143
261
  const deployOpts = {
144
- from: sender,
145
- contractAddressSalt: this.config.tokenSalt,
146
- universalDeploy: true
262
+ from: sender
147
263
  };
264
+ let token;
148
265
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
149
- deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
266
+ deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, {
267
+ salt,
268
+ universalDeploy: true
269
+ });
270
+ const instance = await deploy.getInstance();
271
+ token = TokenContract.at(instance.address, this.wallet);
150
272
  } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
151
- deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender);
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
+ });
152
281
  deployOpts.skipInstancePublication = true;
153
282
  deployOpts.skipClassPublication = true;
154
283
  deployOpts.skipInitialization = false;
284
+ // Register the contract with the secret key before deployment
285
+ const tokenInstance = await deploy.getInstance();
286
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
287
+ await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
288
+ // The contract constructor initializes private storage vars that need the contract's own nullifier key.
289
+ deployOpts.additionalScopes = [
290
+ tokenInstance.address
291
+ ];
155
292
  } else {
156
293
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
157
294
  }
158
- const address = (await deploy.getInstance(deployOpts)).address;
159
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
160
- this.log.info(`Token at ${address.toString()} already deployed`);
161
- return deploy.register();
162
- } else {
163
- this.log.info(`Deploying token contract at ${address.toString()}`);
164
- const sentTx = deploy.send(deployOpts);
165
- const txHash = await sentTx.getTxHash();
166
- this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
167
- return this.withNoMinTxsPerBlock(()=>sentTx.deployed({
168
- timeout: this.config.txMinedWaitSeconds
169
- }));
170
- }
295
+ await this.registerOrDeployContract('token', deploy, deployOpts);
296
+ return token;
171
297
  }
172
298
  /**
173
299
  * Checks if the token contract is deployed and deploys it if necessary.
174
300
  * @param wallet - Wallet to deploy the token contract from.
175
301
  * @returns The TokenContract instance.
176
- */ setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals = 18) {
302
+ */ async setupTokenContract(deployer, salt, name, ticker, decimals = 18) {
177
303
  const deployOpts = {
178
- from: deployer,
179
- contractAddressSalt,
180
- universalDeploy: true
304
+ from: deployer
181
305
  };
182
- const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
183
- return this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
306
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, {
307
+ salt,
308
+ universalDeploy: true
309
+ });
310
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
311
+ return TokenContract.at(instance.address, this.wallet);
184
312
  }
185
- async setupAmmContract(deployer, contractAddressSalt, token0, token1, lpToken) {
313
+ async setupAmmContract(deployer, salt, token0, token1, lpToken) {
186
314
  const deployOpts = {
187
- from: deployer,
188
- contractAddressSalt,
189
- universalDeploy: true
190
- };
191
- const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
192
- const amm = await this.registerOrDeployContract('AMM', deploy, deployOpts);
193
- this.log.info(`AMM deployed at ${amm.address}`);
194
- const minterTx = lpToken.methods.set_minter(amm.address, true).send({
195
315
  from: deployer
316
+ };
317
+ const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address, {
318
+ salt,
319
+ universalDeploy: true
196
320
  });
197
- this.log.info(`Set LP token minter to AMM txHash=${(await minterTx.getTxHash()).toString()}`);
198
- await minterTx.wait({
199
- timeout: this.config.txMinedWaitSeconds
321
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
322
+ const amm = AMMContract.at(instance.address, this.wallet);
323
+ this.log.info(`AMM deployed at ${amm.address}`);
324
+ const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
325
+ const { receipt: minterReceipt } = await setMinterInteraction.send({
326
+ from: deployer,
327
+ wait: {
328
+ timeout: this.config.txMinedWaitSeconds
329
+ }
200
330
  });
331
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
201
332
  this.log.info(`Liquidity token initialized`);
202
333
  return amm;
203
334
  }
@@ -205,13 +336,13 @@ export class BotFactory {
205
336
  const getPrivateBalances = ()=>Promise.all([
206
337
  token0.methods.balance_of_private(liquidityProvider).simulate({
207
338
  from: liquidityProvider
208
- }),
339
+ }).then((r)=>r.result),
209
340
  token1.methods.balance_of_private(liquidityProvider).simulate({
210
341
  from: liquidityProvider
211
- }),
342
+ }).then((r)=>r.result),
212
343
  lpToken.methods.balance_of_private(liquidityProvider).simulate({
213
344
  from: liquidityProvider
214
- })
345
+ }).then((r)=>r.result)
215
346
  ]);
216
347
  const authwitNonce = Fr.random();
217
348
  // keep some tokens for swapping
@@ -230,50 +361,162 @@ export class BotFactory {
230
361
  caller: amm.address,
231
362
  call: await token1.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount1Max, authwitNonce).getFunctionCall()
232
363
  });
233
- const mintTx = new BatchCall(this.wallet, [
364
+ const mintBatch = new BatchCall(this.wallet, [
234
365
  token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
235
366
  token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE)
236
- ]).send({
237
- from: liquidityProvider
238
- });
239
- this.log.info(`Sent mint tx: ${(await mintTx.getTxHash()).toString()}`);
240
- await mintTx.wait({
241
- timeout: this.config.txMinedWaitSeconds
367
+ ]);
368
+ const { receipt: mintReceipt } = await mintBatch.send({
369
+ from: liquidityProvider,
370
+ wait: {
371
+ timeout: this.config.txMinedWaitSeconds
372
+ }
242
373
  });
243
- const addLiquidityTx = amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce).send({
374
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
375
+ const addLiquidityInteraction = amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce);
376
+ const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
244
377
  from: liquidityProvider,
245
378
  authWitnesses: [
246
379
  token0Authwit,
247
380
  token1Authwit
248
- ]
249
- });
250
- this.log.info(`Sent tx to add liquidity to the AMM: ${(await addLiquidityTx.getTxHash()).toString()}`);
251
- await addLiquidityTx.wait({
252
- timeout: this.config.txMinedWaitSeconds
381
+ ],
382
+ wait: {
383
+ timeout: this.config.txMinedWaitSeconds
384
+ }
253
385
  });
386
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
254
387
  this.log.info(`Liquidity added`);
255
388
  const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
256
389
  this.log.info(`Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${newT0Bal}, token1=${newT1Bal}, lp=${newLPBal}`);
257
390
  }
258
391
  async registerOrDeployContract(name, deploy, deployOpts) {
259
- const address = (await deploy.getInstance(deployOpts)).address;
260
- if ((await this.wallet.getContractMetadata(address)).isContractPublished) {
392
+ const instance = await deploy.getInstance();
393
+ const address = instance.address;
394
+ const metadata = await this.wallet.getContractMetadata(address);
395
+ if (metadata.isContractPublished) {
261
396
  this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
262
- return deploy.register();
397
+ await deploy.register();
263
398
  } else {
264
- this.log.info(`Deploying contract ${name} at ${address.toString()}`);
265
- const sentTx = deploy.send(deployOpts);
266
- const txHash = await sentTx.getTxHash();
267
- this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
268
- return this.withNoMinTxsPerBlock(()=>sentTx.deployed({
269
- timeout: this.config.txMinedWaitSeconds
270
- }));
399
+ const sender = deployOpts.from === NO_FROM ? undefined : deployOpts.from;
400
+ const balance = sender ? await getFeeJuiceBalance(sender, this.aztecNode) : 0n;
401
+ const useClaim = sender && balance < FEE_JUICE_TOP_UP_THRESHOLD && this.config.feePaymentMethod === 'fee_juice' && !!this.config.l1RpcUrls?.length;
402
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
403
+ if (useClaim && mnemonicOrPrivateKey) {
404
+ const claim = await this.getOrCreateBridgeClaim(sender);
405
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(sender, claim);
406
+ const { estimatedGas } = await deploy.simulate({
407
+ ...deployOpts,
408
+ fee: {
409
+ estimateGas: true,
410
+ paymentMethod
411
+ }
412
+ });
413
+ const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
414
+ const gasSettings = GasSettings.from({
415
+ ...estimatedGas,
416
+ maxFeesPerGas,
417
+ maxPriorityFeesPerGas: GasFees.empty()
418
+ });
419
+ await this.withNoMinTxsPerBlock(async ()=>{
420
+ const { txHash } = await deploy.send({
421
+ ...deployOpts,
422
+ fee: {
423
+ gasSettings,
424
+ paymentMethod
425
+ },
426
+ wait: NO_WAIT
427
+ });
428
+ this.log.info(`Sent contract ${name} deploy tx ${txHash.toString()} (using bridge claim, balance was ${balance})`);
429
+ return waitForTx(this.aztecNode, txHash, {
430
+ timeout: this.config.txMinedWaitSeconds
431
+ });
432
+ });
433
+ await this.store.deleteBridgeClaim(sender);
434
+ } else {
435
+ const { estimatedGas } = await deploy.simulate({
436
+ ...deployOpts,
437
+ fee: {
438
+ estimateGas: true
439
+ }
440
+ });
441
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`, {
442
+ estimatedGas
443
+ });
444
+ await this.withNoMinTxsPerBlock(async ()=>{
445
+ const { txHash } = await deploy.send({
446
+ ...deployOpts,
447
+ fee: {
448
+ gasSettings: estimatedGas
449
+ },
450
+ wait: NO_WAIT
451
+ });
452
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
453
+ return waitForTx(this.aztecNode, txHash, {
454
+ timeout: this.config.txMinedWaitSeconds
455
+ });
456
+ });
457
+ }
271
458
  }
459
+ return instance;
272
460
  }
273
461
  /**
274
462
  * Mints private and public tokens for the sender if their balance is below the minimum.
275
463
  * @param token - Token contract.
276
- */ async mintTokens(token, minter) {
464
+ */ /**
465
+ * Ensures the account has sufficient fee juice by bridging from L1 if balance is below threshold.
466
+ * Bridges repeatedly until balance reaches the target (10k FJ).
467
+ * Used on startup/restart to top up when the account has run out after previous runs.
468
+ */ async ensureFeeJuiceBalance(account, token) {
469
+ const { feePaymentMethod, l1RpcUrls } = this.config;
470
+ if (feePaymentMethod !== 'fee_juice' || !l1RpcUrls?.length) {
471
+ return;
472
+ }
473
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
474
+ if (!mnemonicOrPrivateKey) {
475
+ return;
476
+ }
477
+ let balance = await getFeeJuiceBalance(account, this.aztecNode);
478
+ if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
479
+ this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
480
+ return;
481
+ }
482
+ this.log.info(`Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1 until ${FEE_JUICE_TOP_UP_TARGET}`);
483
+ const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
484
+ const minimalInteraction = isStandardTokenContract(token) ? token.methods.transfer_in_public(account, account, 0n, 0) : token.methods.transfer(0n, account, account);
485
+ while(balance < FEE_JUICE_TOP_UP_TARGET){
486
+ const claim = await this.bridgeL1FeeJuice(account);
487
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
488
+ const { estimatedGas } = await minimalInteraction.simulate({
489
+ from: account,
490
+ fee: {
491
+ estimateGas: true,
492
+ paymentMethod
493
+ }
494
+ });
495
+ const gasSettings = GasSettings.from({
496
+ ...estimatedGas,
497
+ maxFeesPerGas,
498
+ maxPriorityFeesPerGas: GasFees.empty()
499
+ });
500
+ await this.withNoMinTxsPerBlock(async ()=>{
501
+ const { txHash } = await minimalInteraction.send({
502
+ from: account,
503
+ fee: {
504
+ gasSettings,
505
+ paymentMethod
506
+ },
507
+ wait: NO_WAIT
508
+ });
509
+ this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
510
+ return waitForTx(this.aztecNode, txHash, {
511
+ timeout: this.config.txMinedWaitSeconds
512
+ });
513
+ });
514
+ balance = await getFeeJuiceBalance(account, this.aztecNode);
515
+ this.log.info(`Fee juice balance after top-up: ${balance}`);
516
+ }
517
+ this.log.info(`Fee juice top-up complete for ${account.toString()}`);
518
+ }
519
+ async mintTokens(token, minter) {
277
520
  const isStandardToken = isStandardTokenContract(token);
278
521
  let privateBalance = 0n;
279
522
  let publicBalance = 0n;
@@ -295,14 +538,22 @@ export class BotFactory {
295
538
  this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
296
539
  return;
297
540
  }
298
- const sentTx = new BatchCall(token.wallet, calls).send({
299
- from: minter
300
- });
301
- const txHash = await sentTx.getTxHash();
302
- this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
303
- await this.withNoMinTxsPerBlock(()=>sentTx.wait({
541
+ // PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
542
+ const additionalScopes = isStandardToken ? undefined : [
543
+ token.address
544
+ ];
545
+ const mintBatch = new BatchCall(token.wallet, calls);
546
+ await this.withNoMinTxsPerBlock(async ()=>{
547
+ const { txHash } = await mintBatch.send({
548
+ from: minter,
549
+ additionalScopes,
550
+ wait: NO_WAIT
551
+ });
552
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
553
+ return waitForTx(this.aztecNode, txHash, {
304
554
  timeout: this.config.txMinedWaitSeconds
305
- }));
555
+ });
556
+ });
306
557
  }
307
558
  /**
308
559
  * Gets or creates a bridge claim for the recipient.
@@ -317,8 +568,7 @@ export class BotFactory {
317
568
  try {
318
569
  const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
319
570
  await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
320
- timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
321
- forPublicConsumption: false
571
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
322
572
  }));
323
573
  return existingClaim.claim;
324
574
  } catch (err) {
@@ -346,12 +596,22 @@ export class BotFactory {
346
596
  const mintAmount = await portal.getTokenManager().getMintAmount();
347
597
  const claim = await portal.bridgeTokensPublic(recipient, mintAmount, true);
348
598
  await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
349
- timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
350
- forPublicConsumption: false
599
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds
351
600
  }));
352
601
  this.log.info(`Created a claim for ${mintAmount} L1 fee juice to ${recipient}.`, claim);
353
602
  return claim;
354
603
  }
604
+ /** Returns worst-case min fees across predicted slots, with fallback to current min fees. */ async getMinFees() {
605
+ try {
606
+ const predicted = await this.aztecNode.getPredictedMinFees(ManaUsageEstimate.Limit);
607
+ if (predicted.length === 0) {
608
+ return this.aztecNode.getCurrentMinFees();
609
+ }
610
+ return predicted.reduce((worst, fees)=>fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst);
611
+ } catch {
612
+ return this.aztecNode.getCurrentMinFees();
613
+ }
614
+ }
355
615
  async withNoMinTxsPerBlock(fn) {
356
616
  if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
357
617
  this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
package/dest/index.d.ts CHANGED
@@ -1,8 +1,9 @@
1
1
  export { Bot } from './bot.js';
2
2
  export { AmmBot } from './amm_bot.js';
3
+ export { CrossChainBot } from './cross_chain_bot.js';
3
4
  export { BotRunner } from './runner.js';
4
5
  export { BotStore } from './store/bot_store.js';
5
6
  export { type BotConfig, getBotConfigFromEnv, getBotDefaultConfig, botConfigMappings, SupportedTokenContracts, } from './config.js';
6
7
  export { getBotRunnerApiHandler } from './rpc.js';
7
8
  export * from './interface.js';
8
- //# sourceMappingURL=index.d.ts.map
9
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsR0FBRyxFQUFFLE1BQU0sVUFBVSxDQUFDO0FBQy9CLE9BQU8sRUFBRSxNQUFNLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFDdEMsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ3JELE9BQU8sRUFBRSxTQUFTLEVBQUUsTUFBTSxhQUFhLENBQUM7QUFDeEMsT0FBTyxFQUFFLFFBQVEsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ2hELE9BQU8sRUFDTCxLQUFLLFNBQVMsRUFDZCxtQkFBbUIsRUFDbkIsbUJBQW1CLEVBQ25CLGlCQUFpQixFQUNqQix1QkFBdUIsR0FDeEIsTUFBTSxhQUFhLENBQUM7QUFDckIsT0FBTyxFQUFFLHNCQUFzQixFQUFFLE1BQU0sVUFBVSxDQUFDO0FBQ2xELGNBQWMsZ0JBQWdCLENBQUMifQ==
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EACL,KAAK,SAAS,EACd,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,cAAc,gBAAgB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EACL,KAAK,SAAS,EACd,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,cAAc,gBAAgB,CAAC"}