aprender-ptx-debug 0.36.0

//! trueno-ptx-debug CLI
//!
//! Pure Rust PTX debugging and static analysis tool.
//!
//! Usage:
//!   trueno-ptx-debug analyze <file.ptx> [--falsify] [--min-score N]
//!   trueno-ptx-debug gen-fkr <file.ptx> [-o tests.rs]

use std::env;
use std::fs;
use std::process;

use trueno_ptx_debug::bugs::BugRegistry;
use trueno_ptx_debug::falsification::FalsificationRegistry;
use trueno_ptx_debug::output::{generate_fkr_tests, generate_html_report, AnalysisResult};
use trueno_ptx_debug::parser::Parser;

fn main() {
    let args: Vec<String> = env::args().collect();

    if args.len() < 2 {
        print_usage();
        process::exit(1);
    }

    let result = match args[1].as_str() {
        "analyze" => cmd_analyze(&args[2..]),
        "gen-fkr" => cmd_gen_fkr(&args[2..]),
        "help" | "--help" | "-h" => {
            print_usage();
            Ok(())
        }
        "version" | "--version" | "-V" => {
            println!("trueno-ptx-debug {}", env!("CARGO_PKG_VERSION"));
            Ok(())
        }
        _ => {
            eprintln!("Unknown command: {}", args[1]);
            print_usage();
            process::exit(1);
        }
    };

    if let Err(e) = result {
        eprintln!("Error: {}", e);
        process::exit(1);
    }
}

fn print_usage() {
    println!(
        r"trueno-ptx-debug - Pure Rust PTX debugging and static analysis tool

USAGE:
    trueno-ptx-debug <COMMAND> [OPTIONS]

COMMANDS:
    analyze <file.ptx>    Analyze PTX file for bugs and issues
        --falsify         Run full 100-point falsification framework
        --min-score N     Fail if score < N (default: 70)
        --html <file>     Write HTML report to file
        --json            Output JSON format

    gen-fkr <file.ptx>    Generate FKR tests for jugar-probar
        -o <file.rs>      Output file (default: stdout)

    help                  Show this help message
    version               Show version information

EXIT CODES:
    0 - Analysis passed (score >= 90)
    1 - Analysis passed with warnings (score 70-89)
    2 - Analysis failed (score < 70)
    3 - Critical bugs detected
    10 - Parse error
    11 - I/O error

EXAMPLES:
    trueno-ptx-debug analyze kernel.ptx --falsify
    trueno-ptx-debug analyze kernel.ptx --min-score 90 --html report.html
    trueno-ptx-debug gen-fkr kernel.ptx -o tests/kernel_fkr.rs
"
    );
}

/// Parsed arguments for the `analyze` subcommand.
struct AnalyzeArgs {
    file_path: String,
    #[allow(dead_code)]
    run_falsify: bool,
    min_score: f64,
    html_output: Option<String>,
    json_output: bool,
}

/// Consume the next positional argument from `args[i+1]`, returning an error
/// if the slice is exhausted.  Returns the new index and the consumed value.
fn consume_valued_option(args: &[String], i: usize, flag: &str) -> Result<(usize, String), String> {
    let next = i + 1;
    if next >= args.len() {
        return Err(format!("{} requires a value", flag));
    }
    Ok((next, args[next].clone()))
}

/// Apply a single CLI token to the in-progress `AnalyzeArgs` builder.
/// Returns the (possibly advanced) index after consuming the token.
fn apply_analyze_flag(
    args: &[String],
    i: usize,
    run_falsify: &mut bool,
    min_score: &mut f64,
    html_output: &mut Option<String>,
    json_output: &mut bool,
    file_path: &mut Option<String>,
) -> Result<usize, String> {
    match args[i].as_str() {
        "--falsify" => {
            *run_falsify = true;
            Ok(i)
        }
        "--json" => {
            *json_output = true;
            Ok(i)
        }
        "--min-score" => {
            let (next, val) = consume_valued_option(args, i, "--min-score")?;
            *min_score = val
                .parse()
                .map_err(|_| "Invalid min-score value".to_string())?;
            Ok(next)
        }
        "--html" => {
            let (next, val) = consume_valued_option(args, i, "--html")?;
            *html_output = Some(val);
            Ok(next)
        }
        arg if !arg.starts_with('-') => {
            *file_path = Some(arg.to_string());
            Ok(i)
        }
        arg => Err(format!("Unknown option: {}", arg)),
    }
}

/// Parse the CLI arguments for the `analyze` subcommand.
fn parse_analyze_args(args: &[String]) -> Result<AnalyzeArgs, String> {
    if args.is_empty() {
        return Err("Missing PTX file argument".into());
    }

    let mut file_path = None;
    let mut run_falsify = false;
    let mut min_score = 70.0;
    let mut html_output = None;
    let mut json_output = false;

    let mut i = 0;
    while i < args.len() {
        i = apply_analyze_flag(
            args,
            i,
            &mut run_falsify,
            &mut min_score,
            &mut html_output,
            &mut json_output,
            &mut file_path,
        )?;
        i += 1;
    }

    Ok(AnalyzeArgs {
        file_path: file_path.ok_or("Missing PTX file argument")?,
        run_falsify,
        min_score,
        html_output,
        json_output,
    })
}

