@aztec/bot 0.0.1-commit.43c09e3f → 0.0.1-commit.4d3c002
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.
- package/dest/amm_bot.d.ts +4 -4
- package/dest/amm_bot.d.ts.map +1 -1
- package/dest/amm_bot.js +24 -17
- package/dest/base_bot.d.ts +6 -6
- package/dest/base_bot.d.ts.map +1 -1
- package/dest/base_bot.js +21 -32
- package/dest/bot.d.ts +4 -4
- package/dest/bot.d.ts.map +1 -1
- package/dest/bot.js +5 -8
- package/dest/config.d.ts +34 -18
- package/dest/config.d.ts.map +1 -1
- package/dest/config.js +39 -12
- package/dest/cross_chain_bot.d.ts +54 -0
- package/dest/cross_chain_bot.d.ts.map +1 -0
- package/dest/cross_chain_bot.js +134 -0
- package/dest/factory.d.ts +24 -5
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +282 -72
- package/dest/index.d.ts +2 -1
- package/dest/index.d.ts.map +1 -1
- package/dest/index.js +1 -0
- package/dest/l1_to_l2_seeding.d.ts +8 -0
- package/dest/l1_to_l2_seeding.d.ts.map +1 -0
- package/dest/l1_to_l2_seeding.js +63 -0
- package/dest/runner.d.ts +3 -3
- package/dest/runner.d.ts.map +1 -1
- package/dest/runner.js +17 -1
- package/dest/store/bot_store.d.ts +30 -5
- package/dest/store/bot_store.d.ts.map +1 -1
- package/dest/store/bot_store.js +37 -6
- package/dest/store/index.d.ts +2 -2
- package/dest/store/index.d.ts.map +1 -1
- package/dest/utils.js +3 -3
- package/package.json +16 -13
- package/src/amm_bot.ts +24 -19
- package/src/base_bot.ts +15 -33
- package/src/bot.ts +8 -10
- package/src/config.ts +44 -16
- package/src/cross_chain_bot.ts +203 -0
- package/src/factory.ts +337 -73
- package/src/index.ts +1 -0
- package/src/l1_to_l2_seeding.ts +79 -0
- package/src/runner.ts +18 -5
- package/src/store/bot_store.ts +60 -5
- package/src/store/index.ts +1 -1
- package/src/utils.ts +3 -3
package/src/factory.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
|
|
2
1
|
import { getInitialTestAccountsData } from '@aztec/accounts/testing';
|
|
2
|
+
import { NO_FROM } from '@aztec/aztec.js/account';
|
|
3
3
|
import { AztecAddress } from '@aztec/aztec.js/addresses';
|
|
4
4
|
import {
|
|
5
5
|
BatchCall,
|
|
@@ -9,64 +9,78 @@ import {
|
|
|
9
9
|
type DeployOptions,
|
|
10
10
|
NO_WAIT,
|
|
11
11
|
} from '@aztec/aztec.js/contracts';
|
|
12
|
-
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
13
12
|
import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
|
|
13
|
+
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
14
14
|
import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
|
|
15
15
|
import { deriveKeys } from '@aztec/aztec.js/keys';
|
|
16
16
|
import { createLogger } from '@aztec/aztec.js/log';
|
|
17
17
|
import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
|
|
18
18
|
import { waitForTx } from '@aztec/aztec.js/node';
|
|
19
|
+
import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
|
|
20
|
+
import { ContractInitializationStatus } from '@aztec/aztec.js/wallet';
|
|
19
21
|
import { createEthereumChain } from '@aztec/ethereum/chain';
|
|
20
22
|
import { createExtendedL1Client } from '@aztec/ethereum/client';
|
|
23
|
+
import { RollupContract } from '@aztec/ethereum/contracts';
|
|
24
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
21
25
|
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
26
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
22
27
|
import { Timer } from '@aztec/foundation/timer';
|
|
23
28
|
import { AMMContract } from '@aztec/noir-contracts.js/AMM';
|
|
24
29
|
import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
|
|
25
30
|
import { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
31
|
+
import { TestContract } from '@aztec/noir-test-contracts.js/Test';
|
|
26
32
|
import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
|
|
27
|
-
import { GasSettings } from '@aztec/stdlib/gas';
|
|
33
|
+
import { GasFees, GasSettings } from '@aztec/stdlib/gas';
|
|
28
34
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
29
35
|
import { deriveSigningKey } from '@aztec/stdlib/keys';
|
|
30
|
-
import {
|
|
36
|
+
import { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
31
37
|
|
|
32
38
|
import { type BotConfig, SupportedTokenContracts } from './config.js';
|
|
39
|
+
import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
|
|
33
40
|
import type { BotStore } from './store/index.js';
|
|
34
41
|
import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
|
|
35
42
|
|
|
36
43
|
const MINT_BALANCE = 1e12;
|
|
37
44
|
const MIN_BALANCE = 1e3;
|
|
45
|
+
const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
|
|
46
|
+
const FEE_JUICE_TOP_UP_TARGET = 10_000n * 10n ** 18n;
|
|
38
47
|
|
|
39
48
|
export class BotFactory {
|
|
40
49
|
private log = createLogger('bot');
|
|
41
50
|
|
|
42
51
|
constructor(
|
|
43
52
|
private readonly config: BotConfig,
|
|
44
|
-
private readonly wallet:
|
|
53
|
+
private readonly wallet: EmbeddedWallet,
|
|
45
54
|
private readonly store: BotStore,
|
|
46
55
|
private readonly aztecNode: AztecNode,
|
|
47
56
|
private readonly aztecNodeAdmin?: AztecNodeAdmin,
|
|
48
|
-
) {
|
|
57
|
+
) {
|
|
58
|
+
// Set fee padding on the wallet so that all transactions during setup
|
|
59
|
+
// (token deploy, minting, etc.) use the configured padding, not the default.
|
|
60
|
+
this.wallet.setMinFeePadding(config.minFeePadding);
|
|
61
|
+
}
|
|
49
62
|
|
|
50
63
|
/**
|
|
51
64
|
* Initializes a new bot by setting up the sender account, registering the recipient,
|
|
52
65
|
* deploying the token contract, and minting tokens if necessary.
|
|
53
66
|
*/
|
|
54
67
|
public async setup(): Promise<{
|
|
55
|
-
wallet:
|
|
68
|
+
wallet: EmbeddedWallet;
|
|
56
69
|
defaultAccountAddress: AztecAddress;
|
|
57
70
|
token: TokenContract | PrivateTokenContract;
|
|
58
71
|
node: AztecNode;
|
|
59
72
|
recipient: AztecAddress;
|
|
60
73
|
}> {
|
|
61
74
|
const defaultAccountAddress = await this.setupAccount();
|
|
62
|
-
const recipient = (await this.wallet.
|
|
63
|
-
const token = await this.
|
|
75
|
+
const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
|
|
76
|
+
const token = await this.setupTokenWithOptionalEarlyRefuel(defaultAccountAddress);
|
|
77
|
+
await this.ensureFeeJuiceBalance(defaultAccountAddress, token);
|
|
64
78
|
await this.mintTokens(token, defaultAccountAddress);
|
|
65
79
|
return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
|
|
66
80
|
}
|
|
67
81
|
|
|
68
82
|
public async setupAmm(): Promise<{
|
|
69
|
-
wallet:
|
|
83
|
+
wallet: EmbeddedWallet;
|
|
70
84
|
defaultAccountAddress: AztecAddress;
|
|
71
85
|
amm: AMMContract;
|
|
72
86
|
token0: TokenContract;
|
|
@@ -74,7 +88,13 @@ export class BotFactory {
|
|
|
74
88
|
node: AztecNode;
|
|
75
89
|
}> {
|
|
76
90
|
const defaultAccountAddress = await this.setupAccount();
|
|
77
|
-
const token0 = await this.
|
|
91
|
+
const token0 = await this.setupTokenContractWithOptionalEarlyRefuel(
|
|
92
|
+
defaultAccountAddress,
|
|
93
|
+
this.config.tokenSalt,
|
|
94
|
+
'BotToken0',
|
|
95
|
+
'BOT0',
|
|
96
|
+
);
|
|
97
|
+
await this.ensureFeeJuiceBalance(defaultAccountAddress, token0);
|
|
78
98
|
const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
|
|
79
99
|
const liquidityToken = await this.setupTokenContract(
|
|
80
100
|
defaultAccountAddress,
|
|
@@ -96,6 +116,89 @@ export class BotFactory {
|
|
|
96
116
|
return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
|
|
97
117
|
}
|
|
98
118
|
|
|
119
|
+
/**
|
|
120
|
+
* Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
|
|
121
|
+
* seeding initial L1→L2 messages, and waiting for the first to be ready.
|
|
122
|
+
*/
|
|
123
|
+
public async setupCrossChain(): Promise<{
|
|
124
|
+
wallet: EmbeddedWallet;
|
|
125
|
+
defaultAccountAddress: AztecAddress;
|
|
126
|
+
contract: TestContract;
|
|
127
|
+
node: AztecNode;
|
|
128
|
+
l1Client: ExtendedViemWalletClient;
|
|
129
|
+
rollupVersion: bigint;
|
|
130
|
+
}> {
|
|
131
|
+
const defaultAccountAddress = await this.setupAccount();
|
|
132
|
+
|
|
133
|
+
// Create L1 client (same pattern as bridgeL1FeeJuice)
|
|
134
|
+
const l1RpcUrls = this.config.l1RpcUrls;
|
|
135
|
+
if (!l1RpcUrls?.length) {
|
|
136
|
+
throw new Error('L1 RPC URLs required for cross-chain bot');
|
|
137
|
+
}
|
|
138
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
139
|
+
if (!mnemonicOrPrivateKey) {
|
|
140
|
+
throw new Error('L1 mnemonic or private key required for cross-chain bot');
|
|
141
|
+
}
|
|
142
|
+
const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
|
|
143
|
+
const chain = createEthereumChain(l1RpcUrls, l1ChainId);
|
|
144
|
+
const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
|
|
145
|
+
|
|
146
|
+
// Fetch Rollup version (needed for Inbox L2Actor struct)
|
|
147
|
+
const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
|
|
148
|
+
const rollupVersion = await rollupContract.getVersion();
|
|
149
|
+
|
|
150
|
+
// Deploy TestContract
|
|
151
|
+
const contract = await this.setupTestContract(defaultAccountAddress);
|
|
152
|
+
|
|
153
|
+
// Recover any pending messages from store (clean up stale ones first)
|
|
154
|
+
await this.store.cleanupOldPendingMessages();
|
|
155
|
+
const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
156
|
+
|
|
157
|
+
// Seed initial L1→L2 messages if pipeline is empty
|
|
158
|
+
const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
|
|
159
|
+
for (let i = 0; i < seedCount; i++) {
|
|
160
|
+
await seedL1ToL2Message(
|
|
161
|
+
l1Client,
|
|
162
|
+
EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()),
|
|
163
|
+
contract.address,
|
|
164
|
+
rollupVersion,
|
|
165
|
+
this.store,
|
|
166
|
+
this.log,
|
|
167
|
+
);
|
|
168
|
+
}
|
|
169
|
+
|
|
170
|
+
// Block until at least one message is ready
|
|
171
|
+
const allMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
172
|
+
if (allMessages.length > 0) {
|
|
173
|
+
this.log.info(`Waiting for first L1→L2 message to be ready...`);
|
|
174
|
+
const firstMsg = allMessages[0];
|
|
175
|
+
await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
|
|
176
|
+
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
177
|
+
});
|
|
178
|
+
this.log.info(`First L1→L2 message is ready`);
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
return {
|
|
182
|
+
wallet: this.wallet,
|
|
183
|
+
defaultAccountAddress,
|
|
184
|
+
contract,
|
|
185
|
+
node: this.aztecNode,
|
|
186
|
+
l1Client,
|
|
187
|
+
rollupVersion,
|
|
188
|
+
};
|
|
189
|
+
}
|
|
190
|
+
|
|
191
|
+
private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
|
|
192
|
+
const deployOpts: DeployOptions = {
|
|
193
|
+
from: deployer,
|
|
194
|
+
contractAddressSalt: this.config.tokenSalt,
|
|
195
|
+
universalDeploy: true,
|
|
196
|
+
};
|
|
197
|
+
const deploy = TestContract.deploy(this.wallet);
|
|
198
|
+
const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
|
|
199
|
+
return TestContract.at(instance.address, this.wallet);
|
|
200
|
+
}
|
|
201
|
+
|
|
99
202
|
/**
|
|
100
203
|
* Checks if the sender account contract is initialized, and initializes it if necessary.
|
|
101
204
|
* @returns The sender wallet.
|
|
@@ -114,14 +217,9 @@ export class BotFactory {
|
|
|
114
217
|
private async setupAccountWithPrivateKey(secret: Fr) {
|
|
115
218
|
const salt = this.config.senderSalt ?? Fr.ONE;
|
|
116
219
|
const signingKey = deriveSigningKey(secret);
|
|
117
|
-
const
|
|
118
|
-
secret,
|
|
119
|
-
salt,
|
|
120
|
-
contract: new SchnorrAccountContract(signingKey!),
|
|
121
|
-
};
|
|
122
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
220
|
+
const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
|
|
123
221
|
const metadata = await this.wallet.getContractMetadata(accountManager.address);
|
|
124
|
-
if (metadata.
|
|
222
|
+
if (metadata.initializationStatus === ContractInitializationStatus.INITIALIZED) {
|
|
125
223
|
this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
|
|
126
224
|
const timer = new Timer();
|
|
127
225
|
const address = accountManager.address;
|
|
@@ -136,13 +234,11 @@ export class BotFactory {
|
|
|
136
234
|
|
|
137
235
|
const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
|
|
138
236
|
const deployMethod = await accountManager.getDeployMethod();
|
|
139
|
-
const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
|
|
140
|
-
const gasSettings = GasSettings.default({ maxFeesPerGas });
|
|
141
237
|
|
|
142
238
|
await this.withNoMinTxsPerBlock(async () => {
|
|
143
|
-
const txHash = await deployMethod.send({
|
|
144
|
-
from:
|
|
145
|
-
fee: {
|
|
239
|
+
const { txHash } = await deployMethod.send({
|
|
240
|
+
from: NO_FROM,
|
|
241
|
+
fee: { paymentMethod },
|
|
146
242
|
wait: NO_WAIT,
|
|
147
243
|
});
|
|
148
244
|
this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
|
|
@@ -159,23 +255,87 @@ export class BotFactory {
|
|
|
159
255
|
|
|
160
256
|
private async setupTestAccount() {
|
|
161
257
|
const [initialAccountData] = await getInitialTestAccountsData();
|
|
162
|
-
const
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
258
|
+
const accountManager = await this.wallet.createSchnorrAccount(
|
|
259
|
+
initialAccountData.secret,
|
|
260
|
+
initialAccountData.salt,
|
|
261
|
+
initialAccountData.signingKey,
|
|
262
|
+
);
|
|
168
263
|
return accountManager.address;
|
|
169
264
|
}
|
|
170
265
|
|
|
266
|
+
/**
|
|
267
|
+
* Setup token and refuel first: if the token already exists (restart scenario),
|
|
268
|
+
* run ensureFeeJuiceBalance before any step that might need fee juice. When deploying,
|
|
269
|
+
* use a bridge claim if balance is below threshold.
|
|
270
|
+
*/
|
|
271
|
+
private async setupTokenWithOptionalEarlyRefuel(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
272
|
+
const token = await this.getTokenInstance(sender);
|
|
273
|
+
const address = token.address;
|
|
274
|
+
const metadata = await this.wallet.getContractMetadata(address);
|
|
275
|
+
if (metadata.isContractPublished) {
|
|
276
|
+
this.log.info(`Token at ${address.toString()} already deployed, refueling before setup`);
|
|
277
|
+
await this.ensureFeeJuiceBalance(sender, token);
|
|
278
|
+
}
|
|
279
|
+
return this.setupToken(sender);
|
|
280
|
+
}
|
|
281
|
+
|
|
282
|
+
/**
|
|
283
|
+
* Setup token0 for AMM with refuel-first behaviour when token already exists.
|
|
284
|
+
*/
|
|
285
|
+
private async setupTokenContractWithOptionalEarlyRefuel(
|
|
286
|
+
deployer: AztecAddress,
|
|
287
|
+
contractAddressSalt: Fr,
|
|
288
|
+
name: string,
|
|
289
|
+
ticker: string,
|
|
290
|
+
decimals = 18,
|
|
291
|
+
): Promise<TokenContract> {
|
|
292
|
+
const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
|
|
293
|
+
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
|
|
294
|
+
const instance = await deploy.getInstance(deployOpts);
|
|
295
|
+
const metadata = await this.wallet.getContractMetadata(instance.address);
|
|
296
|
+
if (metadata.isContractPublished) {
|
|
297
|
+
this.log.info(`Token ${name} at ${instance.address.toString()} already deployed, refueling before setup`);
|
|
298
|
+
const token = TokenContract.at(instance.address, this.wallet);
|
|
299
|
+
await this.ensureFeeJuiceBalance(deployer, token);
|
|
300
|
+
}
|
|
301
|
+
return this.setupTokenContract(deployer, contractAddressSalt, name, ticker, decimals);
|
|
302
|
+
}
|
|
303
|
+
|
|
304
|
+
private async getTokenInstance(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
305
|
+
const deployOpts: DeployOptions = {
|
|
306
|
+
from: sender,
|
|
307
|
+
contractAddressSalt: this.config.tokenSalt,
|
|
308
|
+
universalDeploy: true,
|
|
309
|
+
};
|
|
310
|
+
if (this.config.contract === SupportedTokenContracts.TokenContract) {
|
|
311
|
+
const deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
|
|
312
|
+
const instance = await deploy.getInstance(deployOpts);
|
|
313
|
+
return TokenContract.at(instance.address, this.wallet);
|
|
314
|
+
}
|
|
315
|
+
if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
|
|
316
|
+
const tokenSecretKey = Fr.random();
|
|
317
|
+
const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
|
|
318
|
+
const deploy = PrivateTokenContract.deployWithPublicKeys(tokenPublicKeys, this.wallet, MINT_BALANCE, sender);
|
|
319
|
+
const instance = await deploy.getInstance({
|
|
320
|
+
...deployOpts,
|
|
321
|
+
skipInstancePublication: true,
|
|
322
|
+
skipClassPublication: true,
|
|
323
|
+
skipInitialization: false,
|
|
324
|
+
});
|
|
325
|
+
return PrivateTokenContract.at(instance.address, this.wallet);
|
|
326
|
+
}
|
|
327
|
+
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
328
|
+
}
|
|
329
|
+
|
|
171
330
|
/**
|
|
172
331
|
* Checks if the token contract is deployed and deploys it if necessary.
|
|
173
|
-
*
|
|
174
|
-
* @
|
|
332
|
+
* Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
|
|
333
|
+
* @param sender - Aztec address to deploy the token contract from.
|
|
334
|
+
* @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
|
|
335
|
+
* @returns The TokenContract or PrivateTokenContract instance.
|
|
175
336
|
*/
|
|
176
337
|
private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
177
338
|
let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
|
|
178
|
-
let tokenInstance: ContractInstanceWithAddress | undefined;
|
|
179
339
|
const deployOpts: DeployOptions = {
|
|
180
340
|
from: sender,
|
|
181
341
|
contractAddressSalt: this.config.tokenSalt,
|
|
@@ -184,8 +344,8 @@ export class BotFactory {
|
|
|
184
344
|
let token: TokenContract | PrivateTokenContract;
|
|
185
345
|
if (this.config.contract === SupportedTokenContracts.TokenContract) {
|
|
186
346
|
deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
|
|
187
|
-
|
|
188
|
-
token = TokenContract.at(
|
|
347
|
+
const instance = await deploy.getInstance(deployOpts);
|
|
348
|
+
token = TokenContract.at(instance.address, this.wallet);
|
|
189
349
|
} else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
|
|
190
350
|
// Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
|
|
191
351
|
const tokenSecretKey = Fr.random();
|
|
@@ -196,27 +356,16 @@ export class BotFactory {
|
|
|
196
356
|
deployOpts.skipInitialization = false;
|
|
197
357
|
|
|
198
358
|
// Register the contract with the secret key before deployment
|
|
199
|
-
tokenInstance = await deploy.getInstance(deployOpts);
|
|
359
|
+
const tokenInstance = await deploy.getInstance(deployOpts);
|
|
200
360
|
token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
|
|
201
361
|
await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
|
|
362
|
+
// The contract constructor initializes private storage vars that need the contract's own nullifier key.
|
|
363
|
+
deployOpts.additionalScopes = [tokenInstance.address];
|
|
202
364
|
} else {
|
|
203
365
|
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
204
366
|
}
|
|
205
367
|
|
|
206
|
-
|
|
207
|
-
const metadata = await this.wallet.getContractMetadata(address);
|
|
208
|
-
if (metadata.isContractPublished) {
|
|
209
|
-
this.log.info(`Token at ${address.toString()} already deployed`);
|
|
210
|
-
await deploy.register();
|
|
211
|
-
} else {
|
|
212
|
-
this.log.info(`Deploying token contract at ${address.toString()}`);
|
|
213
|
-
const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
214
|
-
this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
|
|
215
|
-
await this.withNoMinTxsPerBlock(async () => {
|
|
216
|
-
await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
217
|
-
return token;
|
|
218
|
-
});
|
|
219
|
-
}
|
|
368
|
+
await this.registerOrDeployContract('token', deploy, deployOpts);
|
|
220
369
|
return token;
|
|
221
370
|
}
|
|
222
371
|
|
|
@@ -251,9 +400,11 @@ export class BotFactory {
|
|
|
251
400
|
const amm = AMMContract.at(instance.address, this.wallet);
|
|
252
401
|
|
|
253
402
|
this.log.info(`AMM deployed at ${amm.address}`);
|
|
254
|
-
const
|
|
255
|
-
|
|
256
|
-
|
|
403
|
+
const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
|
|
404
|
+
const { receipt: minterReceipt } = await setMinterInteraction.send({
|
|
405
|
+
from: deployer,
|
|
406
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
407
|
+
});
|
|
257
408
|
this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
|
|
258
409
|
this.log.info(`Liquidity token initialized`);
|
|
259
410
|
|
|
@@ -270,9 +421,18 @@ export class BotFactory {
|
|
|
270
421
|
): Promise<void> {
|
|
271
422
|
const getPrivateBalances = () =>
|
|
272
423
|
Promise.all([
|
|
273
|
-
token0.methods
|
|
274
|
-
|
|
275
|
-
|
|
424
|
+
token0.methods
|
|
425
|
+
.balance_of_private(liquidityProvider)
|
|
426
|
+
.simulate({ from: liquidityProvider })
|
|
427
|
+
.then(r => r.result),
|
|
428
|
+
token1.methods
|
|
429
|
+
.balance_of_private(liquidityProvider)
|
|
430
|
+
.simulate({ from: liquidityProvider })
|
|
431
|
+
.then(r => r.result),
|
|
432
|
+
lpToken.methods
|
|
433
|
+
.balance_of_private(liquidityProvider)
|
|
434
|
+
.simulate({ from: liquidityProvider })
|
|
435
|
+
.then(r => r.result),
|
|
276
436
|
]);
|
|
277
437
|
|
|
278
438
|
const authwitNonce = Fr.random();
|
|
@@ -313,20 +473,29 @@ export class BotFactory {
|
|
|
313
473
|
.getFunctionCall(),
|
|
314
474
|
});
|
|
315
475
|
|
|
316
|
-
const
|
|
476
|
+
const mintBatch = new BatchCall(this.wallet, [
|
|
317
477
|
token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
318
478
|
token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
319
|
-
])
|
|
479
|
+
]);
|
|
480
|
+
const { receipt: mintReceipt } = await mintBatch.send({
|
|
481
|
+
from: liquidityProvider,
|
|
482
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
483
|
+
});
|
|
320
484
|
|
|
321
485
|
this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
|
|
322
486
|
|
|
323
|
-
const
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
487
|
+
const addLiquidityInteraction = amm.methods.add_liquidity(
|
|
488
|
+
amount0Max,
|
|
489
|
+
amount1Max,
|
|
490
|
+
amount0Min,
|
|
491
|
+
amount1Min,
|
|
492
|
+
authwitNonce,
|
|
493
|
+
);
|
|
494
|
+
const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
|
|
495
|
+
from: liquidityProvider,
|
|
496
|
+
authWitnesses: [token0Authwit, token1Authwit],
|
|
497
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
498
|
+
});
|
|
330
499
|
|
|
331
500
|
this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
|
|
332
501
|
this.log.info(`Liquidity added`);
|
|
@@ -349,12 +518,42 @@ export class BotFactory {
|
|
|
349
518
|
this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
|
|
350
519
|
await deploy.register();
|
|
351
520
|
} else {
|
|
352
|
-
|
|
353
|
-
await this.
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
521
|
+
const sender = deployOpts.from === NO_FROM ? undefined : deployOpts.from;
|
|
522
|
+
const balance = sender ? await getFeeJuiceBalance(sender, this.aztecNode) : 0n;
|
|
523
|
+
const useClaim =
|
|
524
|
+
sender &&
|
|
525
|
+
balance < FEE_JUICE_TOP_UP_THRESHOLD &&
|
|
526
|
+
this.config.feePaymentMethod === 'fee_juice' &&
|
|
527
|
+
!!this.config.l1RpcUrls?.length;
|
|
528
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
529
|
+
|
|
530
|
+
if (useClaim && mnemonicOrPrivateKey) {
|
|
531
|
+
const claim = await this.getOrCreateBridgeClaim(sender!);
|
|
532
|
+
const paymentMethod = new FeeJuicePaymentMethodWithClaim(sender!, claim);
|
|
533
|
+
const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true, paymentMethod } });
|
|
534
|
+
const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
|
|
535
|
+
const gasSettings = GasSettings.from({
|
|
536
|
+
...estimatedGas!,
|
|
537
|
+
maxFeesPerGas,
|
|
538
|
+
maxPriorityFeesPerGas: GasFees.empty(),
|
|
539
|
+
});
|
|
540
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
541
|
+
const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings, paymentMethod }, wait: NO_WAIT });
|
|
542
|
+
this.log.info(
|
|
543
|
+
`Sent contract ${name} deploy tx ${txHash.toString()} (using bridge claim, balance was ${balance})`,
|
|
544
|
+
);
|
|
545
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
546
|
+
});
|
|
547
|
+
await this.store.deleteBridgeClaim(sender!);
|
|
548
|
+
} else {
|
|
549
|
+
const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
|
|
550
|
+
this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
|
|
551
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
552
|
+
const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
|
|
553
|
+
this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
|
|
554
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
555
|
+
});
|
|
556
|
+
}
|
|
358
557
|
}
|
|
359
558
|
return instance;
|
|
360
559
|
}
|
|
@@ -363,6 +562,66 @@ export class BotFactory {
|
|
|
363
562
|
* Mints private and public tokens for the sender if their balance is below the minimum.
|
|
364
563
|
* @param token - Token contract.
|
|
365
564
|
*/
|
|
565
|
+
/**
|
|
566
|
+
* Ensures the account has sufficient fee juice by bridging from L1 if balance is below threshold.
|
|
567
|
+
* Bridges repeatedly until balance reaches the target (10k FJ).
|
|
568
|
+
* Used on startup/restart to top up when the account has run out after previous runs.
|
|
569
|
+
*/
|
|
570
|
+
private async ensureFeeJuiceBalance(
|
|
571
|
+
account: AztecAddress,
|
|
572
|
+
token: TokenContract | PrivateTokenContract,
|
|
573
|
+
): Promise<void> {
|
|
574
|
+
const { feePaymentMethod, l1RpcUrls } = this.config;
|
|
575
|
+
if (feePaymentMethod !== 'fee_juice' || !l1RpcUrls?.length) {
|
|
576
|
+
return;
|
|
577
|
+
}
|
|
578
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
579
|
+
if (!mnemonicOrPrivateKey) {
|
|
580
|
+
return;
|
|
581
|
+
}
|
|
582
|
+
|
|
583
|
+
let balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
584
|
+
if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
|
|
585
|
+
this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
|
|
586
|
+
return;
|
|
587
|
+
}
|
|
588
|
+
|
|
589
|
+
this.log.info(
|
|
590
|
+
`Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1 until ${FEE_JUICE_TOP_UP_TARGET}`,
|
|
591
|
+
);
|
|
592
|
+
const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
|
|
593
|
+
const minimalInteraction = isStandardTokenContract(token)
|
|
594
|
+
? token.methods.transfer_in_public(account, account, 0n, 0)
|
|
595
|
+
: token.methods.transfer(0n, account, account);
|
|
596
|
+
|
|
597
|
+
while (balance < FEE_JUICE_TOP_UP_TARGET) {
|
|
598
|
+
const claim = await this.bridgeL1FeeJuice(account);
|
|
599
|
+
const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
|
|
600
|
+
const { estimatedGas } = await minimalInteraction.simulate({
|
|
601
|
+
from: account,
|
|
602
|
+
fee: { estimateGas: true, paymentMethod },
|
|
603
|
+
});
|
|
604
|
+
const gasSettings = GasSettings.from({
|
|
605
|
+
...estimatedGas!,
|
|
606
|
+
maxFeesPerGas,
|
|
607
|
+
maxPriorityFeesPerGas: GasFees.empty(),
|
|
608
|
+
});
|
|
609
|
+
|
|
610
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
611
|
+
const { txHash } = await minimalInteraction.send({
|
|
612
|
+
from: account,
|
|
613
|
+
fee: { gasSettings, paymentMethod },
|
|
614
|
+
wait: NO_WAIT,
|
|
615
|
+
});
|
|
616
|
+
this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
|
|
617
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
618
|
+
});
|
|
619
|
+
balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
620
|
+
this.log.info(`Fee juice balance after top-up: ${balance}`);
|
|
621
|
+
}
|
|
622
|
+
this.log.info(`Fee juice top-up complete for ${account.toString()}`);
|
|
623
|
+
}
|
|
624
|
+
|
|
366
625
|
private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
|
|
367
626
|
const isStandardToken = isStandardTokenContract(token);
|
|
368
627
|
let privateBalance = 0n;
|
|
@@ -393,8 +652,15 @@ export class BotFactory {
|
|
|
393
652
|
return;
|
|
394
653
|
}
|
|
395
654
|
|
|
655
|
+
// PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
|
|
656
|
+
const additionalScopes = isStandardToken ? undefined : [token.address];
|
|
657
|
+
const mintBatch = new BatchCall(token.wallet, calls);
|
|
396
658
|
await this.withNoMinTxsPerBlock(async () => {
|
|
397
|
-
const txHash = await
|
|
659
|
+
const { txHash } = await mintBatch.send({
|
|
660
|
+
from: minter,
|
|
661
|
+
additionalScopes,
|
|
662
|
+
wait: NO_WAIT,
|
|
663
|
+
});
|
|
398
664
|
this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
|
|
399
665
|
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
400
666
|
});
|
|
@@ -417,7 +683,6 @@ export class BotFactory {
|
|
|
417
683
|
await this.withNoMinTxsPerBlock(() =>
|
|
418
684
|
waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
|
|
419
685
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
420
|
-
forPublicConsumption: false,
|
|
421
686
|
}),
|
|
422
687
|
);
|
|
423
688
|
return existingClaim.claim;
|
|
@@ -456,7 +721,6 @@ export class BotFactory {
|
|
|
456
721
|
await this.withNoMinTxsPerBlock(() =>
|
|
457
722
|
waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
|
|
458
723
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
459
|
-
forPublicConsumption: false,
|
|
460
724
|
}),
|
|
461
725
|
);
|
|
462
726
|
|
package/src/index.ts
CHANGED
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import { generateClaimSecret } from '@aztec/aztec.js/ethereum';
|
|
2
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
3
|
+
import { compactArray } from '@aztec/foundation/collection';
|
|
4
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
5
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
6
|
+
import type { Logger } from '@aztec/foundation/log';
|
|
7
|
+
import { InboxAbi } from '@aztec/l1-artifacts';
|
|
8
|
+
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
9
|
+
|
|
10
|
+
import { decodeEventLog, getContract } from 'viem';
|
|
11
|
+
|
|
12
|
+
import type { BotStore, PendingL1ToL2Message } from './store/index.js';
|
|
13
|
+
|
|
14
|
+
/** Sends an L1→L2 message via the Inbox contract and stores it. */
|
|
15
|
+
export async function seedL1ToL2Message(
|
|
16
|
+
l1Client: ExtendedViemWalletClient,
|
|
17
|
+
inboxAddress: EthAddress,
|
|
18
|
+
l2Recipient: AztecAddress,
|
|
19
|
+
rollupVersion: bigint,
|
|
20
|
+
store: BotStore,
|
|
21
|
+
log: Logger,
|
|
22
|
+
): Promise<PendingL1ToL2Message> {
|
|
23
|
+
log.info('Seeding L1→L2 message');
|
|
24
|
+
const [secret, secretHash] = await generateClaimSecret(log);
|
|
25
|
+
const content = Fr.random();
|
|
26
|
+
|
|
27
|
+
const inbox = getContract({
|
|
28
|
+
address: inboxAddress.toString(),
|
|
29
|
+
abi: InboxAbi,
|
|
30
|
+
client: l1Client,
|
|
31
|
+
});
|
|
32
|
+
|
|
33
|
+
const txHash = await inbox.write.sendL2Message(
|
|
34
|
+
[{ actor: l2Recipient.toString(), version: rollupVersion }, content.toString(), secretHash.toString()],
|
|
35
|
+
{ gas: 1_000_000n },
|
|
36
|
+
);
|
|
37
|
+
log.info(`L1→L2 message sent in tx ${txHash}`);
|
|
38
|
+
|
|
39
|
+
const txReceipt = await l1Client.waitForTransactionReceipt({ hash: txHash });
|
|
40
|
+
if (txReceipt.status !== 'success') {
|
|
41
|
+
throw new Error(`L1→L2 message tx failed: ${txHash}`);
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
// Extract MessageSent event
|
|
45
|
+
const messageSentLogs = compactArray(
|
|
46
|
+
txReceipt.logs
|
|
47
|
+
.filter(l => l.address.toLowerCase() === inboxAddress.toString().toLowerCase())
|
|
48
|
+
.map(l => {
|
|
49
|
+
try {
|
|
50
|
+
return decodeEventLog({ abi: InboxAbi, eventName: 'MessageSent', data: l.data, topics: l.topics });
|
|
51
|
+
} catch {
|
|
52
|
+
return undefined;
|
|
53
|
+
}
|
|
54
|
+
}),
|
|
55
|
+
);
|
|
56
|
+
|
|
57
|
+
if (messageSentLogs.length !== 1) {
|
|
58
|
+
throw new Error(`Expected 1 MessageSent event, got ${messageSentLogs.length}`);
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
const event = messageSentLogs[0];
|
|
62
|
+
|
|
63
|
+
const msgHash = event.args.hash;
|
|
64
|
+
const globalLeafIndex = event.args.index;
|
|
65
|
+
|
|
66
|
+
const msg: PendingL1ToL2Message = {
|
|
67
|
+
content: content.toString(),
|
|
68
|
+
secret: secret.toString(),
|
|
69
|
+
secretHash: secretHash.toString(),
|
|
70
|
+
msgHash,
|
|
71
|
+
sender: l1Client.account!.address,
|
|
72
|
+
globalLeafIndex: globalLeafIndex.toString(),
|
|
73
|
+
timestamp: Date.now(),
|
|
74
|
+
};
|
|
75
|
+
|
|
76
|
+
await store.savePendingL1ToL2Message(msg);
|
|
77
|
+
log.info(`Seeded L1→L2 message msgHash=${msg.msgHash}`);
|
|
78
|
+
return msg;
|
|
79
|
+
}
|