@aztec/aztec 0.0.1-commit.ff7989d6c → 0.0.1-commit.ffe5b04ea

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.
Files changed (89) hide show
  1. package/dest/bin/index.js +4 -2
  2. package/dest/cli/aztec_start_action.d.ts +1 -1
  3. package/dest/cli/aztec_start_action.d.ts.map +1 -1
  4. package/dest/cli/aztec_start_action.js +5 -3
  5. package/dest/cli/aztec_start_options.d.ts +1 -1
  6. package/dest/cli/aztec_start_options.d.ts.map +1 -1
  7. package/dest/cli/aztec_start_options.js +2 -3
  8. package/dest/cli/cli.js +3 -3
  9. package/dest/cli/cmds/compile.d.ts +1 -1
  10. package/dest/cli/cmds/compile.d.ts.map +1 -1
  11. package/dest/cli/cmds/compile.js +100 -35
  12. package/dest/cli/cmds/profile.d.ts +4 -0
  13. package/dest/cli/cmds/profile.d.ts.map +1 -0
  14. package/dest/cli/cmds/profile.js +8 -0
  15. package/dest/cli/cmds/profile_flamegraph.d.ts +4 -0
  16. package/dest/cli/cmds/profile_flamegraph.d.ts.map +1 -0
  17. package/dest/cli/cmds/profile_flamegraph.js +51 -0
  18. package/dest/cli/cmds/profile_gates.d.ts +4 -0
  19. package/dest/cli/cmds/profile_gates.d.ts.map +1 -0
  20. package/dest/cli/cmds/profile_gates.js +57 -0
  21. package/dest/cli/cmds/profile_utils.d.ts +18 -0
  22. package/dest/cli/cmds/profile_utils.d.ts.map +1 -0
  23. package/dest/cli/cmds/profile_utils.js +50 -0
  24. package/dest/cli/cmds/standby.d.ts +27 -0
  25. package/dest/cli/cmds/standby.d.ts.map +1 -0
  26. package/dest/cli/cmds/standby.js +78 -0
  27. package/dest/cli/cmds/start_archiver.d.ts +2 -2
  28. package/dest/cli/cmds/start_archiver.d.ts.map +1 -1
  29. package/dest/cli/cmds/start_archiver.js +1 -1
  30. package/dest/cli/cmds/start_node.d.ts +3 -2
  31. package/dest/cli/cmds/start_node.d.ts.map +1 -1
  32. package/dest/cli/cmds/start_node.js +15 -17
  33. package/dest/cli/cmds/start_prover_broker.d.ts +1 -1
  34. package/dest/cli/cmds/start_prover_broker.d.ts.map +1 -1
  35. package/dest/cli/cmds/start_prover_broker.js +6 -6
  36. package/dest/cli/cmds/utils/artifacts.d.ts +21 -0
  37. package/dest/cli/cmds/utils/artifacts.d.ts.map +1 -0
  38. package/dest/cli/cmds/utils/artifacts.js +24 -0
  39. package/dest/cli/cmds/utils/spawn.d.ts +3 -0
  40. package/dest/cli/cmds/utils/spawn.d.ts.map +1 -0
  41. package/dest/cli/cmds/utils/spawn.js +16 -0
  42. package/dest/cli/util.d.ts +3 -5
  43. package/dest/cli/util.d.ts.map +1 -1
  44. package/dest/cli/util.js +40 -81
  45. package/dest/examples/token.js +4 -4
  46. package/dest/local-network/banana_fpc.js +1 -1
  47. package/dest/local-network/local-network.d.ts +1 -1
  48. package/dest/local-network/local-network.d.ts.map +1 -1
  49. package/dest/local-network/local-network.js +18 -5
  50. package/dest/testing/index.d.ts +2 -1
  51. package/dest/testing/index.d.ts.map +1 -1
  52. package/dest/testing/index.js +1 -0
  53. package/dest/testing/token_allowed_setup.d.ts +7 -0
  54. package/dest/testing/token_allowed_setup.d.ts.map +1 -0
  55. package/dest/testing/token_allowed_setup.js +20 -0
  56. package/package.json +34 -34
  57. package/scripts/add_crate.sh +102 -0
  58. package/scripts/aztec.sh +6 -2
  59. package/scripts/init.sh +23 -19
  60. package/scripts/new.sh +48 -24
  61. package/scripts/setup_workspace.sh +68 -0
  62. package/src/bin/index.ts +4 -2
  63. package/src/cli/aztec_start_action.ts +5 -3
  64. package/src/cli/aztec_start_options.ts +2 -3
  65. package/src/cli/cli.ts +3 -3
  66. package/src/cli/cmds/compile.ts +113 -36
  67. package/src/cli/cmds/profile.ts +25 -0
  68. package/src/cli/cmds/profile_flamegraph.ts +63 -0
  69. package/src/cli/cmds/profile_gates.ts +67 -0
  70. package/src/cli/cmds/profile_utils.ts +58 -0
  71. package/src/cli/cmds/standby.ts +111 -0
  72. package/src/cli/cmds/start_archiver.ts +1 -1
  73. package/src/cli/cmds/start_node.ts +25 -24
  74. package/src/cli/cmds/start_prover_broker.ts +7 -14
  75. package/src/cli/cmds/utils/artifacts.ts +44 -0
  76. package/src/cli/cmds/utils/spawn.ts +16 -0
  77. package/src/cli/util.ts +43 -76
  78. package/src/examples/token.ts +6 -4
  79. package/src/local-network/banana_fpc.ts +1 -1
  80. package/src/local-network/local-network.ts +13 -3
  81. package/src/testing/index.ts +1 -0
  82. package/src/testing/token_allowed_setup.ts +19 -0
  83. package/dest/cli/release_version.d.ts +0 -2
  84. package/dest/cli/release_version.d.ts.map +0 -1
  85. package/dest/cli/release_version.js +0 -14
  86. package/scripts/extract_function.js +0 -47
  87. package/scripts/flamegraph.sh +0 -59
  88. package/scripts/setup_project.sh +0 -31
  89. package/src/cli/release_version.ts +0 -21
