hakorune/src/cli.rs

/*!
 * CLI Argument Parsing Module - Nyash Command Line Interface
 * 
 * This module handles all command-line argument parsing using clap,
 * separating CLI concerns from the main execution logic.
 */

use clap::{Arg, Command, ArgMatches};

/// Command-line configuration structure
#[derive(Debug, Clone)]
pub struct CliConfig {
    pub file: Option<String>,
    pub debug_fuel: Option<usize>,
    pub dump_ast: bool,
    pub dump_mir: bool,
    pub verify_mir: bool,
    pub mir_verbose: bool,
    pub mir_verbose_effects: bool,
    pub no_optimize: bool,
    pub backend: String,
    pub compile_wasm: bool,
    pub compile_native: bool,
    pub output_file: Option<String>,
    pub benchmark: bool,
    pub iterations: u32,
    pub vm_stats: bool,
    pub vm_stats_json: bool,
    // JIT controls
    pub jit_exec: bool,
    pub jit_stats: bool,
    pub jit_stats_json: bool,
    pub jit_dump: bool,
    pub jit_threshold: Option<u32>,
    pub jit_phi_min: bool,
    pub jit_hostcall: bool,
    pub jit_handle_debug: bool,
    pub jit_native_f64: bool,
    pub jit_native_bool: bool,
    pub jit_only: bool,
    pub jit_direct: bool,
    // DOT emit helper
    pub emit_cfg: Option<String>,
}

impl CliConfig {
    /// Parse command-line arguments and return configuration
    pub fn parse() -> Self {
        let matches = Self::build_command().get_matches();
        Self::from_matches(&matches)
    }

