@aztec/bot 0.0.1-commit.96bb3f7 → 0.0.1-commit.993d240
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 +6 -7
- package/dest/amm_bot.d.ts.map +1 -1
- package/dest/amm_bot.js +28 -17
- package/dest/base_bot.d.ts +7 -7
- package/dest/base_bot.d.ts.map +1 -1
- package/dest/base_bot.js +24 -35
- package/dest/bot.d.ts +6 -6
- package/dest/bot.d.ts.map +1 -1
- package/dest/bot.js +9 -9
- package/dest/config.d.ts +47 -82
- 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 +25 -10
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +365 -114
- package/dest/index.d.ts +2 -1
- package/dest/index.d.ts.map +1 -1
- package/dest/index.js +1 -0
- package/dest/interface.d.ts +2 -6
- package/dest/interface.d.ts.map +1 -1
- package/dest/interface.js +30 -7
- 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 +31 -6
- 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 +18 -15
- package/src/amm_bot.ts +27 -22
- package/src/base_bot.ts +18 -40
- package/src/bot.ts +11 -12
- package/src/config.ts +44 -16
- package/src/cross_chain_bot.ts +203 -0
- package/src/factory.ts +390 -95
- package/src/index.ts +1 -0
- package/src/interface.ts +7 -7
- 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,
|
|
@@ -7,64 +7,80 @@ import {
|
|
|
7
7
|
ContractFunctionInteraction,
|
|
8
8
|
type DeployMethod,
|
|
9
9
|
type DeployOptions,
|
|
10
|
+
NO_WAIT,
|
|
10
11
|
} from '@aztec/aztec.js/contracts';
|
|
11
|
-
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
12
12
|
import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
|
|
13
|
+
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
13
14
|
import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
|
|
14
15
|
import { deriveKeys } from '@aztec/aztec.js/keys';
|
|
15
16
|
import { createLogger } from '@aztec/aztec.js/log';
|
|
16
17
|
import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
|
|
18
|
+
import { waitForTx } from '@aztec/aztec.js/node';
|
|
19
|
+
import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
|
|
20
|
+
import { ContractInitializationStatus } from '@aztec/aztec.js/wallet';
|
|
17
21
|
import { createEthereumChain } from '@aztec/ethereum/chain';
|
|
18
22
|
import { createExtendedL1Client } from '@aztec/ethereum/client';
|
|
23
|
+
import { RollupContract } from '@aztec/ethereum/contracts';
|
|
24
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
19
25
|
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
26
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
20
27
|
import { Timer } from '@aztec/foundation/timer';
|
|
21
28
|
import { AMMContract } from '@aztec/noir-contracts.js/AMM';
|
|
22
29
|
import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
|
|
23
30
|
import { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
31
|
+
import { TestContract } from '@aztec/noir-test-contracts.js/Test';
|
|
24
32
|
import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
|
|
25
|
-
import { GasSettings } from '@aztec/stdlib/gas';
|
|
33
|
+
import { GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
|
|
26
34
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
27
35
|
import { deriveSigningKey } from '@aztec/stdlib/keys';
|
|
28
|
-
import {
|
|
36
|
+
import { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
29
37
|
|
|
30
38
|
import { type BotConfig, SupportedTokenContracts } from './config.js';
|
|
39
|
+
import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
|
|
31
40
|
import type { BotStore } from './store/index.js';
|
|
32
41
|
import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
|
|
33
42
|
|
|
34
43
|
const MINT_BALANCE = 1e12;
|
|
35
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;
|
|
36
47
|
|
|
37
48
|
export class BotFactory {
|
|
38
49
|
private log = createLogger('bot');
|
|
39
50
|
|
|
40
51
|
constructor(
|
|
41
52
|
private readonly config: BotConfig,
|
|
42
|
-
private readonly wallet:
|
|
53
|
+
private readonly wallet: EmbeddedWallet,
|
|
43
54
|
private readonly store: BotStore,
|
|
44
55
|
private readonly aztecNode: AztecNode,
|
|
45
56
|
private readonly aztecNodeAdmin?: AztecNodeAdmin,
|
|
46
|
-
) {
|
|
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
|
+
}
|
|
47
62
|
|
|
48
63
|
/**
|
|
49
64
|
* Initializes a new bot by setting up the sender account, registering the recipient,
|
|
50
65
|
* deploying the token contract, and minting tokens if necessary.
|
|
51
66
|
*/
|
|
52
67
|
public async setup(): Promise<{
|
|
53
|
-
wallet:
|
|
68
|
+
wallet: EmbeddedWallet;
|
|
54
69
|
defaultAccountAddress: AztecAddress;
|
|
55
70
|
token: TokenContract | PrivateTokenContract;
|
|
56
71
|
node: AztecNode;
|
|
57
72
|
recipient: AztecAddress;
|
|
58
73
|
}> {
|
|
59
|
-
const recipient = (await this.wallet.createAccount()).address;
|
|
60
74
|
const defaultAccountAddress = await this.setupAccount();
|
|
61
|
-
const
|
|
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);
|
|
62
78
|
await this.mintTokens(token, defaultAccountAddress);
|
|
63
79
|
return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
|
|
64
80
|
}
|
|
65
81
|
|
|
66
82
|
public async setupAmm(): Promise<{
|
|
67
|
-
wallet:
|
|
83
|
+
wallet: EmbeddedWallet;
|
|
68
84
|
defaultAccountAddress: AztecAddress;
|
|
69
85
|
amm: AMMContract;
|
|
70
86
|
token0: TokenContract;
|
|
@@ -72,7 +88,13 @@ export class BotFactory {
|
|
|
72
88
|
node: AztecNode;
|
|
73
89
|
}> {
|
|
74
90
|
const defaultAccountAddress = await this.setupAccount();
|
|
75
|
-
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);
|
|
76
98
|
const token1 = await this.setupTokenContract(defaultAccountAddress, this.config.tokenSalt, 'BotToken1', 'BOT1');
|
|
77
99
|
const liquidityToken = await this.setupTokenContract(
|
|
78
100
|
defaultAccountAddress,
|
|
@@ -94,6 +116,85 @@ export class BotFactory {
|
|
|
94
116
|
return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
|
|
95
117
|
}
|
|
96
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 = { from: deployer };
|
|
193
|
+
const deploy = TestContract.deploy(this.wallet, { salt: this.config.tokenSalt, universalDeploy: true });
|
|
194
|
+
const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
|
|
195
|
+
return TestContract.at(instance.address, this.wallet);
|
|
196
|
+
}
|
|
197
|
+
|
|
97
198
|
/**
|
|
98
199
|
* Checks if the sender account contract is initialized, and initializes it if necessary.
|
|
99
200
|
* @returns The sender wallet.
|
|
@@ -112,14 +213,9 @@ export class BotFactory {
|
|
|
112
213
|
private async setupAccountWithPrivateKey(secret: Fr) {
|
|
113
214
|
const salt = this.config.senderSalt ?? Fr.ONE;
|
|
114
215
|
const signingKey = deriveSigningKey(secret);
|
|
115
|
-
const
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
contract: new SchnorrAccountContract(signingKey!),
|
|
119
|
-
};
|
|
120
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
121
|
-
const isInit = (await this.wallet.getContractMetadata(accountManager.address)).isContractInitialized;
|
|
122
|
-
if (isInit) {
|
|
216
|
+
const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
|
|
217
|
+
const metadata = await this.wallet.getContractMetadata(accountManager.address);
|
|
218
|
+
if (metadata.initializationStatus === ContractInitializationStatus.INITIALIZED) {
|
|
123
219
|
this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
|
|
124
220
|
const timer = new Timer();
|
|
125
221
|
const address = accountManager.address;
|
|
@@ -134,12 +230,16 @@ export class BotFactory {
|
|
|
134
230
|
|
|
135
231
|
const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
|
|
136
232
|
const deployMethod = await accountManager.getDeployMethod();
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
233
|
+
|
|
234
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
235
|
+
const { txHash } = await deployMethod.send({
|
|
236
|
+
from: NO_FROM,
|
|
237
|
+
fee: { paymentMethod },
|
|
238
|
+
wait: NO_WAIT,
|
|
239
|
+
});
|
|
240
|
+
this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
|
|
241
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
242
|
+
});
|
|
143
243
|
this.log.info(`Account deployed at ${address}`);
|
|
144
244
|
|
|
145
245
|
// Clean up the consumed bridge claim
|
|
@@ -151,57 +251,113 @@ export class BotFactory {
|
|
|
151
251
|
|
|
152
252
|
private async setupTestAccount() {
|
|
153
253
|
const [initialAccountData] = await getInitialTestAccountsData();
|
|
154
|
-
const
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
254
|
+
const accountManager = await this.wallet.createSchnorrAccount(
|
|
255
|
+
initialAccountData.secret,
|
|
256
|
+
initialAccountData.salt,
|
|
257
|
+
initialAccountData.signingKey,
|
|
258
|
+
);
|
|
160
259
|
return accountManager.address;
|
|
161
260
|
}
|
|
162
261
|
|
|
262
|
+
/**
|
|
263
|
+
* Setup token and refuel first: if the token already exists (restart scenario),
|
|
264
|
+
* run ensureFeeJuiceBalance before any step that might need fee juice. When deploying,
|
|
265
|
+
* use a bridge claim if balance is below threshold.
|
|
266
|
+
*/
|
|
267
|
+
private async setupTokenWithOptionalEarlyRefuel(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
268
|
+
const token = await this.getTokenInstance(sender);
|
|
269
|
+
const address = token.address;
|
|
270
|
+
const metadata = await this.wallet.getContractMetadata(address);
|
|
271
|
+
if (metadata.isContractPublished) {
|
|
272
|
+
this.log.info(`Token at ${address.toString()} already deployed, refueling before setup`);
|
|
273
|
+
await this.ensureFeeJuiceBalance(sender, token);
|
|
274
|
+
}
|
|
275
|
+
return this.setupToken(sender);
|
|
276
|
+
}
|
|
277
|
+
|
|
278
|
+
/**
|
|
279
|
+
* Setup token0 for AMM with refuel-first behaviour when token already exists.
|
|
280
|
+
*/
|
|
281
|
+
private async setupTokenContractWithOptionalEarlyRefuel(
|
|
282
|
+
deployer: AztecAddress,
|
|
283
|
+
salt: Fr,
|
|
284
|
+
name: string,
|
|
285
|
+
ticker: string,
|
|
286
|
+
decimals = 18,
|
|
287
|
+
): Promise<TokenContract> {
|
|
288
|
+
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
|
|
289
|
+
const instance = await deploy.getInstance();
|
|
290
|
+
const metadata = await this.wallet.getContractMetadata(instance.address);
|
|
291
|
+
if (metadata.isContractPublished) {
|
|
292
|
+
this.log.info(`Token ${name} at ${instance.address.toString()} already deployed, refueling before setup`);
|
|
293
|
+
const token = TokenContract.at(instance.address, this.wallet);
|
|
294
|
+
await this.ensureFeeJuiceBalance(deployer, token);
|
|
295
|
+
}
|
|
296
|
+
return this.setupTokenContract(deployer, salt, name, ticker, decimals);
|
|
297
|
+
}
|
|
298
|
+
|
|
299
|
+
private async getTokenInstance(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
300
|
+
const salt = this.config.tokenSalt;
|
|
301
|
+
if (this.config.contract === SupportedTokenContracts.TokenContract) {
|
|
302
|
+
const deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
|
|
303
|
+
const instance = await deploy.getInstance();
|
|
304
|
+
return TokenContract.at(instance.address, this.wallet);
|
|
305
|
+
}
|
|
306
|
+
if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
|
|
307
|
+
const tokenSecretKey = Fr.random();
|
|
308
|
+
const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
|
|
309
|
+
const deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
|
|
310
|
+
salt,
|
|
311
|
+
universalDeploy: true,
|
|
312
|
+
publicKeys: tokenPublicKeys,
|
|
313
|
+
});
|
|
314
|
+
const instance = await deploy.getInstance();
|
|
315
|
+
return PrivateTokenContract.at(instance.address, this.wallet);
|
|
316
|
+
}
|
|
317
|
+
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
318
|
+
}
|
|
319
|
+
|
|
163
320
|
/**
|
|
164
321
|
* Checks if the token contract is deployed and deploys it if necessary.
|
|
165
|
-
*
|
|
166
|
-
* @
|
|
322
|
+
* Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
|
|
323
|
+
* @param sender - Aztec address to deploy the token contract from.
|
|
324
|
+
* @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
|
|
325
|
+
* @returns The TokenContract or PrivateTokenContract instance.
|
|
167
326
|
*/
|
|
168
327
|
private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
169
328
|
let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
|
|
170
|
-
|
|
171
|
-
const deployOpts: DeployOptions = {
|
|
172
|
-
|
|
173
|
-
contractAddressSalt: this.config.tokenSalt,
|
|
174
|
-
universalDeploy: true,
|
|
175
|
-
};
|
|
329
|
+
const salt = this.config.tokenSalt;
|
|
330
|
+
const deployOpts: DeployOptions = { from: sender };
|
|
331
|
+
let token: TokenContract | PrivateTokenContract;
|
|
176
332
|
if (this.config.contract === SupportedTokenContracts.TokenContract) {
|
|
177
|
-
deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
|
|
333
|
+
deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
|
|
334
|
+
const instance = await deploy.getInstance();
|
|
335
|
+
token = TokenContract.at(instance.address, this.wallet);
|
|
178
336
|
} else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
|
|
179
337
|
// Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
|
|
180
338
|
const tokenSecretKey = Fr.random();
|
|
181
339
|
const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
|
|
182
|
-
deploy = PrivateTokenContract.
|
|
340
|
+
deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
|
|
341
|
+
salt,
|
|
342
|
+
universalDeploy: true,
|
|
343
|
+
publicKeys: tokenPublicKeys,
|
|
344
|
+
});
|
|
183
345
|
deployOpts.skipInstancePublication = true;
|
|
184
346
|
deployOpts.skipClassPublication = true;
|
|
185
347
|
deployOpts.skipInitialization = false;
|
|
186
348
|
|
|
187
349
|
// Register the contract with the secret key before deployment
|
|
188
|
-
tokenInstance = await deploy.getInstance(
|
|
350
|
+
const tokenInstance = await deploy.getInstance();
|
|
351
|
+
token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
|
|
189
352
|
await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
|
|
353
|
+
// The contract constructor initializes private storage vars that need the contract's own nullifier key.
|
|
354
|
+
deployOpts.additionalScopes = [tokenInstance.address];
|
|
190
355
|
} else {
|
|
191
356
|
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
192
357
|
}
|
|
193
358
|
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
this.log.info(`Token at ${address.toString()} already deployed`);
|
|
197
|
-
return deploy.register();
|
|
198
|
-
} else {
|
|
199
|
-
this.log.info(`Deploying token contract at ${address.toString()}`);
|
|
200
|
-
const sentTx = deploy.send(deployOpts);
|
|
201
|
-
const txHash = await sentTx.getTxHash();
|
|
202
|
-
this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
|
|
203
|
-
return this.withNoMinTxsPerBlock(() => sentTx.deployed({ timeout: this.config.txMinedWaitSeconds }));
|
|
204
|
-
}
|
|
359
|
+
await this.registerOrDeployContract('token', deploy, deployOpts);
|
|
360
|
+
return token;
|
|
205
361
|
}
|
|
206
362
|
|
|
207
363
|
/**
|
|
@@ -209,33 +365,41 @@ export class BotFactory {
|
|
|
209
365
|
* @param wallet - Wallet to deploy the token contract from.
|
|
210
366
|
* @returns The TokenContract instance.
|
|
211
367
|
*/
|
|
212
|
-
private setupTokenContract(
|
|
368
|
+
private async setupTokenContract(
|
|
213
369
|
deployer: AztecAddress,
|
|
214
|
-
|
|
370
|
+
salt: Fr,
|
|
215
371
|
name: string,
|
|
216
372
|
ticker: string,
|
|
217
373
|
decimals = 18,
|
|
218
374
|
): Promise<TokenContract> {
|
|
219
|
-
const deployOpts: DeployOptions = { from: deployer
|
|
220
|
-
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
|
|
221
|
-
|
|
375
|
+
const deployOpts: DeployOptions = { from: deployer };
|
|
376
|
+
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
|
|
377
|
+
const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
|
|
378
|
+
return TokenContract.at(instance.address, this.wallet);
|
|
222
379
|
}
|
|
223
380
|
|
|
224
381
|
private async setupAmmContract(
|
|
225
382
|
deployer: AztecAddress,
|
|
226
|
-
|
|
383
|
+
salt: Fr,
|
|
227
384
|
token0: TokenContract,
|
|
228
385
|
token1: TokenContract,
|
|
229
386
|
lpToken: TokenContract,
|
|
230
387
|
): Promise<AMMContract> {
|
|
231
|
-
const deployOpts: DeployOptions = { from: deployer
|
|
232
|
-
const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address
|
|
233
|
-
|
|
388
|
+
const deployOpts: DeployOptions = { from: deployer };
|
|
389
|
+
const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address, {
|
|
390
|
+
salt,
|
|
391
|
+
universalDeploy: true,
|
|
392
|
+
});
|
|
393
|
+
const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
|
|
394
|
+
const amm = AMMContract.at(instance.address, this.wallet);
|
|
234
395
|
|
|
235
396
|
this.log.info(`AMM deployed at ${amm.address}`);
|
|
236
|
-
const
|
|
237
|
-
|
|
238
|
-
|
|
397
|
+
const setMinterInteraction = lpToken.methods.set_minter(amm.address, true);
|
|
398
|
+
const { receipt: minterReceipt } = await setMinterInteraction.send({
|
|
399
|
+
from: deployer,
|
|
400
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
401
|
+
});
|
|
402
|
+
this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
|
|
239
403
|
this.log.info(`Liquidity token initialized`);
|
|
240
404
|
|
|
241
405
|
return amm;
|
|
@@ -251,9 +415,18 @@ export class BotFactory {
|
|
|
251
415
|
): Promise<void> {
|
|
252
416
|
const getPrivateBalances = () =>
|
|
253
417
|
Promise.all([
|
|
254
|
-
token0.methods
|
|
255
|
-
|
|
256
|
-
|
|
418
|
+
token0.methods
|
|
419
|
+
.balance_of_private(liquidityProvider)
|
|
420
|
+
.simulate({ from: liquidityProvider })
|
|
421
|
+
.then(r => r.result),
|
|
422
|
+
token1.methods
|
|
423
|
+
.balance_of_private(liquidityProvider)
|
|
424
|
+
.simulate({ from: liquidityProvider })
|
|
425
|
+
.then(r => r.result),
|
|
426
|
+
lpToken.methods
|
|
427
|
+
.balance_of_private(liquidityProvider)
|
|
428
|
+
.simulate({ from: liquidityProvider })
|
|
429
|
+
.then(r => r.result),
|
|
257
430
|
]);
|
|
258
431
|
|
|
259
432
|
const authwitNonce = Fr.random();
|
|
@@ -294,23 +467,31 @@ export class BotFactory {
|
|
|
294
467
|
.getFunctionCall(),
|
|
295
468
|
});
|
|
296
469
|
|
|
297
|
-
const
|
|
470
|
+
const mintBatch = new BatchCall(this.wallet, [
|
|
298
471
|
token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
299
472
|
token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
300
|
-
])
|
|
473
|
+
]);
|
|
474
|
+
const { receipt: mintReceipt } = await mintBatch.send({
|
|
475
|
+
from: liquidityProvider,
|
|
476
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
477
|
+
});
|
|
301
478
|
|
|
302
|
-
this.log.info(`Sent mint tx: ${
|
|
303
|
-
await mintTx.wait({ timeout: this.config.txMinedWaitSeconds });
|
|
479
|
+
this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
|
|
304
480
|
|
|
305
|
-
const
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
481
|
+
const addLiquidityInteraction = amm.methods.add_liquidity(
|
|
482
|
+
amount0Max,
|
|
483
|
+
amount1Max,
|
|
484
|
+
amount0Min,
|
|
485
|
+
amount1Min,
|
|
486
|
+
authwitNonce,
|
|
487
|
+
);
|
|
488
|
+
const { receipt: addLiquidityReceipt } = await addLiquidityInteraction.send({
|
|
489
|
+
from: liquidityProvider,
|
|
490
|
+
authWitnesses: [token0Authwit, token1Authwit],
|
|
491
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
492
|
+
});
|
|
311
493
|
|
|
312
|
-
this.log.info(`Sent tx to add liquidity to the AMM: ${
|
|
313
|
-
await addLiquidityTx.wait({ timeout: this.config.txMinedWaitSeconds });
|
|
494
|
+
this.log.info(`Sent tx to add liquidity to the AMM: ${addLiquidityReceipt.txHash.toString()}`);
|
|
314
495
|
this.log.info(`Liquidity added`);
|
|
315
496
|
|
|
316
497
|
const [newT0Bal, newT1Bal, newLPBal] = await getPrivateBalances();
|
|
@@ -323,24 +504,118 @@ export class BotFactory {
|
|
|
323
504
|
name: string,
|
|
324
505
|
deploy: DeployMethod<T>,
|
|
325
506
|
deployOpts: DeployOptions,
|
|
326
|
-
): Promise<
|
|
327
|
-
const
|
|
328
|
-
|
|
507
|
+
): Promise<ContractInstanceWithAddress> {
|
|
508
|
+
const instance = await deploy.getInstance();
|
|
509
|
+
const address = instance.address;
|
|
510
|
+
const metadata = await this.wallet.getContractMetadata(address);
|
|
511
|
+
if (metadata.isContractPublished) {
|
|
329
512
|
this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
|
|
330
|
-
|
|
513
|
+
await deploy.register();
|
|
331
514
|
} else {
|
|
332
|
-
|
|
333
|
-
const
|
|
334
|
-
const
|
|
335
|
-
|
|
336
|
-
|
|
515
|
+
const sender = deployOpts.from === NO_FROM ? undefined : deployOpts.from;
|
|
516
|
+
const balance = sender ? await getFeeJuiceBalance(sender, this.aztecNode) : 0n;
|
|
517
|
+
const useClaim =
|
|
518
|
+
sender &&
|
|
519
|
+
balance < FEE_JUICE_TOP_UP_THRESHOLD &&
|
|
520
|
+
this.config.feePaymentMethod === 'fee_juice' &&
|
|
521
|
+
!!this.config.l1RpcUrls?.length;
|
|
522
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
523
|
+
|
|
524
|
+
if (useClaim && mnemonicOrPrivateKey) {
|
|
525
|
+
const claim = await this.getOrCreateBridgeClaim(sender!);
|
|
526
|
+
const paymentMethod = new FeeJuicePaymentMethodWithClaim(sender!, claim);
|
|
527
|
+
const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true, paymentMethod } });
|
|
528
|
+
const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
|
|
529
|
+
const gasSettings = GasSettings.from({
|
|
530
|
+
...estimatedGas!,
|
|
531
|
+
maxFeesPerGas,
|
|
532
|
+
maxPriorityFeesPerGas: GasFees.empty(),
|
|
533
|
+
});
|
|
534
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
535
|
+
const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings, paymentMethod }, wait: NO_WAIT });
|
|
536
|
+
this.log.info(
|
|
537
|
+
`Sent contract ${name} deploy tx ${txHash.toString()} (using bridge claim, balance was ${balance})`,
|
|
538
|
+
);
|
|
539
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
540
|
+
});
|
|
541
|
+
await this.store.deleteBridgeClaim(sender!);
|
|
542
|
+
} else {
|
|
543
|
+
const { estimatedGas } = await deploy.simulate({ ...deployOpts, fee: { estimateGas: true } });
|
|
544
|
+
this.log.info(`Deploying contract ${name} at ${address.toString()}`, { estimatedGas });
|
|
545
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
546
|
+
const { txHash } = await deploy.send({ ...deployOpts, fee: { gasSettings: estimatedGas }, wait: NO_WAIT });
|
|
547
|
+
this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
|
|
548
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
549
|
+
});
|
|
550
|
+
}
|
|
337
551
|
}
|
|
552
|
+
return instance;
|
|
338
553
|
}
|
|
339
554
|
|
|
340
555
|
/**
|
|
341
556
|
* Mints private and public tokens for the sender if their balance is below the minimum.
|
|
342
557
|
* @param token - Token contract.
|
|
343
558
|
*/
|
|
559
|
+
/**
|
|
560
|
+
* Ensures the account has sufficient fee juice by bridging from L1 if balance is below threshold.
|
|
561
|
+
* Bridges repeatedly until balance reaches the target (10k FJ).
|
|
562
|
+
* Used on startup/restart to top up when the account has run out after previous runs.
|
|
563
|
+
*/
|
|
564
|
+
private async ensureFeeJuiceBalance(
|
|
565
|
+
account: AztecAddress,
|
|
566
|
+
token: TokenContract | PrivateTokenContract,
|
|
567
|
+
): Promise<void> {
|
|
568
|
+
const { feePaymentMethod, l1RpcUrls } = this.config;
|
|
569
|
+
if (feePaymentMethod !== 'fee_juice' || !l1RpcUrls?.length) {
|
|
570
|
+
return;
|
|
571
|
+
}
|
|
572
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
573
|
+
if (!mnemonicOrPrivateKey) {
|
|
574
|
+
return;
|
|
575
|
+
}
|
|
576
|
+
|
|
577
|
+
let balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
578
|
+
if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
|
|
579
|
+
this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
|
|
580
|
+
return;
|
|
581
|
+
}
|
|
582
|
+
|
|
583
|
+
this.log.info(
|
|
584
|
+
`Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1 until ${FEE_JUICE_TOP_UP_TARGET}`,
|
|
585
|
+
);
|
|
586
|
+
const maxFeesPerGas = (await this.getMinFees()).mul(1 + this.config.minFeePadding);
|
|
587
|
+
const minimalInteraction = isStandardTokenContract(token)
|
|
588
|
+
? token.methods.transfer_in_public(account, account, 0n, 0)
|
|
589
|
+
: token.methods.transfer(0n, account, account);
|
|
590
|
+
|
|
591
|
+
while (balance < FEE_JUICE_TOP_UP_TARGET) {
|
|
592
|
+
const claim = await this.bridgeL1FeeJuice(account);
|
|
593
|
+
const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
|
|
594
|
+
const { estimatedGas } = await minimalInteraction.simulate({
|
|
595
|
+
from: account,
|
|
596
|
+
fee: { estimateGas: true, paymentMethod },
|
|
597
|
+
});
|
|
598
|
+
const gasSettings = GasSettings.from({
|
|
599
|
+
...estimatedGas!,
|
|
600
|
+
maxFeesPerGas,
|
|
601
|
+
maxPriorityFeesPerGas: GasFees.empty(),
|
|
602
|
+
});
|
|
603
|
+
|
|
604
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
605
|
+
const { txHash } = await minimalInteraction.send({
|
|
606
|
+
from: account,
|
|
607
|
+
fee: { gasSettings, paymentMethod },
|
|
608
|
+
wait: NO_WAIT,
|
|
609
|
+
});
|
|
610
|
+
this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
|
|
611
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
612
|
+
});
|
|
613
|
+
balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
614
|
+
this.log.info(`Fee juice balance after top-up: ${balance}`);
|
|
615
|
+
}
|
|
616
|
+
this.log.info(`Fee juice top-up complete for ${account.toString()}`);
|
|
617
|
+
}
|
|
618
|
+
|
|
344
619
|
private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
|
|
345
620
|
const isStandardToken = isStandardTokenContract(token);
|
|
346
621
|
let privateBalance = 0n;
|
|
@@ -370,10 +645,19 @@ export class BotFactory {
|
|
|
370
645
|
this.log.info(`Skipping minting as ${minter.toString()} has enough tokens`);
|
|
371
646
|
return;
|
|
372
647
|
}
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
648
|
+
|
|
649
|
+
// PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
|
|
650
|
+
const additionalScopes = isStandardToken ? undefined : [token.address];
|
|
651
|
+
const mintBatch = new BatchCall(token.wallet, calls);
|
|
652
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
653
|
+
const { txHash } = await mintBatch.send({
|
|
654
|
+
from: minter,
|
|
655
|
+
additionalScopes,
|
|
656
|
+
wait: NO_WAIT,
|
|
657
|
+
});
|
|
658
|
+
this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
|
|
659
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
660
|
+
});
|
|
377
661
|
}
|
|
378
662
|
|
|
379
663
|
/**
|
|
@@ -393,7 +677,6 @@ export class BotFactory {
|
|
|
393
677
|
await this.withNoMinTxsPerBlock(() =>
|
|
394
678
|
waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
|
|
395
679
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
396
|
-
forPublicConsumption: false,
|
|
397
680
|
}),
|
|
398
681
|
);
|
|
399
682
|
return existingClaim.claim;
|
|
@@ -432,7 +715,6 @@ export class BotFactory {
|
|
|
432
715
|
await this.withNoMinTxsPerBlock(() =>
|
|
433
716
|
waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
|
|
434
717
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
435
|
-
forPublicConsumption: false,
|
|
436
718
|
}),
|
|
437
719
|
);
|
|
438
720
|
|
|
@@ -441,6 +723,19 @@ export class BotFactory {
|
|
|
441
723
|
return claim as L2AmountClaim;
|
|
442
724
|
}
|
|
443
725
|
|
|
726
|
+
/** Returns worst-case min fees across predicted slots, with fallback to current min fees. */
|
|
727
|
+
private async getMinFees(): Promise<GasFees> {
|
|
728
|
+
try {
|
|
729
|
+
const predicted = await this.aztecNode.getPredictedMinFees(ManaUsageEstimate.Limit);
|
|
730
|
+
if (predicted.length === 0) {
|
|
731
|
+
return this.aztecNode.getCurrentMinFees();
|
|
732
|
+
}
|
|
733
|
+
return predicted.reduce((worst, fees) => (fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst));
|
|
734
|
+
} catch {
|
|
735
|
+
return this.aztecNode.getCurrentMinFees();
|
|
736
|
+
}
|
|
737
|
+
}
|
|
738
|
+
|
|
444
739
|
private async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
|
|
445
740
|
if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
|
|
446
741
|
this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
|