package/src/cli/cli.ts CHANGED
@@ -37,9 +37,9 @@ export function injectAztecCommands(program: Command, userLog: LogFn, debugLogge
37
37
  `
38
38
  Additional commands:
39
39
 
40
- init [folder] [options] creates a new Aztec Noir project.
41
- new <path> [options] creates a new Aztec Noir project in a new directory.
42
- test [options] starts a TXE and runs "nargo test" using it as the oracle resolver.
40
+ init creates a new Aztec Noir workspace in the current directory.
41
+ new <name> creates a new Aztec Noir workspace in its own directory (or creates a new contract-test crates pair and adds it to the current workspace if run in workspace).
42
+ test [options] starts a TXE and runs "nargo test" using it as the oracle resolver.
43
43
  `,
44
44
  );
45
45
  }
@@ -1,51 +1,25 @@
1
1
  import type { LogFn } from '@aztec/foundation/log';
2
2
 
3
- import { execFileSync, spawn } from 'child_process';
3
+ import { execFileSync } from 'child_process';
4
4
  import type { Command } from 'commander';
5
- import { readFile, readdir, writeFile } from 'fs/promises';
5
+ import { readFile, writeFile } from 'fs/promises';
6
6
  import { join } from 'path';
7
7
 
8
- /** Spawns a command with inherited stdio and rejects on non-zero exit. */
9
- function run(cmd: string, args: string[]): Promise<void> {
10
- return new Promise((resolve, reject) => {
11
- const child = spawn(cmd, args, { stdio: 'inherit' });
12
- child.on('error', reject);
13
- child.on('close', code => {
14
- if (code !== 0) {
15
- reject(new Error(`${cmd} exited with code ${code}`));
16
- } else {
17
- resolve();
18
- }
19
- });
20
- });
21
- }
8
+ import { readArtifactFiles } from './utils/artifacts.js';
9
+ import { run } from './utils/spawn.js';
22
10
 
23
- /** Returns paths to contract artifacts in the target directory.
24
- * Contract artifacts are identified by having a `functions` array in the JSON.
25
- */
11
+ /** Returns paths to contract artifacts in the target directory. */
26
12
  async function collectContractArtifacts(): Promise<string[]> {
27
- let files: string[];
13
+ let files;
28
14
  try {
29
- files = await readdir('target');
15
+ files = await readArtifactFiles('target');
30
16
  } catch (err: any) {
31
- if (err?.code === 'ENOENT') {
17
+ if (err?.message?.includes('does not exist')) {
32
18
  return [];
33
19
  }
34
- throw new Error(`Failed to read target directory: ${err.message}`);
35
- }
36
-
37
- const artifacts: string[] = [];
38
- for (const file of files) {
39
- if (!file.endsWith('.json')) {
40
- continue;
41
- }
42
- const filePath = join('target', file);
43
- const content = JSON.parse(await readFile(filePath, 'utf-8'));
44
- if (Array.isArray(content.functions)) {
45
- artifacts.push(filePath);
46
- }
20
+ throw err;
47
21
  }
48
- return artifacts;
22
+ return files.filter(f => Array.isArray(f.content.functions)).map(f => f.filePath);
49
23
  }
50
24
 
51
25
  /** Strips the `__aztec_nr_internals__` prefix from function names in contract artifacts. */
@@ -61,6 +35,106 @@ async function stripInternalPrefixes(artifactPaths: string[]): Promise<void> {
61
35
  }
62
36
  }
63
37
 
