@agirails/sdk 2.3.1 → 2.4.0

package/src/cli/commands/init.ts

@@ -10,7 +10,6 @@
 import * as crypto from 'crypto';
 import * as fs from 'fs';
 import * as path from 'path';
-import * as readline from 'readline';
 import { Command } from 'commander';
 import {
   saveConfig,
@@ -21,6 +20,7 @@ import {
   CLIMode,
 } from '../utils/config';
 import { Output, ExitCode } from '../utils/output';
+import { generateWallet, computeSmartWalletInit } from '../utils/wallet';
 import { MockStateManager } from '../../runtime/MockStateManager';
 
 // ============================================================================
@@ -40,13 +40,13 @@ export function createInitCommand(): Command {
     .option('--price <usdc>', 'Base price in USDC (default: 1)')
     .option('--json', 'Output as JSON')
     .option('-q, --quiet', 'Minimal output')
-    .action(async (options) => {
+    .action(async (options, command) => {
       const output = new Output(
         options.json ? 'json' : options.quiet ? 'quiet' : 'human'
       );
 
       try {
-        await runInit(options, output);
+        await runInit(options, output, command);
       } catch (error) {
         output.errorResult({
           code: 'INIT_FAILED',
@@ -76,7 +76,7 @@ interface InitOptions {
   price?: string;
 }
 
-async function runInit(options: InitOptions, output: Output): Promise<void> {
+async function runInit(options: InitOptions, output: Output, cmd?: Command): Promise<void> {
   const projectRoot = process.cwd();
 
   // Check if already initialized
@@ -87,6 +87,65 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
     );
   }
 
+  // ── AGIRAILS.md pre-fill ──────────────────────────────────────────────
+  const agirailsMdPath = path.join(projectRoot, 'AGIRAILS.md');
+  let mdConfig: Record<string, unknown> | null = null;
+
+  if (fs.existsSync(agirailsMdPath)) {
+    try {
+      const { parseAgirailsMd } = await import('../../config/agirailsmd');
+      const parsed = parseAgirailsMd(fs.readFileSync(agirailsMdPath, 'utf-8'));
+      mdConfig = parsed.frontmatter;
+    } catch {
+      output.warning('Found AGIRAILS.md but could not parse it — ignoring');
+    }
+  }
+
+  // Helper: true when the user explicitly passed a flag on the CLI
+  const isExplicit = (flag: string): boolean =>
+    cmd?.getOptionValueSource(flag) === 'cli';
+
+  // Apply AGIRAILS.md values where the user didn't set an explicit flag
+  if (mdConfig) {
+    // network → mode mapping
+    if (!isExplicit('mode') && mdConfig.network) {
+      const net = String(mdConfig.network);
+      if (net === 'base-sepolia' || net === 'testnet') options.mode = 'testnet';
+      else if (net === 'base-mainnet' || net === 'mainnet') options.mode = 'mainnet';
+      else if (net === 'mock') options.mode = 'mock';
+    }
+
+    // intent
+    if (!isExplicit('intent') && mdConfig.intent) {
+      options.intent = String(mdConfig.intent);
+    }
+
+    // capabilities → service (first capability)
+    if (!isExplicit('service') && Array.isArray(mdConfig.capabilities) && mdConfig.capabilities.length > 0) {
+      options.service = String(mdConfig.capabilities[0]);
+    }
+
+    // price
+    if (!isExplicit('price') && mdConfig.price != null) {
+      options.price = String(mdConfig.price);
+    }
+
+    // Log what we pre-filled
+    const lines: string[] = [];
+    if (mdConfig.network) lines.push(`  Mode: ${options.mode}`);
+    if (mdConfig.name) lines.push(`  Agent: ${String(mdConfig.name)}`);
+    if (mdConfig.intent) lines.push(`  Intent: ${options.intent || mdConfig.intent}`);
+    if (Array.isArray(mdConfig.capabilities)) lines.push(`  Capabilities: ${mdConfig.capabilities.join(', ')}`);
+    if (mdConfig.price != null) lines.push(`  Price: $${mdConfig.price} USDC`);
+
+    output.info('Found AGIRAILS.md \u2014 using config from file');
+    for (const line of lines) {
+      output.print(line);
+    }
+    output.blank();
+  }
+  // ── End AGIRAILS.md pre-fill ──────────────────────────────────────────
+
   // Validate mode
   const validModes: CLIMode[] = ['mock', 'testnet', 'mainnet'];
   if (!validModes.includes(options.mode as CLIMode)) {
@@ -108,7 +167,6 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
   // Get or generate address
   let address = options.address;
   let smartWalletAddress: string | undefined;
-  let didRegister = false;
   if (!address) {
     if (mode === 'mock') {
       // Generate a random address for mock mode
@@ -123,16 +181,11 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
       if (walletType === 'auto') {
         // Compute Smart Wallet address from signer
         smartWalletAddress = await computeSmartWalletInit(eoaAddress, mode, output);
+        address = smartWalletAddress;
 
-        // Y/N: Register for gas-free transactions?
-        const shouldRegister = await promptRegister(output);
-        if (shouldRegister) {
-          didRegister = await runInlineRegistration(projectRoot, mode, output);
-        }
-
-        // address = Smart Wallet if registered, EOA if not
-        // This ensures CLI commands (balance, tx list) show the correct address
-        address = didRegister ? smartWalletAddress : eoaAddress;
+        output.info('');
+        output.info('Gas-free transactions enabled (activates on first payment)');
+        output.info('Next: run "actp publish" to publish your agent config');
       } else {
         address = eoaAddress;
       }
@@ -154,7 +207,14 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
     version: '1.0',
     ...(walletType !== 'mock' && { wallet: walletType as 'auto' | 'eoa' }),
     ...(smartWalletAddress && { smartWallet: smartWalletAddress.toLowerCase() }),
-    ...(didRegister && { registered: true }),
+    // AGIRAILS.md-derived values (stored for downstream use)
+    ...(mdConfig && mdConfig.name ? { agentName: String(mdConfig.name) } : {}),
+    ...(mdConfig && mdConfig.intent ? { intent: String(mdConfig.intent) as 'earn' | 'pay' | 'both' } : {}),
+    ...(mdConfig && Array.isArray(mdConfig.capabilities) ? { capabilities: mdConfig.capabilities.map(String) } : {}),
+    ...(mdConfig && mdConfig.price != null ? { price: Number(mdConfig.price) } : {}),
+    ...(mdConfig && mdConfig.concurrency != null ? { concurrency: Number(mdConfig.concurrency) } : {}),
+    ...(mdConfig && mdConfig.payment_mode ? { paymentMode: String(mdConfig.payment_mode) as 'actp' | 'x402' | 'both' } : {}),
+    ...(mdConfig && mdConfig.budget != null ? { budget: Number(mdConfig.budget) } : {}),
   };
 
   // Save configuration
@@ -199,18 +259,12 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
 
   // Generate scaffold if requested
   if (options.scaffold) {
-    await runScaffold(options, mode, output);
+    await runScaffold(options, mode, output, mdConfig);
   } else {
     output.blank();
     output.print('Next steps:');
-    if (walletType === 'auto' && didRegister) {
-      // Already registered — ready to go
-      output.print('  1. Create a payment: actp pay <provider> <amount>');
-      output.print('  2. Check your balance: actp balance');
-      output.print('  3. List transactions: actp tx list');
-    } else if (walletType === 'auto') {
-      // Skipped registration — remind them
-      output.print('  1. Register for gas-free: actp register');
+    if (walletType === 'auto') {
+      output.print('  1. Publish config: actp publish');
       output.print('  2. Create a payment: actp pay <provider> <amount>');
       output.print('  3. Check your balance: actp balance');
     } else {
@@ -223,261 +277,6 @@ async function runInit(options: InitOptions, output: Output): Promise<void> {
   }
 }
 
-// ============================================================================
-// Wallet Generation
-// ============================================================================
-
-async function generateWallet(actpDir: string, output: Output): Promise<string> {
-  const { Wallet } = await import('ethers');
-
-  const wallet = Wallet.createRandom();
-
-  // Get password from env var or interactive prompt
-  let password = process.env.ACTP_KEY_PASSWORD;
-  if (!password) {
-    password = await promptPassword();
-  }
-
-  if (!password || password.length < 8) {
-    throw new Error(
-      'Wallet password required (minimum 8 characters).\n' +
-        'Set ACTP_KEY_PASSWORD env var or enter when prompted.'
-    );
-  }
-
-  // Encrypt with Keystore V3 (scrypt + AES-128-CTR)
-  output.info('Encrypting wallet (this takes a few seconds)...');
-  const keystore = await wallet.encrypt(password);
-
-  // Save with restrictive permissions
-  const keystorePath = path.join(actpDir, 'keystore.json');
-  fs.writeFileSync(keystorePath, keystore, { mode: 0o600 });
-
-  output.success('Key securely saved and encrypted');
-  output.info(`Address: ${wallet.address}`);
-  output.warning('Back up your password — it cannot be recovered.');
-  output.info('');
-  output.info('To start your agent:');
-  output.info('  export ACTP_KEY_PASSWORD="your-password"');
-  output.info('  npx ts-node agent.ts');
-
-  return wallet.address;
-}
-
-/**
- * Compute the Smart Wallet address for an EOA signer.
- * Uses CREATE2 counterfactual derivation — no deployment needed.
- */
-async function computeSmartWalletInit(
-  eoaAddress: string,
-  mode: string,
-  output: Output
-): Promise<string> {
-  const { ethers } = await import('ethers');
-  const { getNetwork } = await import('../../config/networks');
-  const { computeSmartWalletAddress } = await import('../../wallet/aa/UserOpBuilder');
-
-  const network = mode === 'testnet' ? 'base-sepolia' : 'base-mainnet';
-  const networkConfig = getNetwork(network);
-  const rpcUrl = networkConfig.rpcUrl;
-  const provider = new ethers.JsonRpcProvider(rpcUrl);
-
-  output.info('Computing Smart Wallet address...');
-  const smartWalletAddress = await computeSmartWalletAddress(eoaAddress, provider);
-
-  output.success(`Smart Wallet: ${smartWalletAddress}`);
-  output.info('Gas-free transactions enabled (requires registration)');
-  output.info('Register with: actp register');
-
-  return smartWalletAddress;
-}
-
-/**
- * Ask user if they want to register for gas-free transactions.
- * Non-TTY (piped/agent) defaults to yes.
- */
-async function promptRegister(output: Output): Promise<boolean> {
-  output.blank();
-  output.print('Register for gas-free transactions? (recommended)');
-  output.print('  Your agent gets a Smart Wallet with sponsored gas — no ETH needed.');
-  output.print('  Requires on-chain registration on AgentRegistry.');
-  output.blank();
-
-  if (!process.stdin.isTTY) {
-    output.info('Non-interactive mode: auto-registering');
-    return true;
-  }
-
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-  });
-
-  return new Promise((resolve) => {
-    rl.question('  Register now? [Y/n] ', (answer) => {
-      rl.close();
-      const trimmed = answer.trim().toLowerCase();
-      resolve(trimmed === '' || trimmed === 'y' || trimmed === 'yes');
-    });
-  });
-}
-
-/**
- * Run inline registration during init.
- * Reuses the same logic as `actp register` — parses AGIRAILS.md,
- * builds gasless UserOp (testnet: register + mint 1000 USDC).
- *
- * Returns true if registration succeeded, false on failure (non-fatal).
- */
-async function runInlineRegistration(
-  projectRoot: string,
-  mode: string,
-  output: Output
-): Promise<boolean> {
-  try {
-    const { resolvePrivateKey } = await import('../../wallet/keystore');
-    const privateKey = await resolvePrivateKey(projectRoot);
-    if (!privateKey) {
-      output.warning('Could not load wallet key. Run "actp register" later.');
-      return false;
-    }
-
-    const { parseAgirailsMd } = await import('../../config/agirailsmd');
-    const { extractRegistrationParams } = await import('../../config/publishPipeline');
-    const { ethers } = await import('ethers');
-    const { getNetwork } = await import('../../config/networks');
-    const { AutoWalletProvider } = await import('../../wallet/AutoWalletProvider');
-    const { buildRegisterAgentBatch, buildTestnetInitBatch } = await import('../../wallet/aa/TransactionBatcher');
-
-    // Parse AGIRAILS.md if present
-    const agirailsMdPath = path.join(projectRoot, 'AGIRAILS.md');
-    let endpoint = '';
-    let serviceDescriptors;
-
-    if (fs.existsSync(agirailsMdPath)) {
-      const content = fs.readFileSync(agirailsMdPath, 'utf-8');
-      const parsed = parseAgirailsMd(content);
-      const regParams = extractRegistrationParams(parsed.frontmatter);
-      endpoint = regParams.endpoint;
-      serviceDescriptors = regParams.serviceDescriptors;
-      output.info(`Parsed ${serviceDescriptors.length} service(s) from AGIRAILS.md`);
-    } else {
-      const serviceType = 'general';
-      serviceDescriptors = [{
-        serviceTypeHash: ethers.keccak256(ethers.toUtf8Bytes(serviceType)),
-        serviceType,
-        schemaURI: '',
-        minPrice: 0n,
-        maxPrice: 1_000_000_000n,
-        avgCompletionTime: 3600,
-        metadataCID: '',
-      }];
-      output.info('No AGIRAILS.md found. Using default "general" service.');
-    }
-
-    const network = mode === 'testnet' ? 'base-sepolia' : 'base-mainnet';
-    const networkConfig = getNetwork(network);
-
-    if (!networkConfig.aa || !networkConfig.contracts.agentRegistry) {
-      output.warning('AA or AgentRegistry not configured. Run "actp register" later.');
-      return false;
-    }
-
-    // Check for valid bundler/paymaster URL (CDP_API_KEY must be set)
-    const cdpUrl = networkConfig.aa.bundlerUrls.coinbase;
-    const hasPimlico = !!networkConfig.aa.bundlerUrls.pimlico;
-    if (cdpUrl.endsWith('/') && !hasPimlico) {
-      output.warning('CDP_API_KEY not set. Skipping registration.');
-      output.info('Set CDP_API_KEY and run "actp register" later.');
-      return false;
-    }
-
-    const provider = new ethers.JsonRpcProvider(networkConfig.rpcUrl);
-    const signer = new ethers.Wallet(privateKey, provider);
-
-    const autoWallet = await AutoWalletProvider.create({
-      signer,
-      provider,
-      chainId: networkConfig.chainId,
-      actpKernelAddress: networkConfig.contracts.actpKernel,
-      bundler: {
-        primaryUrl: networkConfig.aa.bundlerUrls.coinbase,
-        backupUrl: networkConfig.aa.bundlerUrls.pimlico,
-      },
-      paymaster: {
-        primaryUrl: networkConfig.aa.paymasterUrls.coinbase,
-        backupUrl: networkConfig.aa.paymasterUrls.pimlico,
-      },
-    });
-
-    const smartWalletAddress = autoWallet.getAddress();
-
-    // Build batch
-    let calls;
-    if (mode === 'testnet') {
-      output.info('Testnet: registering + minting 1000 test USDC in one gasless tx...');
-      calls = buildTestnetInitBatch({
-        agentRegistryAddress: networkConfig.contracts.agentRegistry,
-        endpoint,
-        serviceDescriptors,
-        mockUsdcAddress: networkConfig.contracts.usdc,
-        recipient: smartWalletAddress,
-        mintAmount: '1000000000',
-      });
-    } else {
-      output.info('Registering on AgentRegistry (gasless)...');
-      calls = buildRegisterAgentBatch(
-        networkConfig.contracts.agentRegistry,
-        endpoint,
-        serviceDescriptors
-      );
-    }
-
-    const txRequests = calls.map((c) => ({
-      to: c.target,
-      data: c.data,
-      value: c.value.toString(),
-    }));
-
-    const receipt = await autoWallet.sendBatchTransaction(txRequests);
-
-    if (!receipt.success) {
-      output.warning(`Registration failed (tx: ${receipt.hash}). Run "actp register" later.`);
-      return false;
-    }
-
-    output.success('Agent registered on AgentRegistry');
-    if (mode === 'testnet') {
-      output.success('Minted 1,000 test USDC to Smart Wallet');
-    }
-    output.print(`  Tx: ${receipt.hash}`);
-    return true;
-  } catch (error) {
-    output.warning(`Registration failed: ${(error as Error).message}`);
-    output.info('You can register later with: actp register');
-    return false;
-  }
-}
-
-async function promptPassword(): Promise<string> {
-  // If not a TTY (e.g. piped or run by agent), skip prompt
-  if (!process.stdin.isTTY) {
-    return '';
-  }
-
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-  });
-
-  return new Promise((resolve) => {
-    rl.question('Enter password for wallet encryption (min 8 chars): ', (answer) => {
-      rl.close();
-      resolve(answer.trim());
-    });
-  });
-}
-
 // ============================================================================
 // Scaffold
 // ============================================================================
@@ -486,6 +285,7 @@ async function runScaffold(
   options: InitOptions,
   mode: CLIMode,
   output: Output,
+  mdConfig?: Record<string, unknown> | null,
 ): Promise<void> {
   const validIntents: ScaffoldIntent[] = ['earn', 'pay', 'both'];
   const intent: ScaffoldIntent = (options.intent as ScaffoldIntent) || 'earn';
@@ -506,8 +306,8 @@ async function runScaffold(
     return;
   }
 
-  // Derive agent name from directory
-  const agentName = path.basename(process.cwd());
+  // Derive agent name: prefer AGIRAILS.md name, fallback to directory name
+  const agentName = (mdConfig?.name ? String(mdConfig.name) : null) || path.basename(process.cwd());
 
   // Get template and substitute variables
   const template = getTemplate(intent);