@aztec/bot 0.0.0-test.1 → 0.0.1-commit.0b941701

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 (49) hide show
  1. package/dest/amm_bot.d.ts +32 -0
  2. package/dest/amm_bot.d.ts.map +1 -0
  3. package/dest/amm_bot.js +101 -0
  4. package/dest/base_bot.d.ts +21 -0
  5. package/dest/base_bot.d.ts.map +1 -0
  6. package/dest/base_bot.js +80 -0
  7. package/dest/bot.d.ts +13 -18
  8. package/dest/bot.d.ts.map +1 -1
  9. package/dest/bot.js +27 -86
  10. package/dest/config.d.ts +84 -60
  11. package/dest/config.d.ts.map +1 -1
  12. package/dest/config.js +59 -36
  13. package/dest/factory.d.ts +30 -31
  14. package/dest/factory.d.ts.map +1 -1
  15. package/dest/factory.js +318 -139
  16. package/dest/index.d.ts +4 -2
  17. package/dest/index.d.ts.map +1 -1
  18. package/dest/index.js +3 -1
  19. package/dest/interface.d.ts +12 -1
  20. package/dest/interface.d.ts.map +1 -1
  21. package/dest/interface.js +5 -0
  22. package/dest/rpc.d.ts +1 -7
  23. package/dest/rpc.d.ts.map +1 -1
  24. package/dest/rpc.js +0 -11
  25. package/dest/runner.d.ts +15 -11
  26. package/dest/runner.d.ts.map +1 -1
  27. package/dest/runner.js +441 -51
  28. package/dest/store/bot_store.d.ts +44 -0
  29. package/dest/store/bot_store.d.ts.map +1 -0
  30. package/dest/store/bot_store.js +107 -0
  31. package/dest/store/index.d.ts +2 -0
  32. package/dest/store/index.d.ts.map +1 -0
  33. package/dest/store/index.js +1 -0
  34. package/dest/utils.d.ts +8 -5
  35. package/dest/utils.d.ts.map +1 -1
  36. package/dest/utils.js +14 -5
  37. package/package.json +27 -23
  38. package/src/amm_bot.ts +124 -0
  39. package/src/base_bot.ts +92 -0
  40. package/src/bot.ts +53 -103
  41. package/src/config.ts +98 -68
  42. package/src/factory.ts +378 -154
  43. package/src/index.ts +3 -1
  44. package/src/interface.ts +9 -0
  45. package/src/rpc.ts +0 -13
  46. package/src/runner.ts +38 -21
  47. package/src/store/bot_store.ts +141 -0
  48. package/src/store/index.ts +1 -0
  49. package/src/utils.ts +17 -6
package/dest/factory.js CHANGED
@@ -1,229 +1,408 @@
1
- import { getSchnorrAccount } from '@aztec/accounts/schnorr';
2
- import { getDeployedTestAccountsWallets, getInitialTestAccounts } from '@aztec/accounts/testing';
3
- import { BatchCall, FeeJuicePaymentMethodWithClaim, L1FeeJuicePortalManager, createLogger, createPXEClient, retryUntil } from '@aztec/aztec.js';
4
- import { createEthereumChain, createL1Clients } from '@aztec/ethereum';
5
- import { Fr } from '@aztec/foundation/fields';
6
- import { EasyPrivateTokenContract } from '@aztec/noir-contracts.js/EasyPrivateToken';
1
+ import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
2
+ import { getInitialTestAccountsData } from '@aztec/accounts/testing';
3
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
4
+ import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
5
+ import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
6
+ import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
7
+ import { deriveKeys } from '@aztec/aztec.js/keys';
8
+ import { createLogger } from '@aztec/aztec.js/log';
9
+ import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
10
+ import { waitForTx } from '@aztec/aztec.js/node';
11
+ import { createEthereumChain } from '@aztec/ethereum/chain';
12
+ import { createExtendedL1Client } from '@aztec/ethereum/client';
13
+ import { Fr } from '@aztec/foundation/curves/bn254';
14
+ import { Timer } from '@aztec/foundation/timer';
15
+ import { AMMContract } from '@aztec/noir-contracts.js/AMM';
16
+ import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
7
17
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
18
+ import { GasSettings } from '@aztec/stdlib/gas';
8
19
  import { deriveSigningKey } from '@aztec/stdlib/keys';
9
- import { makeTracedFetch } from '@aztec/telemetry-client';
10
- import { SupportedTokenContracts, getVersions } from './config.js';
20
+ import { SupportedTokenContracts } from './config.js';
11
21
  import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
12
22
  const MINT_BALANCE = 1e12;