38
+ /** Returns the set of package names that are contract crates in the current workspace. */
39
+ async function getContractPackageNames(): Promise<Set<string>> {
40
+ const contractNames = new Set<string>();
41
+
42
+ let rootToml: string;
43
+ try {
44
+ rootToml = await readFile('Nargo.toml', 'utf-8');
45
+ } catch {
46
+ return contractNames;
47
+ }
48
+
49
+ const membersMatch = rootToml.match(/members\s*=\s*\[([^\]]*)\]/);
50
+ if (membersMatch) {
51
+ const members = membersMatch[1]
52
+ .split(',')
53
+ .map(m => m.trim().replace(/^"|"$/g, ''))
54
+ .filter(m => m.length > 0);
55
+
56
+ for (const member of members) {
57
+ try {
58
+ const memberToml = await readFile(join(member, 'Nargo.toml'), 'utf-8');
59
+ if (/type\s*=\s*"contract"/.test(memberToml)) {
60
+ const nameMatch = memberToml.match(/name\s*=\s*"([^"]+)"/);
61
+ if (nameMatch) {
62
+ contractNames.add(nameMatch[1]);
63
+ }
64
+ }
65
+ } catch {
66
+ // Member directory might not exist or have no Nargo.toml; skip.
67
+ }
68
+ }
69
+ } else {
70
+ // Single-crate project (no workspace): check if the root Nargo.toml itself is a contract.
71
+ if (/type\s*=\s*"contract"/.test(rootToml)) {
72
+ const nameMatch = rootToml.match(/name\s*=\s*"([^"]+)"/);
73
+ if (nameMatch) {
74
+ contractNames.add(nameMatch[1]);
75
+ }
76
+ }
77
+ }
78
+
79
+ return contractNames;
80
+ }
81
+
82
+ /** Checks that no tests exist in contract crates and fails with a helpful message if they do. */
83
+ async function checkNoTestsInContracts(nargo: string, log: LogFn): Promise<void> {
84
+ const contractPackages = await getContractPackageNames();
85
+ if (contractPackages.size === 0) {
86
+ return;
87
+ }
88
+
89
+ let output: string;
90
+ try {
91
+ // We list tests for all the crates in the workspace
92
+ output = execFileSync(nargo, ['test', '--list-tests', '--silence-warnings'], {
93
+ encoding: 'utf-8',
94
+ stdio: ['pipe', 'pipe', 'inherit'],
95
+ });
96
+ } catch {
97
+ // If listing tests fails (e.g. test crate has compile errors), skip the check.
98
+ return;
99
+ }
100
+
101
+ // The output of the `nargo test --list-tests` command is as follows:
102
+ // ```
103
+ // crate_name_1 test_name_1
104
+ // crate_name_2 test_name_2
105
+ // ...
106
+ // crate_name_n test_name_n
107
+ // ```
108
+ //
109
+ // We parse the individual lines and then we check if any contract crate appeared in the parsed output.
110
+ const lines = output
111
+ .trim()
112
+ .split('\n')
113
+ .filter(line => line.length > 0);
114
+ const testsInContracts: { packageName: string; testName: string }[] = [];
115
+
116
+ for (const line of lines) {
117
+ const spaceIndex = line.indexOf(' ');
118
+ if (spaceIndex === -1) {
119
+ continue;
120
+ }
121
+ const packageName = line.substring(0, spaceIndex);
122
+ const testName = line.substring(spaceIndex + 1);
123
+ if (contractPackages.has(packageName)) {
124
+ testsInContracts.push({ packageName, testName });
125
+ }
126
+ }
127
+
128
+ if (testsInContracts.length > 0) {
129
+ const details = testsInContracts.map(t => ` ${t.packageName}::${t.testName}`).join('\n');
130
+ log(
131
+ `WARNING: Found tests in contract crate(s):\n${details}\n\n` +
132
+ `Tests should be in a dedicated test crate, not in the contract crate.\n` +
133
+ `Learn more: https://docs.aztec.network/errors/1`,
134
+ );
135
+ }
136
+ }
137
+
64
138
  /** Compiles Aztec Noir contracts and postprocesses artifacts. */
