@aztec/aztec 0.0.0-test.0 → 0.0.1-commit.03f7ef2

package/dest/testing/cheat_codes.js

@@ -0,0 +1,62 @@
+import { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
+import { retryUntil } from '@aztec/foundation/retry';
+/**
+ * A class that provides utility functions for interacting with the chain.
+ * @deprecated There used to be 3 kinds of cheat codes: eth, rollup and aztec. We have nuked the Aztec ones because
+ * they became unused (we now have better testing tools). If you are introducing a new functionality to the cheat
+ * codes, please consider whether it makes sense to just introduce new utils in your tests instead.
+ */ export class CheatCodes {
+    eth;
+    rollup;
+    constructor(/** Cheat codes for L1.*/ eth, /** Cheat codes for the Aztec Rollup contract on L1. */ rollup){
+        this.eth = eth;
+        this.rollup = rollup;
+    }
+    static async create(rpcUrls, node, dateProvider) {
+        const ethCheatCodes = new EthCheatCodes(rpcUrls, dateProvider);
+        const rollupCheatCodes = new RollupCheatCodes(ethCheatCodes, await node.getNodeInfo().then((n)=>n.l1ContractAddresses));
+        return new CheatCodes(ethCheatCodes, rollupCheatCodes);
+    }
+    /**
+   * Warps the L1 timestamp to a target timestamp and mines an L2 block that advances the L2 timestamp to at least
+   * the target timestamp. L2 timestamp is not advanced exactly to the target timestamp because it is determined
+   * by the slot number, which advances in fixed intervals.
+   * This is useful for testing time-dependent contract behavior.
+   * @param sequencerClient - The sequencer client to use to force an empty block to be mined.
+   * @param node - The Aztec node used to query if a new block has been mined.
+   * @param targetTimestamp - The target timestamp to warp to (in seconds)
+   */ async warpL2TimeAtLeastTo(sequencerClient, node, targetTimestamp) {
+        const currentL2BlockNumber = await node.getBlockNumber();
+        // We warp the L1 timestamp
+        await this.eth.warp(targetTimestamp, {
+            resetBlockInterval: true
+        });
+        // Wait until an L2 block is mined
+        const sequencer = sequencerClient.getSequencer();
+        const minTxsPerBlock = sequencer.getConfig().minTxsPerBlock;
+        sequencer.updateConfig({
+            minTxsPerBlock: 0
+        });
+        await retryUntil(async ()=>{
+            const newL2BlockNumber = await node.getBlockNumber();
+            return newL2BlockNumber > currentL2BlockNumber;
+        }, 'new block after warping L2 time', 36, 1);
+        // Restore original minTxsPerBlock
+        sequencer.updateConfig({
+            minTxsPerBlock
+        });
+    }
+    /**
+   * Warps the L1 timestamp forward by a specified duration and mines an L2 block that advances the L2 timestamp at
+   * least by the duration. L2 timestamp is not advanced exactly by the duration because it is determined by the slot
+   * number, which advances in fixed intervals.
+   * This is useful for testing time-dependent contract behavior.
+   * @param sequencerClient - The sequencer client to use to force an empty block to be mined.
+   * @param node - The Aztec node used to query if a new block has been mined.
+   * @param duration - The duration to advance time by (in seconds)
+   */ async warpL2TimeAtLeastBy(sequencerClient, node, duration) {
+        const currentTimestamp = await this.eth.timestamp();
+        const targetTimestamp = BigInt(currentTimestamp) + BigInt(duration);
+        await this.warpL2TimeAtLeastTo(sequencerClient, node, targetTimestamp);
+    }
+}

package/dest/testing/index.d.ts

@@ -0,0 +1,4 @@
+export { AnvilTestWatcher } from './anvil_test_watcher.js';
+export { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
+export { CheatCodes } from './cheat_codes.js';
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90ZXN0aW5nL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxnQkFBZ0IsRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQzNELE9BQU8sRUFBRSxhQUFhLEVBQUUsZ0JBQWdCLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUN2RSxPQUFPLEVBQUUsVUFBVSxFQUFFLE1BQU0sa0JBQWtCLENBQUMifQ==

package/dest/testing/index.d.ts.map

@@ -0,0 +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/dest/testing/index.js

@@ -0,0 +1,3 @@
+export { AnvilTestWatcher } from './anvil_test_watcher.js';
+export { EthCheatCodes, RollupCheatCodes } from '@aztec/ethereum/test';
+export { CheatCodes } from './cheat_codes.js';

package/package.json

@@ -1,11 +1,12 @@
 {
   "name": "@aztec/aztec",
-  "version": "0.0.0-test.0",
+  "version": "0.0.1-commit.03f7ef2",
   "type": "module",
   "exports": {
-    ".": "./dest/index.js"
+    ".": "./dest/index.js",
+    "./testing": "./dest/testing/index.js"
   },
-  "bin": "./dest/bin/index.js",
+  "bin": "./scripts/aztec.sh",
   "typedocOptions": {
     "entryPoints": [
       "./src/index.ts"
@@ -14,14 +15,12 @@
     "tsconfig": "./tsconfig.json"
   },
   "scripts": {
-    "build": "yarn clean && tsc -b",
+    "build": "yarn clean && ../scripts/tsc.sh",
     "start": "node --no-warnings ./dest/bin",
     "start:debug": "node --inspect=0.0.0.0:9221 --no-warnings ./dest/bin",
-    "start:sandbox": "ETHEREUM_HOSTS=http://0.0.0.0:8545/ && yarn start start --sandbox",
+    "start:local-network": "ETHEREUM_HOSTS=http://0.0.0.0:8545/ && yarn start start --local-network",
     "clean": "rm -rf ./dest .tsbuildinfo",
-    "formatting": "run -T prettier --check ./src && run -T eslint ./src",
-    "formatting:fix": "run -T eslint --fix ./src && run -T prettier -w ./src",
-    "build:dev": "tsc -b --watch",
+    "build:dev": "../scripts/tsc.sh --watch",
     "test": "NODE_NO_WARNINGS=1 node --experimental-vm-modules ../node_modules/.bin/jest --passWithNoTests --maxWorkers=${JEST_MAX_WORKERS:-8}",
     "run:example:token": "LOG_LEVEL='verbose' node ./dest/examples/token.js"
   },
@@ -29,55 +28,61 @@
     "../package.common.json"
   ],
   "dependencies": {
-    "@aztec/accounts": "0.0.0-test.0",
-    "@aztec/archiver": "0.0.0-test.0",
-    "@aztec/aztec-faucet": "0.0.0-test.0",
-    "@aztec/aztec-node": "0.0.0-test.0",
-    "@aztec/aztec.js": "0.0.0-test.0",
-    "@aztec/bb-prover": "0.0.0-test.0",
-    "@aztec/blob-sink": "0.0.0-test.0",
-    "@aztec/bot": "0.0.0-test.0",
-    "@aztec/builder": "0.0.0-test.0",
-    "@aztec/cli": "0.0.0-test.0",
-    "@aztec/cli-wallet": "0.0.0-test.0",
-    "@aztec/constants": "0.0.0-test.0",
-    "@aztec/entrypoints": "0.0.0-test.0",
-    "@aztec/ethereum": "0.0.0-test.0",
-    "@aztec/foundation": "0.0.0-test.0",
-    "@aztec/kv-store": "0.0.0-test.0",
-    "@aztec/noir-contracts.js": "0.0.0-test.0",
-    "@aztec/noir-protocol-circuits-types": "0.0.0-test.0",
-    "@aztec/p2p": "0.0.0-test.0",
-    "@aztec/p2p-bootstrap": "0.0.0-test.0",
-    "@aztec/protocol-contracts": "0.0.0-test.0",
-    "@aztec/prover-client": "0.0.0-test.0",
-    "@aztec/prover-node": "0.0.0-test.0",
-    "@aztec/pxe": "0.0.0-test.0",
-    "@aztec/stdlib": "0.0.0-test.0",
-    "@aztec/telemetry-client": "0.0.0-test.0",
-    "@aztec/txe": "0.0.0-test.0",
-    "@aztec/world-state": "0.0.0-test.0",
+    "@aztec/accounts": "0.0.1-commit.03f7ef2",
+    "@aztec/archiver": "0.0.1-commit.03f7ef2",
+    "@aztec/aztec-faucet": "0.0.1-commit.03f7ef2",
+    "@aztec/aztec-node": "0.0.1-commit.03f7ef2",
+    "@aztec/aztec.js": "0.0.1-commit.03f7ef2",
+    "@aztec/bb-prover": "0.0.1-commit.03f7ef2",
+    "@aztec/bb.js": "0.0.1-commit.03f7ef2",
+    "@aztec/blob-client": "0.0.1-commit.03f7ef2",
+    "@aztec/bot": "0.0.1-commit.03f7ef2",
+    "@aztec/builder": "0.0.1-commit.03f7ef2",
+    "@aztec/cli": "0.0.1-commit.03f7ef2",
+    "@aztec/constants": "0.0.1-commit.03f7ef2",
+    "@aztec/entrypoints": "0.0.1-commit.03f7ef2",
+    "@aztec/ethereum": "0.0.1-commit.03f7ef2",
+    "@aztec/foundation": "0.0.1-commit.03f7ef2",
+    "@aztec/kv-store": "0.0.1-commit.03f7ef2",
+    "@aztec/l1-artifacts": "0.0.1-commit.03f7ef2",
+    "@aztec/node-lib": "0.0.1-commit.03f7ef2",
+    "@aztec/noir-contracts.js": "0.0.1-commit.03f7ef2",
+    "@aztec/noir-protocol-circuits-types": "0.0.1-commit.03f7ef2",
+    "@aztec/p2p": "0.0.1-commit.03f7ef2",
+    "@aztec/p2p-bootstrap": "0.0.1-commit.03f7ef2",
+    "@aztec/protocol-contracts": "0.0.1-commit.03f7ef2",
+    "@aztec/prover-client": "0.0.1-commit.03f7ef2",
+    "@aztec/prover-node": "0.0.1-commit.03f7ef2",
+    "@aztec/pxe": "0.0.1-commit.03f7ef2",
+    "@aztec/sequencer-client": "0.0.1-commit.03f7ef2",
+    "@aztec/stdlib": "0.0.1-commit.03f7ef2",
+    "@aztec/telemetry-client": "0.0.1-commit.03f7ef2",
+    "@aztec/test-wallet": "0.0.1-commit.03f7ef2",
+    "@aztec/txe": "0.0.1-commit.03f7ef2",
+    "@aztec/world-state": "0.0.1-commit.03f7ef2",
     "@types/chalk": "^2.2.0",
     "abitype": "^0.8.11",
     "chalk": "^5.3.0",
     "commander": "^12.1.0",
-    "koa": "^2.14.2",
-    "koa-router": "^12.0.0",
-    "viem": "2.22.8"
+    "koa": "^2.16.1",
+    "koa-router": "^13.1.1",
+    "viem": "npm:@aztec/viem@2.38.2"
   },
   "files": [
     "dest",
+    "scripts",
     "src",
     "!*.test.*"
   ],
   "types": "./dest/index.d.ts",
   "devDependencies": {
-    "@jest/globals": "^29.5.0",
-    "@types/jest": "^29.5.0",
-    "@types/koa": "^2.13.6",
-    "jest": "^29.5.0",
+    "@jest/globals": "^30.0.0",
+    "@types/jest": "^30.0.0",
+    "@types/koa": "^2.15.0",
+    "@typescript/native-preview": "7.0.0-dev.20251126.1",
+    "jest": "^30.0.0",
     "ts-node": "^10.9.1",
-    "typescript": "^5.0.4"
+    "typescript": "^5.3.3"
   },
   "jest": {
     "moduleNameMapper": {
@@ -110,9 +115,13 @@
     "testTimeout": 120000,
     "setupFiles": [
       "../../foundation/src/jest/setup.mjs"
+    ],
+    "testEnvironment": "../../foundation/src/jest/env.mjs",
+    "setupFilesAfterEnv": [
+      "../../foundation/src/jest/setupAfterEnv.mjs"
     ]
   },
   "engines": {
-    "node": ">=18"
+    "node": ">=20.10"
   }
 }

package/scripts/aztec.sh

@@ -0,0 +1,56 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+script_dir="$(dirname "$(realpath "$0")")"
+
+function aztec {
+  exec node --no-warnings $script_dir/../dest/bin/index.js "$@"
+}
+
+cmd=${1:-}
+[ -n "$cmd" ] && shift
+
+export AZTEC_SHELL_WRAPPER=1
+
+case $cmd in
+  test)
+    export LOG_LEVEL="${LOG_LEVEL:-info}"
+    aztec start --txe --port 8081 &
+    server_pid=$!
+    trap 'kill $server_pid &>/dev/null || true' EXIT
+    while ! nc -z 127.0.0.1 8081 &>/dev/null; do sleep 0.2; done
+    export NARGO_FOREIGN_CALL_TIMEOUT=300000
+    nargo test --silence-warnings --pedantic-solving --oracle-resolver http://127.0.0.1:8081 "$@"
+    ;;
+  start)
+    if [ "${1:-}" == "--local-network" ]; then
+      # TODO: Can these just be set in TS?
+      export ARCHIVER_POLLING_INTERVAL_MS=500
+      export P2P_BLOCK_CHECK_INTERVAL_MS=500
+      export SEQ_TX_POLLING_INTERVAL_MS=500
+      export WS_BLOCK_CHECK_INTERVAL_MS=500
+      export ARCHIVER_VIEM_POLLING_INTERVAL_MS=500
+      export TEST_ACCOUNTS=${TEST_ACCOUNTS:-true}
+      export LOG_LEVEL=${LOG_LEVEL:-info;silent:sequencer;verbose:debug_log}
+      export DEPLOY_AZTEC_CONTRACTS_SALT=${DEPLOY_AZTEC_CONTRACTS_SALT:-$RANDOM}
+
+      ANVIL_PORT=${ANVIL_PORT:-8545}
+
+      export L1_CHAIN_ID=${L1_CHAIN_ID:-31337}
+      export ETHEREUM_HOSTS=${ETHEREUM_HOSTS:-"http://127.0.0.1:${ANVIL_PORT}"}
+
+      anvil --version
+      anvil --silent &
+      anvil_pid=$!
+      trap 'kill $anvil_pid &>/dev/null' EXIT
+    fi
+
+    aztec start "$@"
+    ;;
+  compile|new|init|flamegraph)
+    $script_dir/${cmd}.sh "$@"
+    ;;
+  *)
+    aztec $cmd "$@"
+    ;;
+esac

