@aztec/aztec 0.0.1-commit.7ac86ea28 → 0.0.1-commit.7b86788

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aztec/aztec",
-  "version": "0.0.1-commit.7ac86ea28",
+  "version": "0.0.1-commit.7b86788",
   "type": "module",
   "exports": {
     ".": "./dest/index.js",
@@ -28,39 +28,39 @@
     "../package.common.json"
   ],
   "dependencies": {
-    "@aztec/accounts": "0.0.1-commit.7ac86ea28",
-    "@aztec/archiver": "0.0.1-commit.7ac86ea28",
-    "@aztec/aztec-faucet": "0.0.1-commit.7ac86ea28",
-    "@aztec/aztec-node": "0.0.1-commit.7ac86ea28",
-    "@aztec/aztec.js": "0.0.1-commit.7ac86ea28",
-    "@aztec/bb-prover": "0.0.1-commit.7ac86ea28",
-    "@aztec/bb.js": "0.0.1-commit.7ac86ea28",
-    "@aztec/blob-client": "0.0.1-commit.7ac86ea28",
-    "@aztec/bot": "0.0.1-commit.7ac86ea28",
-    "@aztec/builder": "0.0.1-commit.7ac86ea28",
-    "@aztec/cli": "0.0.1-commit.7ac86ea28",
-    "@aztec/constants": "0.0.1-commit.7ac86ea28",
-    "@aztec/entrypoints": "0.0.1-commit.7ac86ea28",
-    "@aztec/ethereum": "0.0.1-commit.7ac86ea28",
-    "@aztec/foundation": "0.0.1-commit.7ac86ea28",
-    "@aztec/kv-store": "0.0.1-commit.7ac86ea28",
-    "@aztec/l1-artifacts": "0.0.1-commit.7ac86ea28",
-    "@aztec/node-lib": "0.0.1-commit.7ac86ea28",
-    "@aztec/noir-contracts.js": "0.0.1-commit.7ac86ea28",
-    "@aztec/noir-protocol-circuits-types": "0.0.1-commit.7ac86ea28",
-    "@aztec/p2p": "0.0.1-commit.7ac86ea28",
-    "@aztec/p2p-bootstrap": "0.0.1-commit.7ac86ea28",
-    "@aztec/protocol-contracts": "0.0.1-commit.7ac86ea28",
-    "@aztec/prover-client": "0.0.1-commit.7ac86ea28",
-    "@aztec/prover-node": "0.0.1-commit.7ac86ea28",
-    "@aztec/pxe": "0.0.1-commit.7ac86ea28",
-    "@aztec/sequencer-client": "0.0.1-commit.7ac86ea28",
-    "@aztec/stdlib": "0.0.1-commit.7ac86ea28",
-    "@aztec/telemetry-client": "0.0.1-commit.7ac86ea28",
-    "@aztec/txe": "0.0.1-commit.7ac86ea28",
-    "@aztec/validator-ha-signer": "0.0.1-commit.7ac86ea28",
-    "@aztec/wallets": "0.0.1-commit.7ac86ea28",
-    "@aztec/world-state": "0.0.1-commit.7ac86ea28",
+    "@aztec/accounts": "0.0.1-commit.7b86788",
+    "@aztec/archiver": "0.0.1-commit.7b86788",
+    "@aztec/aztec-faucet": "0.0.1-commit.7b86788",
+    "@aztec/aztec-node": "0.0.1-commit.7b86788",
+    "@aztec/aztec.js": "0.0.1-commit.7b86788",
+    "@aztec/bb-prover": "0.0.1-commit.7b86788",
+    "@aztec/bb.js": "0.0.1-commit.7b86788",
+    "@aztec/blob-client": "0.0.1-commit.7b86788",
+    "@aztec/bot": "0.0.1-commit.7b86788",
+    "@aztec/builder": "0.0.1-commit.7b86788",
+    "@aztec/cli": "0.0.1-commit.7b86788",
+    "@aztec/constants": "0.0.1-commit.7b86788",
+    "@aztec/entrypoints": "0.0.1-commit.7b86788",
+    "@aztec/ethereum": "0.0.1-commit.7b86788",
+    "@aztec/foundation": "0.0.1-commit.7b86788",
+    "@aztec/kv-store": "0.0.1-commit.7b86788",
+    "@aztec/l1-artifacts": "0.0.1-commit.7b86788",
+    "@aztec/node-lib": "0.0.1-commit.7b86788",
+    "@aztec/noir-contracts.js": "0.0.1-commit.7b86788",
+    "@aztec/noir-protocol-circuits-types": "0.0.1-commit.7b86788",
+    "@aztec/p2p": "0.0.1-commit.7b86788",
+    "@aztec/p2p-bootstrap": "0.0.1-commit.7b86788",
+    "@aztec/protocol-contracts": "0.0.1-commit.7b86788",
+    "@aztec/prover-client": "0.0.1-commit.7b86788",
+    "@aztec/prover-node": "0.0.1-commit.7b86788",
+    "@aztec/pxe": "0.0.1-commit.7b86788",
+    "@aztec/sequencer-client": "0.0.1-commit.7b86788",
+    "@aztec/stdlib": "0.0.1-commit.7b86788",
+    "@aztec/telemetry-client": "0.0.1-commit.7b86788",
+    "@aztec/txe": "0.0.1-commit.7b86788",
+    "@aztec/validator-ha-signer": "0.0.1-commit.7b86788",
+    "@aztec/wallets": "0.0.1-commit.7b86788",
+    "@aztec/world-state": "0.0.1-commit.7b86788",
     "@types/chalk": "^2.2.0",
     "abitype": "^0.8.11",
     "chalk": "^5.3.0",

package/scripts/aztec.sh

@@ -46,16 +46,20 @@ case $cmd in
       export ETHEREUM_HOSTS=${ETHEREUM_HOSTS:-"http://127.0.0.1:${ANVIL_PORT}"}
 
       anvil --version
-      anvil --silent &
+      anvil --silent --port "$ANVIL_PORT" &
       anvil_pid=$!
       trap 'kill $anvil_pid &>/dev/null' EXIT
     fi
 
     aztec start "$@"
     ;;
-  new|init|flamegraph)
+  new|init)
     $script_dir/${cmd}.sh "$@"
     ;;