13
23
  const MIN_BALANCE = 1e3;
14
24
  export class BotFactory {
15
25
  config;
16
- pxe;
17
- node;
26
+ wallet;
27
+ store;
28
+ aztecNode;
29
+ aztecNodeAdmin;
18
30
  log;
19
- constructor(config, dependencies = {}){
31
+ constructor(config, wallet, store, aztecNode, aztecNodeAdmin){
20
32
  this.config = config;
33
+ this.wallet = wallet;
34
+ this.store = store;
35
+ this.aztecNode = aztecNode;
36
+ this.aztecNodeAdmin = aztecNodeAdmin;
21
37
  this.log = createLogger('bot');
22
- if (config.flushSetupTransactions && !dependencies.node) {
23
- throw new Error(`Either a node client or node url must be provided if transaction flushing is requested`);
24
- }
25
- if (config.senderPrivateKey && !dependencies.node) {
26
- throw new Error(`Either a node client or node url must be provided for bridging L1 fee juice to deploy an account with private key`);
27
- }
28
- if (!dependencies.pxe && !config.pxeUrl) {
29
- throw new Error(`Either a PXE client or a PXE URL must be provided`);
30
- }
31
- this.node = dependencies.node;
32
- if (dependencies.pxe) {
33
- this.log.info(`Using local PXE`);
34
- this.pxe = dependencies.pxe;
35
- return;
36
- }
37
- this.log.info(`Using remote PXE at ${config.pxeUrl}`);
38
- this.pxe = createPXEClient(config.pxeUrl, getVersions(), makeTracedFetch([
39
- 1,
40
- 2,
41
- 3
42
- ], false));
43
38
  }
44
39
  /**
45
40
  * Initializes a new bot by setting up the sender account, registering the recipient,
46
41
  * deploying the token contract, and minting tokens if necessary.
47
42
  */ async setup() {
48
- const recipient = await this.registerRecipient();
49
- const wallet = await this.setupAccount();
50
- const token = await this.setupToken(wallet);
51
- await this.mintTokens(token);
43
+ const defaultAccountAddress = await this.setupAccount();
44
+ const recipient = (await this.wallet.createAccount()).address;
45
+ const token = await this.setupToken(defaultAccountAddress);
46
+ await this.mintTokens(token, defaultAccountAddress);
52
47
  return {
53
- wallet,
48
+ wallet: this.wallet,
49
+ defaultAccountAddress,
54
50
  token,
55
- pxe: this.pxe,
51
+ node: this.aztecNode,
56
52
  recipient
57
53
  };
58
54
  }
55
+ async setupAmm() {
56
+ const defaultAccountAddress = await this.setupAccount();
57
+ const token0 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken0', 'BOT0');
58
+ const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
59
+ const liquidityToken = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotLPToken', 'BOTLP');
60
+ const amm = await this.setupAmmContract(defaultAccountAddress, this.config.tokenSalt, token0, token1, liquidityToken);
61
+ await this.fundAmm(defaultAccountAddress, defaultAccountAddress, amm, token0, token1, liquidityToken);
62
+ this.log.info(`AMM initialized and funded`);
63
+ return {
64
+ wallet: this.wallet,
65
+ defaultAccountAddress,
66
+ amm,
67
+ token0,
68
+ token1,
69
+ node: this.aztecNode
70
+ };
71
+ }
59
72
  /**
60
73
  * Checks if the sender account contract is initialized, and initializes it if necessary.
61
74
  * @returns The sender wallet.
62
75
  */ async setupAccount() {
63
- if (this.config.senderPrivateKey) {
64
- return await this.setupAccountWithPrivateKey(this.config.senderPrivateKey);
76
+ const privateKey = this.config.senderPrivateKey?.getValue();
77
+ if (privateKey) {
78
+ this.log.info(`Setting up account with provided private key`);
79
+ return await this.setupAccountWithPrivateKey(privateKey);
65
80
  } else {
81
+ this.log.info(`Setting up test account`);
66
82
  return await this.setupTestAccount();
67
83
  }
68
84
  }
69
- async setupAccountWithPrivateKey(privateKey) {
70
- const salt = Fr.ONE;
71
- const signingKey = deriveSigningKey(privateKey);
72
- const account = await getSchnorrAccount(this.pxe, privateKey, signingKey, salt);
73
- const isInit = (await this.pxe.getContractMetadata(account.getAddress())).isContractInitialized;
74
- if (isInit) {
75
- this.log.info(`Account at ${account.getAddress().toString()} already initialized`);
76
- const wallet = await account.register();
77
- return wallet;
85
+ async setupAccountWithPrivateKey(secret) {
86
+ const salt = this.config.senderSalt ?? Fr.ONE;
87
+ const signingKey = deriveSigningKey(secret);
88
+ const accountData = {
89
+ secret,
90
+ salt,
91
+ contract: new SchnorrAccountContract(signingKey)
92
+ };
93
+ const accountManager = await this.wallet.createAccount(accountData);
94
+ const metadata = await this.wallet.getContractMetadata(accountManager.address);
95
+ if (metadata.isContractInitialized) {
96
+ this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
97
+ const timer = new Timer();
98
+ const address = accountManager.address;
99
+ this.log.info(`Account at ${address} registered. duration=${timer.ms()}`);
100
+ await this.store.deleteBridgeClaim(address);
101
+ return address;
78
102
  } else {
79
- const address = account.getAddress();
103
+ const address = accountManager.address;
80
104
  this.log.info(`Deploying account at ${address}`);
81
- const claim = await this.bridgeL1FeeJuice(address, 10n ** 22n);
82
- const wallet = await account.getWallet();
83
- const paymentMethod = new FeeJuicePaymentMethodWithClaim(wallet, claim);
84
- const sentTx = account.deploy({
85
- fee: {
86
- paymentMethod
87
- }
105
+ const claim = await this.getOrCreateBridgeClaim(address);
106
+ const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
107
+ const deployMethod = await accountManager.getDeployMethod();
108
+ const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
109
+ const gasSettings = GasSettings.default({
110
+ maxFeesPerGas
88
111
  });
89
- const txHash = await sentTx.getTxHash();
90
- this.log.info(`Sent tx with hash ${txHash.toString()}`);
91
- await this.tryFlushTxs();
92
- this.log.verbose('Waiting for account deployment to settle');
93
- await sentTx.wait({
94
- timeout: this.config.txMinedWaitSeconds
112
+ await this.withNoMinTxsPerBlock(async ()=>{
113
+ const txHash = await deployMethod.send({
114
+ from: AztecAddress.ZERO,
115
+ fee: {
116
+ gasSettings,
117
+ paymentMethod
118
+ },
119
+ wait: NO_WAIT
120
+ });
121
+ this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
122
+ return waitForTx(this.aztecNode, txHash, {
123
+ timeout: this.config.txMinedWaitSeconds
124
+ });
95
125
  });
96
126
  this.log.info(`Account deployed at ${address}`);
97
- return wallet;
127
+ // Clean up the consumed bridge claim
128
+ await this.store.deleteBridgeClaim(address);
129
+ return accountManager.address;
98
130
  }
99
131
  }
100
132
  async setupTestAccount() {
101
- let [wallet] = await getDeployedTestAccountsWallets(this.pxe);
102
- if (wallet) {
103
- this.log.info(`Using funded test account: ${wallet.getAddress()}`);
104
- } else {
105
- this.log.info('Registering funded test account');
106
- const [account] = await getInitialTestAccounts();
107
- const manager = await getSchnorrAccount(this.pxe, account.secret, account.signingKey, account.salt);
108
- wallet = await manager.register();
109
- this.log.info(`Funded test account registered: ${wallet.getAddress()}`);
110
- }
111
- return wallet;
112
- }
113
- /**
114
- * Registers the recipient for txs in the pxe.
115
- */ async registerRecipient() {
116
- const recipient = await this.pxe.registerAccount(this.config.recipientEncryptionSecret, Fr.ONE);
117
- return recipient.address;
133
+ const [initialAccountData] = await getInitialTestAccountsData();
134
+ const accountData = {
135
+ secret: initialAccountData.secret,
136
+ salt: initialAccountData.salt,
137
+ contract: new SchnorrAccountContract(initialAccountData.signingKey)
138
+ };
139
+ const accountManager = await this.wallet.createAccount(accountData);
140
+ return accountManager.address;
118
141
  }
119
142
  /**
120
143
  * Checks if the token contract is deployed and deploys it if necessary.
121
144
  * @param wallet - Wallet to deploy the token contract from.
122
145
  * @returns The TokenContract instance.
123
- */ async setupToken(wallet) {
146
+ */ async setupToken(sender) {
124
147
  let deploy;
148
+ let tokenInstance;
125
149
  const deployOpts = {
150
+ from: sender,
126
151
  contractAddressSalt: this.config.tokenSalt,
127
152
  universalDeploy: true
128
153
  };
154
+ let token;
129
155
  if (this.config.contract === SupportedTokenContracts.TokenContract) {
130
- deploy = TokenContract.deploy(wallet, wallet.getAddress(), 'BotToken', 'BOT', 18);
131
- } else if (this.config.contract === SupportedTokenContracts.EasyPrivateTokenContract) {
132
- deploy = EasyPrivateTokenContract.deploy(wallet, MINT_BALANCE, wallet.getAddress());
133
- deployOpts.skipPublicDeployment = true;
134
- deployOpts.skipClassRegistration = true;
156
+ deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
157
+ tokenInstance = await deploy.getInstance(deployOpts);
158
+ token = TokenContract.at(tokenInstance.address, this.wallet);
159
+ } else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
160
+ // Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
161
+ const tokenSecretKey = Fr.random();
162
+ const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
163
+ deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
164
+ deployOpts.skipInstancePublication = true;
165
+ deployOpts.skipClassPublication = true;
135
166
  deployOpts.skipInitialization = false;
136
- deployOpts.skipPublicSimulation = true;
167
+ // Register the contract with the secret key before deployment
168
+ tokenInstance = await deploy.getInstance(deployOpts);
169
+ token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
170
+ await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
137
171
  } else {
138
172
  throw new Error(`Unsupported token contract type: ${this.config.contract}`);
139
173
  }
140
- const address = (await deploy.getInstance(deployOpts)).address;
141
- if ((await this.pxe.getContractMetadata(address)).isContractPubliclyDeployed) {
174
+ const address = tokenInstance?.address ?? (await deploy.getInstance(deployOpts)).address;
175
+ const metadata = await this.wallet.getContractMetadata(address);
176
+ if (metadata.isContractPublished) {
142
177
  this.log.info(`Token at ${address.toString()} already deployed`);
143
- return deploy.register();
178
+ await deploy.register();
144
179
  } else {
145
180
  this.log.info(`Deploying token contract at ${address.toString()}`);
146
- const sentTx = deploy.send(deployOpts);
147
- const txHash = await sentTx.getTxHash();
148
- this.log.info(`Sent tx with hash ${txHash.toString()}`);
149
- await this.tryFlushTxs();
150
- this.log.verbose('Waiting for token setup to settle');
151
- return sentTx.deployed({
181
+ const txHash = await deploy.send({
182
+ ...deployOpts,
183
+ wait: NO_WAIT
184
+ });
185
+ this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
186
+ await this.withNoMinTxsPerBlock(async ()=>{
187
+ await waitForTx(this.aztecNode, txHash, {
188
+ timeout: this.config.txMinedWaitSeconds
189
+ });
190
+ return token;
191
+ });
192
+ }
193
+ return token;
194
+ }
195
+ /**
196
+ * Checks if the token contract is deployed and deploys it if necessary.
197
+ * @param wallet - Wallet to deploy the token contract from.
198
+ * @returns The TokenContract instance.
199
+ */ async setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals = 18) {
200
+ const deployOpts = {
201
+ from: deployer,
202
+ contractAddressSalt,
203
+ universalDeploy: true
204
+ };
205
+ const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
206
+ const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
207
+ return TokenContract.at(instance.address, this.wallet);
208
+ }
209
+ async setupAmmContract(deployer, contractAddressSalt, token0, token1, lpToken) {
210
+ const deployOpts = {
211
+ from: deployer,
212
+ contractAddressSalt,
213
+ universalDeploy: true
214
+ };
215
+ const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
216
+ const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
217
+ const amm = AMMContract.at(instance.address, this.wallet);
218
+ this.log.info(`AMM deployed at ${amm.address}`);
219
+ const minterReceipt = await lpToken.methods.set_minter(amm.address, true).send({
220
+ from: deployer,
221
+ wait: {
152
222
  timeout: this.config.txMinedWaitSeconds
223
+ }
224
+ });
225
+ this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
226
+ this.log.info(`Liquidity token initialized`);
227
+ return amm;
228
+ }
229
+ async fundAmm(defaultAccountAddress, liquidityProvider, amm, token0, token1, lpToken) {
230
+ const getPrivateBalances = ()=>Promise.all([
231
+ token0.methods.balance_of_private(liquidityProvider).simulate({
232
+ from: liquidityProvider
233
+ }),
234
+ token1.methods.balance_of_private(liquidityProvider).simulate({
235
+ from: liquidityProvider
236
+ }),
237
+ lpToken.methods.balance_of_private(liquidityProvider).simulate({
238
+ from: liquidityProvider
239
+ })
240
+ ]);
241
+ const authwitNonce = Fr.random();
242
+ // keep some tokens for swapping
243
+ const amount0Max = MINT_BALANCE / 2;
244
+ const amount0Min = MINT_BALANCE / 4;
245
+ const amount1Max = MINT_BALANCE / 2;
246
+ const amount1Min = MINT_BALANCE / 4;
247
+ const [t0Bal, t1Bal, lpBal] = await getPrivateBalances();
248
+ this.log.info(`Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1. Current private balances of ${liquidityProvider}: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`);
249
+ // Add authwitnesses for the transfers in AMM::add_liquidity function
250
+ const token0Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
251
+ caller: amm.address,
252
+ call: await token0.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount0Max, authwitNonce).getFunctionCall()
253
+ });
254
+ const token1Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
255
+ caller: amm.address,
256
+ call: await token1.methods.transfer_to_public_and_prepare_private_balance_increase(liquidityProvider, amm.address, amount1Max, authwitNonce).getFunctionCall()
257
+ });
258
+ const mintReceipt = await new BatchCall(this.wallet, [
259
+ token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
260
+ token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE)
261
+ ]).send({
262
+ from: liquidityProvider,
263
+ wait: {
264
+ timeout: this.config.txMinedWaitSeconds
265
+ }
266
+ });
267
+ this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
268
+ const addLiquidityReceipt = await amm.methods.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce).send({
269
+ from: liquidityProvider,
270
+ authWitnesses: [
271
+ token0Authwit,
272
+ token1Authwit
273
+ ],
274
+ wait: {
275
+ timeout: this.config.txMinedWaitSeconds
276
+ }
277
+ });
278
+ this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
279
+ this.log.info(`Liquidity added`);
280
+ const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
281
+ this.log.info(`Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${newT0Bal}, token1=${newT1Bal}, lp=${newLPBal}`);
282
+ }
283
+ async registerOrDeployContract(name, deploy, deployOpts) {
284
+ const instance = await deploy.getInstance(deployOpts);
285
+ const address = instance.address;
286
+ const metadata = await this.wallet.getContractMetadata(address);
287
+ if (metadata.isContractPublished) {
288
+ this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
289
+ await deploy.register();
290
+ } else {
291
+ this.log.info(`Deploying contract ${name} at ${address.toString()}`);
292
+ await this.withNoMinTxsPerBlock(async ()=>{
293
+ const txHash = await deploy.send({
294
+ ...deployOpts,
295
+ wait: NO_WAIT
296
+ });
297
+ this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
298
+ return waitForTx(this.aztecNode, txHash, {
299
+ timeout: this.config.txMinedWaitSeconds
300
+ });
153
301
  });
154
302
  }
303
+ return instance;
155
304
  }
156
305
  /**
157
306
  * Mints private and public tokens for the sender if their balance is below the minimum.
158
307
  * @param token - Token contract.
159
- */ async mintTokens(token) {
160
- const sender = token.wallet.getAddress();
308
+ */ async mintTokens(token, minter) {
161
309
  const isStandardToken = isStandardTokenContract(token);
162
310
  let privateBalance = 0n;
163
311
  let publicBalance = 0n;
164
312
  if (isStandardToken) {
165
- ({ privateBalance, publicBalance } = await getBalances(token, sender));
313
+ ({ privateBalance, publicBalance } = await getBalances(token, minter));
166
314
  } else {
167
- privateBalance = await getPrivateBalance(token, sender);
315
+ privateBalance = await getPrivateBalance(token, minter);
168
316
  }
169
317
  const calls = [];
170
318
  if (privateBalance < MIN_BALANCE) {
171
- this.log.info(`Minting private tokens for ${sender.toString()}`);
172
- const from = sender; // we are setting from to sender here because we need a sender to calculate the tag
173
- calls.push(isStandardToken ? await token.methods.mint_to_private(from, sender, MINT_BALANCE).request() : await token.methods.mint(MINT_BALANCE, sender).request());
319
+ this.log.info(`Minting private tokens for ${minter.toString()}`);
320
+ calls.push(isStandardToken ? token.methods.mint_to_private(minter, MINT_BALANCE) : token.methods.mint(MINT_BALANCE, minter));
174
321
  }
175
322
  if (isStandardToken && publicBalance < MIN_BALANCE) {
176
- this.log.info(`Minting public tokens for ${sender.toString()}`);
177
- calls.push(await token.methods.mint_to_public(sender, MINT_BALANCE).request());
323
+ this.log.info(`Minting public tokens for ${minter.toString()}`);
324
+ calls.push(token.methods.mint_to_public(minter, MINT_BALANCE));
178
325
  }
179
326
  if (calls.length === 0) {
180
- this.log.info(`Skipping minting as ${sender.toString()} has enough tokens`);
327
+ this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
181
328
  return;
182
329
  }
183
- const sentTx = new BatchCall(token.wallet, calls).send();
184
- const txHash = await sentTx.getTxHash();
185
- this.log.info(`Sent tx with hash ${txHash.toString()}`);
186
- await this.tryFlushTxs();
187
- this.log.verbose('Waiting for token mint to settle');
188
- await sentTx.wait({
189
- timeout: this.config.txMinedWaitSeconds
330
+ await this.withNoMinTxsPerBlock(async ()=>{
331
+ const txHash = await new BatchCall(token.wallet, calls).send({
332
+ from: minter,
333
+ wait: NO_WAIT
334
+ });
335
+ this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
336
+ return waitForTx(this.aztecNode, txHash, {
337
+ timeout: this.config.txMinedWaitSeconds
338
+ });
190
339
  });
191
340
  }
192
- async bridgeL1FeeJuice(recipient, amount) {
341
+ /**
342
+ * Gets or creates a bridge claim for the recipient.
343
+ * Checks if a claim already exists in the store and reuses it if valid.
344
+ * Only creates a new bridge if fee juice balance is below threshold.
345
+ */ async getOrCreateBridgeClaim(recipient) {
346
+ // Check if we have an existing claim in the store
347
+ const existingClaim = await this.store.getBridgeClaim(recipient);
348
+ if (existingClaim) {
349
+ this.log.info(`Found existing bridge claim for ${recipient.toString()}, checking validity...`);
350
+ // Check if the message is ready on L2
351
+ try {
352
+ const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
353
+ await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
354
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
355
+ forPublicConsumption: false
356
+ }));
357
+ return existingClaim.claim;
358
+ } catch (err) {
359
+ this.log.warn(`Failed to verify existing claim, creating new one: ${err}`);
360
+ await this.store.deleteBridgeClaim(recipient);
361
+ }
362
+ }
363
+ const claim = await this.bridgeL1FeeJuice(recipient);
364
+ await this.store.saveBridgeClaim(recipient, claim);
365
+ return claim;
366
+ }
367
+ async bridgeL1FeeJuice(recipient) {
193
368
  const l1RpcUrls = this.config.l1RpcUrls;
194
369
  if (!l1RpcUrls?.length) {
195
370
  throw new Error('L1 Rpc url is required to bridge the fee juice to fund the deployment of the account.');
196
371
  }
197
- const mnemonicOrPrivateKey = this.config.l1PrivateKey || this.config.l1Mnemonic;
372
+ const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
198
373
  if (!mnemonicOrPrivateKey) {
199
374
  throw new Error('Either a mnemonic or private key of an L1 account is required to bridge the fee juice to fund the deployment of the account.');
200
375
  }
201
- const { l1ChainId } = await this.pxe.getNodeInfo();
376
+ const { l1ChainId } = await this.aztecNode.getNodeInfo();
202
377
  const chain = createEthereumChain(l1RpcUrls, l1ChainId);
203
- const { publicClient, walletClient } = createL1Clients(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
204
- const portal = await L1FeeJuicePortalManager.new(this.pxe, publicClient, walletClient, this.log);
205
- const claim = await portal.bridgeTokensPublic(recipient, amount, true);
206
- const isSynced = async ()=>await this.pxe.isL1ToL2MessageSynced(Fr.fromHexString(claim.messageHash));
207
- await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 1);
208
- this.log.info(`Created a claim for ${amount} L1 fee juice to ${recipient}.`, claim);
209
- // Progress by 2 L2 blocks so that the l1ToL2Message added above will be available to use on L2.
210
- await this.advanceL2Block();
211
- await this.advanceL2Block();
378
+ const extendedClient = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
379
+ const portal = await L1FeeJuicePortalManager.new(this.aztecNode, extendedClient, this.log);
380
+ const mintAmount = await portal.getTokenManager().getMintAmount();
381
+ const claim = await portal.bridgeTokensPublic(recipient, mintAmount, true);
382
+ await this.withNoMinTxsPerBlock(()=>waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
383
+ timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
384
+ forPublicConsumption: false
385
+ }));
386
+ this.log.info(`Created a claim for ${mintAmount} L1 fee juice to ${recipient}.`, claim);
212
387
  return claim;
213
388
  }
214
- async advanceL2Block() {
215
- const initialBlockNumber = await this.node.getBlockNumber();
216
- await this.tryFlushTxs();
217
- await retryUntil(async ()=>await this.node.getBlockNumber() >= initialBlockNumber + 1);
218
- }
219
- async tryFlushTxs() {
220
- if (this.config.flushSetupTransactions) {
221
- this.log.verbose('Flushing transactions');
222
- try {
223
- await this.node.flushTxs();
224
- } catch (err) {
225
- this.log.error(`Failed to flush transactions: ${err}`);
226
- }
389
+ async withNoMinTxsPerBlock(fn) {
390
+ if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
391
+ this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
392
+ return fn();
393
+ }
394
+ const { minTxsPerBlock } = await this.aztecNodeAdmin.getConfig();
395
+ this.log.warn(`Setting sequencer minTxsPerBlock to 0 from ${minTxsPerBlock} to flush setup transactions`);
396
+ await this.aztecNodeAdmin.setConfig({
397
+ minTxsPerBlock: 0
398
+ });
399
+ try {
400
+ return await fn();
401
+ } finally{
402
+ this.log.warn(`Restoring sequencer minTxsPerBlock to ${minTxsPerBlock}`);
403
+ await this.aztecNodeAdmin.setConfig({
404
+ minTxsPerBlock
405
+ });
227
406
  }
228
407
  }
229
408
  }
package/dest/index.d.ts CHANGED
@@ -1,6 +1,8 @@
1
1
  export { Bot } from './bot.js';
2
+ export { AmmBot } from './amm_bot.js';
2
3
  export { BotRunner } from './runner.js';
4
+ export { BotStore } from './store/bot_store.js';
3
5
  export { type BotConfig, getBotConfigFromEnv, getBotDefaultConfig, botConfigMappings, SupportedTokenContracts, } from './config.js';
4
- export { createBotRunnerRpcServer, getBotRunnerApiHandler } from './rpc.js';
6
+ export { getBotRunnerApiHandler } from './rpc.js';
5
7
  export * from './interface.js';
6
- //# sourceMappingURL=index.d.ts.map
8
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsR0FBRyxFQUFFLE1BQU0sVUFBVSxDQUFDO0FBQy9CLE9BQU8sRUFBRSxNQUFNLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFDdEMsT0FBTyxFQUFFLFNBQVMsRUFBRSxNQUFNLGFBQWEsQ0FBQztBQUN4QyxPQUFPLEVBQUUsUUFBUSxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDaEQsT0FBTyxFQUNMLEtBQUssU0FBUyxFQUNkLG1CQUFtQixFQUNuQixtQkFBbUIsRUFDbkIsaUJBQWlCLEVBQ2pCLHVCQUF1QixHQUN4QixNQUFNLGFBQWEsQ0FBQztBQUNyQixPQUFPLEVBQUUsc0JBQXNCLEVBQUUsTUFBTSxVQUFVLENBQUM7QUFDbEQsY0FBYyxnQkFBZ0IsQ0FBQyJ9
@@ -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,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EACL,KAAK,SAAS,EACd,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAC5E,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,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"}
package/dest/index.js CHANGED
@@ -1,5 +1,7 @@
1
1
  export { Bot } from './bot.js';
2
+ export { AmmBot } from './amm_bot.js';
2
3
  export { BotRunner } from './runner.js';
4
+ export { BotStore } from './store/bot_store.js';
3
5
  export { getBotConfigFromEnv, getBotDefaultConfig, botConfigMappings, SupportedTokenContracts } from './config.js';
4
- export { createBotRunnerRpcServer, getBotRunnerApiHandler } from './rpc.js';
6
+ export { getBotRunnerApiHandler } from './rpc.js';
5
7
  export * from './interface.js';
@@ -1,12 +1,23 @@
1
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
1
2
  import type { ApiSchemaFor } from '@aztec/stdlib/schemas';
3
+ import { z } from 'zod';
2
4
  import { type BotConfig } from './config.js';
5
+ export declare const BotInfoSchema: z.ZodObject<{
6
+ botAddress: import("@aztec/stdlib/schemas").ZodFor<AztecAddress>;
7
+ }, "strip", z.ZodTypeAny, {
8
+ botAddress: AztecAddress;
9
+ }, {
10
+ botAddress?: any;
11
+ }>;
12
+ export type BotInfo = z.infer<typeof BotInfoSchema>;
3
13
  export interface BotRunnerApi {
4
14
  start(): Promise<void>;
5
15
  stop(): Promise<void>;
6
16
  run(): Promise<void>;
7
17
  setup(): Promise<void>;
8
18
  getConfig(): Promise<BotConfig>;
19
+ getInfo(): Promise<BotInfo>;
9
20
  update(config: BotConfig): Promise<void>;
10
21
  }
11
22
  export declare const BotRunnerApiSchema: ApiSchemaFor<BotRunnerApi>;
12
- //# sourceMappingURL=interface.d.ts.map
23
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW50ZXJmYWNlLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvaW50ZXJmYWNlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUN6RCxPQUFPLEtBQUssRUFBRSxZQUFZLEVBQUUsTUFBTSx1QkFBdUIsQ0FBQztBQUUxRCxPQUFPLEVBQUUsQ0FBQyxFQUFFLE1BQU0sS0FBSyxDQUFDO0FBRXhCLE9BQU8sRUFBRSxLQUFLLFNBQVMsRUFBbUIsTUFBTSxhQUFhLENBQUM7QUFFOUQsZUFBTyxNQUFNLGFBQWE7Ozs7OztFQUV4QixDQUFDO0FBRUgsTUFBTSxNQUFNLE9BQU8sR0FBRyxDQUFDLENBQUMsS0FBSyxDQUFDLE9BQU8sYUFBYSxDQUFDLENBQUM7QUFFcEQsTUFBTSxXQUFXLFlBQVk7SUFDM0IsS0FBSyxJQUFJLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQztJQUN2QixJQUFJLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFDO0lBQ3RCLEdBQUcsSUFBSSxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUM7SUFDckIsS0FBSyxJQUFJLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQztJQUN2QixTQUFTLElBQUksT0FBTyxDQUFDLFNBQVMsQ0FBQyxDQUFDO0lBQ2hDLE9BQU8sSUFBSSxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUM7SUFDNUIsTUFBTSxDQUFDLE1BQU0sRUFBRSxTQUFTLEdBQUcsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFDO0NBQzFDO0FBRUQsZUFBTyxNQUFNLGtCQUFrQixFQUFFLFlBQVksQ0FBQyxZQUFZLENBUXpELENBQUMifQ==
@@ -1 +1 @@
1
- {"version":3,"file":"interface.d.ts","sourceRoot":"","sources":["../src/interface.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAI1D,OAAO,EAAE,KAAK,SAAS,EAAmB,MAAM,aAAa,CAAC;AAE9D,MAAM,WAAW,YAAY;IAC3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,eAAO,MAAM,kBAAkB,EAAE,YAAY,CAAC,YAAY,CAOzD,CAAC"}
1
+ {"version":3,"file":"interface.d.ts","sourceRoot":"","sources":["../src/interface.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAE1D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,KAAK,SAAS,EAAmB,MAAM,aAAa,CAAC;AAE9D,eAAO,MAAM,aAAa;;;;;;EAExB,CAAC;AAEH,MAAM,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAEpD,MAAM,WAAW,YAAY;IAC3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5B,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,eAAO,MAAM,kBAAkB,EAAE,YAAY,CAAC,YAAY,CAQzD,CAAC"}
package/dest/interface.js CHANGED
@@ -1,10 +1,15 @@
1
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
1
2
  import { z } from 'zod';
2
3
  import { BotConfigSchema } from './config.js';
4
+ export const BotInfoSchema = z.object({
5
+ botAddress: AztecAddress.schema
6
+ });
3
7
  export const BotRunnerApiSchema = {
4
8
  start: z.function().args().returns(z.void()),
5
9
  stop: z.function().args().returns(z.void()),
6
10
  run: z.function().args().returns(z.void()),
7
11
  setup: z.function().args().returns(z.void()),
12
+ getInfo: z.function().args().returns(BotInfoSchema),
8
13
  getConfig: z.function().args().returns(BotConfigSchema),
9
14
  update: z.function().args(BotConfigSchema).returns(z.void())
10
15
  };
package/dest/rpc.d.ts CHANGED
@@ -1,10 +1,4 @@
1
1
  import type { ApiHandler } from '@aztec/foundation/json-rpc/server';
2
2
  import type { BotRunner } from './runner.js';
3
- /**
4
- * Wraps a bot runner with a JSON RPC HTTP server.
5
- * @param botRunner - The BotRunner.
6
- * @returns An JSON-RPC HTTP server
7
- */
8
- export declare function createBotRunnerRpcServer(botRunner: BotRunner): void;
9
3
  export declare function getBotRunnerApiHandler(botRunner: BotRunner): ApiHandler;
10
- //# sourceMappingURL=rpc.d.ts.map
4
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicnBjLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvcnBjLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBSyxFQUFFLFVBQVUsRUFBRSxNQUFNLG1DQUFtQyxDQUFDO0FBR3BFLE9BQU8sS0FBSyxFQUFFLFNBQVMsRUFBRSxNQUFNLGFBQWEsQ0FBQztBQUU3Qyx3QkFBZ0Isc0JBQXNCLENBQUMsU0FBUyxFQUFFLFNBQVMsR0FBRyxVQUFVLENBRXZFIn0=