pmat 3.15.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

// Code quality analyzer methods
// Included by refactor.rs — shares parent module scope

impl CodeAnalyzer {
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Create a new instance.
    pub fn new(_profile: QualityProfile) -> Self {
        Self {}
    }

    /// Analyze code quality
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn analyze(&self, code: &str) -> Result<CodeAnalysis> {
        Ok(CodeAnalysis {
            complexity: self.calculate_complexity(code),
            coverage: self.estimate_coverage(code),
            tdg: self.calculate_tdg(code),
            satd_count: self.count_satd(code),
            function_count: self.count_functions(code),
            quality_score: self.calculate_quality_score(code),
        })
    }

    fn calculate_complexity(&self, code: &str) -> u32 {
        let if_count = code.matches("if ").count() as u32;
        let match_count = code.matches("match ").count() as u32;
        let loop_count =
            code.matches("for ").count() as u32 + code.matches("while ").count() as u32;

        1 + if_count + match_count + loop_count
    }

    fn estimate_coverage(&self, code: &str) -> f64 {
        let test_lines = code.matches("#[test]").count() * 10; // Rough estimate
        let total_lines = code.lines().count().max(1);
        (test_lines as f64 / total_lines as f64 * 100.0).min(100.0)
    }

    fn calculate_tdg(&self, code: &str) -> u32 {
        let todo_count = code.matches("todo!").count() as u32;
        let unwrap_count = code.matches("unwrap").count() as u32;
        todo_count + unwrap_count
    }

    fn count_satd(&self, code: &str) -> u32 {
        (code.matches("TODO").count()
            + code.matches("FIXME").count()
            + code.matches("HACK").count()) as u32
    }

    fn count_functions(&self, code: &str) -> usize {
        code.matches("fn ").count()
    }

    fn calculate_quality_score(&self, code: &str) -> f64 {
        let complexity = f64::from(self.calculate_complexity(code));
        let coverage = self.estimate_coverage(code);
        let tdg = f64::from(self.calculate_tdg(code));

        let complexity_score = (20.0 - complexity).max(0.0) / 20.0 * 40.0;
        let coverage_score = coverage * 0.4;
        let tdg_score = (10.0 - tdg).max(0.0) / 10.0 * 20.0;

        complexity_score + coverage_score + tdg_score
    }
}