+  flamegraph)
+    echo "Warning: 'aztec flamegraph' is deprecated. Use 'aztec profile flamegraph' instead." >&2
+    aztec profile flamegraph "$@"
+    ;;
   *)
     aztec $cmd "$@"
     ;;

package/scripts/init.sh

@@ -1,9 +1,11 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
-NARGO=${NARGO:-nargo}
 script_path=$(realpath $(dirname "$0"))
 
+name_arg=""
+
+# Check for help first
 for arg in "$@"; do
   if [ "$arg" == "--help" ] || [ "$arg" == "-h" ]; then
     cat << 'EOF'
@@ -13,23 +15,31 @@ Usage: aztec init [OPTIONS]
 
 Options:
   --name <NAME>  Name of the package [default: current directory name]
-  --lib          Use a library template
   -h, --help     Print help
 
-This command creates a new Aztec Noir project in the current directory using nargo
-and automatically adds the Aztec.nr dependency to your Nargo.toml file.
-
+This command creates a new Aztec Noir project in the current directory with
+a workspace containing a contract crate and a test crate, and automatically
+adds the Aztec.nr dependency to both.
 EOF
     exit 0
   fi
-  if [ "$arg" == "--lib" ]; then
-    is_contract=0
-  fi
 done
 
-echo "Initializing Noir project..."
-$NARGO init "$@"
+# Parse arguments
+while [[ $# -gt 0 ]]; do
+  case $1 in
+    --name)
+      name_arg="$2"
+      shift 2
+      ;;
+    *)
+      shift
+      ;;
+  esac
+done
+
+# Derive package name: use --name if provided, otherwise use current directory name
+package_name="${name_arg:-$(basename $(pwd))}"
 
-if [ "${is_contract:-1}" -eq 1 ]; then
-  $script_path/setup_project.sh
-fi
+echo "Initializing Aztec contract project..."
+$script_path/setup_workspace.sh "$package_name"

package/scripts/new.sh

@@ -1,10 +1,10 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
-NARGO=${NARGO:-nargo}
 script_path=$(realpath $(dirname "$0"))
 
