foxguard 0.8.0

use crate::compliance::{MigrationLevel, MigrationReport};
use crate::{Finding, Severity};
use colored::Colorize;
use std::path::Path;

const MAX_SNIPPET_WIDTH: usize = 120;

pub fn print_banner() {
    // Print on same line so we can overwrite with clear_banner
    eprint!(
        "\r  {} {} {} ",
        "foxguard".truecolor(245, 158, 11).bold(),
        format!("v{}", env!("CARGO_PKG_VERSION")).dimmed(),
        "\u{00b7} scanning...".dimmed(),
    );
}

pub fn clear_banner() {
    // Clear the scanning line and move cursor back
    eprint!("\r{}\r", " ".repeat(60));
}

/// Shorten a file path for display. If the path is absolute and starts with
/// the current working directory, strip the prefix to show a relative path.
fn display_path(path: &str) -> String {
    if let Ok(cwd) = std::env::current_dir() {
        if let Ok(rel) = Path::new(path).strip_prefix(&cwd) {
            return rel.display().to_string();
        }
    }
    // If it's still long, show just the last 3 components
    let parts: Vec<&str> = path.split('/').collect();
    if parts.len() > 4 {
        format!(".../{}", parts[parts.len() - 3..].join("/"))
    } else {
        path.to_string()
    }
}

pub fn print_findings(findings: &[Finding], files_scanned: usize, duration: std::time::Duration) {
    print_findings_with_options(findings, files_scanned, duration, false);
}

pub fn print_findings_with_options(
    findings: &[Finding],
    files_scanned: usize,
    duration: std::time::Duration,
    explain: bool,
) {
    print_findings_with_options_and_confidence(findings, files_scanned, duration, explain, false);
}

/// All the knobs the scan pipeline needs to control terminal rendering.
///
/// Introduced when the `--cnsa2` flag was added (issue #241) to keep
/// `print_findings_*` callsites readable rather than growing yet another
/// positional-bool variant. Existing public entry points remain for
/// back-compat with callers that don't care about CNSA annotations.
#[derive(Debug, Clone, Copy, Default)]
pub struct ReportOptions {
    pub explain: bool,
    pub show_confidence: bool,
    pub cnsa2: bool,
}

/// Variant of [`print_findings_with_options`] that also controls whether
/// per-finding confidence scores are displayed. Terminal output keeps
/// them hidden by default because they're too noisy; callers that want
/// them pass `show_confidence: true` (typically from the
/// `--show-confidence` CLI flag).
pub fn print_findings_with_options_and_confidence(
    findings: &[Finding],
    files_scanned: usize,
    duration: std::time::Duration,
    explain: bool,
    show_confidence: bool,
) {
    print_findings_full(
        findings,
        files_scanned,
        duration,
        ReportOptions {
            explain,
            show_confidence,
            cnsa2: false,
        },
    );
}

/// New preferred entry point. Accepts a [`ReportOptions`] bundle so call
/// sites can opt into CNSA 2.0 rendering without touching the other
/// overloads.
pub fn print_findings_full(
    findings: &[Finding],
    files_scanned: usize,
    duration: std::time::Duration,
    opts: ReportOptions,
) {
    let ReportOptions {
        explain,
        show_confidence,
        cnsa2,
    } = opts;
    if findings.is_empty() {
        if files_scanned > 0 {
            println!(
                "\n  {} Scanned {} files in {:.2}s.\n",
                "\u{2714}".green(),
                files_scanned,
                duration.as_secs_f64(),
            );
        } else {
            println!("\n  {} No security issues found.\n", "\u{2714}".green());
        }
        return;
    }

    // Group by file
    let mut by_file: Vec<(&str, Vec<&Finding>)> = Vec::new();
    let mut current_file = "";
    for f in findings {
        if f.file != current_file {
            current_file = &f.file;
            by_file.push((&f.file, Vec::new()));
        }
        by_file.last_mut().expect("just pushed").1.push(f);
    }

    println!();

    for (file, file_findings) in &by_file {
        let count = file_findings.len();
        let label = if count == 1 { "issue" } else { "issues" };
        let short_path = display_path(file);
        println!(
            "  {} {} {}",
            short_path.bold(),
            "\u{00b7}".dimmed(),
            format!("{count} {label}").dimmed(),
        );
        println!();

        for f in file_findings {
            print_finding(f, explain, show_confidence, cnsa2);
        }
        println!();
    }

    print_summary(findings, files_scanned, duration);

    if cnsa2 {
        print_cnsa2_summary(findings);
    }
}

