@aztec/bot 0.0.1-commit.f2ce05ee → 0.0.1-commit.f81dbcf
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dest/amm_bot.d.ts +4 -4
- package/dest/amm_bot.d.ts.map +1 -1
- package/dest/amm_bot.js +23 -16
- 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 +4 -3
- 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 +134 -0
- package/dest/factory.d.ts +20 -5
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +87 -28
- package/dest/index.d.ts +2 -1
- package/dest/index.d.ts.map +1 -1
- package/dest/index.js +1 -0
- package/dest/l1_to_l2_seeding.d.ts +8 -0
- package/dest/l1_to_l2_seeding.d.ts.map +1 -0
- package/dest/l1_to_l2_seeding.js +63 -0
- package/dest/runner.d.ts +3 -3
- package/dest/runner.d.ts.map +1 -1
- package/dest/runner.js +17 -1
- package/dest/store/bot_store.d.ts +30 -5
- package/dest/store/bot_store.d.ts.map +1 -1
- package/dest/store/bot_store.js +37 -6
- package/dest/store/index.d.ts +2 -2
- package/dest/store/index.d.ts.map +1 -1
- package/dest/utils.js +3 -3
- package/package.json +16 -13
- package/src/amm_bot.ts +23 -18
- package/src/base_bot.ts +8 -18
- package/src/bot.ts +7 -6
- package/src/config.ts +39 -10
- package/src/cross_chain_bot.ts +203 -0
- package/src/factory.ts +127 -31
- package/src/index.ts +1 -0
- package/src/l1_to_l2_seeding.ts +79 -0
- package/src/runner.ts +18 -5
- package/src/store/bot_store.ts +60 -5
- package/src/store/index.ts +1 -1
- package/src/utils.ts +3 -3
|
@@ -0,0 +1,203 @@
|
|
|
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
|
+
const { txHash } = await batch.send({ ...opts, wait: NO_WAIT });
|
|
144
|
+
return txHash;
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
protected override async onTxMined(receipt: TxReceipt, logCtx: object): Promise<void> {
|
|
148
|
+
// Verify L2→L1 messages appeared in this tx's effects
|
|
149
|
+
const indexed = await this.node.getTxEffect(receipt.txHash);
|
|
150
|
+
if (indexed) {
|
|
151
|
+
const l2ToL1Msgs = indexed.data.l2ToL1Msgs.filter(m => !m.isZero());
|
|
152
|
+
if (l2ToL1Msgs.length >= this.config.l2ToL1MessagesPerTx) {
|
|
153
|
+
this.l2ToL1Sent += l2ToL1Msgs.length;
|
|
154
|
+
} else {
|
|
155
|
+
this.log.error(`Expected ${this.config.l2ToL1MessagesPerTx} L2→L1 messages but found ${l2ToL1Msgs.length}`, {
|
|
156
|
+
...logCtx,
|
|
157
|
+
blockNumber: receipt.blockNumber,
|
|
158
|
+
txHash: receipt.txHash.toString(),
|
|
159
|
+
});
|
|
160
|
+
}
|
|
161
|
+
}
|
|
162
|
+
|
|
163
|
+
const pendingCount = (await this.store.getUnconsumedL1ToL2Messages()).length;
|
|
164
|
+
this.log.info(`CrossChainBot txs mined`, {
|
|
165
|
+
...logCtx,
|
|
166
|
+
l2ToL1Sent: this.l2ToL1Sent,
|
|
167
|
+
l1ToL2Consumed: this.l1ToL2Consumed,
|
|
168
|
+
l1ToL2Pending: pendingCount,
|
|
169
|
+
});
|
|
170
|
+
}
|
|
171
|
+
|
|
172
|
+
/** Finds the oldest pending message that is ready for consumption. */
|
|
173
|
+
private async getReadyL1ToL2Message(
|
|
174
|
+
pendingMessages: PendingL1ToL2Message[],
|
|
175
|
+
): Promise<PendingL1ToL2Message | undefined> {
|
|
176
|
+
const now = Date.now();
|
|
177
|
+
for (const msg of pendingMessages) {
|
|
178
|
+
const ready = await isL1ToL2MessageReady(this.node, Fr.fromHexString(msg.msgHash));
|
|
179
|
+
if (ready) {
|
|
180
|
+
return msg;
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
// Time-based stale detection: if the message is old and still not ready, remove it
|
|
184
|
+
if (now - msg.timestamp > STALE_MESSAGE_THRESHOLD_MS) {
|
|
185
|
+
await this.store.deleteL1ToL2Message(msg.msgHash);
|
|
186
|
+
this.log.warn(`Removed stale L1→L2 message ${msg.msgHash}`);
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
return undefined;
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
/** Seeds a new L1→L2 message on L1 and stores it. */
|
|
193
|
+
private async seedNewL1ToL2Message(): Promise<void> {
|
|
194
|
+
await seedL1ToL2Message(
|
|
195
|
+
this.l1Client,
|
|
196
|
+
this.inboxAddress,
|
|
197
|
+
this.contract.address,
|
|
198
|
+
this.rollupVersion,
|
|
199
|
+
this.store,
|
|
200
|
+
this.log,
|
|
201
|
+
);
|
|
202
|
+
}
|
|
203
|
+
}
|
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,89 @@ 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
|
+
});
|
|
162
|
+
this.log.info(`First L1→L2 message is ready`);
|
|
163
|
+
}
|
|
164
|
+
|
|
165
|
+
return {
|
|
166
|
+
wallet: this.wallet,
|
|
167
|
+
defaultAccountAddress,
|
|
168
|
+
contract,
|
|
169
|
+
node: this.aztecNode,
|
|
170
|
+
l1Client,
|
|
171
|
+
rollupVersion,
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
|
|
175
|
+
private async setupTestContract(deployer: AztecAddress): Promise<TestContract> {
|
|
176
|
+
const deployOpts: DeployOptions = {
|
|
177
|
+
from: deployer,
|
|
178
|
+
contractAddressSalt: this.config.tokenSalt,
|
|
179
|
+
universalDeploy: true,
|
|
180
|
+
};
|
|
181
|
+
const deploy = TestContract.deploy(this.wallet);
|
|
182
|
+
const instance = await this.registerOrDeployContract('TestContract', deploy, deployOpts);
|
|
183
|
+
return TestContract.at(instance.address, this.wallet);
|
|
184
|
+
}
|
|
185
|
+
|
|
99
186
|
/**
|
|
100
187
|
* Checks if the sender account contract is initialized, and initializes it if necessary.
|
|
101
188
|
* @returns The sender wallet.
|
|
@@ -114,12 +201,7 @@ export class BotFactory {
|
|
|
114
201
|
private async setupAccountWithPrivateKey(secret: Fr) {
|
|
115
202
|
const salt = this.config.senderSalt ?? Fr.ONE;
|
|
116
203
|
const signingKey = deriveSigningKey(secret);
|
|
117
|
-
const
|
|
118
|
-
secret,
|
|
119
|
-
salt,
|
|
120
|
-
contract: new SchnorrAccountContract(signingKey!),
|
|
121
|
-
};
|
|
122
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
204
|
+
const accountManager = await this.wallet.createSchnorrAccount(secret, salt, signingKey);
|
|
123
205
|
const metadata = await this.wallet.getContractMetadata(accountManager.address);
|
|
124
206
|
if (metadata.isContractInitialized) {
|
|
125
207
|
this.log.info(`Account at ${accountManager.address.toString()} already initialized`);
|
|
@@ -140,7 +222,7 @@ export class BotFactory {
|
|
|
140
222
|
const gasSettings = GasSettings.default({ maxFeesPerGas });
|
|
141
223
|
|
|
142
224
|
await this.withNoMinTxsPerBlock(async () => {
|
|
143
|
-
const txHash = await deployMethod.send({
|
|
225
|
+
const { txHash } = await deployMethod.send({
|
|
144
226
|
from: AztecAddress.ZERO,
|
|
145
227
|
fee: { gasSettings, paymentMethod },
|
|
146
228
|
wait: NO_WAIT,
|
|
@@ -159,12 +241,11 @@ export class BotFactory {
|
|
|
159
241
|
|
|
160
242
|
private async setupTestAccount() {
|
|
161
243
|
const [initialAccountData] = await getInitialTestAccountsData();
|
|
162
|
-
const
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
const accountManager = await this.wallet.createAccount(accountData);
|
|
244
|
+
const accountManager = await this.wallet.createSchnorrAccount(
|
|
245
|
+
initialAccountData.secret,
|
|
246
|
+
initialAccountData.salt,
|
|
247
|
+
initialAccountData.signingKey,
|
|
248
|
+
);
|
|
168
249
|
return accountManager.address;
|
|
169
250
|
}
|
|
170
251
|
|
|
@@ -199,6 +280,8 @@ export class BotFactory {
|
|
|
199
280
|
tokenInstance = await deploy.getInstance(deployOpts);
|
|
200
281
|
token = PrivateTokenContract.at(tokenInstance.address, this.wallet);
|
|
201
282
|
await this.wallet.registerContract(tokenInstance, PrivateTokenContract.artifact, tokenSecretKey);
|
|
283
|
+
// The contract constructor initializes private storage vars that need the contract's own nullifier key.
|
|
284
|
+
deployOpts.additionalScopes = [tokenInstance.address];
|
|
202
285
|
} else {
|
|
203
286
|
throw new Error(`Unsupported token contract type: ${this.config.contract}`);
|
|
204
287
|
}
|
|
@@ -210,7 +293,7 @@ export class BotFactory {
|
|
|
210
293
|
await deploy.register();
|
|
211
294
|
} else {
|
|
212
295
|
this.log.info(`Deploying token contract at ${address.toString()}`);
|
|
213
|
-
const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
296
|
+
const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
214
297
|
this.log.info(`Sent tx for token setup with hash ${txHash.toString()}`);
|
|
215
298
|
await this.withNoMinTxsPerBlock(async () => {
|
|
216
299
|
await waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
@@ -251,7 +334,7 @@ export class BotFactory {
|
|
|
251
334
|
const amm = AMMContract.at(instance.address, this.wallet);
|
|
252
335
|
|
|
253
336
|
this.log.info(`AMM deployed at ${amm.address}`);
|
|
254
|
-
const minterReceipt = await lpToken.methods
|
|
337
|
+
const { receipt: minterReceipt } = await lpToken.methods
|
|
255
338
|
.set_minter(amm.address, true)
|
|
256
339
|
.send({ from: deployer, wait: { timeout: this.config.txMinedWaitSeconds } });
|
|
257
340
|
this.log.info(`Set LP token minter to AMM txHash=${minterReceipt.txHash.toString()}`);
|
|
@@ -270,9 +353,18 @@ export class BotFactory {
|
|
|
270
353
|
): Promise<void> {
|
|
271
354
|
const getPrivateBalances = () =>
|
|
272
355
|
Promise.all([
|
|
273
|
-
token0.methods
|
|
274
|
-
|
|
275
|
-
|
|
356
|
+
token0.methods
|
|
357
|
+
.balance_of_private(liquidityProvider)
|
|
358
|
+
.simulate({ from: liquidityProvider })
|
|
359
|
+
.then(r => r.result),
|
|
360
|
+
token1.methods
|
|
361
|
+
.balance_of_private(liquidityProvider)
|
|
362
|
+
.simulate({ from: liquidityProvider })
|
|
363
|
+
.then(r => r.result),
|
|
364
|
+
lpToken.methods
|
|
365
|
+
.balance_of_private(liquidityProvider)
|
|
366
|
+
.simulate({ from: liquidityProvider })
|
|
367
|
+
.then(r => r.result),
|
|
276
368
|
]);
|
|
277
369
|
|
|
278
370
|
const authwitNonce = Fr.random();
|
|
@@ -313,14 +405,14 @@ export class BotFactory {
|
|
|
313
405
|
.getFunctionCall(),
|
|
314
406
|
});
|
|
315
407
|
|
|
316
|
-
const mintReceipt = await new BatchCall(this.wallet, [
|
|
408
|
+
const { receipt: mintReceipt } = await new BatchCall(this.wallet, [
|
|
317
409
|
token0.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
318
410
|
token1.methods.mint_to_private(liquidityProvider, MINT_BALANCE),
|
|
319
411
|
]).send({ from: liquidityProvider, wait: { timeout: this.config.txMinedWaitSeconds } });
|
|
320
412
|
|
|
321
413
|
this.log.info(`Sent mint tx: ${mintReceipt.txHash.toString()}`);
|
|
322
414
|
|
|
323
|
-
const addLiquidityReceipt = await amm.methods
|
|
415
|
+
const { receipt: addLiquidityReceipt } = await amm.methods
|
|
324
416
|
.add_liquidity(amount0Max, amount1Max, amount0Min, amount1Min, authwitNonce)
|
|
325
417
|
.send({
|
|
326
418
|
from: liquidityProvider,
|
|
@@ -351,7 +443,7 @@ export class BotFactory {
|
|
|
351
443
|
} else {
|
|
352
444
|
this.log.info(`Deploying contract ${name} at ${address.toString()}`);
|
|
353
445
|
await this.withNoMinTxsPerBlock(async () => {
|
|
354
|
-
const txHash = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
446
|
+
const { txHash } = await deploy.send({ ...deployOpts, wait: NO_WAIT });
|
|
355
447
|
this.log.info(`Sent contract ${name} setup tx with hash ${txHash.toString()}`);
|
|
356
448
|
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
357
449
|
});
|
|
@@ -393,8 +485,14 @@ export class BotFactory {
|
|
|
393
485
|
return;
|
|
394
486
|
}
|
|
395
487
|
|
|
488
|
+
// PrivateToken's mint accesses contract-level private storage vars (admin, total_supply).
|
|
489
|
+
const additionalScopes = isStandardToken ? undefined : [token.address];
|
|
396
490
|
await this.withNoMinTxsPerBlock(async () => {
|
|
397
|
-
const txHash = await new BatchCall(token.wallet, calls).send({
|
|
491
|
+
const { txHash } = await new BatchCall(token.wallet, calls).send({
|
|
492
|
+
from: minter,
|
|
493
|
+
additionalScopes,
|
|
494
|
+
wait: NO_WAIT,
|
|
495
|
+
});
|
|
398
496
|
this.log.info(`Sent token mint tx with hash ${txHash.toString()}`);
|
|
399
497
|
return waitForTx(this.aztecNode, txHash, { timeout: this.config.txMinedWaitSeconds });
|
|
400
498
|
});
|
|
@@ -417,7 +515,6 @@ export class BotFactory {
|
|
|
417
515
|
await this.withNoMinTxsPerBlock(() =>
|
|
418
516
|
waitForL1ToL2MessageReady(this.aztecNode, messageHash, {
|
|
419
517
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
420
|
-
forPublicConsumption: false,
|
|
421
518
|
}),
|
|
422
519
|
);
|
|
423
520
|
return existingClaim.claim;
|
|
@@ -456,7 +553,6 @@ export class BotFactory {
|
|
|
456
553
|
await this.withNoMinTxsPerBlock(() =>
|
|
457
554
|
waitForL1ToL2MessageReady(this.aztecNode, Fr.fromHexString(claim.messageHash), {
|
|
458
555
|
timeoutSeconds: this.config.l1ToL2MessageTimeoutSeconds,
|
|
459
|
-
forPublicConsumption: false,
|
|
460
556
|
}),
|
|
461
557
|
);
|
|
462
558
|
|
package/src/index.ts
CHANGED
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import { generateClaimSecret } from '@aztec/aztec.js/ethereum';
|
|
2
|
+
import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
|
|
3
|
+
import { compactArray } from '@aztec/foundation/collection';
|
|
4
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
5
|
+
import { EthAddress } from '@aztec/foundation/eth-address';
|
|
6
|
+
import type { Logger } from '@aztec/foundation/log';
|
|
7
|
+
import { InboxAbi } from '@aztec/l1-artifacts';
|
|
8
|
+
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
9
|
+
|
|
10
|
+
import { decodeEventLog, getContract } from 'viem';
|
|
11
|
+
|
|
12
|
+
import type { BotStore, PendingL1ToL2Message } from './store/index.js';
|
|
13
|
+
|
|
14
|
+
/** Sends an L1→L2 message via the Inbox contract and stores it. */
|
|
15
|
+
export async function seedL1ToL2Message(
|
|
16
|
+
l1Client: ExtendedViemWalletClient,
|
|
17
|
+
inboxAddress: EthAddress,
|
|
18
|
+
l2Recipient: AztecAddress,
|
|
19
|
+
rollupVersion: bigint,
|
|
20
|
+
store: BotStore,
|
|
21
|
+
log: Logger,
|
|
22
|
+
): Promise<PendingL1ToL2Message> {
|
|
23
|
+
log.info('Seeding L1→L2 message');
|
|
24
|
+
const [secret, secretHash] = await generateClaimSecret(log);
|
|
25
|
+
const content = Fr.random();
|
|
26
|
+
|
|
27
|
+
const inbox = getContract({
|
|
28
|
+
address: inboxAddress.toString(),
|
|
29
|
+
abi: InboxAbi,
|
|
30
|
+
client: l1Client,
|
|
31
|
+
});
|
|
32
|
+
|
|
33
|
+
const txHash = await inbox.write.sendL2Message(
|
|
34
|
+
[{ actor: l2Recipient.toString(), version: rollupVersion }, content.toString(), secretHash.toString()],
|
|
35
|
+
{ gas: 1_000_000n },
|
|
36
|
+
);
|
|
37
|
+
log.info(`L1→L2 message sent in tx ${txHash}`);
|
|
38
|
+
|
|
39
|
+
const txReceipt = await l1Client.waitForTransactionReceipt({ hash: txHash });
|
|
40
|
+
if (txReceipt.status !== 'success') {
|
|
41
|
+
throw new Error(`L1→L2 message tx failed: ${txHash}`);
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
// Extract MessageSent event
|
|
45
|
+
const messageSentLogs = compactArray(
|
|
46
|
+
txReceipt.logs
|
|
47
|
+
.filter(l => l.address.toLowerCase() === inboxAddress.toString().toLowerCase())
|
|
48
|
+
.map(l => {
|
|
49
|
+
try {
|
|
50
|
+
return decodeEventLog({ abi: InboxAbi, eventName: 'MessageSent', data: l.data, topics: l.topics });
|
|
51
|
+
} catch {
|
|
52
|
+
return undefined;
|
|
53
|
+
}
|
|
54
|
+
}),
|
|
55
|
+
);
|
|
56
|
+
|
|
57
|
+
if (messageSentLogs.length !== 1) {
|
|
58
|
+
throw new Error(`Expected 1 MessageSent event, got ${messageSentLogs.length}`);
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
const event = messageSentLogs[0];
|
|
62
|
+
|
|
63
|
+
const msgHash = event.args.hash;
|
|
64
|
+
const globalLeafIndex = event.args.index;
|
|
65
|
+
|
|
66
|
+
const msg: PendingL1ToL2Message = {
|
|
67
|
+
content: content.toString(),
|
|
68
|
+
secret: secret.toString(),
|
|
69
|
+
secretHash: secretHash.toString(),
|
|
70
|
+
msgHash,
|
|
71
|
+
sender: l1Client.account!.address,
|
|
72
|
+
globalLeafIndex: globalLeafIndex.toString(),
|
|
73
|
+
timestamp: Date.now(),
|
|
74
|
+
};
|
|
75
|
+
|
|
76
|
+
await store.savePendingL1ToL2Message(msg);
|
|
77
|
+
log.info(`Seeded L1→L2 message msgHash=${msg.msgHash}`);
|
|
78
|
+
return msg;
|
|
79
|
+
}
|
package/src/runner.ts
CHANGED
|
@@ -4,12 +4,13 @@ import { omit } from '@aztec/foundation/collection';
|
|
|
4
4
|
import { RunningPromise } from '@aztec/foundation/running-promise';
|
|
5
5
|
import type { AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
|
|
6
6
|
import { type TelemetryClient, type Traceable, type Tracer, trackSpan } from '@aztec/telemetry-client';
|
|
7
|
-
import type {
|
|
7
|
+
import type { EmbeddedWallet } from '@aztec/wallets/embedded';
|
|
8
8
|
|
|
9
9
|
import { AmmBot } from './amm_bot.js';
|
|
10
10
|
import type { BaseBot } from './base_bot.js';
|
|
11
11
|
import { Bot } from './bot.js';
|
|
12
12
|
import type { BotConfig } from './config.js';
|
|
13
|
+
import { CrossChainBot } from './cross_chain_bot.js';
|
|
13
14
|
import type { BotInfo, BotRunnerApi } from './interface.js';
|
|
14
15
|
import { BotStore } from './store/index.js';
|
|
15
16
|
|
|
@@ -24,7 +25,7 @@ export class BotRunner implements BotRunnerApi, Traceable {
|
|
|
24
25
|
|
|
25
26
|
public constructor(
|
|
26
27
|
private config: BotConfig,
|
|
27
|
-
private readonly wallet:
|
|
28
|
+
private readonly wallet: EmbeddedWallet,
|
|
28
29
|
private readonly aztecNode: AztecNode,
|
|
29
30
|
private readonly telemetry: TelemetryClient,
|
|
30
31
|
private readonly aztecNodeAdmin: AztecNodeAdmin | undefined,
|
|
@@ -146,9 +147,21 @@ export class BotRunner implements BotRunnerApi, Traceable {
|
|
|
146
147
|
|
|
147
148
|
async #createBot() {
|
|
148
149
|
try {
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
150
|
+
switch (this.config.botMode) {
|
|
151
|
+
case 'crosschain':
|
|
152
|
+
this.bot = CrossChainBot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
|
|
153
|
+
break;
|
|
154
|
+
case 'amm':
|
|
155
|
+
this.bot = AmmBot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
|
|
156
|
+
break;
|
|
157
|
+
case 'transfer':
|
|
158
|
+
this.bot = Bot.create(this.config, this.wallet, this.aztecNode, this.aztecNodeAdmin, this.store);
|
|
159
|
+
break;
|
|
160
|
+
default: {
|
|
161
|
+
const _exhaustive: never = this.config.botMode;
|
|
162
|
+
throw new Error(`Unsupported bot mode: [${_exhaustive}]`);
|
|
163
|
+
}
|
|
164
|
+
}
|
|
152
165
|
await this.bot;
|
|
153
166
|
} catch (err) {
|
|
154
167
|
this.log.error(`Error setting up bot: ${err}`);
|