package/scripts/compile.sh

@@ -0,0 +1,44 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+NARGO=${NARGO:-nargo}
+BB=${BB:-bb}
+
+# If help is requested, show Aztec-specific info then run nargo compile help and then exit in order to not trigger
+# transpilation
+for arg in "$@"; do
+  if [ "$arg" == "--help" ] || [ "$arg" == "-h" ]; then
+    cat << 'EOF'
+Aztec Compile - Compile Aztec Noir contracts
+
+This command compiles Aztec Noir contracts using nargo and then automatically
+postprocesses them to generate Aztec specific artifacts including:
+- Transpiled contract artifacts
+- Verification keys
+
+The compiled contracts will be placed in the target/ directory by default.
+
+---
+Underlying nargo compile options:
+
+EOF
+    nargo compile --help
+    exit 0
+  fi
+done
+
+# Run nargo compile.
+$NARGO compile "$@"
+
+echo "Postprocessing contract..."
+$BB aztec_process
+
+# Strip internal prefixes from all compiled contract JSONs in target directory
+# TODO: This should be part of bb aztec_process!
+for json in target/*.json; do
+  temp_file="${json}.tmp"
+  jq '.functions |= map(.name |= sub("^__aztec_nr_internals__"; ""))' "$json" > "$temp_file"
+  mv "$temp_file" "$json"
+done
+
+echo "Compilation complete!"

package/scripts/extract_function.js

@@ -0,0 +1,47 @@
+#!/usr/bin/env node
+import fs from 'fs/promises';
+import path from 'path';
+
+// Simple script to extract a contract function as a separate Noir artifact.
+// We need to use this since the transpiling that we do on public functions make the contract artifacts
+// unreadable by noir tooling, since they are no longer following the noir artifact format.
+async function main() {
+  let [contractArtifactPath, functionName] = process.argv.slice(2);
+  if (!contractArtifactPath || !functionName) {
+    console.log('Usage: node extractFunctionAsNoirArtifact.js <contractArtifactPath> <functionName>');
+    return;
+  }
+
+  const contractArtifact = JSON.parse(await fs.readFile(contractArtifactPath, 'utf8'));
+  const func = contractArtifact.functions.find(f => f.name === functionName);
+  if (!func) {
+    console.error(`Function ${functionName} not found in ${contractArtifactPath}`);
+    return;
+  }
+
+  const artifact = {
+    noir_version: contractArtifact.noir_version,
+    hash: 0,
+    abi: func.abi,
+    bytecode: func.bytecode,
+    debug_symbols: func.debug_symbols,
+    file_map: contractArtifact.file_map,
+    expression_width: {
+      Bounded: {
+        width: 4,
+      },
+    },
+  };
+
+  const outputDir = path.dirname(contractArtifactPath);
+  const outputName = path.basename(contractArtifactPath, '.json') + `-${functionName}.json`;
+
+  const outPath = path.join(outputDir, outputName);
+
+  await fs.writeFile(outPath, JSON.stringify(artifact, null, 2));
+}
+
+main().catch(err => {
+  console.error(err);
+  process.exit(1);
+});

package/scripts/flamegraph.sh

@@ -0,0 +1,59 @@
+#!/usr/bin/env bash
+set -eu
+
+# If first arg is -h or --help, print usage.
+if [ $# -lt 2 ] || [ "$1" == "-h" ] || [ "$1" == "--help" ]; then
+    cat << 'EOF'
+Aztec Flamegraph - Generate a gate count flamegraph for an aztec contract function.
+
+Usage: aztec flamegraph <contract_artifact> <function>
+
+Options:
+  -h, --help     Print help
+
+Will output an svg at <artifact_path>/<contract>-<function>-flamegraph.svg.
+You can open it in your browser to view it.
+
+EOF
+    exit 0
+fi
+
+cleanup() {
+  set +e
+  if [ -f "$function_artifact" ]; then
+    rm -f "$function_artifact"
+  fi
+}
+
+trap cleanup EXIT
+
+# Get the directory of the script
+script_dir=$(realpath $(dirname $0))
+
+PROFILER=${PROFILER_PATH:-noir-profiler}
+BB=${BB:-bb}
+
+# first console arg is contract name in camel case or path to contract artifact
+contract=$1
+
+# second console arg is the contract function
+function=$2
+
+if [ ! -f "$contract" ]; then
+  echo "Error: Contract artifact not found at: $contract"
+  exit 1
+fi
+artifact_path=$contract
+function_artifact="${artifact_path%%.json}-${function}.json"
+output_dir=$(dirname "$artifact_path")
+
+# Extract artifact for the specific function.
+node $script_dir/extract_function.js "$artifact_path" $function
+
+# Generate the flamegraph
+$PROFILER gates --artifact-path "$function_artifact" --backend-path "$BB" --backend-gates-command "gates" --output "$output_dir" --scheme chonk --include_gates_per_opcode
+
+# Save as $artifact_name-$function-flamegraph.svg
+output_file="${function_artifact%%.json}-flamegraph.svg"
+mv "$output_dir/__aztec_nr_internals__${function}_gates.svg" "$output_file"
+echo "Flamegraph generated at: $output_file"

package/scripts/init.sh

@@ -0,0 +1,35 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+NARGO=${NARGO:-nargo}
+script_path=$(realpath $(dirname "$0"))
+
+for arg in "$@"; do
+  if [ "$arg" == "--help" ] || [ "$arg" == "-h" ]; then
+    cat << 'EOF'
+Aztec Init - Create a new Aztec Noir project in the current directory
+
+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.
+
+EOF
+    exit 0
+  fi
+  if [ "$arg" == "--lib" ]; then
+    is_contract=0
+  fi
+done
+
+echo "Initializing Noir project..."
+$NARGO init "$@"
+
+if [ "${is_contract:-1}" -eq 1 ]; then
+  $script_path/setup_project.sh
+fi

package/scripts/new.sh

@@ -0,0 +1,59 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+NARGO=${NARGO:-nargo}
+script_path=$(realpath $(dirname "$0"))
+
+type_arg="--contract"
+
+while [[ $# -gt 0 ]]; do
+  case $1 in
+    --help|-h)
+      cat << 'EOF'
+Aztec New - Create a new Aztec Noir project in a new directory
+
+Usage: aztec new [OPTIONS] <PATH>
+
+Arguments:
+  <PATH>  The path to save the new project
+
+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.
+EOF
+      exit 0
+      ;;
+    --lib)
+      type_arg="--lib"
+      shift
+      ;;
+    --name)
+      name_arg="--name $2"
+      shift 2
+      ;;
+    *)
+      project_path=$1
+      shift
+      break
+      ;;
+  esac
+done
+
+if [ -z "$project_path" ]; then
+  echo "Error: PATH argument is required"
+  echo "Usage: aztec new [OPTIONS] <PATH>"
+  echo "Run 'aztec new --help' for more information"
+  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
+fi

package/scripts/setup_project.sh

@@ -0,0 +1,31 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Get the actual aztec version for the git tag.
+AZTEC_VERSION=$(jq -r '.version' $(dirname $0)/../package.json)
+NARGO_TOML_PATH="Nargo.toml"
+MAIN_NR_PATH="src/main.nr"
+
+if [ ! -f "$NARGO_TOML_PATH" ]; then
+  >&2 echo "Warning: Could not find Nargo.toml at $NARGO_TOML_PATH to add aztec dependency"
+  exit 1
+fi
+
+if [ ! -f "$MAIN_NR_PATH" ]; then
+  >&2 echo "Warning: Could not find main.nr at $MAIN_NR_PATH"
+  exit 1
+fi
+
+# Add aztec dependency to Nargo.toml
+echo "" >> "$NARGO_TOML_PATH"
+echo "aztec = { git=\"https://github.com/AztecProtocol/aztec-nr\", tag=\"v${AZTEC_VERSION}\", directory=\"aztec\" }" >> "$NARGO_TOML_PATH"
+echo "Added aztec dependency (v${AZTEC_VERSION}) to Nargo.toml"
+
+# Replace the contents of main.nr with the Aztec contract template
+cat > "$MAIN_NR_PATH" << 'EOF'
+use aztec::macros::aztec;
+
+#[aztec]
+contract Main {}
+EOF
+echo "Created main.nr with Aztec contract template"

package/src/bin/index.ts

@@ -1,21 +1,23 @@
 #!/usr/bin/env node
 //
 import { injectCommands as injectBuilderCommands } from '@aztec/builder';
-import { injectCommands as injectWalletCommands } from '@aztec/cli-wallet';
+import { injectCommands as injectAztecNodeCommands } from '@aztec/cli/aztec_node';
+import { enrichEnvironmentWithChainName } from '@aztec/cli/config/chain';
+import { enrichEnvironmentWithNetworkConfig } from '@aztec/cli/config/network';
 import { injectCommands as injectContractCommands } from '@aztec/cli/contracts';
-import { injectCommands as injectDevnetCommands } from '@aztec/cli/devnet';
 import { injectCommands as injectInfrastructureCommands } from '@aztec/cli/infrastructure';
 import { injectCommands as injectL1Commands } from '@aztec/cli/l1';
 import { injectCommands as injectMiscCommands } from '@aztec/cli/misc';
-import { injectCommands as injectPXECommands } from '@aztec/cli/pxe';
+import { injectCommands as injectValidatorKeysCommands } from '@aztec/cli/validator_keys';
+import { getActiveNetworkName } from '@aztec/foundation/config';
 import { createConsoleLogger, createLogger } from '@aztec/foundation/log';
-import { fileURLToPath } from '@aztec/foundation/url';
 
 import { Command } from 'commander';
-import { readFileSync } from 'fs';
-import { dirname, resolve } from 'path';
 
 import { injectAztecCommands } from '../cli/index.js';
+import { getCliVersion } from '../cli/release_version.js';
+
+const NETWORK_FLAG = 'network';
 
 const userLog = createConsoleLogger();
 const debugLogger = createLogger('cli');
@@ -28,19 +30,31 @@ async function main() {
   process.once('SIGINT', shutdown);
   process.once('SIGTERM', shutdown);
 
-  const packageJsonPath = resolve(dirname(fileURLToPath(import.meta.url)), '../../package.json');
-  const cliVersion: string = JSON.parse(readFileSync(packageJsonPath).toString()).version;
+  // Intercept the setting of a network and enrich the environment with defaults for that network
+  let networkValue: string | undefined;
+
+  const args = process.argv.slice(2);
+  const networkIndex = args.findIndex(arg => arg.startsWith(`--${NETWORK_FLAG}=`) || arg === `--${NETWORK_FLAG}`);
+
+  if (networkIndex !== -1) {
+    networkValue = args[networkIndex].split('=')[1] || args[networkIndex + 1];
+  }
+
+  const networkName = getActiveNetworkName(networkValue);
+  await enrichEnvironmentWithNetworkConfig(networkName);
+  enrichEnvironmentWithChainName(networkName);
+
+  const cliVersion = getCliVersion();
   let program = new Command('aztec');
   program.description('Aztec command line interface').version(cliVersion);
   program = injectAztecCommands(program, userLog, debugLogger);
   program = injectBuilderCommands(program);
   program = injectContractCommands(program, userLog, debugLogger);
-  program = injectInfrastructureCommands(program, userLog, debugLogger);
+  program = injectInfrastructureCommands(program, userLog);
   program = injectL1Commands(program, userLog, debugLogger);
-  program = injectPXECommands(program, userLog, debugLogger);
+  program = injectAztecNodeCommands(program, userLog, debugLogger);
   program = injectMiscCommands(program, userLog);
-  program = injectDevnetCommands(program, userLog, debugLogger);
-  program = injectWalletCommands(program, userLog, debugLogger);
+  program = injectValidatorKeysCommands(program, userLog);
 
   await program.parseAsync(process.argv);
 }