fn severity_badge(severity: Severity) -> colored::ColoredString {
    match severity {
        Severity::Critical => " CRITICAL ".on_truecolor(130, 50, 180).white().bold(),
        Severity::High => " HIGH ".on_red().white().bold(),
        Severity::Medium => " MEDIUM ".on_yellow().black().bold(),
        Severity::Low => " LOW ".on_blue().white().bold(),
    }
}

fn severity_accent(severity: Severity) -> colored::ColoredString {
    match severity {
        Severity::Critical => "\u{2588}".truecolor(130, 50, 180),
        Severity::High => "\u{2588}".red(),
        Severity::Medium => "\u{2588}".yellow(),
        Severity::Low => "\u{2588}".blue(),
    }
}

fn tag_badges(tags: &[String]) -> String {
    if tags.is_empty() {
        return String::new();
    }
    let mut out = String::new();
    for tag in tags {
        out.push_str(&format!(
            " {}",
            format!(" {} ", tag).on_cyan().black().bold()
        ));
    }
    out
}

fn truncate_snippet(line: &str) -> String {
    let trimmed = line.trim();
    if trimmed.len() <= MAX_SNIPPET_WIDTH {
        return trimmed.to_string();
    }
    let mut end = MAX_SNIPPET_WIDTH;
    while end > 0 && !trimmed.is_char_boundary(end) {
        end -= 1;
    }
    format!("{}...", &trimmed[..end])
}

fn print_finding(f: &Finding, explain: bool, show_confidence: bool, cnsa2: bool) {
    let accent = severity_accent(f.severity);
    let badge = severity_badge(f.severity);
    let cwe = f
        .cwe
        .as_ref()
        .map(|c| format!(" ({c})"))
        .unwrap_or_default();

    // Line 1: badge + tags + description (the main thing you read)
    let tag_str = tag_badges(&f.tags);
    println!("    {accent} {badge}{tag_str} {}", f.description);

    // Line 2: rule ID + CWE + location (secondary info, dimmed)
    // Confidence is intentionally off by default — it's noisy for most
    // users. `--show-confidence` opts in.
    let conf_suffix = if show_confidence {
        format!("  conf {:.2}", f.confidence.clamp(0.0, 1.0))
    } else {
        String::new()
    };
    println!(
        "    {accent}   {}{}  {}{}",
        f.rule_id.cyan().dimmed(),
        cwe.dimmed(),
        format!("line {}:{}", f.line, f.column).dimmed(),
        conf_suffix.dimmed(),
    );

    // Code snippet (recessed further)
    if !f.snippet.is_empty() {
        for line in f.snippet.lines() {
            let truncated = truncate_snippet(line);
            println!("    {accent}   {}", truncated.dimmed());
        }
    }

    // Source/sink trace
    if explain {
        if let (Some(src_line), Some(src_desc)) = (f.source_line, f.source_description.as_ref()) {
            let src_path = display_path(&f.file);
            println!(
                "    {accent} {} {}:{}  {}",
                "source \u{2192}".yellow(),
                src_path.dimmed(),
                src_line.to_string().dimmed(),
                src_desc,
            );
        }
        if let (Some(snk_line), Some(snk_desc)) = (f.sink_line, f.sink_description.as_ref()) {
            let snk_path = display_path(&f.file);
            println!(
                "    {accent} {} {}:{}  {}",
                "sink   \u{2192}".red(),
                snk_path.dimmed(),
                snk_line.to_string().dimmed(),
                snk_desc,
            );
        }
    }

    // Fix suggestion
    if let Some(fix) = f.fix_suggestion.as_ref() {
        println!("    {accent} {} {}", "Fix:".green().bold(), fix);
    }

    // CNSA 2.0 deadline annotation. Only rendered when `--cnsa2` is on,
    // so the default terminal view stays unchanged for users who don't
    // care about NSS compliance. When on, the field is always populated
    // for any PQ-vulnerable finding; None means the rule is simply not
    // CNSA-relevant.
    if cnsa2 {
        if let Some(deadline) = &f.cnsa2_deadline {
            println!(
                "    {accent} {} {}",
                "CNSA 2.0:".truecolor(245, 158, 11).bold(),
                format!("migrate before end of {deadline}").dimmed(),
            );
        }
    }

    // Blank line between findings
    println!();
}

