@aztec/bot 0.0.1-commit.96dac018d → 0.0.1-commit.9a89641
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 -3
- package/dest/amm_bot.d.ts.map +1 -1
- package/dest/amm_bot.js +26 -19
- package/dest/base_bot.d.ts +4 -4
- package/dest/base_bot.d.ts.map +1 -1
- package/dest/base_bot.js +12 -22
- package/dest/bot.d.ts +3 -2
- package/dest/bot.d.ts.map +1 -1
- package/dest/bot.js +5 -8
- package/dest/config.d.ts +31 -82
- package/dest/config.d.ts.map +1 -1
- package/dest/config.js +6 -7
- package/dest/cross_chain_bot.d.ts +6 -4
- package/dest/cross_chain_bot.d.ts.map +1 -1
- package/dest/cross_chain_bot.js +16 -18
- package/dest/factory.d.ts +17 -7
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +235 -177
- package/dest/interface.d.ts +2 -6
- package/dest/interface.d.ts.map +1 -1
- package/dest/interface.js +30 -7
- package/dest/runner.d.ts +3 -2
- package/dest/runner.d.ts.map +1 -1
- package/dest/runner.js +6 -4
- package/dest/utils.js +3 -3
- package/package.json +16 -16
- package/src/amm_bot.ts +25 -17
- package/src/base_bot.ts +10 -17
- package/src/bot.ts +6 -5
- package/src/config.ts +8 -8
- package/src/cross_chain_bot.ts +14 -15
- package/src/factory.ts +242 -187
- package/src/interface.ts +7 -7
- package/src/runner.ts +26 -3
- package/src/utils.ts +3 -3
package/src/factory.ts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { getInitialTestAccountsData } from '@aztec/accounts/testing';
|
|
2
|
+
import { deriveSecretKeyFromSigningKey } from '@aztec/accounts/utils';
|
|
2
3
|
import { AztecAddress } from '@aztec/aztec.js/addresses';
|
|
3
4
|
import {
|
|
4
5
|
BatchCall,
|
|
@@ -15,21 +16,21 @@ 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';
|
|
17
18
|
import { waitForTx } from '@aztec/aztec.js/node';
|
|
19
|
+
import { getFeeJuiceBalance } from '@aztec/aztec.js/utils';
|
|
18
20
|
import { createEthereumChain } from '@aztec/ethereum/chain';
|
|
19
21
|
import { createExtendedL1Client } from '@aztec/ethereum/client';
|
|
20
22
|
import { RollupContract } from '@aztec/ethereum/contracts';
|
|
21
23
|
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
22
24
|
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
25
|
+
import { GrumpkinScalar } from '@aztec/foundation/curves/grumpkin';
|
|
23
26
|
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
24
|
-
import { Timer } from '@aztec/foundation/timer';
|
|
25
27
|
import { AMMContract } from '@aztec/noir-contracts.js/AMM';
|
|
26
28
|
import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
|
|
27
29
|
import { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
28
30
|
import { TestContract } from '@aztec/noir-test-contracts.js/Test';
|
|
31
|
+
import type { BlockTag } from '@aztec/stdlib/block';
|
|
29
32
|
import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
|
|
30
|
-
import { GasSettings } from '@aztec/stdlib/gas';
|
|
31
33
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
32
|
-
import { deriveSigningKey } from '@aztec/stdlib/keys';
|
|
33
34
|
import { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
34
35
|
|
|
35
36
|
import { type BotConfig, SupportedTokenContracts } from './config.js';
|
|
@@ -39,17 +40,28 @@ import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils
|
|
|
39
40
|
|
|
40
41
|
const MINT_BALANCE = 1e12;
|
|
41
42
|
const MIN_BALANCE = 1e3;
|
|
43
|
+
const FEE_JUICE_TOP_UP_THRESHOLD = 100n * 10n ** 18n;
|
|
42
44
|
|
|
43
45
|
export class BotFactory {
|
|
44
46
|
private log = createLogger('bot');
|
|
45
47
|
|
|
48
|
+
/** Number of in-flight withNoMinTxsPerBlock calls; see that method for why they are counted. */
|
|
49
|
+
private noMinTxsPerBlockDepth = 0;
|
|
50
|
+
/** Set by the first withNoMinTxsPerBlock entrant; resolves to the minTxsPerBlock value to restore. */
|
|
51
|
+
private savedMinTxsPerBlock?: Promise<{ minTxsPerBlock?: number }>;
|
|
52
|
+
|
|
46
53
|
constructor(
|
|
47
54
|
private readonly config: BotConfig,
|
|
48
55
|
private readonly wallet: EmbeddedWallet,
|
|
49
56
|
private readonly store: BotStore,
|
|
50
57
|
private readonly aztecNode: AztecNode,
|
|
51
58
|
private readonly aztecNodeAdmin?: AztecNodeAdmin,
|
|
52
|
-
|
|
59
|
+
private readonly syncChainTip?: BlockTag,
|
|
60
|
+
) {
|
|
61
|
+
// Set fee padding on the wallet so that all transactions during setup
|
|
62
|
+
// (token deploy, minting, etc.) use the configured padding, not the default.
|
|
63
|
+
this.wallet.setMinFeePadding(config.minFeePadding);
|
|
64
|
+
}
|
|
53
65
|
|
|
54
66
|
/**
|
|
55
67
|
* Initializes a new bot by setting up the sender account, registering the recipient,
|
|
@@ -63,7 +75,9 @@ export class BotFactory {
|
|
|
63
75
|
recipient: AztecAddress;
|
|
64
76
|
}> {
|
|
65
77
|
const defaultAccountAddress = await this.setupAccount();
|
|
66
|
-
const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random()))
|
|
78
|
+
const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random(), GrumpkinScalar.random()))
|
|
79
|
+
.address;
|
|
80
|
+
await this.ensureFeeJuiceBalance(defaultAccountAddress);
|
|
67
81
|
const token = await this.setupToken(defaultAccountAddress);
|
|
68
82
|
await this.mintTokens(token, defaultAccountAddress);
|
|
69
83
|
return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
|
|
@@ -78,23 +92,36 @@ export class BotFactory {
|
|
|
78
92
|
node: AztecNode;
|
|
79
93
|
}> {
|
|
80
94
|
const defaultAccountAddress = await this.setupAccount();
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
const
|
|
84
|
-
defaultAccountAddress,
|
|
85
|
-
this.config.tokenSalt,
|
|
86
|
-
'BotLPToken',
|
|
87
|
-
'BOTLP',
|
|
88
|
-
);
|
|
89
|
-
const amm = await this.setupAmmContract(
|
|
90
|
-
defaultAccountAddress,
|
|
91
|
-
this.config.tokenSalt,
|
|
92
|
-
token0,
|
|
93
|
-
token1,
|
|
94
|
-
liquidityToken,
|
|
95
|
-
);
|
|
95
|
+
await this.ensureFeeJuiceBalance(defaultAccountAddress);
|
|
96
|
+
|
|
97
|
+
const salt = this.config.tokenSalt;
|
|
96
98
|
|
|
97
|
-
|
|
99
|
+
// token0, token1 and the LP token are independent contracts with no shared state, so deploy them
|
|
100
|
+
// concurrently rather than one slot at a time.
|
|
101
|
+
const [token0, token1, liquidityToken] = await Promise.all([
|
|
102
|
+
this.setupTokenContract(defaultAccountAddress, salt, 'BotToken0', 'BOT0'),
|
|
103
|
+
this.setupTokenContract(defaultAccountAddress, salt, 'BotToken1', 'BOT1'),
|
|
104
|
+
this.setupTokenContract(defaultAccountAddress, salt, 'BotLPToken', 'BOTLP'),
|
|
105
|
+
]);
|
|
106
|
+
|
|
107
|
+
const ammDeploy = AMMContract.deploy(this.wallet, token0.address, token1.address, liquidityToken.address, {
|
|
108
|
+
salt,
|
|
109
|
+
universalDeploy: true,
|
|
110
|
+
});
|
|
111
|
+
const ammAddress = (await ammDeploy.getInstance()).address;
|
|
112
|
+
|
|
113
|
+
// The AMM constructor only stores the (already-derived) token addresses, and set_minter only records
|
|
114
|
+
// the AMM address on the LP token: neither reads the other's on-chain state, so the AMM deploy, the
|
|
115
|
+
// LP-minter grant, and the token0/token1 mints are mutually independent and run concurrently.
|
|
116
|
+
const [amm] = await Promise.all([
|
|
117
|
+
this.deployAmmContract(defaultAccountAddress, ammDeploy),
|
|
118
|
+
this.grantLpTokenMinter(defaultAccountAddress, liquidityToken, ammAddress),
|
|
119
|
+
this.mintAmmLiquidity(defaultAccountAddress, token0, token1),
|
|
120
|
+
]);
|
|
121
|
+
|
|
122
|
+
// add_liquidity spends the minted token0/token1 balances and mints LP tokens, so it must follow both
|
|
123
|
+
// the mints and the minter grant, and target the deployed AMM.
|
|
124
|
+
await this.addAmmLiquidity(defaultAccountAddress, defaultAccountAddress, amm, token0, token1, liquidityToken);
|
|
98
125
|
this.log.info(`AMM initialized and funded`);
|
|
99
126
|
|
|
100
127
|
return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
|
|
@@ -113,6 +140,7 @@ export class BotFactory {
|
|
|
113
140
|
rollupVersion: bigint;
|
|
114
141
|
}> {
|
|
115
142
|
const defaultAccountAddress = await this.setupAccount();
|
|
143
|
+
await this.ensureFeeJuiceBalance(defaultAccountAddress);
|
|
116
144
|
|
|
117
145
|
// Create L1 client (same pattern as bridgeL1FeeJuice)
|
|
118
146
|
const l1RpcUrls = this.config.l1RpcUrls;
|
|
@@ -131,25 +159,32 @@ export class BotFactory {
|
|
|
131
159
|
const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
|
|
132
160
|
const rollupVersion = await rollupContract.getVersion();
|
|
133
161
|
|
|
134
|
-
//
|
|
135
|
-
|
|
162
|
+
// Derive the TestContract address up front (deterministic from the salt). Seeding L1→L2 messages only
|
|
163
|
+
// needs the L2 recipient address — the messages are queued on L1 and don't require the L2 contract to
|
|
164
|
+
// exist yet (they're consumed later, after setup completes) — so the deploy (an L2 tx paying from the
|
|
165
|
+
// standing balance funded above) and the L1 seeding run concurrently.
|
|
166
|
+
const testContractDeploy = TestContract.deploy(this.wallet, {
|
|
167
|
+
salt: this.config.tokenSalt,
|
|
168
|
+
universalDeploy: true,
|
|
169
|
+
});
|
|
170
|
+
const contractAddress = (await testContractDeploy.getInstance()).address;
|
|
136
171
|
|
|
137
172
|
// Recover any pending messages from store (clean up stale ones first)
|
|
138
173
|
await this.store.cleanupOldPendingMessages();
|
|
139
174
|
const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
140
175
|
|
|
141
|
-
// Seed initial L1→L2 messages if pipeline is empty
|
|
176
|
+
// Seed initial L1→L2 messages if pipeline is empty. The seeds are sent one at a time: they share the
|
|
177
|
+
// bot's L1 account, so concurrent sends would race on the L1 nonce.
|
|
142
178
|
const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
}
|
|
179
|
+
const inboxAddress = EthAddress.fromString(l1ContractAddresses.inboxAddress.toString());
|
|
180
|
+
const [contract] = await Promise.all([
|
|
181
|
+
this.deployTestContract(defaultAccountAddress, testContractDeploy),
|
|
182
|
+
(async () => {
|
|
183
|
+
for (let i = 0; i < seedCount; i++) {
|
|
184
|
+
await seedL1ToL2Message(l1Client, inboxAddress, contractAddress, rollupVersion, this.store, this.log);
|
|
185
|
+
}
|
|
186
|
+
})(),
|
|
187
|
+
]);
|
|
153
188
|
|
|
154
189
|
// Block until at least one message is ready
|
|
155
190
|
const allMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
@@ -158,11 +193,7 @@ export class BotFactory {
|
|
|
158
193
|
const firstMsg = allMessages[0];
|
|
159
194
|
await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
|
|
160
195
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
161
|
-
|
|
162
|
-
// state. With true, it returns one block early which causes gas estimation simulation to
|
|
163
|
-
// fail since it runs against the current state.
|
|
164
|
-
// See https://linear.app/aztec-labs/issue/A-548 for details.
|
|
165
|
-
forPublicConsumption: false,
|
|
196
|
+
chainTip: this.syncChainTip,
|
|
166
197
|
});
|
|
167
198
|
this.log.info(`First L1→L2 message is ready`);
|
|
168
199
|
}
|
|
@@ -177,14 +208,8 @@ export class BotFactory {
|
|
|
177
208
|
};
|
|
178
209
|
}
|
|
179
210
|
|
|
180
|
-
private async
|
|
181
|
-
const
|
|
182
|
-
from: deployer,
|
|
183
|
-
contractAddressSalt: this.config.tokenSalt,
|
|
184
|
-
universalDeploy: true,
|
|
185
|
-
};
|
|
186
|
-
const deploy = TestContract.deploy(this.wallet);
|
|
187
|
-
const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
|
|
211
|
+
private async deployTestContract(deployer: AztecAddress, deploy: DeployMethod<TestContract>): Promise<TestContract> {
|
|
212
|
+
const instance = await this.registerOrDeployContract('TestContract', deploy, { from: deployer });
|
|
188
213
|
return TestContract.at(instance.address, this.wallet);
|
|
189
214
|
}
|
|
190
215
|
|
|
@@ -203,50 +228,16 @@ export class BotFactory {
|
|
|
203
228
|
}
|
|
204
229
|
}
|
|
205
230
|
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
const timer = new Timer();
|
|
214
|
-
const address = accountManager.address;
|
|
215
|
-
this.log.info(`Account at ${address} registered. duration=${timer.ms()}`);
|
|
216
|
-
await this.store.deleteBridgeClaim(address);
|
|
217
|
-
return address;
|
|
218
|
-
} else {
|
|
219
|
-
const address = accountManager.address;
|
|
220
|
-
this.log.info(`Deploying account at ${address}`);
|
|
221
|
-
|
|
222
|
-
const claim = await this.getOrCreateBridgeClaim(address);
|
|
223
|
-
|
|
224
|
-
const paymentMethod = new FeeJuicePaymentMethodWithClaim(accountManager.address, claim);
|
|
225
|
-
const deployMethod = await accountManager.getDeployMethod();
|
|
226
|
-
const maxFeesPerGas = (await this.aztecNode.getCurrentMinFees()).mul(1 + this.config.minFeePadding);
|
|
227
|
-
const gasSettings = GasSettings.default({ maxFeesPerGas });
|
|
228
|
-
|
|
229
|
-
await this.withNoMinTxsPerBlock(async () => {
|
|
230
|
-
const txHash = await deployMethod.send({
|
|
231
|
-
from: AztecAddress.ZERO,
|
|
232
|
-
fee: { gasSettings, paymentMethod },
|
|
233
|
-
wait: NO_WAIT,
|
|
234
|
-
});
|
|
235
|
-
this.log.info(`Sent tx for account deployment with hash ${txHash.toString()}`);
|
|
236
|
-
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
237
|
-
});
|
|
238
|
-
this.log.info(`Account deployed at ${address}`);
|
|
239
|
-
|
|
240
|
-
// Clean up the consumed bridge claim
|
|
241
|
-
await this.store.deleteBridgeClaim(address);
|
|
242
|
-
|
|
243
|
-
return accountManager.address;
|
|
244
|
-
}
|
|
245
|
-
}
|
|
246
|
-
|
|
231
|
+
/**
|
|
232
|
+
* Keyless fallback for tests and local dev: reuses the first genesis test account, whose address is
|
|
233
|
+
* pre-funded with fee juice via `initialFundedAccounts`. The test accounts are initializerless, so this
|
|
234
|
+
* must create an initializerless account for the address to match the funded one. Production bots set a
|
|
235
|
+
* sender private key and fund the resulting initializerless account from L1 instead; see
|
|
236
|
+
* setupAccountWithPrivateKey.
|
|
237
|
+
*/
|
|
247
238
|
private async setupTestAccount() {
|
|
248
239
|
const [initialAccountData] = await getInitialTestAccountsData();
|
|
249
|
-
const accountManager = await this.wallet.
|
|
240
|
+
const accountManager = await this.wallet.createSchnorrInitializerlessAccount(
|
|
250
241
|
initialAccountData.secret,
|
|
251
242
|
initialAccountData.salt,
|
|
252
243
|
initialAccountData.signingKey,
|
|
@@ -254,35 +245,45 @@ export class BotFactory {
|
|
|
254
245
|
return accountManager.address;
|
|
255
246
|
}
|
|
256
247
|
|
|
248
|
+
private async setupAccountWithPrivateKey(privateKey: Fr) {
|
|
249
|
+
const salt = this.config.senderSalt ?? Fr.ONE;
|
|
250
|
+
const signingKey = GrumpkinScalar.fromBuffer(privateKey.toBuffer());
|
|
251
|
+
const secret = await deriveSecretKeyFromSigningKey(signingKey);
|
|
252
|
+
const accountManager = await this.wallet.createSchnorrInitializerlessAccount(secret, salt, signingKey);
|
|
253
|
+
return accountManager.address;
|
|
254
|
+
}
|
|
255
|
+
|
|
257
256
|
/**
|
|
258
257
|
* Checks if the token contract is deployed and deploys it if necessary.
|
|
259
|
-
*
|
|
260
|
-
* @
|
|
258
|
+
* Uses a bridge claim for deploy when balance is below threshold to avoid failing before refuel.
|
|
259
|
+
* @param sender - Aztec address to deploy the token contract from.
|
|
260
|
+
* @param existingToken - Optional token instance when called from setupTokenWithOptionalEarlyRefuel.
|
|
261
|
+
* @returns The TokenContract or PrivateTokenContract instance.
|
|
261
262
|
*/
|
|
262
263
|
private async setupToken(sender: AztecAddress): Promise<TokenContract | PrivateTokenContract> {
|
|
263
264
|
let deploy: DeployMethod<TokenContract | PrivateTokenContract>;
|
|
264
|
-
|
|
265
|
-
const deployOpts: DeployOptions = {
|
|
266
|
-
from: sender,
|
|
267
|
-
contractAddressSalt: this.config.tokenSalt,
|
|
268
|
-
universalDeploy: true,
|
|
269
|
-
};
|
|
265
|
+
const salt = this.config.tokenSalt;
|
|
266
|
+
const deployOpts: DeployOptions = { from: sender };
|
|
270
267
|
let token: TokenContract | PrivateTokenContract;
|
|
271
268
|
if (this.config.contract === SupportedTokenContracts.TokenContract) {
|
|
272
|
-
deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18);
|
|
273
|
-
|
|
274
|
-
token = TokenContract.at(
|
|
269
|
+
deploy = TokenContract.deploy(this.wallet, sender, 'BotToken', 'BOT', 18, { salt, universalDeploy: true });
|
|
270
|
+
const instance = await deploy.getInstance();
|
|
271
|
+
token = TokenContract.at(instance.address, this.wallet);
|
|
275
272
|
} else if (this.config.contract === SupportedTokenContracts.PrivateTokenContract) {
|
|
276
273
|
// Generate keys for the contract since PrivateToken uses SinglePrivateMutable which requires keys
|
|
277
274
|
const tokenSecretKey = Fr.random();
|
|
278
275
|
const tokenPublicKeys = (await deriveKeys(tokenSecretKey)).publicKeys;
|
|
279
|
-
deploy = PrivateTokenContract.
|
|
276
|
+
deploy = PrivateTokenContract.deploy(this.wallet, MINT_BALANCE, sender, {
|
|
277
|
+
salt,
|
|
278
|
+
universalDeploy: true,
|
|
279
|
+
publicKeys: tokenPublicKeys,
|
|
280
|
+
});
|
|
280
281
|
deployOpts.skipInstancePublication = true;
|
|
281
282
|
deployOpts.skipClassPublication = true;
|
|
282
283
|
deployOpts.skipInitialization = false;
|
|
283
284
|
|
|
284
285
|
// Register the contract with the secret key before deployment
|
|
285
|
-
tokenInstance = await deploy.getInstance(
|
|
286
|
+
const tokenInstance = await deploy.getInstance();
|
|
286
287
|
token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
|
|
287
288
|
await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
|
|
288
289
|
// The contract constructor initializes private storage vars that need the contract's own nullifier key.
|
|
@@ -291,20 +292,7 @@ export class BotFactory {
|
|
|
291
292
|
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
292
293
|
}
|
|
293
294
|
|
|
294
|
-
|
|
295
|
-
const metadata = await this.wallet.getContractMetadata(address);
|
|
296
|
-
if (metadata.isContractPublished) {
|
|
297
|
-
this.log.info(`Token at ${address.toString()} already deployed`);
|
|
298
|
-
await deploy.register();
|
|
299
|
-
} else {
|
|
300
|
-
this.log.info(`Deploying token contract at ${address.toString()}`);
|
|
301
|
-
const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
302
|
-
this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
|
|
303
|
-
await this.withNoMinTxsPerBlock(async () => {
|
|
304
|
-
await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
305
|
-
return token;
|
|
306
|
-
});
|
|
307
|
-
}
|
|
295
|
+
await this.registerOrDeployContract('token', deploy, deployOpts);
|
|
308
296
|
return token;
|
|
309
297
|
}
|
|
310
298
|
|
|
@@ -315,40 +303,48 @@ export class BotFactory {
|
|
|
315
303
|
*/
|
|
316
304
|
private async setupTokenContract(
|
|
317
305
|
deployer: AztecAddress,
|
|
318
|
-
|
|
306
|
+
salt: Fr,
|
|
319
307
|
name: string,
|
|
320
308
|
ticker: string,
|
|
321
309
|
decimals = 18,
|
|
322
310
|
): Promise<TokenContract> {
|
|
323
|
-
const deployOpts: DeployOptions = { from: deployer
|
|
324
|
-
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals);
|
|
311
|
+
const deployOpts: DeployOptions = { from: deployer };
|
|
312
|
+
const deploy = TokenContract.deploy(this.wallet, deployer, name, ticker, decimals, { salt, universalDeploy: true });
|
|
325
313
|
const instance = await this.registerOrDeployContract('Token - ' + name, deploy, deployOpts);
|
|
326
314
|
return TokenContract.at(instance.address, this.wallet);
|
|
327
315
|
}
|
|
328
316
|
|
|
329
|
-
private async
|
|
330
|
-
|
|
331
|
-
contractAddressSalt: Fr,
|
|
332
|
-
token0: TokenContract,
|
|
333
|
-
token1: TokenContract,
|
|
334
|
-
lpToken: TokenContract,
|
|
335
|
-
): Promise<AMMContract> {
|
|
336
|
-
const deployOpts: DeployOptions = { from: deployer, contractAddressSalt, universalDeploy: true };
|
|
337
|
-
const deploy = AMMContract.deploy(this.wallet, token0.address, token1.address, lpToken.address);
|
|
338
|
-
const instance = await this.registerOrDeployContract('AMM', deploy, deployOpts);
|
|
317
|
+
private async deployAmmContract(deployer: AztecAddress, deploy: DeployMethod<AMMContract>): Promise<AMMContract> {
|
|
318
|
+
const instance = await this.registerOrDeployContract('AMM', deploy, { from: deployer });
|
|
339
319
|
const amm = AMMContract.at(instance.address, this.wallet);
|
|
340
|
-
|
|
341
320
|
this.log.info(`AMM deployed at ${amm.address}`);
|
|
342
|
-
const minterReceipt = await lpToken.methods
|
|
343
|
-
.set_minter(amm.address, true)
|
|
344
|
-
.send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
|
|
345
|
-
this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
|
|
346
|
-
this.log.info(`Liquidity token initialized`);
|
|
347
|
-
|
|
348
321
|
return amm;
|
|
349
322
|
}
|
|
350
323
|
|
|
351
|
-
|
|
324
|
+
/** Grants the AMM minting rights over the LP token. set_minter only records the address, so it does not
|
|
325
|
+
* require the AMM contract to be deployed first. */
|
|
326
|
+
private async grantLpTokenMinter(deployer: AztecAddress, lpToken: TokenContract, amm: AztecAddress): Promise<void> {
|
|
327
|
+
const { receipt } = await lpToken.methods.set_minter(amm, true).send({
|
|
328
|
+
from: deployer,
|
|
329
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
330
|
+
});
|
|
331
|
+
this.log.info(`Set LP token minter to AMM txHash=${receipt.txHash.toString()}`);
|
|
332
|
+
}
|
|
333
|
+
|
|
334
|
+
private async mintAmmLiquidity(minter: AztecAddress, token0: TokenContract, token1: TokenContract): Promise<void> {
|
|
335
|
+
this.log.info(`Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1 for ${minter}`);
|
|
336
|
+
const mintBatch = new BatchCall(this.wallet, [
|
|
337
|
+
token0.methods.mint_to_private(minter, MINT_BALANCE),
|
|
338
|
+
token1.methods.mint_to_private(minter, MINT_BALANCE),
|
|
339
|
+
]);
|
|
340
|
+
const { receipt } = await mintBatch.send({
|
|
341
|
+
from: minter,
|
|
342
|
+
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
343
|
+
});
|
|
344
|
+
this.log.info(`Sent mint tx: ${receipt.txHash.toString()}`);
|
|
345
|
+
}
|
|
346
|
+
|
|
347
|
+
private async addAmmLiquidity(
|
|
352
348
|
defaultAccountAddress: AztecAddress,
|
|
353
349
|
liquidityProvider: AztecAddress,
|
|
354
350
|
amm: AMMContract,
|
|
@@ -356,13 +352,6 @@ export class BotFactory {
|
|
|
356
352
|
token1: TokenContract,
|
|
357
353
|
lpToken: TokenContract,
|
|
358
354
|
): Promise<void> {
|
|
359
|
-
const getPrivateBalances = () =>
|
|
360
|
-
Promise.all([
|
|
361
|
-
token0.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
|
|
362
|
-
token1.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
|
|
363
|
-
lpToken.methods.balance_of_private(liquidityProvider).simulate({ from: liquidityProvider }),
|
|
364
|
-
]);
|
|
365
|
-
|
|
366
355
|
const authwitNonce = Fr.random();
|
|
367
356
|
|
|
368
357
|
// keep some tokens for swapping
|
|
@@ -371,12 +360,6 @@ export class BotFactory {
|
|
|
371
360
|
const amount1Max = MINT_BALANCE / 2;
|
|
372
361
|
const amount1Min = MINT_BALANCE / 4;
|
|
373
362
|
|
|
374
|
-
const [t0Bal, t1Bal, lpBal] = await getPrivateBalances();
|
|
375
|
-
|
|
376
|
-
this.log.info(
|
|
377
|
-
`Minting ${MINT_BALANCE} tokens of each BotToken0 and BotToken1. Current private balances of ${liquidityProvider}: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`,
|
|
378
|
-
);
|
|
379
|
-
|
|
380
363
|
// Add authwitnesses for the transfers in AMM::add_liquidity function
|
|
381
364
|
const token0Authwit = await this.wallet.createAuthWit(defaultAccountAddress, {
|
|
382
365
|
caller: amm.address,
|
|
@@ -401,14 +384,7 @@ export class BotFactory {
|
|
|
401
384
|
.getFunctionCall(),
|
|
402
385
|
});
|
|
403
386
|
|
|
404
|
-
const
|
|
405
|
-
token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
406
|
-
token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
407
|
-
]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
|
|
408
|
-
|
|
409
|
-
this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
|
|
410
|
-
|
|
411
|
-
const addLiquidityReceipt = await amm.methods
|
|
387
|
+
const { receipt } = await amm.methods
|
|
412
388
|
.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
|
|
413
389
|
.send({
|
|
414
390
|
from: liquidityProvider,
|
|
@@ -416,12 +392,25 @@ export class BotFactory {
|
|
|
416
392
|
wait: { timeout: this.config.txMinedWaitSeconds },
|
|
417
393
|
});
|
|
418
394
|
|
|
419
|
-
this.log.info(`Sent tx to add liquidity to the AMM: ${
|
|
395
|
+
this.log.info(`Sent tx to add liquidity to the AMM: ${receipt.txHash.toString()}`);
|
|
420
396
|
this.log.info(`Liquidity added`);
|
|
421
397
|
|
|
422
|
-
const [
|
|
398
|
+
const [t0Bal, t1Bal, lpBal] = await Promise.all([
|
|
399
|
+
token0.methods
|
|
400
|
+
.balance_of_private(liquidityProvider)
|
|
401
|
+
.simulate({ from: liquidityProvider })
|
|
402
|
+
.then(r => r.result),
|
|
403
|
+
token1.methods
|
|
404
|
+
.balance_of_private(liquidityProvider)
|
|
405
|
+
.simulate({ from: liquidityProvider })
|
|
406
|
+
.then(r => r.result),
|
|
407
|
+
lpToken.methods
|
|
408
|
+
.balance_of_private(liquidityProvider)
|
|
409
|
+
.simulate({ from: liquidityProvider })
|
|
410
|
+
.then(r => r.result),
|
|
411
|
+
]);
|
|
423
412
|
this.log.info(
|
|
424
|
-
`Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${
|
|
413
|
+
`Updated private balances of ${defaultAccountAddress} after minting and funding AMM: token0=${t0Bal}, token1=${t1Bal}, lp=${lpBal}`,
|
|
425
414
|
);
|
|
426
415
|
}
|
|
427
416
|
|
|
@@ -430,27 +419,75 @@ export class BotFactory {
|
|
|
430
419
|
deploy: DeployMethod<T>,
|
|
431
420
|
deployOpts: DeployOptions,
|
|
432
421
|
): Promise<ContractInstanceWithAddress> {
|
|
433
|
-
const instance = await deploy.getInstance(
|
|
422
|
+
const instance = await deploy.getInstance();
|
|
434
423
|
const address = instance.address;
|
|
435
424
|
const metadata = await this.wallet.getContractMetadata(address);
|
|
436
425
|
if (metadata.isContractPublished) {
|
|
437
426
|
this.log.info(`Contract ${name} at ${address.toString()} already deployed`);
|
|
438
427
|
await deploy.register();
|
|
439
|
-
|
|
440
|
-
this.log.info(`Deploying contract ${name} at ${address.toString()}`);
|
|
441
|
-
await this.withNoMinTxsPerBlock(async () => {
|
|
442
|
-
const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
443
|
-
this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
|
|
444
|
-
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
445
|
-
});
|
|
428
|
+
return instance;
|
|
446
429
|
}
|
|
430
|
+
|
|
431
|
+
// Setup always runs ensureFeeJuiceBalance before any deploy, so the account pays from its standing
|
|
432
|
+
// balance here. No manual gas estimation: the embedded wallet simulates before sending and derives
|
|
433
|
+
// the gas limits and padded maxFeesPerGas itself.
|
|
434
|
+
this.log.info(`Deploying contract ${name} at ${address.toString()}`);
|
|
435
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
436
|
+
const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
437
|
+
this.log.info(`Sent contract ${name} deploy tx ${txHash.toString()}`);
|
|
438
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
439
|
+
});
|
|
440
|
+
|
|
447
441
|
return instance;
|
|
448
442
|
}
|
|
449
443
|
|
|
444
|
+
/** True when the config allows bridging fee juice from L1 (fee_juice mode, an L1 RPC, and an L1 key). */
|
|
445
|
+
private isL1BridgingConfigured(): boolean {
|
|
446
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
447
|
+
return this.config.feePaymentMethod === 'fee_juice' && !!this.config.l1RpcUrls?.length && !!mnemonicOrPrivateKey;
|
|
448
|
+
}
|
|
449
|
+
|
|
450
450
|
/**
|
|
451
|
-
*
|
|
452
|
-
*
|
|
451
|
+
* Ensures the account holds enough fee juice before any other setup step. The account starts empty
|
|
452
|
+
* (initializerless accounts have no deployment tx) and the runtime loop pays fees from this balance and
|
|
453
|
+
* never refuels itself, so every flow funds the account up front. Bridges claims from L1 and consumes
|
|
454
|
+
* each with a claim-only tx until the balance clears the threshold, working from a zero (fresh run) or
|
|
455
|
+
* drained (restart) balance. Each bridge mints a fixed amount well above the threshold, so this is a
|
|
456
|
+
* single bridge in practice. No-op when L1 bridging is not configured or the balance is already above
|
|
457
|
+
* the threshold.
|
|
453
458
|
*/
|
|
459
|
+
private async ensureFeeJuiceBalance(account: AztecAddress): Promise<void> {
|
|
460
|
+
if (!this.isL1BridgingConfigured()) {
|
|
461
|
+
return;
|
|
462
|
+
}
|
|
463
|
+
|
|
464
|
+
let balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
465
|
+
if (balance >= FEE_JUICE_TOP_UP_THRESHOLD) {
|
|
466
|
+
this.log.info(`Fee juice balance ${balance} above threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, skipping top-up`);
|
|
467
|
+
return;
|
|
468
|
+
}
|
|
469
|
+
|
|
470
|
+
this.log.info(`Fee juice balance ${balance} below threshold ${FEE_JUICE_TOP_UP_THRESHOLD}, bridging from L1`);
|
|
471
|
+
|
|
472
|
+
while (balance < FEE_JUICE_TOP_UP_THRESHOLD) {
|
|
473
|
+
// Persist the claim before consuming it: if the top-up tx fails or the bot crashes mid-loop, the
|
|
474
|
+
// next run reuses the pending claim instead of bridging again (and wasting the bridged funds).
|
|
475
|
+
const claim = await this.getOrCreateBridgeClaim(account);
|
|
476
|
+
const paymentMethod = new FeeJuicePaymentMethodWithClaim(account, claim);
|
|
477
|
+
|
|
478
|
+
await this.withNoMinTxsPerBlock(async () => {
|
|
479
|
+
const executionPayload = await paymentMethod.getExecutionPayload();
|
|
480
|
+
const { txHash } = await this.wallet.sendTx(executionPayload, { from: account, wait: NO_WAIT });
|
|
481
|
+
this.log.info(`Sent fee juice top-up tx ${txHash.toString()}`);
|
|
482
|
+
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
483
|
+
});
|
|
484
|
+
await this.store.deleteBridgeClaim(account);
|
|
485
|
+
balance = await getFeeJuiceBalance(account, this.aztecNode);
|
|
486
|
+
this.log.info(`Fee juice balance after top-up: ${balance}`);
|
|
487
|
+
}
|
|
488
|
+
this.log.info(`Fee juice top-up complete for ${account.toString()}`);
|
|
489
|
+
}
|
|
490
|
+
|
|
454
491
|
private async mintTokens(token: TokenContract | PrivateTokenContract, minter: AztecAddress) {
|
|
455
492
|
const isStandardToken = isStandardTokenContract(token);
|
|
456
493
|
let privateBalance = 0n;
|
|
@@ -483,31 +520,33 @@ export class BotFactory {
|
|
|
483
520
|
|
|
484
521
|
// PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
|
|
485
522
|
const additionalScopes = isStandardToken ? undefined : [token.address];
|
|
523
|
+
const mintBatch = new BatchCall(token.wallet, calls);
|
|
486
524
|
await this.withNoMinTxsPerBlock(async () => {
|
|
487
|
-
const txHash = await
|
|
525
|
+
const { txHash } = await mintBatch.send({
|
|
526
|
+
from: minter,
|
|
527
|
+
additionalScopes,
|
|
528
|
+
wait: NO_WAIT,
|
|
529
|
+
});
|
|
488
530
|
this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
|
|
489
531
|
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
490
532
|
});
|
|
491
533
|
}
|
|
492
534
|
|
|
493
535
|
/**
|
|
494
|
-
*
|
|
495
|
-
*
|
|
496
|
-
*
|
|
536
|
+
* Returns a usable bridge claim for the recipient, reusing a persisted one when its L1→L2 message is
|
|
537
|
+
* still available (resuming a top-up that failed or crashed before the claim was consumed) and bridging
|
|
538
|
+
* a fresh claim otherwise. The caller deletes the claim from the store once it has been consumed.
|
|
497
539
|
*/
|
|
498
540
|
private async getOrCreateBridgeClaim(recipient: AztecAddress): Promise<L2AmountClaim> {
|
|
499
|
-
// Check if we have an existing claim in the store
|
|
500
541
|
const existingClaim = await this.store.getBridgeClaim(recipient);
|
|
501
542
|
if (existingClaim) {
|
|
502
543
|
this.log.info(`Found existing bridge claim for ${recipient.toString()}, checking validity...`);
|
|
503
|
-
|
|
504
|
-
// Check if the message is ready on L2
|
|
505
544
|
try {
|
|
506
545
|
const messageHash = Fr.fromHexString(existingClaim.claim.messageHash);
|
|
507
546
|
await this.withNoMinTxsPerBlock(() =>
|
|
508
547
|
waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
|
|
509
548
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
510
|
-
|
|
549
|
+
chainTip: this.syncChainTip,
|
|
511
550
|
}),
|
|
512
551
|
);
|
|
513
552
|
return existingClaim.claim;
|
|
@@ -519,7 +558,6 @@ export class BotFactory {
|
|
|
519
558
|
|
|
520
559
|
const claim = await this.bridgeL1FeeJuice(recipient);
|
|
521
560
|
await this.store.saveBridgeClaim(recipient, claim);
|
|
522
|
-
|
|
523
561
|
return claim;
|
|
524
562
|
}
|
|
525
563
|
|
|
@@ -546,7 +584,7 @@ export class BotFactory {
|
|
|
546
584
|
await this.withNoMinTxsPerBlock(() =>
|
|
547
585
|
waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
|
|
548
586
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
549
|
-
|
|
587
|
+
chainTip: this.syncChainTip,
|
|
550
588
|
}),
|
|
551
589
|
);
|
|
552
590
|
|
|
@@ -555,19 +593,36 @@ export class BotFactory {
|
|
|
555
593
|
return claim as L2AmountClaim;
|
|
556
594
|
}
|
|
557
595
|
|
|
558
|
-
|
|
596
|
+
protected async withNoMinTxsPerBlock<T>(fn: () => Promise<T>): Promise<T> {
|
|
559
597
|
if (!this.aztecNodeAdmin || !this.config.flushSetupTransactions) {
|
|
560
598
|
this.log.verbose(`No node admin client or flushing not requested (not setting minTxsPerBlock to 0)`);
|
|
561
599
|
return fn();
|
|
562
600
|
}
|
|
563
|
-
const
|
|
564
|
-
this
|
|
565
|
-
|
|
601
|
+
const aztecNodeAdmin = this.aztecNodeAdmin;
|
|
602
|
+
// Setup steps run concurrently, so this wrapper can be re-entered while another call is in flight.
|
|
603
|
+
// Reference-count the entrants: the first saves the current value and zeroes it, the last restores it.
|
|
604
|
+
// A naive save/zero/restore per call could interleave, with a late entrant reading the already-zeroed
|
|
605
|
+
// value and "restoring" 0 at the end.
|
|
606
|
+
if (this.noMinTxsPerBlockDepth++ === 0) {
|
|
607
|
+
this.savedMinTxsPerBlock = (async () => {
|
|
608
|
+
const { minTxsPerBlock } = await aztecNodeAdmin.getConfig();
|
|
609
|
+
this.log.warn(`Setting sequencer minTxsPerBlock to 0 from ${minTxsPerBlock} to flush setup transactions`);
|
|
610
|
+
await aztecNodeAdmin.setConfig({ minTxsPerBlock: 0 });
|
|
611
|
+
return { minTxsPerBlock };
|
|
612
|
+
})();
|
|
613
|
+
}
|
|
566
614
|
try {
|
|
615
|
+
await this.savedMinTxsPerBlock;
|
|
567
616
|
return await fn();
|
|
568
617
|
} finally {
|
|
569
|
-
this.
|
|
570
|
-
|
|
618
|
+
if (--this.noMinTxsPerBlockDepth === 0) {
|
|
619
|
+
// If saving/zeroing itself failed there is nothing to restore.
|
|
620
|
+
const saved = await this.savedMinTxsPerBlock!.catch(() => undefined);
|
|
621
|
+
if (saved) {
|
|
622
|
+
this.log.warn(`Restoring sequencer minTxsPerBlock to ${saved.minTxsPerBlock}`);
|
|
623
|
+
await aztecNodeAdmin.setConfig({ minTxsPerBlock: saved.minTxsPerBlock });
|
|
624
|
+
}
|
|
625
|
+
}
|
|
571
626
|
}
|
|
572
627
|
}
|
|
573
628
|
}
|