@aztec/bot 0.0.1-commit.e61ad554 → 0.0.1-commit.ec5f612
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/base_bot.d.ts +4 -4
- package/dest/base_bot.d.ts.map +1 -1
- package/dest/base_bot.js +9 -10
- package/dest/bot.d.ts +4 -4
- package/dest/bot.d.ts.map +1 -1
- package/dest/bot.js +2 -2
- package/dest/config.d.ts +30 -14
- package/dest/config.d.ts.map +1 -1
- package/dest/config.js +36 -8
- 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 +140 -0
- package/dest/factory.d.ts +20 -5
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +80 -14
- 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/package.json +16 -13
- package/src/amm_bot.ts +3 -3
- package/src/base_bot.ts +8 -18
- package/src/bot.ts +5 -5
- package/src/config.ts +39 -10
- package/src/cross_chain_bot.ts +209 -0
- package/src/factory.ts +110 -20
- 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/amm_bot.ts
CHANGED
|
@@ -6,7 +6,7 @@ import { jsonStringify } from '@aztec/foundation/json-rpc';
|
|
|
6
6
|
import type { AMMContract } from '@aztec/noir-contracts.js/AMM';
|
|
7
7
|
import type { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
8
8
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
9
|
-
import type {
|
|
9
|
+
import type { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
10
10
|
|
|
11
11
|
import { BaseBot } from './base_bot.js';
|
|
12
12
|
import type { BotConfig } from './config.js';
|
|
@@ -21,7 +21,7 @@ type Balances = { token0: bigint; token1: bigint };
|
|
|
21
21
|
export class AmmBot extends BaseBot {
|
|
22
22
|
protected constructor(
|
|
23
23
|
node: AztecNode,
|
|
24
|
-
wallet:
|
|
24
|
+
wallet: EmbeddedWallet,
|
|
25
25
|
defaultAccountAddress: AztecAddress,
|
|
26
26
|
public readonly amm: AMMContract,
|
|
27
27
|
public readonly token0: TokenContract,
|
|
@@ -33,7 +33,7 @@ export class AmmBot extends BaseBot {
|
|
|
33
33
|
|
|
34
34
|
static async create(
|
|
35
35
|
config: BotConfig,
|
|
36
|
-
wallet:
|
|
36
|
+
wallet: EmbeddedWallet,
|
|
37
37
|
aztecNode: AztecNode,
|
|
38
38
|
aztecNodeAdmin: AztecNodeAdmin | undefined,
|
|
39
39
|
store: BotStore,
|
package/src/base_bot.ts
CHANGED
|
@@ -1,16 +1,11 @@
|
|
|
1
1
|
import { AztecAddress } from '@aztec/aztec.js/addresses';
|
|
2
|
-
import {
|
|
3
|
-
BatchCall,
|
|
4
|
-
ContractFunctionInteraction,
|
|
5
|
-
type SendInteractionOptions,
|
|
6
|
-
waitForProven,
|
|
7
|
-
} from '@aztec/aztec.js/contracts';
|
|
2
|
+
import { BatchCall, ContractFunctionInteraction, type SendInteractionOptions } from '@aztec/aztec.js/contracts';
|
|
8
3
|
import { createLogger } from '@aztec/aztec.js/log';
|
|
9
4
|
import { waitForTx } from '@aztec/aztec.js/node';
|
|
10
|
-
import { TxHash, TxReceipt } from '@aztec/aztec.js/tx';
|
|
5
|
+
import { TxHash, TxReceipt, TxStatus } from '@aztec/aztec.js/tx';
|
|
11
6
|
import { Gas } from '@aztec/stdlib/gas';
|
|
12
7
|
import type { AztecNode } from '@aztec/stdlib/interfaces/client';
|
|
13
|
-
import type {
|
|
8
|
+
import type { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
14
9
|
|
|
15
10
|
import type { BotConfig } from './config.js';
|
|
16
11
|
|
|
@@ -22,15 +17,15 @@ export abstract class BaseBot {
|
|
|
22
17
|
|
|
23
18
|
protected constructor(
|
|
24
19
|
public readonly node: AztecNode,
|
|
25
|
-
public readonly wallet:
|
|
20
|
+
public readonly wallet: EmbeddedWallet,
|
|
26
21
|
public readonly defaultAccountAddress: AztecAddress,
|
|
27
22
|
public config: BotConfig,
|
|
28
23
|
) {}
|
|
29
24
|
|
|
30
25
|
public async run(): Promise<TxReceipt | TxHash> {
|
|
31
26
|
this.attempts++;
|
|
32
|
-
const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000) };
|
|
33
27
|
const { followChain, txMinedWaitSeconds } = this.config;
|
|
28
|
+
const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000), followChain, txMinedWaitSeconds };
|
|
34
29
|
|
|
35
30
|
this.log.verbose(`Creating tx`, logCtx);
|
|
36
31
|
const txHash = await this.createAndSendTx(logCtx);
|
|
@@ -40,14 +35,9 @@ export abstract class BaseBot {
|
|
|
40
35
|
return txHash;
|
|
41
36
|
}
|
|
42
37
|
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
);
|
|
47
|
-
const receipt = await waitForTx(this.node, txHash, { timeout: txMinedWaitSeconds });
|
|
48
|
-
if (followChain === 'PROVEN') {
|
|
49
|
-
await waitForProven(this.node, receipt, { provenTimeout: txMinedWaitSeconds });
|
|
50
|
-
}
|
|
38
|
+
const waitForStatus = TxStatus[followChain];
|
|
39
|
+
this.log.verbose(`Awaiting tx ${txHash.toString()} to be on the ${followChain} chain`, logCtx);
|
|
40
|
+
const receipt = await waitForTx(this.node, txHash, { timeout: txMinedWaitSeconds, waitForStatus });
|
|
51
41
|
this.successes++;
|
|
52
42
|
this.log.info(
|
|
53
43
|
`Tx #${this.attempts} ${receipt.txHash} successfully mined in block ${receipt.blockNumber} (stats: ${this.successes}/${this.attempts} success)`,
|
package/src/bot.ts
CHANGED
|
@@ -5,7 +5,7 @@ import { times } from '@aztec/foundation/collection';
|
|
|
5
5
|
import type { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
|
|
6
6
|
import type { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
7
7
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
8
|
-
import type {
|
|
8
|
+
import type { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
9
9
|
|
|
10
10
|
import { BaseBot } from './base_bot.js';
|
|
11
11
|
import type { BotConfig } from './config.js';
|
|
@@ -18,7 +18,7 @@ const TRANSFER_AMOUNT = 1;
|
|
|
18
18
|
export class Bot extends BaseBot {
|
|
19
19
|
protected constructor(
|
|
20
20
|
node: AztecNode,
|
|
21
|
-
wallet:
|
|
21
|
+
wallet: EmbeddedWallet,
|
|
22
22
|
defaultAccountAddress: AztecAddress,
|
|
23
23
|
public readonly token: TokenContract | PrivateTokenContract,
|
|
24
24
|
public readonly recipient: AztecAddress,
|
|
@@ -29,7 +29,7 @@ export class Bot extends BaseBot {
|
|
|
29
29
|
|
|
30
30
|
static async create(
|
|
31
31
|
config: BotConfig,
|
|
32
|
-
wallet:
|
|
32
|
+
wallet: EmbeddedWallet,
|
|
33
33
|
aztecNode: AztecNode,
|
|
34
34
|
aztecNodeAdmin: AztecNodeAdmin | undefined,
|
|
35
35
|
store: BotStore,
|
|
@@ -83,7 +83,7 @@ export class Bot extends BaseBot {
|
|
|
83
83
|
if (isStandardTokenContract(this.token)) {
|
|
84
84
|
return {
|
|
85
85
|
sender: await getBalances(this.token, this.defaultAccountAddress),
|
|
86
|
-
recipient: await getBalances(this.token, this.recipient
|
|
86
|
+
recipient: await getBalances(this.token, this.recipient),
|
|
87
87
|
};
|
|
88
88
|
} else {
|
|
89
89
|
return {
|
|
@@ -92,7 +92,7 @@ export class Bot extends BaseBot {
|
|
|
92
92
|
publicBalance: 0n,
|
|
93
93
|
},
|
|
94
94
|
recipient: {
|
|
95
|
-
privateBalance: await getPrivateBalance(this.token, this.recipient
|
|
95
|
+
privateBalance: await getPrivateBalance(this.token, this.recipient),
|
|
96
96
|
publicBalance: 0n,
|
|
97
97
|
},
|
|
98
98
|
};
|
package/src/config.ts
CHANGED
|
@@ -19,9 +19,12 @@ import type { ComponentsVersions } from '@aztec/stdlib/versioning';
|
|
|
19
19
|
|
|
20
20
|
import { z } from 'zod';
|
|
21
21
|
|
|
22
|
-
const BotFollowChain = ['NONE', '
|
|
22
|
+
const BotFollowChain = ['NONE', 'PROPOSED', 'CHECKPOINTED', 'PROVEN'] as const;
|
|
23
23
|
type BotFollowChain = (typeof BotFollowChain)[number];
|
|
24
24
|
|
|
25
|
+
const BotMode = ['transfer', 'amm', 'crosschain'] as const;
|
|
26
|
+
type BotMode = (typeof BotMode)[number];
|
|
27
|
+
|
|
25
28
|
export enum SupportedTokenContracts {
|
|
26
29
|
TokenContract = 'TokenContract',
|
|
27
30
|
PrivateTokenContract = 'PrivateTokenContract',
|
|
@@ -76,8 +79,12 @@ export type BotConfig = {
|
|
|
76
79
|
maxConsecutiveErrors: number;
|
|
77
80
|
/** Stops the bot if service becomes unhealthy */
|
|
78
81
|
stopWhenUnhealthy: boolean;
|
|
79
|
-
/**
|
|
80
|
-
|
|
82
|
+
/** Bot mode: transfer, amm, or crosschain. */
|
|
83
|
+
botMode: BotMode;
|
|
84
|
+
/** Number of L2→L1 messages per tx (crosschain mode). */
|
|
85
|
+
l2ToL1MessagesPerTx: number;
|
|
86
|
+
/** Max L1→L2 messages to keep in-flight (crosschain mode). */
|
|
87
|
+
l1ToL2SeedCount: number;
|
|
81
88
|
} & Pick<DataStoreConfig, 'dataDirectory' | 'dataStoreMapSizeKb'>;
|
|
82
89
|
|
|
83
90
|
export const BotConfigSchema = zodFor<BotConfig>()(
|
|
@@ -107,7 +114,9 @@ export const BotConfigSchema = zodFor<BotConfig>()(
|
|
|
107
114
|
contract: z.nativeEnum(SupportedTokenContracts),
|
|
108
115
|
maxConsecutiveErrors: z.number().int().nonnegative(),
|
|
109
116
|
stopWhenUnhealthy: z.boolean(),
|
|
110
|
-
|
|
117
|
+
botMode: z.enum(BotMode).default('transfer'),
|
|
118
|
+
l2ToL1MessagesPerTx: z.number().int().nonnegative().default(1),
|
|
119
|
+
l1ToL2SeedCount: z.number().int().nonnegative().default(1),
|
|
111
120
|
dataDirectory: z.string().optional(),
|
|
112
121
|
dataStoreMapSizeKb: z.number().optional(),
|
|
113
122
|
})
|
|
@@ -213,10 +222,14 @@ export const botConfigMappings: ConfigMappingsType<BotConfig> = {
|
|
|
213
222
|
description: 'Which chain the bot follows',
|
|
214
223
|
defaultValue: 'NONE',
|
|
215
224
|
parseEnv(val) {
|
|
216
|
-
|
|
225
|
+
const upper = val.toUpperCase();
|
|
226
|
+
if (upper === 'PENDING') {
|
|
227
|
+
return 'CHECKPOINTED';
|
|
228
|
+
}
|
|
229
|
+
if (!(BotFollowChain as readonly string[]).includes(upper)) {
|
|
217
230
|
throw new Error(`Invalid value for BOT_FOLLOW_CHAIN: ${val}`);
|
|
218
231
|
}
|
|
219
|
-
return
|
|
232
|
+
return upper as BotFollowChain;
|
|
220
233
|
},
|
|
221
234
|
},
|
|
222
235
|
maxPendingTxs: {
|
|
@@ -264,10 +277,26 @@ export const botConfigMappings: ConfigMappingsType<BotConfig> = {
|
|
|
264
277
|
description: 'Stops the bot if service becomes unhealthy',
|
|
265
278
|
...booleanConfigHelper(false),
|
|
266
279
|
},
|
|
267
|
-
|
|
268
|
-
env: '
|
|
269
|
-
description: '
|
|
270
|
-
|
|
280
|
+
botMode: {
|
|
281
|
+
env: 'BOT_MODE',
|
|
282
|
+
description: 'Bot mode: transfer, amm, or crosschain',
|
|
283
|
+
defaultValue: 'transfer' as BotMode,
|
|
284
|
+
parseEnv(val: string) {
|
|
285
|
+
if (!(BotMode as readonly string[]).includes(val)) {
|
|
286
|
+
throw new Error(`Invalid value for BOT_MODE: ${val}`);
|
|
287
|
+
}
|
|
288
|
+
return val as BotMode;
|
|
289
|
+
},
|
|
290
|
+
},
|
|
291
|
+
l2ToL1MessagesPerTx: {
|
|
292
|
+
env: 'BOT_L2_TO_L1_MESSAGES_PER_TX',
|
|
293
|
+
description: 'Number of L2→L1 messages per tx (crosschain mode)',
|
|
294
|
+
...numberConfigHelper(1),
|
|
295
|
+
},
|
|
296
|
+
l1ToL2SeedCount: {
|
|
297
|
+
env: 'BOT_L1_TO_L2_SEED_COUNT',
|
|
298
|
+
description: 'Max L1→L2 messages to keep in-flight (crosschain mode)',
|
|
299
|
+
...numberConfigHelper(1),
|
|
271
300
|
},
|
|
272
301
|
...pickConfigMappings(dataConfigMappings, ['dataStoreMapSizeKb', 'dataDirectory']),
|
|
273
302
|
};
|
|
@@ -0,0 +1,209 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* CrossChainBot exercises L2->L1 and L1->L2 messaging.
|
|
3
|
+
*
|
|
4
|
+
* createAndSendTx onTxMined
|
|
5
|
+
* ────────────────────────────────────── ──────────────────────────────
|
|
6
|
+
*
|
|
7
|
+
* 1. SEED (fire-and-forget) 3. VERIFY L2->L1
|
|
8
|
+
* if store has fewer pending messages Query getTxEffect, confirm
|
|
9
|
+
* than seedCount and no seed is the expected L2->L1 messages
|
|
10
|
+
* in-flight: appeared in tx effects.
|
|
11
|
+
* * kick off L1 inbox tx
|
|
12
|
+
* * store msg on completion
|
|
13
|
+
*
|
|
14
|
+
* 2. BUILD & SEND BATCH
|
|
15
|
+
* Always:
|
|
16
|
+
* N x create_l2_to_l1_message
|
|
17
|
+
* (random content, fixed
|
|
18
|
+
* L1 recipient)
|
|
19
|
+
* If a ready L1->L2 msg exists:
|
|
20
|
+
* 1 x consume_message_from_
|
|
21
|
+
* arbitrary_sender_public
|
|
22
|
+
* delete consumed msg from store
|
|
23
|
+
* Send batch tx (no wait)
|
|
24
|
+
*
|
|
25
|
+
*/
|
|
26
|
+
import { AztecAddress } from '@aztec/aztec.js/addresses';
|
|
27
|
+
import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
|
|
28
|
+
import { isL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
|
|
29
|
+
import { TxHash, TxReceipt } from '@aztec/aztec.js/tx';
|
|
30
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
31
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
32
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
33
|
+
import type { TestContract } from '@aztec/noir-test-contracts.js/Test';
|
|
34
|
+
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
35
|
+
import type { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
36
|
+
|
|
37
|
+
import { BaseBot } from './base_bot.js';
|
|
38
|
+
import type { BotConfig } from './config.js';
|
|
39
|
+
import { BotFactory } from './factory.js';
|
|
40
|
+
import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
|
|
41
|
+
import type { BotStore, PendingL1ToL2Message } from './store/index.js';
|
|
42
|
+
|
|
43
|
+
/** Stale message threshold: messages older than this are removed. */
|
|
44
|
+
const STALE_MESSAGE_THRESHOLD_MS = 2 * 60 * 60 * 1000; // 2 hours
|
|
45
|
+
|
|
46
|
+
/** Bot that exercises both L2→L1 and L1→L2 cross-chain messaging. */
|
|
47
|
+
export class CrossChainBot extends BaseBot {
|
|
48
|
+
private l2ToL1Sent = 0;
|
|
49
|
+
private l1ToL2Consumed = 0;
|
|
50
|
+
private pendingSeedPromise: Promise<void> | undefined;
|
|
51
|
+
|
|
52
|
+
protected constructor(
|
|
53
|
+
node: AztecNode,
|
|
54
|
+
wallet: EmbeddedWallet,
|
|
55
|
+
defaultAccountAddress: AztecAddress,
|
|
56
|
+
private readonly contract: TestContract,
|
|
57
|
+
private readonly l1Client: ExtendedViemWalletClient,
|
|
58
|
+
private readonly l1Recipient: EthAddress,
|
|
59
|
+
private readonly inboxAddress: EthAddress,
|
|
60
|
+
private readonly rollupVersion: bigint,
|
|
61
|
+
private readonly store: BotStore,
|
|
62
|
+
config: BotConfig,
|
|
63
|
+
) {
|
|
64
|
+
super(node, wallet, defaultAccountAddress, config);
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
static async create(
|
|
68
|
+
config: BotConfig,
|
|
69
|
+
wallet: EmbeddedWallet,
|
|
70
|
+
aztecNode: AztecNode,
|
|
71
|
+
aztecNodeAdmin: AztecNodeAdmin | undefined,
|
|
72
|
+
store: BotStore,
|
|
73
|
+
): Promise<CrossChainBot> {
|
|
74
|
+
if (config.followChain === 'NONE') {
|
|
75
|
+
throw new Error(`CrossChainBot requires followChain to be set (got NONE)`);
|
|
76
|
+
}
|
|
77
|
+
const factory = new BotFactory(config, wallet, store, aztecNode, aztecNodeAdmin);
|
|
78
|
+
const { defaultAccountAddress, contract, l1Client, rollupVersion } = await factory.setupCrossChain();
|
|
79
|
+
const l1Recipient = EthAddress.fromString(l1Client.account!.address);
|
|
80
|
+
const { l1ContractAddresses } = await aztecNode.getNodeInfo();
|
|
81
|
+
const inboxAddress = EthAddress.fromString(l1ContractAddresses.inboxAddress.toString());
|
|
82
|
+
return new CrossChainBot(
|
|
83
|
+
aztecNode,
|
|
84
|
+
wallet,
|
|
85
|
+
defaultAccountAddress,
|
|
86
|
+
contract,
|
|
87
|
+
l1Client,
|
|
88
|
+
l1Recipient,
|
|
89
|
+
inboxAddress,
|
|
90
|
+
rollupVersion,
|
|
91
|
+
store,
|
|
92
|
+
config,
|
|
93
|
+
);
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
protected async createAndSendTx(logCtx: object): Promise<TxHash> {
|
|
97
|
+
const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
98
|
+
|
|
99
|
+
// Send an L1→L2 message if we're below the threshold and not already seeding one
|
|
100
|
+
if (pendingMessages.length < this.config.l1ToL2SeedCount && !this.pendingSeedPromise) {
|
|
101
|
+
this.pendingSeedPromise = this.seedNewL1ToL2Message()
|
|
102
|
+
.catch(err => this.log.warn(`Failed to seed L1→L2 message: ${err}`, logCtx))
|
|
103
|
+
.finally(() => {
|
|
104
|
+
this.pendingSeedPromise = undefined;
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
// Build batch: always L2→L1, optionally consume L1→L2
|
|
109
|
+
const calls = [];
|
|
110
|
+
|
|
111
|
+
// L2→L1: create messages with random content
|
|
112
|
+
for (let i = 0; i < this.config.l2ToL1MessagesPerTx; i++) {
|
|
113
|
+
calls.push(
|
|
114
|
+
this.contract.methods.create_l2_to_l1_message_arbitrary_recipient_public(Fr.random(), this.l1Recipient),
|
|
115
|
+
);
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
// L1→L2: consume oldest ready message if available
|
|
119
|
+
const readyMsg = await this.getReadyL1ToL2Message(pendingMessages);
|
|
120
|
+
if (readyMsg) {
|
|
121
|
+
calls.push(
|
|
122
|
+
this.contract.methods.consume_message_from_arbitrary_sender_public(
|
|
123
|
+
Fr.fromHexString(readyMsg.content),
|
|
124
|
+
Fr.fromHexString(readyMsg.secret),
|
|
125
|
+
EthAddress.fromString(readyMsg.sender),
|
|
126
|
+
new Fr(BigInt(readyMsg.globalLeafIndex)),
|
|
127
|
+
),
|
|
128
|
+
);
|
|
129
|
+
// Delete consumed message immediately so it works with FOLLOW_CHAIN=NONE
|
|
130
|
+
await this.store.deleteL1ToL2Message(readyMsg.msgHash);
|
|
131
|
+
this.l1ToL2Consumed++;
|
|
132
|
+
} else {
|
|
133
|
+
this.log.warn(`No ready L1→L2 message to consume`, {
|
|
134
|
+
...logCtx,
|
|
135
|
+
pendingCount: pendingMessages.length,
|
|
136
|
+
});
|
|
137
|
+
}
|
|
138
|
+
|
|
139
|
+
const batch = new BatchCall(this.wallet, calls);
|
|
140
|
+
const opts = await this.getSendMethodOpts(batch);
|
|
141
|
+
|
|
142
|
+
this.log.verbose(`Sending cross-chain batch with ${calls.length} calls`, logCtx);
|
|
143
|
+
return batch.send({ ...opts, wait: NO_WAIT });
|
|
144
|
+
}
|
|
145
|
+
|
|
146
|
+
protected override async onTxMined(receipt: TxReceipt, logCtx: object): Promise<void> {
|
|
147
|
+
// Verify L2→L1 messages appeared in this tx's effects
|
|
148
|
+
const indexed = await this.node.getTxEffect(receipt.txHash);
|
|
149
|
+
if (indexed) {
|
|
150
|
+
const l2ToL1Msgs = indexed.data.l2ToL1Msgs.filter(m => !m.isZero());
|
|
151
|
+
if (l2ToL1Msgs.length >= this.config.l2ToL1MessagesPerTx) {
|
|
152
|
+
this.l2ToL1Sent += l2ToL1Msgs.length;
|
|
153
|
+
} else {
|
|
154
|
+
this.log.error(`Expected ${this.config.l2ToL1MessagesPerTx} L2→L1 messages but found ${l2ToL1Msgs.length}`, {
|
|
155
|
+
...logCtx,
|
|
156
|
+
blockNumber: receipt.blockNumber,
|
|
157
|
+
txHash: receipt.txHash.toString(),
|
|
158
|
+
});
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
const pendingCount = (await this.store.getUnconsumedL1ToL2Messages()).length;
|
|
163
|
+
this.log.info(`CrossChainBot txs mined`, {
|
|
164
|
+
...logCtx,
|
|
165
|
+
l2ToL1Sent: this.l2ToL1Sent,
|
|
166
|
+
l1ToL2Consumed: this.l1ToL2Consumed,
|
|
167
|
+
l1ToL2Pending: pendingCount,
|
|
168
|
+
});
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
/** Finds the oldest pending message that is ready for consumption. */
|
|
172
|
+
private async getReadyL1ToL2Message(
|
|
173
|
+
pendingMessages: PendingL1ToL2Message[],
|
|
174
|
+
): Promise<PendingL1ToL2Message | undefined> {
|
|
175
|
+
const now = Date.now();
|
|
176
|
+
for (const msg of pendingMessages) {
|
|
177
|
+
const ready = await isL1ToL2MessageReady(this.node, Fr.fromHexString(msg.msgHash), {
|
|
178
|
+
// Use forPublicConsumption: false so we wait until blockNumber >= messageBlockNumber.
|
|
179
|
+
// With forPublicConsumption: true, the check returns true one block early (the sequencer
|
|
180
|
+
// includes L1→L2 messages before executing the block's txs), but gas estimation simulates
|
|
181
|
+
// against the current world state which doesn't yet have the message.
|
|
182
|
+
// See https://linear.app/aztec-labs/issue/A-548 for details.
|
|
183
|
+
forPublicConsumption: false,
|
|
184
|
+
});
|
|
185
|
+
if (ready) {
|
|
186
|
+
return msg;
|
|
187
|
+
}
|
|
188
|
+
|
|
189
|
+
// Time-based stale detection: if the message is old and still not ready, remove it
|
|
190
|
+
if (now - msg.timestamp > STALE_MESSAGE_THRESHOLD_MS) {
|
|
191
|
+
await this.store.deleteL1ToL2Message(msg.msgHash);
|
|
192
|
+
this.log.warn(`Removed stale L1→L2 message ${msg.msgHash}`);
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
return undefined;
|
|
196
|
+
}
|
|
197
|
+
|
|
198
|
+
/** Seeds a new L1→L2 message on L1 and stores it. */
|
|
199
|
+
private async seedNewL1ToL2Message(): Promise<void> {
|
|
200
|
+
await seedL1ToL2Message(
|
|
201
|
+
this.l1Client,
|
|
202
|
+
this.inboxAddress,
|
|
203
|
+
this.contract.address,
|
|
204
|
+
this.rollupVersion,
|
|
205
|
+
this.store,
|
|
206
|
+
this.log,
|
|
207
|
+
);
|
|
208
|
+
}
|
|
209
|
+
}
|
package/src/factory.ts
CHANGED
|
@@ -1,4 +1,3 @@
|
|
|
1
|
-
import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
|
|
2
1
|
import { getInitialTestAccountsData } from '@aztec/accounts/testing';
|
|
3
2
|
import { AztecAddress } from '@aztec/aztec.js/addresses';
|
|
4
3
|
import {
|
|
@@ -9,8 +8,8 @@ import {
|
|
|
9
8
|
type DeployOptions,
|
|
10
9
|
NO_WAIT,
|
|
11
10
|
} from '@aztec/aztec.js/contracts';
|
|
12
|
-
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
13
11
|
import type { L2AmountClaim } from '@aztec/aztec.js/ethereum';
|
|
12
|
+
import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
|
|
14
13
|
import { FeeJuicePaymentMethodWithClaim } from '@aztec/aztec.js/fee';
|
|
15
14
|
import { deriveKeys } from '@aztec/aztec.js/keys';
|
|
16
15
|
import { createLogger } from '@aztec/aztec.js/log';
|
|
@@ -18,18 +17,23 @@ import { waitForL1ToL2MessageReady } from '@aztec/aztec.js/messaging';
|
|
|
18
17
|
import { waitForTx } from '@aztec/aztec.js/node';
|
|
19
18
|
import { createEthereumChain } from '@aztec/ethereum/chain';
|
|
20
19
|
import { createExtendedL1Client } from '@aztec/ethereum/client';
|
|
20
|
+
import { RollupContract } from '@aztec/ethereum/contracts';
|
|
21
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
21
22
|
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
23
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
22
24
|
import { Timer } from '@aztec/foundation/timer';
|
|
23
25
|
import { AMMContract } from '@aztec/noir-contracts.js/AMM';
|
|
24
26
|
import { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
|
|
25
27
|
import { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
28
|
+
import { TestContract } from '@aztec/noir-test-contracts.js/Test';
|
|
26
29
|
import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
|
|
27
30
|
import { GasSettings } from '@aztec/stdlib/gas';
|
|
28
31
|
import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
29
32
|
import { deriveSigningKey } from '@aztec/stdlib/keys';
|
|
30
|
-
import {
|
|
33
|
+
import { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
31
34
|
|
|
32
35
|
import { type BotConfig, SupportedTokenContracts } from './config.js';
|
|
36
|
+
import { seedL1ToL2Message } from './l1_to_l2_seeding.js';
|
|
33
37
|
import type { BotStore } from './store/index.js';
|
|
34
38
|
import { getBalances, getPrivateBalance, isStandardTokenContract } from './utils.js';
|
|
35
39
|
|
|
@@ -41,7 +45,7 @@ export class BotFactory {
|
|
|
41
45
|
|
|
42
46
|
constructor(
|
|
43
47
|
private readonly config: BotConfig,
|
|
44
|
-
private readonly wallet:
|
|
48
|
+
private readonly wallet: EmbeddedWallet,
|
|
45
49
|
private readonly store: BotStore,
|
|
46
50
|
private readonly aztecNode: AztecNode,
|
|
47
51
|
private readonly aztecNodeAdmin?: AztecNodeAdmin,
|
|
@@ -52,21 +56,21 @@ export class BotFactory {
|
|
|
52
56
|
* deploying the token contract, and minting tokens if necessary.
|
|
53
57
|
*/
|
|
54
58
|
public async setup(): Promise<{
|
|
55
|
-
wallet:
|
|
59
|
+
wallet: EmbeddedWallet;
|
|
56
60
|
defaultAccountAddress: AztecAddress;
|
|
57
61
|
token: TokenContract | PrivateTokenContract;
|
|
58
62
|
node: AztecNode;
|
|
59
63
|
recipient: AztecAddress;
|
|
60
64
|
}> {
|
|
61
65
|
const defaultAccountAddress = await this.setupAccount();
|
|
62
|
-
const recipient = (await this.wallet.
|
|
66
|
+
const recipient = (await this.wallet.createSchnorrAccount(Fr.random(), Fr.random())).address;
|
|
63
67
|
const token = await this.setupToken(defaultAccountAddress);
|
|
64
68
|
await this.mintTokens(token, defaultAccountAddress);
|
|
65
69
|
return { wallet: this.wallet, defaultAccountAddress, token, node: this.aztecNode, recipient };
|
|
66
70
|
}
|
|
67
71
|
|
|
68
72
|
public async setupAmm(): Promise<{
|
|
69
|
-
wallet:
|
|
73
|
+
wallet: EmbeddedWallet;
|
|
70
74
|
defaultAccountAddress: AztecAddress;
|
|
71
75
|
amm: AMMContract;
|
|
72
76
|
token0: TokenContract;
|
|
@@ -96,6 +100,94 @@ export class BotFactory {
|
|
|
96
100
|
return { wallet: this.wallet, defaultAccountAddress, amm, token0, token1, node: this.aztecNode };
|
|
97
101
|
}
|
|
98
102
|
|
|
103
|
+
/**
|
|
104
|
+
* Initializes the cross-chain bot by deploying TestContract, creating an L1 client,
|
|
105
|
+
* seeding initial L1→L2 messages, and waiting for the first to be ready.
|
|
106
|
+
*/
|
|
107
|
+
public async setupCrossChain(): Promise<{
|
|
108
|
+
wallet: EmbeddedWallet;
|
|
109
|
+
defaultAccountAddress: AztecAddress;
|
|
110
|
+
contract: TestContract;
|
|
111
|
+
node: AztecNode;
|
|
112
|
+
l1Client: ExtendedViemWalletClient;
|
|
113
|
+
rollupVersion: bigint;
|
|
114
|
+
}> {
|
|
115
|
+
const defaultAccountAddress = await this.setupAccount();
|
|
116
|
+
|
|
117
|
+
// Create L1 client (same pattern as bridgeL1FeeJuice)
|
|
118
|
+
const l1RpcUrls = this.config.l1RpcUrls;
|
|
119
|
+
if (!l1RpcUrls?.length) {
|
|
120
|
+
throw new Error('L1 RPC URLs required for cross-chain bot');
|
|
121
|
+
}
|
|
122
|
+
const mnemonicOrPrivateKey = this.config.l1PrivateKey?.getValue() ?? this.config.l1Mnemonic?.getValue();
|
|
123
|
+
if (!mnemonicOrPrivateKey) {
|
|
124
|
+
throw new Error('L1 mnemonic or private key required for cross-chain bot');
|
|
125
|
+
}
|
|
126
|
+
const { l1ChainId, l1ContractAddresses } = await this.aztecNode.getNodeInfo();
|
|
127
|
+
const chain = createEthereumChain(l1RpcUrls, l1ChainId);
|
|
128
|
+
const l1Client = createExtendedL1Client(chain.rpcUrls, mnemonicOrPrivateKey, chain.chainInfo);
|
|
129
|
+
|
|
130
|
+
// Fetch Rollup version (needed for Inbox L2Actor struct)
|
|
131
|
+
const rollupContract = new RollupContract(l1Client, l1ContractAddresses.rollupAddress.toString());
|
|
132
|
+
const rollupVersion = await rollupContract.getVersion();
|
|
133
|
+
|
|
134
|
+
// Deploy TestContract
|
|
135
|
+
const contract = await this.setupTestContract(defaultAccountAddress);
|
|
136
|
+
|
|
137
|
+
// Recover any pending messages from store (clean up stale ones first)
|
|
138
|
+
await this.store.cleanupOldPendingMessages();
|
|
139
|
+
const pendingMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
140
|
+
|
|
141
|
+
// Seed initial L1→L2 messages if pipeline is empty
|
|
142
|
+
const seedCount = Math.max(0, this.config.l1ToL2SeedCount - pendingMessages.length);
|
|
143
|
+
for (let i = 0; i < seedCount; i++) {
|
|
144
|
+
await seedL1ToL2Message(
|
|
145
|
+
l1Client,
|
|
146
|
+
EthAddress.fromString(l1ContractAddresses.inboxAddress.toString()),
|
|
147
|
+
contract.address,
|
|
148
|
+
rollupVersion,
|
|
149
|
+
this.store,
|
|
150
|
+
this.log,
|
|
151
|
+
);
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
// Block until at least one message is ready
|
|
155
|
+
const allMessages = await this.store.getUnconsumedL1ToL2Messages();
|
|
156
|
+
if (allMessages.length > 0) {
|
|
157
|
+
this.log.info(`Waiting for first L1→L2 message to be ready...`);
|
|
158
|
+
const firstMsg = allMessages[0];
|
|
159
|
+
await waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(firstMsg.msgHash), {
|
|
160
|
+
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
161
|
+
// Use forPublicConsumption: false so we wait until the message is in the current world
|
|
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,
|
|
166
|
+
});
|
|
167
|
+
this.log.info(`First L1→L2 message is ready`);
|
|
168
|
+
}
|
|
169
|
+
|
|
170
|
+
return {
|
|
171
|
+
wallet: this.wallet,
|
|
172
|
+
defaultAccountAddress,
|
|
173
|
+
contract,
|
|
174
|
+
node: this.aztecNode,
|
|
175
|
+
l1Client,
|
|
176
|
+
rollupVersion,
|
|
177
|
+
};
|
|
178
|
+
}
|
|
179
|
+
|
|
180
|
+
private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
|
|
181
|
+
const deployOpts: DeployOptions = {
|
|
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);
|
|
188
|
+
return TestContract.at(instance.address, this.wallet);
|
|
189
|
+
}
|
|
190
|
+
|
|
99
191
|
/**
|
|
100
192
|
* Checks if the sender account contract is initialized, and initializes it if necessary.
|
|
101
193
|
* @returns The sender wallet.
|
|
@@ -114,12 +206,7 @@ export class BotFactory {
|
|
|
114
206
|
private async setupAccountWithPrivateKey(secret: Fr) {
|
|
115
207
|
const salt = this.config.senderSalt ?? Fr.ONE;
|
|
116
208
|
const signingKey = deriveSigningKey(secret);
|
|
117
|
-
const
|
|
118
|
-
secret,
|
|
119
|
-
salt,
|
|
120
|
-
contract: new SchnorrAccountContract(signingKey!),
|
|
121
|
-
};
|
|
122
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
209
|
+
const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
|
|
123
210
|
const metadata = await this.wallet.getContractMetadata(accountManager.address);
|
|
124
211
|
if (metadata.isContractInitialized) {
|
|
125
212
|
this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
|
|
@@ -159,12 +246,11 @@ export class BotFactory {
|
|
|
159
246
|
|
|
160
247
|
private async setupTestAccount() {
|
|
161
248
|
const [initialAccountData] = await getInitialTestAccountsData();
|
|
162
|
-
const
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
249
|
+
const accountManager = await this.wallet.createSchnorrAccount(
|
|
250
|
+
initialAccountData.secret,
|
|
251
|
+
initialAccountData.salt,
|
|
252
|
+
initialAccountData.signingKey,
|
|
253
|
+
);
|
|
168
254
|
return accountManager.address;
|
|
169
255
|
}
|
|
170
256
|
|
|
@@ -199,6 +285,8 @@ export class BotFactory {
|
|
|
199
285
|
tokenInstance = await deploy.getInstance(deployOpts);
|
|
200
286
|
token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
|
|
201
287
|
await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
|
|
288
|
+
// The contract constructor initializes private storage vars that need the contract's own nullifier key.
|
|
289
|
+
deployOpts.additionalScopes = [tokenInstance.address];
|
|
202
290
|
} else {
|
|
203
291
|
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
204
292
|
}
|
|
@@ -393,8 +481,10 @@ export class BotFactory {
|
|
|
393
481
|
return;
|
|
394
482
|
}
|
|
395
483
|
|
|
484
|
+
// PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
|
|
485
|
+
const additionalScopes = isStandardToken ? undefined : [token.address];
|
|
396
486
|
await this.withNoMinTxsPerBlock(async () => {
|
|
397
|
-
const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, wait: NO_WAIT });
|
|
487
|
+
const txHash = await new BatchCall(token.wallet, calls).send({ from: minter, additionalScopes, wait: NO_WAIT });
|
|
398
488
|
this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
|
|
399
489
|
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
400
490
|
});
|