/// CNSA 2.0 migration-readiness block. Appears only when the caller set
/// `ReportOptions::cnsa2 = true`. Reads each finding's pre-annotated
/// `cnsa2_deadline` rather than recomputing, so the block matches what
/// SARIF emits exactly.
fn print_cnsa2_summary(findings: &[Finding]) {
    let report = MigrationReport::from_findings(findings);
    let level_label = match report.level {
        MigrationLevel::Clean => " clean ".on_green().black().bold(),
        MigrationLevel::OnTrack => " on-track ".on_blue().white().bold(),
        MigrationLevel::AtRisk => " at-risk ".on_red().white().bold(),
    };

    println!(
        "  {} {}  {}",
        "CNSA 2.0".truecolor(245, 158, 11).bold(),
        level_label,
        format!(
            "{} finding{} with NSA transition deadlines",
            report.annotated,
            if report.annotated == 1 { "" } else { "s" }
        )
        .dimmed(),
    );

    if !report.by_deadline.is_empty() {
        let mut entries: Vec<(&String, &usize)> = report.by_deadline.iter().collect();
        entries.sort_by(|a, b| a.0.cmp(b.0));
        let bullets: Vec<String> = entries
            .iter()
            .map(|(year, count)| format!("{} by {}", count, year))
            .collect();
        println!("  {}", bullets.join("  \u{00b7}  ").dimmed());
    }
    println!();
}

fn print_summary(findings: &[Finding], files_scanned: usize, duration: std::time::Duration) {
    let critical = findings
        .iter()
        .filter(|f| f.severity == Severity::Critical)
        .count();
    let high = findings
        .iter()
        .filter(|f| f.severity == Severity::High)
        .count();
    let medium = findings
        .iter()
        .filter(|f| f.severity == Severity::Medium)
        .count();
    let low = findings
        .iter()
        .filter(|f| f.severity == Severity::Low)
        .count();

    let mut badges = Vec::new();
    if critical > 0 {
        badges.push(format!(
            "{}",
            format!(" {critical} critical ")
                .on_truecolor(130, 50, 180)
                .white()
                .bold()
        ));
    }
    if high > 0 {
        badges.push(format!(
            "{}",
            format!(" {high} high ").on_red().white().bold()
        ));
    }
    if medium > 0 {
        badges.push(format!(
            "{}",
            format!(" {medium} medium ").on_yellow().black().bold()
        ));
    }
    if low > 0 {
        badges.push(format!(
            "{}",
            format!(" {low} low ").on_blue().white().bold()
        ));
    }

    let total = findings.len();
    let secs = duration.as_secs_f64();

    println!("  {}", "\u{2500}".repeat(50).dimmed());
    println!();
    println!(
        "  {} {}  {}",
        format!("{total}").bold(),
        "issues".dimmed(),
        format!("{files_scanned} files \u{00b7} {secs:.2}s").dimmed(),
    );
    println!();
    println!("  {}", badges.join("  "));
    println!();
}