@aixyz/cli 0.10.0 → 0.11.0

package/build/index.ts

@@ -49,6 +49,8 @@ Examples:
 async function action(options: BuildOptions = {}): Promise<void> {
   const cwd = process.cwd();
   loadEnvConfig(cwd, false);
+  process.env.NODE_ENV = "production";
+  process.env.AIXYZ_ENV = "production";
   const entrypoint = getEntrypointMayGenerate(cwd, "build");
 
   // Determine output target: explicit CLI flag takes precedence, then config file, then auto-detect VERCEL env
@@ -79,6 +81,10 @@ async function buildBun(entrypoint: string): Promise<void> {
     target: "bun",
     format: "esm",
     sourcemap: "linked",
+    define: {
+      "process.env.NODE_ENV": JSON.stringify("production"),
+      "process.env.AIXYZ_ENV": JSON.stringify("production"),
+    },
     plugins: [AixyzConfigPlugin(), AixyzServerPlugin(entrypoint, "standalone")],
   });
 
@@ -136,6 +142,10 @@ async function buildVercel(entrypoint: string): Promise<void> {
     target: "bun",
     format: "esm",
     sourcemap: "linked",
+    define: {
+      "process.env.NODE_ENV": JSON.stringify("production"),
+      "process.env.AIXYZ_ENV": JSON.stringify("production"),
+    },
     plugins: [AixyzConfigPlugin(), AixyzServerPlugin(entrypoint, "vercel")],
   });
 

package/dev/index.ts