-type_arg="--contract"
+name_arg=""
+project_path=""
 
 while [[ $# -gt 0 ]]; do
   case $1 in
@@ -19,20 +19,16 @@ Arguments:
 
 Options:
   --name <NAME>  Name of the package [default: package directory name]
-  --lib          Create a library template instead of a contract
   -h, --help     Print help
 
-This command creates a new Aztec Noir project using nargo and automatically
-adds the Aztec.nr dependency to your Nargo.toml file.
+This command creates a new Aztec Noir project with a workspace containing
+a contract crate and a test crate, and automatically adds the Aztec.nr
+dependency to both.
 EOF
       exit 0
       ;;
-    --lib)
-      type_arg="--lib"
-      shift
-      ;;
     --name)
-      name_arg="--name $2"
+      name_arg="$2"
       shift 2
       ;;
     *)
@@ -50,10 +46,15 @@ if [ -z "$project_path" ]; then
   exit 1
 fi
 
-echo "Creating new Noir project at $project_path..."
-$NARGO new $type_arg ${name_arg:-} $project_path
-
-if [ "$type_arg" == "--contract" ]; then
-  cd $project_path
-  $script_path/setup_project.sh
+if [ -d "$project_path" ] && [ "$(ls -A $project_path 2>/dev/null)" ]; then
+  echo "Error: $project_path already exists and is not empty"
+  exit 1
 fi
+
+# Derive package name: use --name if provided, otherwise use directory basename
+package_name="${name_arg:-$(basename $project_path)}"
+
+echo "Creating new Aztec contract project at $project_path..."
+mkdir -p "$project_path"
+cd "$project_path"
+$script_path/setup_workspace.sh "$package_name"

package/scripts/setup_workspace.sh

