lean-ctx 3.9.3

//! Native code-health engine — clean code as a token-cost lever.
//!
//! Computes the structural signals the Sonar study links to agent token cost —
//! **cognitive complexity** (S3776-style), **naming quality**, and **module
//! coupling** — once during indexing, then fans them out across the data fabric
//! and every agent surface (see the Code Health Engine plan).
//!
//! Naming note: this is distinct from [`crate::core::quality`] (compression
//! fidelity) and from [`crate::core::gain`]'s usage-quality component; this
//! module scores the *source code's* navigability.

pub mod annotate;
#[cfg(feature = "tree-sitter")]
pub(crate) mod astutil;
pub mod cognitive;
pub mod coupling;
pub mod delta;
pub mod fabric;
pub mod gate;
pub mod naming;
pub mod persist;
pub mod report;
pub mod scan;
pub mod score;

pub use annotate::{ReadAnnotation, annotations_for_file};
pub use cognitive::{FunctionCognitive, cognitive_per_function};
pub use coupling::{ModuleCoupling, module_coupling};
pub use delta::{CognitiveDelta, cognitive_delta, format_gate_notice, worst_regression};
pub use naming::{NamingFinding, cryptic_reason, naming_findings};
pub use scan::{FileReport, ProjectHealth, scan_project};
pub use score::{Hotspot, NavigabilityInputs, NavigabilityScore, grade, navigability};

use serde::Serialize;

/// Default cognitive-complexity threshold (SonarQube S3776 "HIGH" default).
/// A function at or below this is considered navigable. Mirrored by
/// `CodeHealthConfig::default().cognitive_threshold`.
pub const DEFAULT_COGNITIVE_THRESHOLD: u32 = 15;

/// Edit-gate behavior when an edit increases cognitive complexity.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum GateMode {
    /// Never emit a code-health gate notice.
    Off,
    /// Append an advisory `[CODE HEALTH]` notice (default).
    #[default]
    Warn,
    /// Refuse edits that push a clean function over the threshold.
    Block,
}

impl GateMode {
    /// Parse a config string; unknown values fall back to [`GateMode::Warn`].
    pub fn parse(value: &str) -> Self {
        match value.trim().to_ascii_lowercase().as_str() {
            "off" | "false" | "none" | "disabled" => GateMode::Off,
            "block" | "hard" | "error" => GateMode::Block,
            _ => GateMode::Warn,
        }
    }
}

/// Combined per-file health: cognitive scores plus naming findings. This is the
/// single entry point used by the edit-gate, read annotations, and the
/// `ctx_quality` tool.
#[derive(Debug, Clone, Default, PartialEq, Serialize)]
pub struct FileHealth {
    pub functions: Vec<FunctionCognitive>,
    pub naming: Vec<NamingFinding>,
}

impl FileHealth {
    /// Functions whose cognitive complexity exceeds `threshold`.
    pub fn over_threshold(&self, threshold: u32) -> impl Iterator<Item = &FunctionCognitive> {
        self.functions
            .iter()
            .filter(move |f| f.cognitive > threshold)
    }

    /// The single worst cognitive complexity in the file (0 if none).
    pub fn worst_cognitive(&self) -> u32 {
        self.functions
            .iter()
            .map(|f| f.cognitive)
            .max()
            .unwrap_or(0)
    }
}

/// Analyze one file's `source` for the given file `extension`.
///
/// Returns `None` only when tree-sitter is disabled or the extension is
/// unsupported (i.e. no functions could be parsed). Naming findings default to
/// empty when the language has no analyzable identifiers.
pub fn analyze_file(source: &str, extension: &str) -> Option<FileHealth> {
    let functions = cognitive_per_function(source, extension)?;
    let naming = naming_findings(source, extension).unwrap_or_default();
    Some(FileHealth { functions, naming })
}

#[cfg(all(test, feature = "tree-sitter"))]
mod tests {
    use super::*;

    #[test]
    fn analyze_file_combines_signals() {
        let src = "fn _xfm_q2(a: bool, b: bool) { if a { if b {} } }\n";
        let health = analyze_file(src, "rs").unwrap();
        assert_eq!(health.functions.len(), 1);
        assert_eq!(health.worst_cognitive(), 3);
        assert_eq!(health.naming.len(), 1, "cryptic name flagged");
    }

    #[test]
    fn analyze_file_unsupported_ext_is_none() {
        assert!(analyze_file("plain text", "txt").is_none());
    }

    #[test]
    fn over_threshold_filters() {
        let src = "fn deep(a: bool) { if a { if a { if a { if a {} } } } }\n";
        let health = analyze_file(src, "rs").unwrap();
        // 1+2+3+4 = 10 cognitive; above 5, below 15.
        assert_eq!(health.over_threshold(5).count(), 1);
        assert_eq!(health.over_threshold(15).count(), 0);
    }
}