@@ -15,6 +15,8 @@ async function action(options: { port?: string }): Promise<void> {
 
   // Load environment config
   const { loadedEnvFiles } = loadEnvConfig(cwd, true);
+  process.env.NODE_ENV = "development";
+  process.env.AIXYZ_ENV = "development";
   const envFileNames = loadedEnvFiles.map((f) => relative(cwd, f.path));
 
   const port = options.port || process.env.PORT || "3000";
@@ -41,7 +43,7 @@ async function action(options: { port?: string }): Promise<void> {
       cwd,
       stdout: "inherit",
       stderr: "inherit",
-      env: process.env,
+      env: { ...process.env, NODE_ENV: "development", AIXYZ_ENV: "development" },
     });
     child.exited.then((code) => {
       if (!restarting && code !== 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aixyz/cli",
-  "version": "0.10.0",
+  "version": "0.11.0",
   "description": "Payment-native SDK for AI Agent",
   "keywords": [
     "ai",
@@ -27,8 +27,8 @@
     "bin.ts"
   ],
   "dependencies": {
-    "@aixyz/config": "0.10.0",
-    "@aixyz/erc-8004": "0.10.0",
+    "@aixyz/config": "0.11.0",
+    "@aixyz/erc-8004": "0.11.0",
     "@inquirer/prompts": "^8.3.0",
     "@next/env": "^16.1.6",
     "boxen": "^8.0.1",

package/register/index.ts

@@ -4,7 +4,6 @@ import { update } from "./update";
 import type { WalletOptions } from "./wallet";
 
 export interface BaseOptions extends WalletOptions {
-  chain?: string;
   rpcUrl?: string;
   registry?: string;
   outDir?: string;
@@ -16,9 +15,9 @@ erc8004Command
   .command("register")
   .description("Register a new agent to the ERC-8004 IdentityRegistry")
   .option("--url <url>", "Agent deployment URL (e.g., https://my-agent.example.com)")
-  .option("--chain <chain>", "Target chain (mainnet, sepolia, base-sepolia, localhost)")
+  .option("--chain-id <chainId>", "Target chain by numeric chain ID", parseInt)
   .option("--rpc-url <url>", "Custom RPC URL (uses default if not provided)")
-  .option("--registry <address>", "Contract address of the IdentityRegistry (required for localhost)")
+  .option("--registry <address>", "Contract address of the IdentityRegistry (required for localhost and custom chains)")
   .option("--keystore <path>", "Path to Ethereum keystore (V3) JSON file for local signing")
   .option("--browser", "Use browser extension wallet (any extension)")
   .option("--broadcast", "Sign and broadcast the transaction (default: dry-run)")
@@ -32,22 +31,23 @@ Option Details:
       The registration URI will be derived as <url>/_aixyz/erc-8004.json.
       If omitted, you will be prompted to enter the URL interactively.
 
-  --chain <chain>
-      Target chain for registration. Supported values:
-        mainnet       Ethereum mainnet (chain ID 1)
-        sepolia       Ethereum Sepolia testnet (chain ID 11155111)
-        base-sepolia  Base Sepolia testnet (chain ID 84532)
-        localhost     Local Foundry/Anvil node (chain ID 31337)
+  --chain-id <chainId>
+      Target chain by numeric chain ID. Supported chain IDs include:
+        1 (mainnet), 8453 (base), 42161 (arbitrum), 10 (optimism),
+        137 (polygon), 56 (bsc), 43114 (avalanche), 11155111 (sepolia),
+        84532 (base-sepolia), and more (see @aixyz/erc-8004).
       If omitted, you will be prompted to select a chain interactively.
+      Use any custom EVM chain ID with --rpc-url for BYO chains.
 
   --rpc-url <url>
       Custom RPC endpoint URL. Overrides the default RPC for the selected
-      chain. Cannot be used with --browser since the browser wallet manages
+      chain. Required when using a custom chain ID with no default deployment.
+      Cannot be used with --browser since the browser wallet manages
       its own RPC connection.
 
   --registry <address>
-      Contract address of the ERC-8004 IdentityRegistry. Only required for
-      localhost, where there is no default deployment.
+      Contract address of the ERC-8004 IdentityRegistry. Required for
+      localhost and custom chains, where there is no default deployment.
 
   --keystore <path>
       Path to an Ethereum keystore (V3) JSON file. You will be prompted for
@@ -74,11 +74,14 @@ Environment Variables:
 
 Examples:
   # Dry-run (default)
-  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain sepolia
+  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain-id 11155111
 
-  # Sign and broadcast
-  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain sepolia --keystore ~/.foundry/keystores/default --broadcast
-  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain sepolia --browser --broadcast`,
+  # Sign and broadcast (known chain)
+  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain-id 84532 --keystore ~/.foundry/keystores/default --broadcast
+  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain-id 84532 --browser --broadcast
+
+  # BYO: register on any custom EVM chain
+  $ aixyz erc-8004 register --url "https://my-agent.example.com" --chain-id 999999 --rpc-url https://my-rpc.example.com --registry 0xABCD... --broadcast`,
   )
   .action(register);
 

package/register/register.ts

@@ -3,15 +3,16 @@ import { IdentityRegistryAbi } from "@aixyz/erc-8004";
 import { selectWalletMethod } from "./wallet";
 import { signTransaction } from "./wallet/sign";
 import {
-  resolveChainConfig,
+  resolveChainConfigById,
   selectChain,
   resolveRegistryAddress,
   validateBrowserRpcConflict,
   getExplorerUrl,
+  CHAINS,
 } from "./utils/chain";
 import { writeResultJson } from "./utils/result";
 import { label, truncateUri, broadcastAndConfirm, logSignResult } from "./utils/transaction";
-import { promptAgentUrl, promptSupportedTrust, deriveAgentUri } from "./utils/prompt";
+import { promptAgentUrl, promptSupportedTrust, promptRegistryAddress, deriveAgentUri } from "./utils/prompt";
 import { hasErc8004File, createErc8004File, writeRegistrationEntry } from "./utils/erc8004-file";
 import { confirm } from "@inquirer/prompts";
 import chalk from "chalk";
@@ -20,7 +21,7 @@ import type { BaseOptions } from "./index";
 
 export interface RegisterOptions extends BaseOptions {
   url?: string;
-  chain?: string;
+  chainId?: number;
 }
 
 export async function register(options: RegisterOptions): Promise<void> {
@@ -47,9 +48,10 @@ export async function register(options: RegisterOptions): Promise<void> {
   }
 
   // Step 3: Select chain
-  const chainName = options.chain ?? (await selectChain());
-  const chainConfig = resolveChainConfig(chainName);
-  const registryAddress = resolveRegistryAddress(chainName, chainConfig.chainId, options.registry);
+  const chainId = options.chainId ?? (await selectChain());
+  const chainConfig = resolveChainConfigById(chainId, options.rpcUrl);
+  const chainName = Object.entries(CHAINS).find(([, c]) => c.chainId === chainId)?.[0] ?? `chain-${chainId}`;
+  const registryAddress = resolveRegistryAddress(chainId, options.registry) ?? (await promptRegistryAddress());
 
   // Step 4: Encode transaction
   const data = encodeFunctionData({

package/register/update.ts

@@ -2,7 +2,7 @@ import { encodeFunctionData, formatEther, parseEventLogs, type Chain, type Log }
 import { IdentityRegistryAbi } from "@aixyz/erc-8004";
 import { selectWalletMethod } from "./wallet";
 import { signTransaction } from "./wallet/sign";
-import { resolveChainConfig, validateBrowserRpcConflict, getExplorerUrl, CHAINS } from "./utils/chain";
+import { resolveChainConfigById, validateBrowserRpcConflict, getExplorerUrl, CHAINS } from "./utils/chain";
 import { writeResultJson } from "./utils/result";
 import { label, truncateUri, broadcastAndConfirm, logSignResult } from "./utils/transaction";
 import { promptAgentUrl, promptSelectRegistration, deriveAgentUri } from "./utils/prompt";
@@ -36,7 +36,7 @@ export async function update(options: UpdateOptions): Promise<void> {
   const chainId = Number(parts[1]);
   const registryAddress = parts.slice(2).join(":") as `0x${string}`;
   const chainName = Object.entries(CHAINS).find(([, config]) => config.chainId === chainId)?.[0] ?? `chain-${chainId}`;
-  const chainConfig = resolveChainConfig(chainName);
+  const chainConfig = resolveChainConfigById(chainId, options.rpcUrl);
 
   // Step 4: Get new agent URL and derive URI
   const agentUrl = options.url ?? (await promptAgentUrl());

package/register/utils/chain.ts

@@ -1,20 +1,128 @@
-import { isAddress, type Chain } from "viem";
-import { mainnet, sepolia, baseSepolia, foundry } from "viem/chains";
+import { isAddress, defineChain, type Chain } from "viem";
+import {
+  abstract,
+  abstractTestnet,
+  arbitrum,
+  arbitrumSepolia,
+  avalanche,
+  avalancheFuji,
+  base,
+  baseSepolia,
+  bsc,
+  bscTestnet,
+  celo,
+  celoSepolia,
+  foundry,
+  gnosis,
+  linea,
+  lineaSepolia,
+  mainnet,
+  mantle,
+  mantleSepoliaTestnet,
+  megaeth,
+  monad,
+  monadTestnet,
+  optimism,
+  optimismSepolia,
+  polygon,
+  polygonAmoy,
+  scroll,
+  scrollSepolia,
+  sepolia,
+  taiko,
+} from "viem/chains";
 import { CHAIN_ID, getIdentityRegistryAddress } from "@aixyz/erc-8004";
-import { select } from "@inquirer/prompts";
+import { input, select } from "@inquirer/prompts";
 
 export interface ChainConfig {
   chain: Chain;
   chainId: number;
 }
 
+// Maps supported chain IDs to viem Chain objects (derived from @aixyz/erc-8004 CHAIN_ID)
+const VIEM_CHAIN_BY_ID: Record<number, Chain> = {
+  [CHAIN_ID.ABSTRACT]: abstract,
+  [CHAIN_ID.ARBITRUM]: arbitrum,
+  [CHAIN_ID.AVALANCHE]: avalanche,
+  [CHAIN_ID.BASE]: base,
+  [CHAIN_ID.BSC]: bsc,
+  [CHAIN_ID.CELO]: celo,
+  [CHAIN_ID.GNOSIS]: gnosis,
+  [CHAIN_ID.LINEA]: linea,
+  [CHAIN_ID.MAINNET]: mainnet,
+  [CHAIN_ID.MANTLE]: mantle,
+  [CHAIN_ID.MEGAETH]: megaeth,
+  [CHAIN_ID.MONAD]: monad,
+  [CHAIN_ID.OPTIMISM]: optimism,
+  [CHAIN_ID.POLYGON]: polygon,
+  [CHAIN_ID.SCROLL]: scroll,
+  [CHAIN_ID.TAIKO]: taiko,
+  [CHAIN_ID.ABSTRACT_TESTNET]: abstractTestnet,
+  [CHAIN_ID.ARBITRUM_SEPOLIA]: arbitrumSepolia,
+  [CHAIN_ID.AVALANCHE_FUJI]: avalancheFuji,
+  [CHAIN_ID.BASE_SEPOLIA]: baseSepolia,
+  [CHAIN_ID.BSC_TESTNET]: bscTestnet,
+  [CHAIN_ID.CELO_SEPOLIA]: celoSepolia,
+  [CHAIN_ID.LINEA_SEPOLIA]: lineaSepolia,
+  [CHAIN_ID.MANTLE_SEPOLIA]: mantleSepoliaTestnet,
+  [CHAIN_ID.MONAD_TESTNET]: monadTestnet,
+  [CHAIN_ID.OPTIMISM_SEPOLIA]: optimismSepolia,
+  [CHAIN_ID.POLYGON_AMOY]: polygonAmoy,
+  [CHAIN_ID.SCROLL_SEPOLIA]: scrollSepolia,
+  [CHAIN_ID.SEPOLIA]: sepolia,
+  31337: foundry,
+};
+
+// Build CHAINS from CHAIN_ID as the single source of truth from @aixyz/erc-8004.
+// Chain names are derived from CHAIN_ID keys: lowercase with underscores replaced by hyphens.
 export const CHAINS: Record<string, ChainConfig> = {
-  mainnet: { chain: mainnet, chainId: CHAIN_ID.MAINNET },
-  sepolia: { chain: sepolia, chainId: CHAIN_ID.SEPOLIA },
-  "base-sepolia": { chain: baseSepolia, chainId: CHAIN_ID.BASE_SEPOLIA },
+  ...Object.fromEntries(
+    (Object.entries(CHAIN_ID) as [string, number][])
+      .filter(([, id]) => id in VIEM_CHAIN_BY_ID)
+      .map(([key, id]) => [key.toLowerCase().replace(/_/g, "-"), { chain: VIEM_CHAIN_BY_ID[id]!, chainId: id }]),
+  ),
   localhost: { chain: foundry, chainId: 31337 },
 };
 
+// Priority-ordered chain IDs for interactive selection — most popular first.
+const CHAIN_SELECTION_ORDER: number[] = [
+  // Popular mainnets
+  CHAIN_ID.MAINNET,
+  CHAIN_ID.BASE,
+  CHAIN_ID.ARBITRUM,
+  CHAIN_ID.OPTIMISM,
+  CHAIN_ID.POLYGON,
+  CHAIN_ID.BSC,
+  CHAIN_ID.AVALANCHE,
+  CHAIN_ID.SCROLL,
+  CHAIN_ID.LINEA,
+  CHAIN_ID.CELO,
+  CHAIN_ID.GNOSIS,
+  CHAIN_ID.TAIKO,
+  CHAIN_ID.MANTLE,
+  CHAIN_ID.MONAD,
+  CHAIN_ID.MEGAETH,
+  CHAIN_ID.ABSTRACT,
+  // Popular testnets
+  CHAIN_ID.SEPOLIA,
+  CHAIN_ID.BASE_SEPOLIA,
+  CHAIN_ID.ARBITRUM_SEPOLIA,
+  CHAIN_ID.OPTIMISM_SEPOLIA,
+  CHAIN_ID.POLYGON_AMOY,
+  CHAIN_ID.AVALANCHE_FUJI,
+  CHAIN_ID.BSC_TESTNET,
+  CHAIN_ID.SCROLL_SEPOLIA,
+  CHAIN_ID.LINEA_SEPOLIA,
+  CHAIN_ID.CELO_SEPOLIA,
+  CHAIN_ID.MANTLE_SEPOLIA,
+  CHAIN_ID.MONAD_TESTNET,
+  CHAIN_ID.ABSTRACT_TESTNET,
+  // Special
+  31337,
+];
+
+const OTHER_CHAIN_ID = -1;
+
 export function resolveChainConfig(chainName: string): ChainConfig {
   const config = CHAINS[chainName];
   if (!config) {
@@ -23,24 +131,73 @@ export function resolveChainConfig(chainName: string): ChainConfig {
   return config;
 }
 
-export async function selectChain(): Promise<string> {
-  return select({
-    message: "Select target chain:",
-    choices: Object.keys(CHAINS).map((name) => ({ name, value: name })),
-  });
+// Resolve chain config by numeric chain ID, supporting BYO chains via rpcUrl.
+// For known chain IDs the viem chain object is used directly.
+// For unknown chain IDs, a minimal chain is constructed using defineChain (requires rpcUrl).
+export function resolveChainConfigById(chainId: number, rpcUrl?: string): ChainConfig {
+  const chain = VIEM_CHAIN_BY_ID[chainId];
+  if (chain) {
+    return { chain, chainId };
+  }
+  if (!rpcUrl) {
+    throw new Error(`Unknown chain ID ${chainId}. Provide --rpc-url to register on a custom chain.`);
+  }
+  return {
+    chain: defineChain({
+      id: chainId,
+      name: `chain-${chainId}`,
+      nativeCurrency: { name: "Ether", symbol: "ETH", decimals: 18 },
+      rpcUrls: { default: { http: [rpcUrl] } },
+    }),
+    chainId,
+  };
+}
+
+// Prompt user to select a chain interactively, returning the numeric chain ID.
+// Chains are sorted by popularity. "Other" allows entering any custom chain ID.
+export async function selectChain(): Promise<number> {
+  const chainById = new Map(Object.values(CHAINS).map((c) => [c.chainId, c]));
+  const nameById = new Map(Object.entries(CHAINS).map(([name, c]) => [c.chainId, name]));
+
+  const choices = CHAIN_SELECTION_ORDER.filter((id) => chainById.has(id)).map((id) => ({
+    name: `${nameById.get(id) ?? `chain-${id}`} (${id})`,
+    value: id,
+  }));
+  choices.push({ name: "Other (enter chain ID)", value: OTHER_CHAIN_ID });
+
+  const selected = await select({ message: "Select target chain:", choices });
+
+  if (selected === OTHER_CHAIN_ID) {
+    const raw = await input({
+      message: "Enter chain ID:",
+      validate: (v) => {
+        const n = parseInt(v, 10);
+        return Number.isInteger(n) && n > 0 ? true : "Must be a positive integer";
+      },
+    });
+    return parseInt(raw, 10);
+  }
+
+  return selected;
 }
 
-export function resolveRegistryAddress(chainName: string, chainId: number, registry?: string): `0x${string}` {
+// Returns the registry address if known, or null if no default exists for the chain (requires interactive prompt).
+// Throws only for an explicitly invalid registry address.
+export function resolveRegistryAddress(chainId: number, registry?: string): `0x${string}` | null {
   if (registry) {
     if (!isAddress(registry)) {
       throw new Error(`Invalid registry address: ${registry}`);
     }
     return registry as `0x${string}`;
   }
-  if (chainName === "localhost") {
-    throw new Error("--registry is required for localhost (no default contract deployment)");
+  if (chainId === 31337) {
+    return null;
+  }
+  try {
+    return getIdentityRegistryAddress(chainId) as `0x${string}`;
+  } catch {
+    return null;
   }
-  return getIdentityRegistryAddress(chainId) as `0x${string}`;
 }
 
 export function validateBrowserRpcConflict(browser: boolean | undefined, rpcUrl: string | undefined): void {

package/register/utils/prompt.ts

@@ -1,4 +1,5 @@
 import { checkbox, confirm, input, select } from "@inquirer/prompts";
+import { isAddress } from "viem";
 import type { RegistrationEntry } from "@aixyz/erc-8004/schemas/registration";
 
 export async function promptAgentUrl(): Promise<string> {
@@ -54,6 +55,14 @@ export async function promptSelectRegistration(registrations: RegistrationEntry[
   });
 }
 
+export async function promptRegistryAddress(): Promise<`0x${string}`> {
+  const value = await input({
+    message: "IdentityRegistry contract address (no default for this chain):",
+    validate: (v) => (isAddress(v) ? true : "Must be a valid Ethereum address (0x…)"),
+  });
+  return value as `0x${string}`;
+}
+
 export function deriveAgentUri(url: string): string {
   // Ensure no trailing slash before appending path
   const base = url.replace(/\/+$/, "");