/// Print analysis results as JSON.
fn print_json_report(
    result: &AnalysisResult,
    report: &trueno_ptx_debug::falsification::FalsificationReport,
) {
    println!("{{");
    println!("  \"module\": \"{}\",", result.module_name);
    println!("  \"score\": {:.1},", result.falsification_score);
    println!("  \"confidence\": {:.2},", result.confidence);
    println!("  \"earned_points\": {},", report.earned_points);
    println!("  \"total_points\": {},", report.total_points);
    println!(
        "  \"critical_bugs_absent\": {}",
        report.critical_bugs_absent()
    );
    println!("}}");
}

/// Print analysis results as human-readable text.
fn print_text_report(
    result: &AnalysisResult,
    report: &trueno_ptx_debug::falsification::FalsificationReport,
) {
    println!("PTX Analysis Report: {}", result.module_name);
    println!("=========================================");
    println!("Score: {:.1}/100", result.falsification_score);
    println!("Confidence: {:.1}%", result.confidence * 100.0);
    println!("Points: {}/{}", report.earned_points, report.total_points);
    println!();

    let failed = report.failed_tests();
    if failed.is_empty() {
        println!("All tests passed!");
    } else {
        println!("Failed tests ({}):", failed.len());
        for (id, category, desc, _result) in failed {
            println!("  {} [{}]: {}", id, category, desc);
        }
    }
}

/// Determine the process exit code from the analysis results.
fn exit_for_score(
    report: &trueno_ptx_debug::falsification::FalsificationReport,
    score: f64,
    min_score: f64,
) {
    if report.has_critical_bugs() {
        process::exit(3);
    } else if score < min_score {
        process::exit(2);
    } else if score < 90.0 {
        process::exit(1);
    }
}

fn cmd_analyze(args: &[String]) -> Result<(), String> {
    let opts = parse_analyze_args(args)?;
    let result = analyze_ptx_file(&opts.file_path)?;

    // Output results
    if opts.json_output {
        print_json_report(&result, &result.falsification_report);
    } else {
        print_text_report(&result, &result.falsification_report);
    }

    // Write HTML report if requested
    if let Some(html_path) = opts.html_output {
        let html = generate_html_report(&result);
        fs::write(&html_path, html).map_err(|e| format!("Failed to write {}: {}", html_path, e))?;
        println!("\nHTML report written to: {}", html_path);
    }

    exit_for_score(
        &result.falsification_report,
        result.falsification_score,
        opts.min_score,
    );

    Ok(())
}

/// Parsed arguments for the `gen-fkr` subcommand.
struct GenFkrArgs {
    file_path: String,
    output_file: Option<String>,
}

/// Apply a single CLI token to the in-progress `GenFkrArgs` builder.
/// Returns the (possibly advanced) index after consuming the token.
fn apply_gen_fkr_flag(
    args: &[String],
    i: usize,
    output_file: &mut Option<String>,
    file_path: &mut Option<String>,
) -> Result<usize, String> {
    match args[i].as_str() {
        "-o" => {
            let (next, val) = consume_valued_option(args, i, "-o")?;
            *output_file = Some(val);
            Ok(next)
        }
        arg if !arg.starts_with('-') => {
            *file_path = Some(arg.to_string());
            Ok(i)
        }
        arg => Err(format!("Unknown option: {}", arg)),
    }
}

/// Parse the CLI arguments for the `gen-fkr` subcommand.
fn parse_gen_fkr_args(args: &[String]) -> Result<GenFkrArgs, String> {
    if args.is_empty() {
        return Err("Missing PTX file argument".into());
    }

    let mut file_path = None;
    let mut output_file = None;

    let mut i = 0;
    while i < args.len() {
        i = apply_gen_fkr_flag(args, i, &mut output_file, &mut file_path)?;
        i += 1;
    }

    Ok(GenFkrArgs {
        file_path: file_path.ok_or("Missing PTX file argument")?,
        output_file,
    })
}

/// Read a PTX file, parse it, run analysis, and return the result.
fn analyze_ptx_file(file_path: &str) -> Result<AnalysisResult, String> {
    let ptx_source = fs::read_to_string(file_path)
        .map_err(|e| format!("Failed to read {}: {}", file_path, e))?;

    let mut parser = Parser::new(&ptx_source).map_err(|e| format!("Parse error: {}", e))?;
    let module = parser.parse().map_err(|e| format!("Parse error: {}", e))?;

    let module_name = std::path::Path::new(file_path)
        .file_stem()
        .and_then(|s| s.to_str())
        .unwrap_or("unknown")
        .to_string();

    let registry = FalsificationRegistry::new();
    let report = registry.evaluate(&module);
    let bugs = BugRegistry::new();
    Ok(AnalysisResult::new(&module_name, report, bugs))
}

/// Write generated content to a file, or print to stdout if no path is given.
fn write_or_print(content: &str, output_path: Option<String>, label: &str) -> Result<(), String> {
    match output_path {
        Some(path) => {
            fs::write(&path, content).map_err(|e| format!("Failed to write {}: {}", path, e))?;
            println!("{} written to: {}", label, path);
        }
        None => println!("{}", content),
    }
    Ok(())
}

fn cmd_gen_fkr(args: &[String]) -> Result<(), String> {
    let opts = parse_gen_fkr_args(args)?;
    let result = analyze_ptx_file(&opts.file_path)?;
    let fkr_tests = generate_fkr_tests(&result);
    write_or_print(&fkr_tests, opts.output_file, "FKR tests")
}