65
139
  async function compileAztecContract(nargoArgs: string[], log: LogFn): Promise<void> {
66
140
  const nargo = process.env.NARGO ?? 'nargo';
@@ -68,6 +142,9 @@ async function compileAztecContract(nargoArgs: string[], log: LogFn): Promise<vo
68
142
 
69
143
  await run(nargo, ['compile', ...nargoArgs]);
70
144
 
145
+ // Ensure contract crates contain no tests (tests belong in the test crate).
146
+ await checkNoTestsInContracts(nargo, log);
147
+
71
148
  const artifacts = await collectContractArtifacts();
72
149
 
73
150
  if (artifacts.length > 0) {
@@ -0,0 +1,25 @@
1
+ import type { LogFn } from '@aztec/foundation/log';
2
+
3
+ import type { Command } from 'commander';
4
+
5
+ import { profileFlamegraph } from './profile_flamegraph.js';
6
+ import { profileGates } from './profile_gates.js';
7
+
8
+ export function injectProfileCommand(program: Command, log: LogFn): Command {
9
+ const profile = program.command('profile').description('Profile compiled Aztec artifacts.');
10
+
11
+ profile
12
+ .command('gates')
13
+ .argument('[target-dir]', 'Path to the compiled artifacts directory', './target')
14
+ .description('Display gate counts for all compiled Aztec artifacts in a target directory.')
15
+ .action((targetDir: string) => profileGates(targetDir, log));
16
+
17
+ profile
18
+ .command('flamegraph')
19
+ .argument('<contract-artifact>', 'Path to the compiled contract artifact JSON')
20
+ .argument('<function>', 'Name of the contract function to profile')
21
+ .description('Generate a gate count flamegraph SVG for a contract function.')
22
+ .action((artifactPath: string, functionName: string) => profileFlamegraph(artifactPath, functionName, log));
23
+
24
+ return program;
25
+ }
@@ -0,0 +1,63 @@
1
+ import type { LogFn } from '@aztec/foundation/log';
2
+
3
+ import { readFile, rename, rm, writeFile } from 'fs/promises';
4
+ import { basename, dirname, join } from 'path';
5
+
6
+ import { makeFunctionArtifact } from './profile_utils.js';
7
+ import type { CompiledArtifact } from './utils/artifacts.js';
8
+ import { run } from './utils/spawn.js';
9
+
10
+ /** Generates a gate count flamegraph SVG for a single contract function. */
11
+ export async function profileFlamegraph(artifactPath: string, functionName: string, log: LogFn): Promise<void> {
12
+ const raw = await readFile(artifactPath, 'utf-8');
13
+ const artifact: CompiledArtifact = JSON.parse(raw);
14
+
15
+ if (!Array.isArray(artifact.functions)) {
16
+ throw new Error(`${artifactPath} does not appear to be a contract artifact (no functions array)`);
17
+ }
18
+
19
+ const func = artifact.functions.find(f => f.name === functionName);
20
+ if (!func) {
21
+ const available = artifact.functions.map(f => f.name).join(', ');
22
+ throw new Error(`Function "${functionName}" not found in artifact. Available: ${available}`);
23
+ }
24
+ if (func.is_unconstrained) {
25
+ throw new Error(`Function "${functionName}" is unconstrained and cannot be profiled`);
26
+ }
27
+
28
+ const outputDir = dirname(artifactPath);
29
+ const contractName = basename(artifactPath, '.json');
30
+ const functionArtifact = join(outputDir, `${contractName}-${functionName}.json`);
31
+
32
+ try {
33
+ await writeFile(functionArtifact, makeFunctionArtifact(artifact, func));
34
+
35
+ const profiler = process.env.PROFILER_PATH ?? 'noir-profiler';
36
+ const bb = process.env.BB ?? 'bb';
37
+
38
+ await run(profiler, [
39
+ 'gates',
40
+ '--artifact-path',
41
+ functionArtifact,
42
+ '--backend-path',
43
+ bb,
44
+ '--backend-gates-command',
45
+ 'gates',
46
+ '--output',
47
+ outputDir,
48
+ '--scheme',
49
+ 'chonk',
50
+ '--include_gates_per_opcode',
51
+ ]);
52
+
53
+ // noir-profiler names the SVG using the internal function name which
54
+ // retains the __aztec_nr_internals__ prefix in the bytecode metadata.
55
+ const srcSvg = join(outputDir, `__aztec_nr_internals__${functionName}_gates.svg`);
56
+ const destSvg = join(outputDir, `${contractName}-${functionName}-flamegraph.svg`);
57
+ await rename(srcSvg, destSvg);
58
+
59
+ log(`Flamegraph written to ${destSvg}`);
60
+ } finally {
61
+ await rm(functionArtifact, { force: true });
62
+ }
63
+ }
@@ -0,0 +1,67 @@
1
+ import { asyncPool } from '@aztec/foundation/async-pool';
2
+ import type { LogFn } from '@aztec/foundation/log';
3
+
4
+ import { execFile as execFileCb } from 'child_process';
5
+ import { rm } from 'fs/promises';
6
+ import { promisify } from 'util';
7
+
8
+ import { MAX_CONCURRENT, discoverArtifacts } from './profile_utils.js';
9
+
10
+ const execFile = promisify(execFileCb);
11
+
12
+ interface GateCountResult {
13
+ name: string;
14
+ gateCount: number;
15
+ }
16
+
17
+ /** Parses circuit_size from bb gates JSON output: { "functions": [{ "circuit_size": N }] } */
18
+ function parseGateCount(stdout: string): number {
19
+ const parsed = JSON.parse(stdout);
20
+ const size = parsed?.functions?.[0]?.circuit_size;
21
+ if (typeof size !== 'number') {
22
+ throw new Error('Failed to parse circuit_size from bb gates output');
23
+ }
24
+ return size;
25
+ }
26
+
27
+ /** Runs bb gates on a single artifact file and returns the gate count. */
28
+ async function getGateCount(bb: string, artifactPath: string): Promise<number> {
29
+ const { stdout } = await execFile(bb, ['gates', '--scheme', 'chonk', '-b', artifactPath]);
30
+ return parseGateCount(stdout);
31
+ }
32
+
33
+ /** Profiles all compiled artifacts in a target directory and prints gate counts. */
34
+ export async function profileGates(targetDir: string, log: LogFn): Promise<void> {
35
+ const bb = process.env.BB ?? 'bb';
36
+ const { artifacts, tmpDir } = await discoverArtifacts(targetDir);
37
+
38
+ if (artifacts.length === 0) {
39
+ log('No artifacts found in target directory.');
40
+ return;
41
+ }
42
+
43
+ try {
44
+ const results: GateCountResult[] = await asyncPool(MAX_CONCURRENT, artifacts, async artifact => ({
45
+ name: artifact.name,
46
+ gateCount: await getGateCount(bb, artifact.filePath),
47
+ }));
48
+ results.sort((a, b) => a.name.localeCompare(b.name));
49
+
50
+ if (results.length === 0) {
51
+ log('No constrained circuits found.');
52
+ return;
53
+ }
54
+
55
+ const maxNameLen = Math.max(...results.map(r => r.name.length));
56
+ log('');
57
+ log('Gate counts:');
58
+ log('-'.repeat(maxNameLen + 16));
59
+ for (const { name, gateCount } of results) {
60
+ log(`${name.padEnd(maxNameLen)} ${gateCount.toLocaleString().padStart(12)}`);
61
+ }
62
+ log('-'.repeat(maxNameLen + 16));
63
+ log(`Total: ${results.length} circuit(s)`);
64
+ } finally {
65
+ await rm(tmpDir, { recursive: true, force: true });
66
+ }
67
+ }
@@ -0,0 +1,58 @@
1
+ import { mkdtemp, writeFile } from 'fs/promises';
2
+ import { tmpdir } from 'os';
3
+ import { join } from 'path';
4
+
5
+ import type { CompiledArtifact, ContractFunction } from './utils/artifacts.js';
6
+ import { readArtifactFiles } from './utils/artifacts.js';
7
+
8
+ export const MAX_CONCURRENT = 4;
9
+
10
+ export interface DiscoveredArtifact {
11
+ name: string;
12
+ filePath: string;
13
+ type: 'contract-function' | 'program';
14
+ }
15
+
16
+ /**
17
+ * Reads a target directory and returns a list of discovered artifacts with temp files
18
+ * created for contract functions. Caller must clean up tmpDir when done.
19
+ */
20
+ export async function discoverArtifacts(
21
+ targetDir: string,
22
+ ): Promise<{ artifacts: DiscoveredArtifact[]; tmpDir: string }> {
23
+ const files = await readArtifactFiles(targetDir);
24
+ const tmpDir = await mkdtemp(join(tmpdir(), 'aztec-profile-'));
25
+ const artifacts: DiscoveredArtifact[] = [];
26
+
27
+ for (const file of files) {
28
+ if (Array.isArray(file.content.functions)) {
29
+ for (const func of file.content.functions) {
30
+ if (!func.bytecode || func.is_unconstrained) {
31
+ continue;
32
+ }
33
+ const name = `${file.name}::${func.name}`;
34
+ const tmpPath = join(tmpDir, `${file.name}-${func.name}.json`);
35
+ await writeFile(tmpPath, makeFunctionArtifact(file.content, func));
36
+ artifacts.push({ name, filePath: tmpPath, type: 'contract-function' });
37
+ }
38
+ } else if (file.content.bytecode) {
39
+ artifacts.push({ name: file.name, filePath: file.filePath, type: 'program' });
40
+ }
41
+ }
42
+
43
+ return { artifacts, tmpDir };
44
+ }
45
+
46
+ /** Extracts a contract function as a standalone program artifact JSON string. */
47
+ export function makeFunctionArtifact(artifact: CompiledArtifact, func: ContractFunction) {
48
+ /* eslint-disable camelcase */
49
+ return JSON.stringify({
50
+ noir_version: artifact.noir_version,
51
+ hash: 0,
52
+ abi: func.abi,
53
+ bytecode: func.bytecode,
54
+ debug_symbols: func.debug_symbols,
55
+ file_map: artifact.file_map,
56
+ });
57
+ /* eslint-enable camelcase */
58
+ }
@@ -0,0 +1,111 @@
1
+ import { getInitialTestAccountsData } from '@aztec/accounts/testing';
2
+ import type { Fr } from '@aztec/aztec.js/fields';
3
+ import { getSponsoredFPCAddress } from '@aztec/cli/cli-utils';
4
+ import { getPublicClient } from '@aztec/ethereum/client';
5
+ import type { GenesisStateConfig } from '@aztec/ethereum/config';
6
+ import { RegistryContract, RollupContract } from '@aztec/ethereum/contracts';
7
+ import type { EthAddress } from '@aztec/foundation/eth-address';
8
+ import { startHttpRpcServer } from '@aztec/foundation/json-rpc/server';
9
+ import type { LogFn } from '@aztec/foundation/log';
10
+ import { retryUntil } from '@aztec/foundation/retry';
11
+ import { AztecAddress } from '@aztec/stdlib/aztec-address';
12
+ import { getGenesisValues } from '@aztec/world-state/testing';
13
+
14
+ import Koa from 'koa';
15
+
16
+ const ROLLUP_POLL_INTERVAL_S = 60;
17
+
18
+ /**
19
+ * Computes the expected genesis archive root from the genesis state config.
20
+ * Reads test accounts and sponsored FPC addresses as specified, then computes
21
+ * the genesis values including the archive root and prefilled public data.
22
+ */
23
+ export async function computeExpectedGenesisRoot(config: GenesisStateConfig, userLog: LogFn) {
24
+ const testAccounts = config.testAccounts ? (await getInitialTestAccountsData()).map(a => a.address) : [];
25
+ const sponsoredFPCAccounts = config.sponsoredFPC ? [await getSponsoredFPCAddress()] : [];
26
+ const prefundAddresses = (config.prefundAddresses ?? []).map(a => AztecAddress.fromString(a));
27
+ const initialFundedAccounts = testAccounts.concat(sponsoredFPCAccounts).concat(prefundAddresses);
28
+
29
+ userLog(`Initial funded accounts: ${initialFundedAccounts.map(a => a.toString()).join(', ')}`);
30
+
31
+ const { genesisArchiveRoot, prefilledPublicData } = await getGenesisValues(initialFundedAccounts);
32
+
33
+ userLog(`Genesis archive root: ${genesisArchiveRoot.toString()}`);
34
+
35
+ return { genesisArchiveRoot, prefilledPublicData };
36
+ }
37
+
38
+ /**
39
+ * Waits until the canonical rollup's genesis archive root matches the expected local genesis root.
40
+ * If the rollup is not yet compatible (e.g. during L1 contract upgrades), enters standby mode:
41
+ * starts a lightweight HTTP server for K8s liveness probes and polls every 60s until a compatible rollup appears.
42
+ */
43
+ export async function waitForCompatibleRollup(
44
+ config: {
45
+ l1RpcUrls: string[];
46
+ l1ChainId: number;
47
+ l1Contracts: { registryAddress: EthAddress };
48
+ rollupVersion?: number;
49
+ },
50
+ expectedGenesisRoot: Fr,
51
+ port: number | undefined,
52
+ userLog: LogFn,
53
+ ): Promise<void> {
54
+ const publicClient = getPublicClient(config);
55
+ const rollupVersion: number | 'canonical' = config.rollupVersion ?? 'canonical';
56
+
57
+ const registry = new RegistryContract(publicClient, config.l1Contracts.registryAddress);
58
+ const rollupAddress = await registry.getRollupAddress(rollupVersion);
59
+ const rollup = new RollupContract(publicClient, rollupAddress.toString());
60
+
61
+ let l1GenesisRoot: Fr;
62
+ try {
63
+ l1GenesisRoot = await rollup.getGenesisArchiveTreeRoot();
64
+ } catch (err: any) {
65
+ throw new Error(
66
+ `Could not retrieve genesis archive root from canonical rollup at ${rollupAddress}: ${err.message}`,
67
+ );
68
+ }
69
+
70
+ if (l1GenesisRoot.equals(expectedGenesisRoot)) {
71
+ return;
72
+ }
73
+
74
+ userLog(
75
+ `Genesis root mismatch: expected ${expectedGenesisRoot}, got ${l1GenesisRoot} from rollup at ${rollupAddress}. ` +
76
+ `Entering standby mode. Will poll every ${ROLLUP_POLL_INTERVAL_S}s for a compatible rollup...`,
77
+ );
78
+
79
+ const standbyServer = await startHttpRpcServer({ getApp: () => new Koa(), isHealthy: () => true }, { port });
80
+ userLog(`Standby status server listening on port ${standbyServer.port}`);
81
+
82
+ try {
83
+ await retryUntil(
84
+ async () => {
85
+ const currentRollupAddress = await registry.getRollupAddress(rollupVersion);
86
+ const currentRollup = new RollupContract(publicClient, currentRollupAddress.toString());
87
+
88
+ let currentGenesisRoot: Fr;
89
+ try {
90
+ currentGenesisRoot = await currentRollup.getGenesisArchiveTreeRoot();
91
+ } catch {
92
+ userLog(`Failed to fetch genesis root from rollup at ${currentRollupAddress}. Retrying...`);
93
+ return undefined;
94
+ }
95
+
96
+ if (currentGenesisRoot.equals(expectedGenesisRoot)) {
97
+ userLog(`Compatible rollup found at ${currentRollupAddress}. Exiting standby mode.`);
98
+ return true;
99
+ }
100
+
101
+ userLog(`Still waiting. Rollup at ${currentRollupAddress} has genesis root ${currentGenesisRoot}.`);
102
+ return undefined;
103
+ },
104
+ 'compatible rollup',
105
+ 0,
106
+ ROLLUP_POLL_INTERVAL_S,
107
+ );
108
+ } finally {
109
+ await new Promise<void>((resolve, reject) => standbyServer.close(err => (err ? reject(err) : resolve())));
110
+ }
111
+ }
@@ -3,8 +3,8 @@ import { createLogger } from '@aztec/aztec.js/log';
3
3
  import { type BlobClientConfig, blobClientConfigMapping, createBlobClient } from '@aztec/blob-client/client';
4
4
  import { getL1Config } from '@aztec/cli/config';
5
5
  import type { NamespacedApiHandlers } from '@aztec/foundation/json-rpc/server';
6
- import { type DataStoreConfig, dataConfigMappings } from '@aztec/kv-store/config';
7
6
  import { ArchiverApiSchema } from '@aztec/stdlib/interfaces/server';
7
+ import { type DataStoreConfig, dataConfigMappings } from '@aztec/stdlib/kv-store';
8
8
  import { getConfigEnvVars as getTelemetryClientConfig, initTelemetryClient } from '@aztec/telemetry-client';
9
9
 
10
10
  import { extractRelevantOptions } from '../util.js';
@@ -1,10 +1,9 @@
1
- import { getInitialTestAccountsData } from '@aztec/accounts/testing';
2
1
  import { type AztecNodeConfig, aztecNodeConfigMappings, getConfigEnvVars } from '@aztec/aztec-node';
3
2
  import { Fr } from '@aztec/aztec.js/fields';
4
- import { getSponsoredFPCAddress } from '@aztec/cli/cli-utils';
5
3
  import { getL1Config } from '@aztec/cli/config';
6
4
  import { getPublicClient } from '@aztec/ethereum/client';
7
- import { SecretValue } from '@aztec/foundation/config';
5
+ import { getGenesisStateConfigEnvVars } from '@aztec/ethereum/config';
6
+ import { type NetworkNames, SecretValue } from '@aztec/foundation/config';
8
7
  import type { NamespacedApiHandlers } from '@aztec/foundation/json-rpc/server';
9
8
  import { Agent, makeUndiciFetch } from '@aztec/foundation/json-rpc/undici';
10
9
  import type { LogFn } from '@aztec/foundation/log';
@@ -19,16 +18,16 @@ import {
19
18
  telemetryClientConfigMappings,
20
19
  } from '@aztec/telemetry-client';
21
20
  import { EmbeddedWallet } from '@aztec/wallets/embedded';
22
- import { getGenesisValues } from '@aztec/world-state/testing';
23
21
 
24
22
  import { createAztecNode } from '../../local-network/index.js';
25
23
  import {
26
24
  extractNamespacedOptions,
27
25
  extractRelevantOptions,
28
26
  preloadCrsDataForVerifying,
29
- setupUpdateMonitor,
27
+ setupVersionChecker,
30
28
  } from '../util.js';
31
29
  import { getVersions } from '../versioning.js';
30
+ import { computeExpectedGenesisRoot, waitForCompatibleRollup } from './standby.js';
32
31
  import { startProverBroker } from './start_prover_broker.js';
33
32
 
34
33
  export async function startNode(
@@ -37,6 +36,7 @@ export async function startNode(
37
36
  services: NamespacedApiHandlers,
38
37
  adminServices: NamespacedApiHandlers,
39
38
  userLog: LogFn,
39
+ networkName: NetworkNames,
40
40
  ): Promise<{ config: AztecNodeConfig }> {
41
41
  // All options set from environment variables
42
42
  const configFromEnvVars = getConfigEnvVars();
@@ -80,15 +80,8 @@ export async function startNode(
80
80
 
81
81
  await preloadCrsDataForVerifying(nodeConfig, userLog);
82
82
 
83
- const testAccounts = nodeConfig.testAccounts ? (await getInitialTestAccountsData()).map(a => a.address) : [];
84
- const sponsoredFPCAccounts = nodeConfig.sponsoredFPC ? [await getSponsoredFPCAddress()] : [];
85
- const initialFundedAccounts = testAccounts.concat(sponsoredFPCAccounts);
86
-
87
- userLog(`Initial funded accounts: ${initialFundedAccounts.map(a => a.toString()).join(', ')}`);
88
-
89
- const { genesisArchiveRoot, prefilledPublicData } = await getGenesisValues(initialFundedAccounts);
90
-
91
- userLog(`Genesis archive root: ${genesisArchiveRoot.toString()}`);
83
+ const genesisConfig = getGenesisStateConfigEnvVars();
84
+ const { genesisArchiveRoot, prefilledPublicData } = await computeExpectedGenesisRoot(genesisConfig, userLog);
92
85
 
93
86
  const followsCanonicalRollup =
94
87
  typeof nodeConfig.rollupVersion !== 'number' || (nodeConfig.rollupVersion as unknown as string) === 'canonical';
@@ -96,6 +89,11 @@ export async function startNode(
96
89
  if (!nodeConfig.l1Contracts.registryAddress || nodeConfig.l1Contracts.registryAddress.isZero()) {
97
90
  throw new Error('L1 registry address is required to start Aztec Node');
98
91
  }
92
+
93
+ // Wait for a compatible rollup before proceeding with full L1 config fetch.
94
+ // This prevents crashes when the canonical rollup hasn't been upgraded yet.
95
+ await waitForCompatibleRollup(nodeConfig, genesisArchiveRoot, options.port, userLog);
96
+
99
97
  const { addresses, config } = await getL1Config(
100
98
  nodeConfig.l1Contracts.registryAddress,
101
99
  nodeConfig.l1RpcUrls,
@@ -182,16 +180,19 @@ export async function startNode(
182
180
  await addBot(options, signalHandlers, services, wallet, node, telemetry, undefined);
183
181
  }
184
182
 
185
- if (nodeConfig.autoUpdate !== 'disabled' && nodeConfig.autoUpdateUrl) {
186
- await setupUpdateMonitor(
187
- nodeConfig.autoUpdate,
188
- new URL(nodeConfig.autoUpdateUrl),
189
- followsCanonicalRollup,
190
- getPublicClient(nodeConfig!),
191
- nodeConfig.l1Contracts.registryAddress,
192
- signalHandlers,
193
- async config => node.setConfig((await AztecNodeAdminApiSchema.setConfig.parameters().parseAsync([config]))[0]),
194
- );
183
+ if (nodeConfig.enableVersionCheck && networkName !== 'local') {
184
+ const cacheDir = process.env.DATA_DIRECTORY ? `${process.env.DATA_DIRECTORY}/cache` : undefined;
185
+ try {
186
+ await setupVersionChecker(
187
+ networkName,
188
+ followsCanonicalRollup,
189
+ getPublicClient(nodeConfig!),
190
+ signalHandlers,
191
+ cacheDir,
192
+ );
193
+ } catch {
194
+ /* no-op */
195
+ }
195
196
  }
196
197
 
197
198
  return { config: nodeConfig };
@@ -1,5 +1,5 @@
1
1
  import { getL1Config } from '@aztec/cli/config';
2
- import { getPublicClient } from '@aztec/ethereum/client';
2
+ import { getGenesisStateConfigEnvVars } from '@aztec/ethereum/config';
3
3
  import type { NamespacedApiHandlers } from '@aztec/foundation/json-rpc/server';
4
4
  import type { LogFn } from '@aztec/foundation/log';
5
5
  import {
@@ -13,7 +13,8 @@ import { getProverNodeBrokerConfigFromEnv } from '@aztec/prover-node';
13
13
  import type { ProvingJobBroker } from '@aztec/stdlib/interfaces/server';
14
14
  import { getConfigEnvVars as getTelemetryClientConfig, initTelemetryClient } from '@aztec/telemetry-client';
15
15
 
16
- import { extractRelevantOptions, setupUpdateMonitor } from '../util.js';
16
+ import { extractRelevantOptions } from '../util.js';
17
+ import { computeExpectedGenesisRoot, waitForCompatibleRollup } from './standby.js';
17
18
 
18
19
  export async function startProverBroker(
19
20
  options: any,
@@ -35,7 +36,10 @@ export async function startProverBroker(
35
36
  throw new Error('L1 registry address is required to start Aztec Node without --deploy-aztec-contracts option');
36
37
  }
37
38
 
38
- const followsCanonicalRollup = typeof config.rollupVersion !== 'number';
39
+ const genesisConfig = getGenesisStateConfigEnvVars();
40
+ const { genesisArchiveRoot } = await computeExpectedGenesisRoot(genesisConfig, userLog);
41
+ await waitForCompatibleRollup(config, genesisArchiveRoot, options.port, userLog);
42
+
39
43
  const { addresses, config: rollupConfig } = await getL1Config(
40
44
  config.l1Contracts.registryAddress,
41
45
  config.l1RpcUrls,
@@ -49,17 +53,6 @@ export async function startProverBroker(
49
53
  const client = await initTelemetryClient(getTelemetryClientConfig());
50
54
  const broker = await createAndStartProvingBroker(config, client);
51
55
 
52
- if (options.autoUpdate !== 'disabled' && options.autoUpdateUrl) {
53
- await setupUpdateMonitor(
54
- options.autoUpdate,
55
- new URL(options.autoUpdateUrl),
56
- followsCanonicalRollup,
57
- getPublicClient(config),
58
- config.l1Contracts.registryAddress,
59
- signalHandlers,
60
- );
61
- }
62
-
63
56
  services.proverBroker = [
64
57
  broker,
65
58
  config.proverBrokerDebugReplayEnabled ? ProvingJobBrokerSchemaWithDebug : ProvingJobBrokerSchema,