    /// Build the clap Command structure
    fn build_command() -> Command {
        Command::new("nyash")
            .version("1.0")
            .author("Claude Code <claude@anthropic.com>")
            .about("🦀 Nyash Programming Language - Everything is Box in Rust! 🦀")
            .arg(
                Arg::new("file")
                    .help("Nyash file to execute")
                    .value_name("FILE")
                    .index(1)
            )
            .arg(
                Arg::new("debug-fuel")
                    .long("debug-fuel")
                    .value_name("ITERATIONS")
                    .help("Set parser debug fuel limit (default: 100000, 'unlimited' for no limit)")
                    .default_value("100000")
            )
            .arg(
                Arg::new("dump-ast")
                    .long("dump-ast")
                    .help("Dump parsed AST and exit")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("dump-mir")
                    .long("dump-mir")
                    .help("Dump MIR (Mid-level Intermediate Representation) instead of executing")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("verify")
                    .long("verify")
                    .help("Verify MIR integrity and exit")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("mir-verbose")
                    .long("mir-verbose")
                    .help("Show verbose MIR output with statistics")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("mir-verbose-effects")
                    .long("mir-verbose-effects")
                    .help("Show per-instruction effect category (pure/readonly/side)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("no-optimize")
                    .long("no-optimize")
                    .help("Disable MIR optimizer passes (dump raw Builder MIR)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("backend")
                    .long("backend")
                    .value_name("BACKEND")
                    .help("Choose execution backend: 'interpreter' (default), 'vm', or 'llvm'")
                    .default_value("interpreter")
            )
            .arg(
                Arg::new("compile-wasm")
                    .long("compile-wasm")
                    .help("Compile to WebAssembly (WAT format) instead of executing")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("compile-native")
                    .long("compile-native")
                    .help("Compile to native AOT executable using wasmtime precompilation")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("aot")
                    .long("aot")
                    .help("Short form of --compile-native")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("output")
                    .long("output")
                    .short('o')
                    .value_name("FILE")
                    .help("Output file (for WASM compilation or AOT executable)")
            )
            .arg(
                Arg::new("benchmark")
                    .long("benchmark")
                    .help("Run performance benchmarks across all backends")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("iterations")
                    .long("iterations")
                    .value_name("COUNT")
                    .help("Number of iterations for benchmarks (default: 10)")
                    .default_value("10")
            )
            .arg(
                Arg::new("vm-stats")
                    .long("vm-stats")
                    .help("Enable VM instruction statistics (equivalent to NYASH_VM_STATS=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("vm-stats-json")
                    .long("vm-stats-json")
                    .help("Output VM statistics in JSON format")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-exec")
                    .long("jit-exec")
                    .help("Enable JIT execution where available (NYASH_JIT_EXEC=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-stats")
                    .long("jit-stats")
                    .help("Print JIT compilation/execution statistics (NYASH_JIT_STATS=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-stats-json")
                    .long("jit-stats-json")
                    .help("Output JIT statistics in JSON format (NYASH_JIT_STATS_JSON=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-dump")
                    .long("jit-dump")
                    .help("Dump JIT lowering summary (NYASH_JIT_DUMP=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-threshold")
                    .long("jit-threshold")
                    .value_name("N")
                    .help("Set hotness threshold for JIT compilation (NYASH_JIT_THRESHOLD)")
            )
            .arg(
                Arg::new("jit-phi-min")
                    .long("jit-phi-min")
                    .help("Enable minimal PHI path for branches (NYASH_JIT_PHI_MIN=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-hostcall")
                    .long("jit-hostcall")
                    .help("Enable JIT hostcall bridge for Array/Map (NYASH_JIT_HOSTCALL=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-handle-debug")
                    .long("jit-handle-debug")
                    .help("Print JIT handle allocation debug logs (NYASH_JIT_HANDLE_DEBUG=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-native-f64")
                    .long("jit-native-f64")
                    .help("Enable native f64 ABI path in JIT (NYASH_JIT_NATIVE_F64=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-native-bool")
                    .long("jit-native-bool")
                    .help("Enable native bool ABI path in JIT (NYASH_JIT_NATIVE_BOOL=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-only")
                    .long("jit-only")
                    .help("Run JIT only (no VM fallback). Fails if JIT is unavailable (NYASH_JIT_ONLY=1)")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("jit-direct")
                    .long("jit-direct")
                    .help("Run program via independent JIT engine (no VM interpreter/executor). Requires --features cranelift-jit")
                    .action(clap::ArgAction::SetTrue)
            )
            .arg(
                Arg::new("emit-cfg")
                    .long("emit-cfg")
                    .value_name("DOT_FILE")
                    .help("Emit JIT CFG as DOT to file (equivalent to setting NYASH_JIT_DOT)")
            )
    }

    /// Convert ArgMatches to CliConfig
    fn from_matches(matches: &ArgMatches) -> Self {
        Self {
            file: matches.get_one::<String>("file").cloned(),
            debug_fuel: parse_debug_fuel(matches.get_one::<String>("debug-fuel").unwrap()),
            dump_ast: matches.get_flag("dump-ast"),
            dump_mir: matches.get_flag("dump-mir"),
            verify_mir: matches.get_flag("verify"),
            mir_verbose: matches.get_flag("mir-verbose"),
            mir_verbose_effects: matches.get_flag("mir-verbose-effects"),
            no_optimize: matches.get_flag("no-optimize"),
            backend: matches.get_one::<String>("backend").unwrap().clone(),
            compile_wasm: matches.get_flag("compile-wasm"),
            compile_native: matches.get_flag("compile-native") || matches.get_flag("aot"),
            output_file: matches.get_one::<String>("output").cloned(),
            benchmark: matches.get_flag("benchmark"),
            iterations: matches.get_one::<String>("iterations").unwrap().parse().unwrap_or(10),
            vm_stats: matches.get_flag("vm-stats"),
            vm_stats_json: matches.get_flag("vm-stats-json"),
            jit_exec: matches.get_flag("jit-exec"),
            jit_stats: matches.get_flag("jit-stats"),
            jit_stats_json: matches.get_flag("jit-stats-json"),
            jit_dump: matches.get_flag("jit-dump"),
            jit_threshold: matches.get_one::<String>("jit-threshold").and_then(|s| s.parse::<u32>().ok()),
            jit_phi_min: matches.get_flag("jit-phi-min"),
            jit_hostcall: matches.get_flag("jit-hostcall"),
            jit_handle_debug: matches.get_flag("jit-handle-debug"),
            jit_native_f64: matches.get_flag("jit-native-f64"),
            jit_native_bool: matches.get_flag("jit-native-bool"),
            emit_cfg: matches.get_one::<String>("emit-cfg").cloned(),
            jit_only: matches.get_flag("jit-only"),
            jit_direct: matches.get_flag("jit-direct"),
        }
    }
}

/// Parse debug fuel value ("unlimited" or numeric)
fn parse_debug_fuel(value: &str) -> Option<usize> {
    if value == "unlimited" {
        None  // No limit
    } else {
        value.parse::<usize>().ok()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_parse_debug_fuel() {
        assert_eq!(parse_debug_fuel("unlimited"), None);
        assert_eq!(parse_debug_fuel("1000"), Some(1000));
        assert_eq!(parse_debug_fuel("invalid"), None);
    }

    #[test]
    fn test_default_config() {
        // This test would require mocking clap's behavior
        // For now, we just ensure the structure is valid
        let config = CliConfig {
            file: None,
            debug_fuel: Some(100000),
            dump_ast: false,
            dump_mir: false,
            verify_mir: false,
            mir_verbose: false,
            mir_verbose_effects: false,
            no_optimize: false,
            backend: "interpreter".to_string(),
            compile_wasm: false,
            compile_native: false,
            output_file: None,
            benchmark: false,
            iterations: 10,
            vm_stats: false,
            vm_stats_json: false,
            jit_exec: false,
            jit_stats: false,
            jit_stats_json: false,
            jit_dump: false,
            jit_threshold: None,
            jit_phi_min: false,
            jit_hostcall: false,
            jit_handle_debug: false,
            jit_native_f64: false,
            jit_native_bool: false,
        };
        
        assert_eq!(config.backend, "interpreter");
        assert_eq!(config.iterations, 10);
    }
}