@@ -0,0 +1,124 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Creates an Aztec contract workspace with a contract crate and a test crate.
+# Usage: setup_workspace.sh <package_name>
+# Must be called from the workspace root directory.
+
+package_name=$1
+
+if [ -z "$package_name" ]; then
+  echo "Error: package name is required"
+  exit 1
+fi
+
+if [ -f "Nargo.toml" ]; then
+  echo "Error: Nargo.toml already exists in the current directory"
+  exit 1
+fi
+
+# Get the actual aztec version for the git tag.
+AZTEC_VERSION=$(jq -r '.version' $(dirname $0)/../package.json)
+
+# Create workspace root Nargo.toml
+cat > Nargo.toml << 'EOF'
+[workspace]
+members = ["contract", "test"]
+EOF
+
+# Create contract crate
+mkdir -p contract/src
+cat > contract/Nargo.toml << CEOF
+[package]
+name = "${package_name}"
+type = "contract"
+
+[dependencies]
+aztec = { git="https://github.com/AztecProtocol/aztec-nr", tag="v${AZTEC_VERSION}", directory="aztec" }
+CEOF
+
+cat > contract/src/main.nr << 'EOF'
+use aztec::macros::aztec;
+
+#[aztec]
+pub contract Main {
+    use aztec::macros::functions::{external, initializer};
+
+    #[initializer]
+    #[external("private")]
+    fn constructor() {}
+}
+EOF
+
+# Create test crate
+mkdir -p test/src
+cat > test/Nargo.toml << TEOF
+[package]
+name = "${package_name}_test"
+type = "lib"
+
+[dependencies]
+aztec = { git="https://github.com/AztecProtocol/aztec-nr", tag="v${AZTEC_VERSION}", directory="aztec" }
+${package_name} = { path = "../contract" }
+TEOF
+
+cat > test/src/lib.nr << 'NOIR'
+use aztec::test::helpers::test_environment::TestEnvironment;
+use __PACKAGE_NAME__::Main;
+
+#[test]
+unconstrained fn test_constructor() {
+    let mut env = TestEnvironment::new();
+    let deployer = env.create_light_account();
+
+    // Deploy the contract with the default constructor:
+    let contract_address = env.deploy("@__PACKAGE_NAME__/Main").with_private_initializer(
+        deployer,
+        Main::interface().constructor(),
+    );
+
+    // Deploy without an initializer:
+    let contract_address = env.deploy("@__PACKAGE_NAME__/Main").without_initializer();
+}
+NOIR
+
+sed -i "s/__PACKAGE_NAME__/${package_name}/g" test/src/lib.nr
+
+# Create README
+cat > README.md << REOF
+# ${package_name}
+
+An Aztec Noir contract project.
+
+## Compile
+
+\`\`\`bash
+aztec compile
+\`\`\`
+
+This compiles the contract in \`contract/\` and outputs artifacts to \`target/\`.
+
+## Test
+
+\`\`\`bash
+aztec test
+\`\`\`
+
+This runs the tests in \`test/\`.
+
+## Generate TypeScript bindings
+
+\`\`\`bash
+aztec codegen target -o src/artifacts
+\`\`\`
+
+This generates TypeScript contract artifacts from the compiled output in \`target/\` into \`src/artifacts/\`.
+REOF
+
+# Create .gitignore
+cat > .gitignore << 'GEOF'
+target/
+codegenCache.json
+GEOF
+
+echo "Created Aztec contract workspace with crates '${package_name}' and '${package_name}_test'"

package/src/bin/index.ts

@@ -16,6 +16,7 @@ import { Command } from 'commander';
 
 import { injectCompileCommand } from '../cli/cmds/compile.js';
 import { injectMigrateCommand } from '../cli/cmds/migrate_ha_db.js';
+import { injectProfileCommand } from '../cli/cmds/profile.js';
 import { injectAztecCommands } from '../cli/index.js';
 import { getCliVersion } from '../cli/release_version.js';
 
@@ -58,6 +59,7 @@ async function main() {
   program = injectMiscCommands(program, userLog);
   program = injectValidatorKeysCommands(program, userLog);
   program = injectCompileCommand(program, userLog);
+  program = injectProfileCommand(program, userLog);
   program = injectMigrateCommand(program, userLog);
 
   await program.parseAsync(process.argv);

package/src/cli/admin_api_key_store.ts

@@ -28,7 +28,7 @@ export interface ResolveAdminApiKeyOptions {
   /** SHA-256 hex hash of a pre-generated API key. When set, the node uses this hash directly. */
   adminApiKeyHash?: string;
   /** If true, disable admin API key auth entirely. */
-  noAdminApiKey?: boolean;
+  disableAdminApiKey?: boolean;
   /** If true, force-generate a new key even if one is already persisted. */
   resetAdminApiKey?: boolean;
   /** Root data directory for persistent storage. */
@@ -39,7 +39,7 @@ export interface ResolveAdminApiKeyOptions {
  * Resolves the admin API key for the admin RPC endpoint.
  *
  * Strategy:
- * 1. If opt-out flag is set (`noAdminApiKey`), return undefined (no auth).
+ * 1. If opt-out flag is set (`disableAdminApiKey`), return undefined (no auth).
  * 2. If a pre-generated hash is provided (`adminApiKeyHash`), use it directly.
  * 3. If a data directory exists, look for a persisted hash file
  *    at `<dataDirectory>/admin/api_key_hash`:
@@ -58,8 +58,8 @@ export async function resolveAdminApiKey(
   log: Logger,
 ): Promise<AdminApiKeyResolution | undefined> {
   // Operator explicitly opted out of admin auth
-  if (options.noAdminApiKey) {
-    log.warn('Admin API key authentication is DISABLED (--no-admin-api-key / AZTEC_NO_ADMIN_API_KEY)');
+  if (options.disableAdminApiKey) {
+    log.warn('Admin API key authentication is DISABLED (--disable-admin-api-key / AZTEC_DISABLE_ADMIN_API_KEY)');
     return undefined;
   }
 

package/src/cli/aztec_start_action.ts

@@ -109,7 +109,7 @@ export async function aztecStart(options: any, userLog: LogFn, debugLogger: Logg
     const apiKeyResolution = await resolveAdminApiKey(
       {
         adminApiKeyHash: options.adminApiKeyHash,
-        noAdminApiKey: options.noAdminApiKey,
+        disableAdminApiKey: options.disableAdminApiKey,
         resetAdminApiKey: options.resetAdminApiKey,
         dataDirectory: options.dataDirectory,
       },
@@ -148,7 +148,7 @@ export async function aztecStart(options: any, userLog: LogFn, debugLogger: Logg
         userLog('  The key hash has been persisted — on next restart, the same key will be used.');
       }
       userLog('');
-      userLog('  To disable admin auth: --no-admin-api-key or AZTEC_NO_ADMIN_API_KEY=true');
+      userLog('  To disable admin auth: --disable-admin-api-key or AZTEC_DISABLE_ADMIN_API_KEY=true');
       userLog(separator);
       userLog('');
     }

package/src/cli/aztec_start_options.ts

@@ -150,12 +150,13 @@ export const aztecStartOptions: { [key: string]: AztecStartOption[] } = {
       env: 'AZTEC_ADMIN_API_KEY_HASH',
     },
     {
-      flag: '--no-admin-api-key',
+      flag: '--disable-admin-api-key',
       description:
         'Disable API key authentication on the admin RPC endpoint. By default, a key is auto-generated, displayed once, and its hash is persisted.',
       defaultValue: false,
-      env: 'AZTEC_NO_ADMIN_API_KEY',
-      parseVal: val => val === 'true' || val === '1',
+      env: 'AZTEC_DISABLE_ADMIN_API_KEY',
+      // undefined means the flag was passed without a value (boolean toggle), treat as true.
+      parseVal: val => val === undefined || val === 'true' || val === '1',
     },
     {
       flag: '--reset-admin-api-key',

package/src/cli/cmds/compile.ts

@@ -1,51 +1,25 @@
 import type { LogFn } from '@aztec/foundation/log';
 
-import { execFileSync, spawn } from 'child_process';
+import { execFileSync } from 'child_process';
 import type { Command } from 'commander';
-import { readFile, readdir, writeFile } from 'fs/promises';
+import { readFile, writeFile } from 'fs/promises';
 import { join } from 'path';
 
-/** Spawns a command with inherited stdio and rejects on non-zero exit. */
-function run(cmd: string, args: string[]): Promise<void> {
-  return new Promise((resolve, reject) => {
-    const child = spawn(cmd, args, { stdio: 'inherit' });
-    child.on('error', reject);
-    child.on('close', code => {
-      if (code !== 0) {
-        reject(new Error(`${cmd} exited with code ${code}`));
-      } else {
-        resolve();
-      }
-    });
-  });
-}
+import { readArtifactFiles } from './utils/artifacts.js';
+import { run } from './utils/spawn.js';
 
-/** Returns paths to contract artifacts in the target directory.
- *  Contract artifacts are identified by having a `functions` array in the JSON.
- */
+/** Returns paths to contract artifacts in the target directory. */
 async function collectContractArtifacts(): Promise<string[]> {
-  let files: string[];
+  let files;
   try {
-    files = await readdir('target');
+    files = await readArtifactFiles('target');
   } catch (err: any) {
-    if (err?.code === 'ENOENT') {
+    if (err?.message?.includes('does not exist')) {
       return [];
     }
-    throw new Error(`Failed to read target directory: ${err.message}`);
-  }
-
-  const artifacts: string[] = [];
-  for (const file of files) {
-    if (!file.endsWith('.json')) {
-      continue;
-    }
-    const filePath = join('target', file);
-    const content = JSON.parse(await readFile(filePath, 'utf-8'));
-    if (Array.isArray(content.functions)) {
-      artifacts.push(filePath);
-    }
+    throw err;
   }
-  return artifacts;
+  return files.filter(f => Array.isArray(f.content.functions)).map(f => f.filePath);
 }
 
 /** Strips the `__aztec_nr_internals__` prefix from function names in contract artifacts. */
@@ -61,6 +35,106 @@ async function stripInternalPrefixes(artifactPaths: string[]): Promise<void> {
   }
 }
 
+/** Returns the set of package names that are contract crates in the current workspace. */
+async function getContractPackageNames(): Promise<Set<string>> {
+  const contractNames = new Set<string>();
+
+  let rootToml: string;
+  try {
+    rootToml = await readFile('Nargo.toml', 'utf-8');
+  } catch {
+    return contractNames;
+  }
+
+  const membersMatch = rootToml.match(/members\s*=\s*\[([^\]]*)\]/);
+  if (membersMatch) {
+    const members = membersMatch[1]
+      .split(',')
+      .map(m => m.trim().replace(/^"|"$/g, ''))
+      .filter(m => m.length > 0);
+
+    for (const member of members) {
+      try {
+        const memberToml = await readFile(join(member, 'Nargo.toml'), 'utf-8');
+        if (/type\s*=\s*"contract"/.test(memberToml)) {
+          const nameMatch = memberToml.match(/name\s*=\s*"([^"]+)"/);
+          if (nameMatch) {
+            contractNames.add(nameMatch[1]);
+          }
+        }
+      } catch {
+        // Member directory might not exist or have no Nargo.toml; skip.
+      }
+    }
+  } else {
+    // Single-crate project (no workspace): check if the root Nargo.toml itself is a contract.
+    if (/type\s*=\s*"contract"/.test(rootToml)) {
+      const nameMatch = rootToml.match(/name\s*=\s*"([^"]+)"/);
+      if (nameMatch) {
+        contractNames.add(nameMatch[1]);
+      }
+    }
+  }
+
+  return contractNames;
+}
+
+/** Checks that no tests exist in contract crates and fails with a helpful message if they do. */
+async function checkNoTestsInContracts(nargo: string, log: LogFn): Promise<void> {
+  const contractPackages = await getContractPackageNames();
+  if (contractPackages.size === 0) {
+    return;
+  }
+
+  let output: string;
+  try {
+    // We list tests for all the crates in the workspace
+    output = execFileSync(nargo, ['test', '--list-tests', '--silence-warnings'], {
+      encoding: 'utf-8',
+      stdio: ['pipe', 'pipe', 'inherit'],
+    });
+  } catch {
+    // If listing tests fails (e.g. test crate has compile errors), skip the check.
+    return;
+  }
+
+  // The output of the `nargo test --list-tests` command is as follows:
+  // ```
+  // crate_name_1 test_name_1
+  // crate_name_2 test_name_2
+  // ...
+  // crate_name_n test_name_n
+  // ```
+  //
+  // We parse the individual lines and then we check if any contract crate appeared in the parsed output.
+  const lines = output
+    .trim()
+    .split('\n')
+    .filter(line => line.length > 0);
+  const testsInContracts: { packageName: string; testName: string }[] = [];
+
+  for (const line of lines) {
+    const spaceIndex = line.indexOf(' ');
+    if (spaceIndex === -1) {
+      continue;
+    }
+    const packageName = line.substring(0, spaceIndex);
+    const testName = line.substring(spaceIndex + 1);
+    if (contractPackages.has(packageName)) {
+      testsInContracts.push({ packageName, testName });
+    }
+  }
+
+  if (testsInContracts.length > 0) {
+    const details = testsInContracts.map(t => `  ${t.packageName}::${t.testName}`).join('\n');
+    log(
+      `WARNING: Found tests in contract crate(s):\n${details}\n\n` +
+        `Tests should be in a dedicated test crate, not in the contract crate.\n` +
+        `Learn more: https://docs.aztec.network/errors/1`,
+    );
+  }
+}
+
 /** Compiles Aztec Noir contracts and postprocesses artifacts. */
 async function compileAztecContract(nargoArgs: string[], log: LogFn): Promise<void> {
   const nargo = process.env.NARGO ?? 'nargo';
@@ -68,6 +142,9 @@ async function compileAztecContract(nargoArgs: string[], log: LogFn): Promise<vo
 
   await run(nargo, ['compile', ...nargoArgs]);
 
+  // Ensure contract crates contain no tests (tests belong in the test crate).
+  await checkNoTestsInContracts(nargo, log);
+
   const artifacts = await collectContractArtifacts();
 
   if (artifacts.length > 0) {

package/src/cli/cmds/profile.ts

@@ -0,0 +1,25 @@
+import type { LogFn } from '@aztec/foundation/log';
+
+import type { Command } from 'commander';
+
+import { profileFlamegraph } from './profile_flamegraph.js';
+import { profileGates } from './profile_gates.js';
+
+export function injectProfileCommand(program: Command, log: LogFn): Command {
+  const profile = program.command('profile').description('Profile compiled Aztec artifacts.');
+
+  profile
+    .command('gates')
+    .argument('[target-dir]', 'Path to the compiled artifacts directory', './target')
+    .description('Display gate counts for all compiled Aztec artifacts in a target directory.')
+    .action((targetDir: string) => profileGates(targetDir, log));
+
+  profile
+    .command('flamegraph')
+    .argument('<contract-artifact>', 'Path to the compiled contract artifact JSON')
+    .argument('<function>', 'Name of the contract function to profile')
+    .description('Generate a gate count flamegraph SVG for a contract function.')
+    .action((artifactPath: string, functionName: string) => profileFlamegraph(artifactPath, functionName, log));
+
+  return program;
+}