@aztec/aztec 0.0.0-test.1 → 0.0.1-commit.21caa21
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/README.md +1 -1
- package/dest/bin/index.d.ts +1 -1
- package/dest/bin/index.js +20 -12
- package/dest/cli/aztec_start_action.d.ts +1 -1
- package/dest/cli/aztec_start_action.d.ts.map +1 -1
- package/dest/cli/aztec_start_action.js +36 -37
- package/dest/cli/aztec_start_options.d.ts +5 -3
- package/dest/cli/aztec_start_options.d.ts.map +1 -1
- package/dest/cli/aztec_start_options.js +142 -210
- package/dest/cli/cli.d.ts +1 -1
- package/dest/cli/cli.d.ts.map +1 -1
- package/dest/cli/cli.js +39 -2
- package/dest/cli/cmds/start_archiver.d.ts +1 -1
- package/dest/cli/cmds/start_archiver.d.ts.map +1 -1
- package/dest/cli/cmds/start_archiver.js +18 -10
- package/dest/cli/cmds/start_blob_sink.d.ts +1 -1
- package/dest/cli/cmds/start_blob_sink.d.ts.map +1 -1
- package/dest/cli/cmds/start_blob_sink.js +17 -1
- package/dest/cli/cmds/start_bot.d.ts +4 -7
- package/dest/cli/cmds/start_bot.d.ts.map +1 -1
- package/dest/cli/cmds/start_bot.js +24 -13
- package/dest/cli/cmds/start_node.d.ts +2 -2
- package/dest/cli/cmds/start_node.d.ts.map +1 -1
- package/dest/cli/cmds/start_node.js +67 -80
- package/dest/cli/cmds/start_p2p_bootstrap.d.ts +1 -1
- package/dest/cli/cmds/start_p2p_bootstrap.d.ts.map +1 -1
- package/dest/cli/cmds/start_p2p_bootstrap.js +9 -4
- package/dest/cli/cmds/start_prover_agent.d.ts +1 -1
- package/dest/cli/cmds/start_prover_agent.d.ts.map +1 -1
- package/dest/cli/cmds/start_prover_agent.js +31 -2
- package/dest/cli/cmds/start_prover_broker.d.ts +1 -1
- package/dest/cli/cmds/start_prover_broker.d.ts.map +1 -1
- package/dest/cli/cmds/start_prover_broker.js +9 -3
- package/dest/cli/cmds/start_prover_node.d.ts +1 -1
- package/dest/cli/cmds/start_prover_node.d.ts.map +1 -1
- package/dest/cli/cmds/start_prover_node.js +43 -45
- package/dest/cli/cmds/start_txe.d.ts +1 -1
- package/dest/cli/index.d.ts +1 -1
- package/dest/cli/preload_crs.d.ts +3 -0
- package/dest/cli/preload_crs.d.ts.map +1 -0
- package/dest/cli/preload_crs.js +6 -0
- package/dest/cli/release_version.d.ts +2 -0
- package/dest/cli/release_version.d.ts.map +1 -0
- package/dest/cli/release_version.js +14 -0
- package/dest/cli/util.d.ts +38 -7
- package/dest/cli/util.d.ts.map +1 -1
- package/dest/cli/util.js +198 -28
- package/dest/cli/versioning.d.ts +1 -1
- package/dest/cli/versioning.js +3 -3
- package/dest/examples/token.d.ts +1 -1
- package/dest/examples/token.js +31 -18
- package/dest/examples/util.d.ts +5 -6
- package/dest/examples/util.d.ts.map +1 -1
- package/dest/examples/util.js +5 -6
- package/dest/index.d.ts +2 -2
- package/dest/index.d.ts.map +1 -1
- package/dest/index.js +1 -1
- package/dest/local-network/banana_fpc.d.ts +10 -0
- package/dest/local-network/banana_fpc.d.ts.map +1 -0
- package/dest/{sandbox → local-network}/banana_fpc.js +19 -21
- package/dest/local-network/index.d.ts +4 -0
- package/dest/local-network/index.d.ts.map +1 -0
- package/dest/local-network/index.js +3 -0
- package/dest/local-network/local-network.d.ts +74 -0
- package/dest/local-network/local-network.d.ts.map +1 -0
- package/dest/{sandbox/sandbox.js → local-network/local-network.js} +59 -57
- package/dest/local-network/sponsored_fpc.d.ts +5 -0
- package/dest/local-network/sponsored_fpc.d.ts.map +1 -0
- package/dest/local-network/sponsored_fpc.js +18 -0
- package/dest/mnemonic.d.ts +1 -1
- package/dest/splash.d.ts +1 -1
- package/dest/testing/anvil_test_watcher.d.ts +34 -0
- package/dest/testing/anvil_test_watcher.d.ts.map +1 -0
- package/dest/testing/anvil_test_watcher.js +144 -0
- package/dest/testing/cheat_codes.d.ts +41 -0
- package/dest/testing/cheat_codes.d.ts.map +1 -0
- package/dest/testing/cheat_codes.js +62 -0
- package/dest/testing/index.d.ts +4 -0
- package/dest/testing/index.d.ts.map +1 -0
- package/dest/testing/index.js +3 -0
- package/package.json +53 -45
- package/scripts/aztec.sh +60 -0
- package/scripts/compile.sh +44 -0
- package/scripts/extract_function.js +47 -0
- package/scripts/flamegraph.sh +59 -0
- package/scripts/init.sh +35 -0
- package/scripts/new.sh +59 -0
- package/scripts/setup_project.sh +31 -0
- package/src/bin/index.ts +24 -12
- package/src/cli/aztec_start_action.ts +32 -35
- package/src/cli/aztec_start_options.ts +161 -210
- package/src/cli/cli.ts +46 -5
- package/src/cli/cmds/start_archiver.ts +19 -13
- package/src/cli/cmds/start_blob_sink.ts +27 -1
- package/src/cli/cmds/start_bot.ts +35 -12
- package/src/cli/cmds/start_node.ts +90 -85
- package/src/cli/cmds/start_p2p_bootstrap.ts +12 -4
- package/src/cli/cmds/start_prover_agent.ts +22 -2
- package/src/cli/cmds/start_prover_broker.ts +23 -3
- package/src/cli/cmds/start_prover_node.ts +53 -50
- package/src/cli/preload_crs.ts +7 -0
- package/src/cli/release_version.ts +21 -0
- package/src/cli/util.ts +208 -34
- package/src/cli/versioning.ts +3 -3
- package/src/examples/token.ts +23 -19
- package/src/examples/util.ts +6 -8
- package/src/index.ts +5 -6
- package/src/{sandbox → local-network}/banana_fpc.ts +20 -25
- package/src/local-network/index.ts +7 -0
- package/src/{sandbox/sandbox.ts → local-network/local-network.ts} +81 -68
- package/src/local-network/sponsored_fpc.ts +26 -0
- package/src/testing/anvil_test_watcher.ts +166 -0
- package/src/testing/cheat_codes.ts +78 -0
- package/src/testing/index.ts +3 -0
- package/dest/cli/chain_l2_config.d.ts +0 -19
- package/dest/cli/chain_l2_config.d.ts.map +0 -1
- package/dest/cli/chain_l2_config.js +0 -56
- package/dest/cli/cmds/start_faucet.d.ts +0 -4
- package/dest/cli/cmds/start_faucet.d.ts.map +0 -1
- package/dest/cli/cmds/start_faucet.js +0 -20
- package/dest/cli/cmds/start_pxe.d.ts +0 -16
- package/dest/cli/cmds/start_pxe.d.ts.map +0 -1
- package/dest/cli/cmds/start_pxe.js +0 -95
- package/dest/cli/get_l1_config.d.ts +0 -7
- package/dest/cli/get_l1_config.d.ts.map +0 -1
- package/dest/cli/get_l1_config.js +0 -13
- package/dest/sandbox/banana_fpc.d.ts +0 -11
- package/dest/sandbox/banana_fpc.d.ts.map +0 -1
- package/dest/sandbox/index.d.ts +0 -5
- package/dest/sandbox/index.d.ts.map +0 -1
- package/dest/sandbox/index.js +0 -4
- package/dest/sandbox/sandbox.d.ts +0 -76
- package/dest/sandbox/sandbox.d.ts.map +0 -1
- package/dest/sandbox/sponsored_fee_payment_method.d.ts +0 -23
- package/dest/sandbox/sponsored_fee_payment_method.d.ts.map +0 -1
- package/dest/sandbox/sponsored_fee_payment_method.js +0 -36
- package/dest/sandbox/sponsored_fpc.d.ts +0 -6
- package/dest/sandbox/sponsored_fpc.d.ts.map +0 -1
- package/dest/sandbox/sponsored_fpc.js +0 -26
- package/src/cli/chain_l2_config.ts +0 -74
- package/src/cli/cmds/start_faucet.ts +0 -34
- package/src/cli/cmds/start_pxe.ts +0 -129
- package/src/cli/get_l1_config.ts +0 -18
- package/src/sandbox/index.ts +0 -5
- package/src/sandbox/sponsored_fee_payment_method.ts +0 -46
- package/src/sandbox/sponsored_fpc.ts +0 -38
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { RollupCheatCodes } from '@aztec/ethereum/test';
|
|
2
|
+
import { SlotNumber } from '@aztec/foundation/branded-types';
|
|
3
|
+
import { createLogger } from '@aztec/foundation/log';
|
|
4
|
+
import { RunningPromise } from '@aztec/foundation/running-promise';
|
|
5
|
+
import { RollupAbi } from '@aztec/l1-artifacts/RollupAbi';
|
|
6
|
+
import { getAddress, getContract } from 'viem';
|
|
7
|
+
/**
|
|
8
|
+
* Represents a watcher for a rollup contract.
|
|
9
|
+
*
|
|
10
|
+
* It started on a network like anvil where time traveling is allowed, and auto-mine is turned on
|
|
11
|
+
* it will periodically check if the current slot have already been filled, e.g., there was an L2
|
|
12
|
+
* block within the slot. And if so, it will time travel into the next slot.
|
|
13
|
+
*/ export class AnvilTestWatcher {
|
|
14
|
+
cheatcodes;
|
|
15
|
+
dateProvider;
|
|
16
|
+
isLocalNetwork;
|
|
17
|
+
rollup;
|
|
18
|
+
rollupCheatCodes;
|
|
19
|
+
l2SlotDuration;
|
|
20
|
+
filledRunningPromise;
|
|
21
|
+
syncDateProviderPromise;
|
|
22
|
+
markingAsProvenRunningPromise;
|
|
23
|
+
logger;
|
|
24
|
+
isMarkingAsProven;
|
|
25
|
+
constructor(cheatcodes, rollupAddress, l1Client, dateProvider){
|
|
26
|
+
this.cheatcodes = cheatcodes;
|
|
27
|
+
this.dateProvider = dateProvider;
|
|
28
|
+
this.isLocalNetwork = false;
|
|
29
|
+
this.logger = createLogger(`aztecjs:utils:watcher`);
|
|
30
|
+
this.isMarkingAsProven = true;
|
|
31
|
+
this.rollup = getContract({
|
|
32
|
+
address: getAddress(rollupAddress.toString()),
|
|
33
|
+
abi: RollupAbi,
|
|
34
|
+
client: l1Client
|
|
35
|
+
});
|
|
36
|
+
this.rollupCheatCodes = new RollupCheatCodes(this.cheatcodes, {
|
|
37
|
+
rollupAddress
|
|
38
|
+
});
|
|
39
|
+
this.logger.debug(`Watcher created for rollup at ${rollupAddress}`);
|
|
40
|
+
}
|
|
41
|
+
setIsMarkingAsProven(isMarkingAsProven) {
|
|
42
|
+
this.logger.warn(`Watcher is now ${isMarkingAsProven ? 'marking' : 'not marking'} blocks as proven`);
|
|
43
|
+
this.isMarkingAsProven = isMarkingAsProven;
|
|
44
|
+
}
|
|
45
|
+
setisLocalNetwork(isLocalNetwork) {
|
|
46
|
+
this.isLocalNetwork = isLocalNetwork;
|
|
47
|
+
}
|
|
48
|
+
async start() {
|
|
49
|
+
if (this.filledRunningPromise) {
|
|
50
|
+
throw new Error('Watcher already watching for filled slot');
|
|
51
|
+
}
|
|
52
|
+
const config = await this.rollupCheatCodes.getConfig();
|
|
53
|
+
this.l2SlotDuration = config.slotDuration;
|
|
54
|
+
// If auto mining is not supported (e.g., we are on a real network), then we
|
|
55
|
+
// will simple do nothing. But if on an anvil or the like, this make sure that
|
|
56
|
+
// the local network and tests don't break because time is frozen and we never get to
|
|
57
|
+
// the next slot.
|
|
58
|
+
const isAutoMining = await this.cheatcodes.isAutoMining();
|
|
59
|
+
if (isAutoMining) {
|
|
60
|
+
this.filledRunningPromise = new RunningPromise(()=>this.warpTimeIfNeeded(), this.logger, 200);
|
|
61
|
+
this.filledRunningPromise.start();
|
|
62
|
+
this.syncDateProviderPromise = new RunningPromise(()=>this.syncDateProviderToL1IfBehind(), this.logger, 200);
|
|
63
|
+
this.syncDateProviderPromise.start();
|
|
64
|
+
this.markingAsProvenRunningPromise = new RunningPromise(()=>this.markAsProven(), this.logger, 200);
|
|
65
|
+
this.markingAsProvenRunningPromise.start();
|
|
66
|
+
this.logger.info(`Watcher started for rollup at ${this.rollup.address}`);
|
|
67
|
+
} else {
|
|
68
|
+
this.logger.info(`Watcher not started because not auto mining`);
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
async stop() {
|
|
72
|
+
await this.filledRunningPromise?.stop();
|
|
73
|
+
await this.syncDateProviderPromise?.stop();
|
|
74
|
+
await this.markingAsProvenRunningPromise?.stop();
|
|
75
|
+
}
|
|
76
|
+
async trigger() {
|
|
77
|
+
await this.filledRunningPromise?.trigger();
|
|
78
|
+
await this.syncDateProviderPromise?.trigger();
|
|
79
|
+
await this.markingAsProvenRunningPromise?.trigger();
|
|
80
|
+
}
|
|
81
|
+
async markAsProven() {
|
|
82
|
+
if (!this.isMarkingAsProven) {
|
|
83
|
+
return;
|
|
84
|
+
}
|
|
85
|
+
await this.rollupCheatCodes.markAsProven();
|
|
86
|
+
}
|
|
87
|
+
async syncDateProviderToL1IfBehind() {
|
|
88
|
+
// this doesn't apply to the local network, because we don't have a date provider in the local network
|
|
89
|
+
if (!this.dateProvider) {
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
const l1Time = await this.cheatcodes.timestamp() * 1000;
|
|
93
|
+
const wallTime = this.dateProvider.now();
|
|
94
|
+
if (l1Time > wallTime) {
|
|
95
|
+
this.logger.warn(`L1 is ahead of wall time. Syncing wall time to L1 time`);
|
|
96
|
+
this.dateProvider.setTime(l1Time);
|
|
97
|
+
} else if (l1Time + Number(this.l2SlotDuration) * 1000 < wallTime) {
|
|
98
|
+
this.logger.warn(`L1 is more than 1 L2 slot behind wall time. Warping to wall time`);
|
|
99
|
+
await this.cheatcodes.warp(Math.ceil(wallTime / 1000));
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
async warpTimeIfNeeded() {
|
|
103
|
+
try {
|
|
104
|
+
const currentSlot = SlotNumber.fromBigInt(await this.rollup.read.getCurrentSlot());
|
|
105
|
+
const pendingCheckpointNumber = BigInt(await this.rollup.read.getPendingCheckpointNumber());
|
|
106
|
+
const checkpointLog = await this.rollup.read.getCheckpoint([
|
|
107
|
+
pendingCheckpointNumber
|
|
108
|
+
]);
|
|
109
|
+
const nextSlot = SlotNumber(currentSlot + 1);
|
|
110
|
+
const nextSlotTimestamp = Number(await this.rollup.read.getTimestampForSlot([
|
|
111
|
+
BigInt(nextSlot)
|
|
112
|
+
]));
|
|
113
|
+
if (BigInt(currentSlot) === checkpointLog.slotNumber) {
|
|
114
|
+
// We should jump to the next slot
|
|
115
|
+
try {
|
|
116
|
+
await this.cheatcodes.warp(nextSlotTimestamp, {
|
|
117
|
+
resetBlockInterval: true
|
|
118
|
+
});
|
|
119
|
+
} catch (e) {
|
|
120
|
+
this.logger.error(`Failed to warp to timestamp ${nextSlotTimestamp}: ${e}`);
|
|
121
|
+
}
|
|
122
|
+
this.logger.info(`Slot ${currentSlot} was filled, jumped to next slot`);
|
|
123
|
+
return;
|
|
124
|
+
}
|
|
125
|
+
// If we are not in local network, we don't need to warp time
|
|
126
|
+
if (!this.isLocalNetwork) {
|
|
127
|
+
return;
|
|
128
|
+
}
|
|
129
|
+
const currentTimestamp = this.dateProvider?.now() ?? Date.now();
|
|
130
|
+
if (currentTimestamp > nextSlotTimestamp * 1000) {
|
|
131
|
+
try {
|
|
132
|
+
await this.cheatcodes.warp(nextSlotTimestamp, {
|
|
133
|
+
resetBlockInterval: true
|
|
134
|
+
});
|
|
135
|
+
} catch (e) {
|
|
136
|
+
this.logger.error(`Failed to warp to timestamp ${nextSlotTimestamp}: ${e}`);
|
|
137
|
+
}
|
|
138
|
+
this.logger.info(`Slot ${currentSlot} was missed, jumped to next slot`);
|
|
139
|
+
}
|
|
140
|
+
} catch {
|
|
141
|
+
this.logger.error('mineIfSlotFilled failed');
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
import { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
|
|
2
|
+
import type { DateProvider } from '@aztec/foundation/timer';
|
|
3
|
+
import type { SequencerClient } from '@aztec/sequencer-client';
|
|
4
|
+
import type { AztecNode } from '@aztec/stdlib/interfaces/client';
|
|
5
|
+
/**
|
|
6
|
+
* A class that provides utility functions for interacting with the chain.
|
|
7
|
+
* @deprecated There used to be 3 kinds of cheat codes: eth, rollup and aztec. We have nuked the Aztec ones because
|
|
8
|
+
* they became unused (we now have better testing tools). If you are introducing a new functionality to the cheat
|
|
9
|
+
* codes, please consider whether it makes sense to just introduce new utils in your tests instead.
|
|
10
|
+
*/
|
|
11
|
+
export declare class CheatCodes {
|
|
12
|
+
eth: EthCheatCodes;
|
|
13
|
+
rollup: RollupCheatCodes;
|
|
14
|
+
constructor(
|
|
15
|
+
/** Cheat codes for L1.*/
|
|
16
|
+
eth: EthCheatCodes,
|
|
17
|
+
/** Cheat codes for the Aztec Rollup contract on L1. */
|
|
18
|
+
rollup: RollupCheatCodes);
|
|
19
|
+
static create(rpcUrls: string[], node: AztecNode, dateProvider: DateProvider): Promise<CheatCodes>;
|
|
20
|
+
/**
|
|
21
|
+
* Warps the L1 timestamp to a target timestamp and mines an L2 block that advances the L2 timestamp to at least
|
|
22
|
+
* the target timestamp. L2 timestamp is not advanced exactly to the target timestamp because it is determined
|
|
23
|
+
* by the slot number, which advances in fixed intervals.
|
|
24
|
+
* This is useful for testing time-dependent contract behavior.
|
|
25
|
+
* @param sequencerClient - The sequencer client to use to force an empty block to be mined.
|
|
26
|
+
* @param node - The Aztec node used to query if a new block has been mined.
|
|
27
|
+
* @param targetTimestamp - The target timestamp to warp to (in seconds)
|
|
28
|
+
*/
|
|
29
|
+
warpL2TimeAtLeastTo(sequencerClient: SequencerClient, node: AztecNode, targetTimestamp: bigint | number): Promise<void>;
|
|
30
|
+
/**
|
|
31
|
+
* Warps the L1 timestamp forward by a specified duration and mines an L2 block that advances the L2 timestamp at
|
|
32
|
+
* least by the duration. L2 timestamp is not advanced exactly by the duration because it is determined by the slot
|
|
33
|
+
* number, which advances in fixed intervals.
|
|
34
|
+
* This is useful for testing time-dependent contract behavior.
|
|
35
|
+
* @param sequencerClient - The sequencer client to use to force an empty block to be mined.
|
|
36
|
+
* @param node - The Aztec node used to query if a new block has been mined.
|
|
37
|
+
* @param duration - The duration to advance time by (in seconds)
|
|
38
|
+
*/
|
|
39
|
+
warpL2TimeAtLeastBy(sequencerClient: SequencerClient, node: AztecNode, duration: bigint | number): Promise<void>;
|
|
40
|
+
}
|
|
41
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiY2hlYXRfY29kZXMuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90ZXN0aW5nL2NoZWF0X2NvZGVzLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxhQUFhLEVBQUUsZ0JBQWdCLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUV2RSxPQUFPLEtBQUssRUFBRSxZQUFZLEVBQUUsTUFBTSx5QkFBeUIsQ0FBQztBQUM1RCxPQUFPLEtBQUssRUFBRSxlQUFlLEVBQUUsTUFBTSx5QkFBeUIsQ0FBQztBQUMvRCxPQUFPLEtBQUssRUFBRSxTQUFTLEVBQUUsTUFBTSxpQ0FBaUMsQ0FBQztBQUVqRTs7Ozs7R0FLRztBQUNILHFCQUFhLFVBQVU7SUFHWixHQUFHLEVBQUUsYUFBYTtJQUVsQixNQUFNLEVBQUUsZ0JBQWdCO0lBSmpDO0lBQ0UseUJBQXlCO0lBQ2xCLEdBQUcsRUFBRSxhQUFhO0lBQ3pCLHVEQUF1RDtJQUNoRCxNQUFNLEVBQUUsZ0JBQWdCLEVBQzdCO0lBRUosT0FBYSxNQUFNLENBQUMsT0FBTyxFQUFFLE1BQU0sRUFBRSxFQUFFLElBQUksRUFBRSxTQUFTLEVBQUUsWUFBWSxFQUFFLFlBQVksR0FBRyxPQUFPLENBQUMsVUFBVSxDQUFDLENBT3ZHO0lBRUQ7Ozs7Ozs7O09BUUc7SUFDRyxtQkFBbUIsQ0FBQyxlQUFlLEVBQUUsZUFBZSxFQUFFLElBQUksRUFBRSxTQUFTLEVBQUUsZUFBZSxFQUFFLE1BQU0sR0FBRyxNQUFNLGlCQXVCNUc7SUFFRDs7Ozs7Ozs7T0FRRztJQUNHLG1CQUFtQixDQUFDLGVBQWUsRUFBRSxlQUFlLEVBQUUsSUFBSSxFQUFFLFNBQVMsRUFBRSxRQUFRLEVBQUUsTUFBTSxHQUFHLE1BQU0saUJBSXJHO0NBQ0YifQ==
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cheat_codes.d.ts","sourceRoot":"","sources":["../../src/testing/cheat_codes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAEvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAEjE;;;;;GAKG;AACH,qBAAa,UAAU;IAGZ,GAAG,EAAE,aAAa;IAElB,MAAM,EAAE,gBAAgB;IAJjC;IACE,yBAAyB;IAClB,GAAG,EAAE,aAAa;IACzB,uDAAuD;IAChD,MAAM,EAAE,gBAAgB,EAC7B;IAEJ,OAAa,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,CAOvG;IAED;;;;;;;;OAQG;IACG,mBAAmB,CAAC,eAAe,EAAE,eAAe,EAAE,IAAI,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,GAAG,MAAM,iBAuB5G;IAED;;;;;;;;OAQG;IACG,mBAAmB,CAAC,eAAe,EAAE,eAAe,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,iBAIrG;CACF"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
|
|
2
|
+
import { retryUntil } from '@aztec/foundation/retry';
|
|
3
|
+
/**
|
|
4
|
+
* A class that provides utility functions for interacting with the chain.
|
|
5
|
+
* @deprecated There used to be 3 kinds of cheat codes: eth, rollup and aztec. We have nuked the Aztec ones because
|
|
6
|
+
* they became unused (we now have better testing tools). If you are introducing a new functionality to the cheat
|
|
7
|
+
* codes, please consider whether it makes sense to just introduce new utils in your tests instead.
|
|
8
|
+
*/ export class CheatCodes {
|
|
9
|
+
eth;
|
|
10
|
+
rollup;
|
|
11
|
+
constructor(/** Cheat codes for L1.*/ eth, /** Cheat codes for the Aztec Rollup contract on L1. */ rollup){
|
|
12
|
+
this.eth = eth;
|
|
13
|
+
this.rollup = rollup;
|
|
14
|
+
}
|
|
15
|
+
static async create(rpcUrls, node, dateProvider) {
|
|
16
|
+
const ethCheatCodes = new EthCheatCodes(rpcUrls, dateProvider);
|
|
17
|
+
const rollupCheatCodes = new RollupCheatCodes(ethCheatCodes, await node.getNodeInfo().then((n)=>n.l1ContractAddresses));
|
|
18
|
+
return new CheatCodes(ethCheatCodes, rollupCheatCodes);
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Warps the L1 timestamp to a target timestamp and mines an L2 block that advances the L2 timestamp to at least
|
|
22
|
+
* the target timestamp. L2 timestamp is not advanced exactly to the target timestamp because it is determined
|
|
23
|
+
* by the slot number, which advances in fixed intervals.
|
|
24
|
+
* This is useful for testing time-dependent contract behavior.
|
|
25
|
+
* @param sequencerClient - The sequencer client to use to force an empty block to be mined.
|
|
26
|
+
* @param node - The Aztec node used to query if a new block has been mined.
|
|
27
|
+
* @param targetTimestamp - The target timestamp to warp to (in seconds)
|
|
28
|
+
*/ async warpL2TimeAtLeastTo(sequencerClient, node, targetTimestamp) {
|
|
29
|
+
const currentL2BlockNumber = await node.getBlockNumber();
|
|
30
|
+
// We warp the L1 timestamp
|
|
31
|
+
await this.eth.warp(targetTimestamp, {
|
|
32
|
+
resetBlockInterval: true
|
|
33
|
+
});
|
|
34
|
+
// Wait until an L2 block is mined
|
|
35
|
+
const sequencer = sequencerClient.getSequencer();
|
|
36
|
+
const minTxsPerBlock = sequencer.getConfig().minTxsPerBlock;
|
|
37
|
+
sequencer.updateConfig({
|
|
38
|
+
minTxsPerBlock: 0
|
|
39
|
+
});
|
|
40
|
+
await retryUntil(async ()=>{
|
|
41
|
+
const newL2BlockNumber = await node.getBlockNumber();
|
|
42
|
+
return newL2BlockNumber > currentL2BlockNumber;
|
|
43
|
+
}, 'new block after warping L2 time', 36, 1);
|
|
44
|
+
// Restore original minTxsPerBlock
|
|
45
|
+
sequencer.updateConfig({
|
|
46
|
+
minTxsPerBlock
|
|
47
|
+
});
|
|
48
|
+
}
|
|
49
|
+
/**
|
|
50
|
+
* Warps the L1 timestamp forward by a specified duration and mines an L2 block that advances the L2 timestamp at
|
|
51
|
+
* least by the duration. L2 timestamp is not advanced exactly by the duration because it is determined by the slot
|
|
52
|
+
* number, which advances in fixed intervals.
|
|
53
|
+
* This is useful for testing time-dependent contract behavior.
|
|
54
|
+
* @param sequencerClient - The sequencer client to use to force an empty block to be mined.
|
|
55
|
+
* @param node - The Aztec node used to query if a new block has been mined.
|
|
56
|
+
* @param duration - The duration to advance time by (in seconds)
|
|
57
|
+
*/ async warpL2TimeAtLeastBy(sequencerClient, node, duration) {
|
|
58
|
+
const currentTimestamp = await this.eth.timestamp();
|
|
59
|
+
const targetTimestamp = BigInt(currentTimestamp) + BigInt(duration);
|
|
60
|
+
await this.warpL2TimeAtLeastTo(sequencerClient, node, targetTimestamp);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
export { AnvilTestWatcher } from './anvil_test_watcher.js';
|
|
2
|
+
export { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
|
|
3
|
+
export { CheatCodes } from './cheat_codes.js';
|
|
4
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90ZXN0aW5nL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxnQkFBZ0IsRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQzNELE9BQU8sRUFBRSxhQUFhLEVBQUUsZ0JBQWdCLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUN2RSxPQUFPLEVBQUUsVUFBVSxFQUFFLE1BQU0sa0JBQWtCLENBQUMifQ==
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/testing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,11 +1,12 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@aztec/aztec",
|
|
3
|
-
"version": "0.0.
|
|
3
|
+
"version": "0.0.1-commit.21caa21",
|
|
4
4
|
"type": "module",
|
|
5
5
|
"exports": {
|
|
6
|
-
".": "./dest/index.js"
|
|
6
|
+
".": "./dest/index.js",
|
|
7
|
+
"./testing": "./dest/testing/index.js"
|
|
7
8
|
},
|
|
8
|
-
"bin": "./
|
|
9
|
+
"bin": "./scripts/aztec.sh",
|
|
9
10
|
"typedocOptions": {
|
|
10
11
|
"entryPoints": [
|
|
11
12
|
"./src/index.ts"
|
|
@@ -14,14 +15,12 @@
|
|
|
14
15
|
"tsconfig": "./tsconfig.json"
|
|
15
16
|
},
|
|
16
17
|
"scripts": {
|
|
17
|
-
"build": "yarn clean &&
|
|
18
|
+
"build": "yarn clean && tsgo -b",
|
|
18
19
|
"start": "node --no-warnings ./dest/bin",
|
|
19
20
|
"start:debug": "node --inspect=0.0.0.0:9221 --no-warnings ./dest/bin",
|
|
20
|
-
"start:
|
|
21
|
+
"start:local-network": "ETHEREUM_HOSTS=http://0.0.0.0:8545/ && yarn start start --local-network",
|
|
21
22
|
"clean": "rm -rf ./dest .tsbuildinfo",
|
|
22
|
-
"
|
|
23
|
-
"formatting:fix": "run -T eslint --fix ./src && run -T prettier -w ./src",
|
|
24
|
-
"build:dev": "tsc -b --watch",
|
|
23
|
+
"build:dev": "tsgo -b --watch",
|
|
25
24
|
"test": "NODE_NO_WARNINGS=1 node --experimental-vm-modules ../node_modules/.bin/jest --passWithNoTests --maxWorkers=${JEST_MAX_WORKERS:-8}",
|
|
26
25
|
"run:example:token": "LOG_LEVEL='verbose' node ./dest/examples/token.js"
|
|
27
26
|
},
|
|
@@ -29,55 +28,60 @@
|
|
|
29
28
|
"../package.common.json"
|
|
30
29
|
],
|
|
31
30
|
"dependencies": {
|
|
32
|
-
"@aztec/accounts": "0.0.
|
|
33
|
-
"@aztec/archiver": "0.0.
|
|
34
|
-
"@aztec/aztec-faucet": "0.0.
|
|
35
|
-
"@aztec/aztec-node": "0.0.
|
|
36
|
-
"@aztec/aztec.js": "0.0.
|
|
37
|
-
"@aztec/bb-prover": "0.0.
|
|
38
|
-
"@aztec/
|
|
39
|
-
"@aztec/
|
|
40
|
-
"@aztec/
|
|
41
|
-
"@aztec/
|
|
42
|
-
"@aztec/cli
|
|
43
|
-
"@aztec/constants": "0.0.
|
|
44
|
-
"@aztec/entrypoints": "0.0.
|
|
45
|
-
"@aztec/ethereum": "0.0.
|
|
46
|
-
"@aztec/foundation": "0.0.
|
|
47
|
-
"@aztec/kv-store": "0.0.
|
|
48
|
-
"@aztec/
|
|
49
|
-
"@aztec/
|
|
50
|
-
"@aztec/
|
|
51
|
-
"@aztec/
|
|
52
|
-
"@aztec/
|
|
53
|
-
"@aztec/
|
|
54
|
-
"@aztec/
|
|
55
|
-
"@aztec/
|
|
56
|
-
"@aztec/
|
|
57
|
-
"@aztec/
|
|
58
|
-
"@aztec/
|
|
59
|
-
"@aztec/
|
|
31
|
+
"@aztec/accounts": "0.0.1-commit.21caa21",
|
|
32
|
+
"@aztec/archiver": "0.0.1-commit.21caa21",
|
|
33
|
+
"@aztec/aztec-faucet": "0.0.1-commit.21caa21",
|
|
34
|
+
"@aztec/aztec-node": "0.0.1-commit.21caa21",
|
|
35
|
+
"@aztec/aztec.js": "0.0.1-commit.21caa21",
|
|
36
|
+
"@aztec/bb-prover": "0.0.1-commit.21caa21",
|
|
37
|
+
"@aztec/bb.js": "0.0.1-commit.21caa21",
|
|
38
|
+
"@aztec/blob-sink": "0.0.1-commit.21caa21",
|
|
39
|
+
"@aztec/bot": "0.0.1-commit.21caa21",
|
|
40
|
+
"@aztec/builder": "0.0.1-commit.21caa21",
|
|
41
|
+
"@aztec/cli": "0.0.1-commit.21caa21",
|
|
42
|
+
"@aztec/constants": "0.0.1-commit.21caa21",
|
|
43
|
+
"@aztec/entrypoints": "0.0.1-commit.21caa21",
|
|
44
|
+
"@aztec/ethereum": "0.0.1-commit.21caa21",
|
|
45
|
+
"@aztec/foundation": "0.0.1-commit.21caa21",
|
|
46
|
+
"@aztec/kv-store": "0.0.1-commit.21caa21",
|
|
47
|
+
"@aztec/l1-artifacts": "0.0.1-commit.21caa21",
|
|
48
|
+
"@aztec/node-lib": "0.0.1-commit.21caa21",
|
|
49
|
+
"@aztec/noir-contracts.js": "0.0.1-commit.21caa21",
|
|
50
|
+
"@aztec/noir-protocol-circuits-types": "0.0.1-commit.21caa21",
|
|
51
|
+
"@aztec/p2p": "0.0.1-commit.21caa21",
|
|
52
|
+
"@aztec/p2p-bootstrap": "0.0.1-commit.21caa21",
|
|
53
|
+
"@aztec/protocol-contracts": "0.0.1-commit.21caa21",
|
|
54
|
+
"@aztec/prover-client": "0.0.1-commit.21caa21",
|
|
55
|
+
"@aztec/prover-node": "0.0.1-commit.21caa21",
|
|
56
|
+
"@aztec/pxe": "0.0.1-commit.21caa21",
|
|
57
|
+
"@aztec/stdlib": "0.0.1-commit.21caa21",
|
|
58
|
+
"@aztec/telemetry-client": "0.0.1-commit.21caa21",
|
|
59
|
+
"@aztec/test-wallet": "0.0.1-commit.21caa21",
|
|
60
|
+
"@aztec/txe": "0.0.1-commit.21caa21",
|
|
61
|
+
"@aztec/world-state": "0.0.1-commit.21caa21",
|
|
60
62
|
"@types/chalk": "^2.2.0",
|
|
61
63
|
"abitype": "^0.8.11",
|
|
62
64
|
"chalk": "^5.3.0",
|
|
63
65
|
"commander": "^12.1.0",
|
|
64
|
-
"koa": "^2.
|
|
65
|
-
"koa-router": "^
|
|
66
|
-
"viem": "2.
|
|
66
|
+
"koa": "^2.16.1",
|
|
67
|
+
"koa-router": "^13.1.1",
|
|
68
|
+
"viem": "npm:@aztec/viem@2.38.2"
|
|
67
69
|
},
|
|
68
70
|
"files": [
|
|
69
71
|
"dest",
|
|
72
|
+
"scripts",
|
|
70
73
|
"src",
|
|
71
74
|
"!*.test.*"
|
|
72
75
|
],
|
|
73
76
|
"types": "./dest/index.d.ts",
|
|
74
77
|
"devDependencies": {
|
|
75
|
-
"@jest/globals": "^
|
|
76
|
-
"@types/jest": "^
|
|
77
|
-
"@types/koa": "^2.
|
|
78
|
-
"
|
|
78
|
+
"@jest/globals": "^30.0.0",
|
|
79
|
+
"@types/jest": "^30.0.0",
|
|
80
|
+
"@types/koa": "^2.15.0",
|
|
81
|
+
"@typescript/native-preview": "7.0.0-dev.20251126.1",
|
|
82
|
+
"jest": "^30.0.0",
|
|
79
83
|
"ts-node": "^10.9.1",
|
|
80
|
-
"typescript": "^5.
|
|
84
|
+
"typescript": "^5.3.3"
|
|
81
85
|
},
|
|
82
86
|
"jest": {
|
|
83
87
|
"moduleNameMapper": {
|
|
@@ -110,9 +114,13 @@
|
|
|
110
114
|
"testTimeout": 120000,
|
|
111
115
|
"setupFiles": [
|
|
112
116
|
"../../foundation/src/jest/setup.mjs"
|
|
117
|
+
],
|
|
118
|
+
"testEnvironment": "../../foundation/src/jest/env.mjs",
|
|
119
|
+
"setupFilesAfterEnv": [
|
|
120
|
+
"../../foundation/src/jest/setupAfterEnv.mjs"
|
|
113
121
|
]
|
|
114
122
|
},
|
|
115
123
|
"engines": {
|
|
116
|
-
"node": ">=
|
|
124
|
+
"node": ">=20.10"
|
|
117
125
|
}
|
|
118
126
|
}
|
package/scripts/aztec.sh
ADDED
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
#!/usr/bin/env bash
|
|
2
|
+
set -euo pipefail
|
|
3
|
+
|
|
4
|
+
script_dir="$(dirname "$(realpath "$0")")"
|
|
5
|
+
|
|
6
|
+
function aztec {
|
|
7
|
+
node --no-warnings $script_dir/../dest/bin/index.js "$@"
|
|
8
|
+
}
|
|
9
|
+
|
|
10
|
+
cmd=${1:-}
|
|
11
|
+
[ -n "$cmd" ] && shift
|
|
12
|
+
|
|
13
|
+
export AZTEC_SHELL_WRAPPER=1
|
|
14
|
+
|
|
15
|
+
case $cmd in
|
|
16
|
+
test)
|
|
17
|
+
export LOG_LEVEL="${LOG_LEVEL:-info}"
|
|
18
|
+
aztec start --txe --port 8081 &
|
|
19
|
+
server_pid=$!
|
|
20
|
+
trap 'kill $server_pid &>/dev/null || true' EXIT
|
|
21
|
+
while ! nc -z 127.0.0.1 8081 &>/dev/null; do sleep 0.2; done
|
|
22
|
+
export NARGO_FOREIGN_CALL_TIMEOUT=300000
|
|
23
|
+
nargo test --silence-warnings --pedantic-solving --oracle-resolver http://127.0.0.1:8081 "$@"
|
|
24
|
+
;;
|
|
25
|
+
start)
|
|
26
|
+
if [ "${1:-}" == "--local-network" ]; then
|
|
27
|
+
# TODO: Can these just be set in TS?
|
|
28
|
+
export ARCHIVER_POLLING_INTERVAL_MS=500
|
|
29
|
+
export P2P_BLOCK_CHECK_INTERVAL_MS=500
|
|
30
|
+
export SEQ_TX_POLLING_INTERVAL_MS=500
|
|
31
|
+
export WS_BLOCK_CHECK_INTERVAL_MS=500
|
|
32
|
+
export ARCHIVER_VIEM_POLLING_INTERVAL_MS=500
|
|
33
|
+
export TEST_ACCOUNTS=${TEST_ACCOUNTS:-true}
|
|
34
|
+
export LOG_LEVEL=${LOG_LEVEL:-info;silent:sequencer;verbose:debug_log}
|
|
35
|
+
export DEPLOY_AZTEC_CONTRACTS_SALT=${DEPLOY_AZTEC_CONTRACTS_SALT:-$RANDOM}
|
|
36
|
+
|
|
37
|
+
ANVIL_PORT=${ANVIL_PORT:-8545}
|
|
38
|
+
|
|
39
|
+
export L1_CHAIN_ID=${L1_CHAIN_ID:-31337}
|
|
40
|
+
export ETHEREUM_HOSTS=${ETHEREUM_HOSTS:-"http://127.0.0.1:${ANVIL_PORT}"}
|
|
41
|
+
|
|
42
|
+
anvil --version
|
|
43
|
+
anvil --silent &
|
|
44
|
+
anvil_pid=$!
|
|
45
|
+
trap 'kill $anvil_pid &>/dev/null' EXIT
|
|
46
|
+
fi
|
|
47
|
+
|
|
48
|
+
aztec start "$@"
|
|
49
|
+
;;
|
|
50
|
+
compile|new|init|flamegraph)
|
|
51
|
+
$script_dir/${cmd}.sh "$@"
|
|
52
|
+
;;
|
|
53
|
+
fmt|check|lsp)
|
|
54
|
+
# TODO: These should be removed, just use nargo directly.
|
|
55
|
+
nargo $cmd "$@"
|
|
56
|
+
;;
|
|
57
|
+
*)
|
|
58
|
+
aztec $cmd "$@"
|
|
59
|
+
;;
|
|
60
|
+
esac
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
#!/usr/bin/env bash
|
|
2
|
+
set -euo pipefail
|
|
3
|
+
|
|
4
|
+
NARGO=${NARGO:-nargo}
|
|
5
|
+
BB=${BB:-bb}
|
|
6
|
+
|
|
7
|
+
# If help is requested, show Aztec-specific info then run nargo compile help and then exit in order to not trigger
|
|
8
|
+
# transpilation
|
|
9
|
+
for arg in "$@"; do
|
|
10
|
+
if [ "$arg" == "--help" ] || [ "$arg" == "-h" ]; then
|
|
11
|
+
cat << 'EOF'
|
|
12
|
+
Aztec Compile - Compile Aztec Noir contracts
|
|
13
|
+
|
|
14
|
+
This command compiles Aztec Noir contracts using nargo and then automatically
|
|
15
|
+
postprocesses them to generate Aztec specific artifacts including:
|
|
16
|
+
- Transpiled contract artifacts
|
|
17
|
+
- Verification keys
|
|
18
|
+
|
|
19
|
+
The compiled contracts will be placed in the target/ directory by default.
|
|
20
|
+
|
|
21
|
+
---
|
|
22
|
+
Underlying nargo compile options:
|
|
23
|
+
|
|
24
|
+
EOF
|
|
25
|
+
nargo compile --help
|
|
26
|
+
exit 0
|
|
27
|
+
fi
|
|
28
|
+
done
|
|
29
|
+
|
|
30
|
+
# Run nargo compile.
|
|
31
|
+
$NARGO compile "$@"
|
|
32
|
+
|
|
33
|
+
echo "Postprocessing contract..."
|
|
34
|
+
$BB aztec_process
|
|
35
|
+
|
|
36
|
+
# Strip internal prefixes from all compiled contract JSONs in target directory
|
|
37
|
+
# TODO: This should be part of bb aztec_process!
|
|
38
|
+
for json in target/*.json; do
|
|
39
|
+
temp_file="${json}.tmp"
|
|
40
|
+
jq '.functions |= map(.name |= sub("^__aztec_nr_internals__"; ""))' "$json" > "$temp_file"
|
|
41
|
+
mv "$temp_file" "$json"
|
|
42
|
+
done
|
|
43
|
+
|
|
44
|
+
echo "Compilation complete!"
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import fs from 'fs/promises';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
|
|
5
|
+
// Simple script to extract a contract function as a separate Noir artifact.
|
|
6
|
+
// We need to use this since the transpiling that we do on public functions make the contract artifacts
|
|
7
|
+
// unreadable by noir tooling, since they are no longer following the noir artifact format.
|
|
8
|
+
async function main() {
|
|
9
|
+
let [contractArtifactPath, functionName] = process.argv.slice(2);
|
|
10
|
+
if (!contractArtifactPath || !functionName) {
|
|
11
|
+
console.log('Usage: node extractFunctionAsNoirArtifact.js <contractArtifactPath> <functionName>');
|
|
12
|
+
return;
|
|
13
|
+
}
|
|
14
|
+
|
|
15
|
+
const contractArtifact = JSON.parse(await fs.readFile(contractArtifactPath, 'utf8'));
|
|
16
|
+
const func = contractArtifact.functions.find(f => f.name === functionName);
|
|
17
|
+
if (!func) {
|
|
18
|
+
console.error(`Function ${functionName} not found in ${contractArtifactPath}`);
|
|
19
|
+
return;
|
|
20
|
+
}
|
|
21
|
+
|
|
22
|
+
const artifact = {
|
|
23
|
+
noir_version: contractArtifact.noir_version,
|
|
24
|
+
hash: 0,
|
|
25
|
+
abi: func.abi,
|
|
26
|
+
bytecode: func.bytecode,
|
|
27
|
+
debug_symbols: func.debug_symbols,
|
|
28
|
+
file_map: contractArtifact.file_map,
|
|
29
|
+
expression_width: {
|
|
30
|
+
Bounded: {
|
|
31
|
+
width: 4,
|
|
32
|
+
},
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
|
|
36
|
+
const outputDir = path.dirname(contractArtifactPath);
|
|
37
|
+
const outputName = path.basename(contractArtifactPath, '.json') + `-${functionName}.json`;
|
|
38
|
+
|
|
39
|
+
const outPath = path.join(outputDir, outputName);
|
|
40
|
+
|
|
41
|
+
await fs.writeFile(outPath, JSON.stringify(artifact, null, 2));
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
main().catch(err => {
|
|
45
|
+
console.error(err);
|
|
46
|
+
process.exit(1);
|
|
47
|
+
});
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
#!/usr/bin/env bash
|
|
2
|
+
set -eu
|
|
3
|
+
|
|
4
|
+
# If first arg is -h or --help, print usage.
|
|
5
|
+
if [ $# -lt 2 ] || [ "$1" == "-h" ] || [ "$1" == "--help" ]; then
|
|
6
|
+
cat << 'EOF'
|
|
7
|
+
Aztec Flamegraph - Generate a gate count flamegraph for an aztec contract function.
|
|
8
|
+
|
|
9
|
+
Usage: aztec flamegraph <contract_artifact> <function>
|
|
10
|
+
|
|
11
|
+
Options:
|
|
12
|
+
-h, --help Print help
|
|
13
|
+
|
|
14
|
+
Will output an svg at <artifact_path>/<contract>-<function>-flamegraph.svg.
|
|
15
|
+
You can open it in your browser to view it.
|
|
16
|
+
|
|
17
|
+
EOF
|
|
18
|
+
exit 0
|
|
19
|
+
fi
|
|
20
|
+
|
|
21
|
+
cleanup() {
|
|
22
|
+
set +e
|
|
23
|
+
if [ -f "$function_artifact" ]; then
|
|
24
|
+
rm -f "$function_artifact"
|
|
25
|
+
fi
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
trap cleanup EXIT
|
|
29
|
+
|
|
30
|
+
# Get the directory of the script
|
|
31
|
+
script_dir=$(realpath $(dirname $0))
|
|
32
|
+
|
|
33
|
+
PROFILER=${PROFILER_PATH:-noir-profiler}
|
|
34
|
+
BB=${BB:-bb}
|
|
35
|
+
|
|
36
|
+
# first console arg is contract name in camel case or path to contract artifact
|
|
37
|
+
contract=$1
|
|
38
|
+
|
|
39
|
+
# second console arg is the contract function
|
|
40
|
+
function=$2
|
|
41
|
+
|
|
42
|
+
if [ ! -f "$contract" ]; then
|
|
43
|
+
echo "Error: Contract artifact not found at: $contract"
|
|
44
|
+
exit 1
|
|
45
|
+
fi
|
|
46
|
+
artifact_path=$contract
|
|
47
|
+
function_artifact="${artifact_path%%.json}-${function}.json"
|
|
48
|
+
output_dir=$(dirname "$artifact_path")
|
|
49
|
+
|
|
50
|
+
# Extract artifact for the specific function.
|
|
51
|
+
node $script_dir/extract_function.js "$artifact_path" $function
|
|
52
|
+
|
|
53
|
+
# Generate the flamegraph
|
|
54
|
+
$PROFILER gates --artifact-path "$function_artifact" --backend-path "$BB" --backend-gates-command "gates" --output "$output_dir" --scheme chonk --include_gates_per_opcode
|
|
55
|
+
|
|
56
|
+
# Save as $artifact_name-$function-flamegraph.svg
|
|
57
|
+
output_file="${function_artifact%%.json}-flamegraph.svg"
|
|
58
|
+
mv "$output_dir/__aztec_nr_internals__${function}_gates.svg" "$output_file"
|
|
59
|
+
echo "Flamegraph generated at: $output_file"
|