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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

impl IncrementalCoverageAnalyzer {
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "path_exists")]
    /// Create a new instance.
    pub fn new(_db_path: &Path) -> Result<Self> {
        Ok(Self {
            coverage_cache: Arc::new(DashMap::new()),
            ast_cache: Arc::new(DashMap::new()),
            call_graph: Arc::new(CallGraph::new()),
            semaphore: Arc::new(Semaphore::new(num_cpus::get())),
        })
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub async fn analyze_changes(&self, changeset: &ChangeSet) -> Result<CoverageUpdate> {
        let affected_files = self.compute_affected_files(changeset).await?;

        // Parallel analysis with bounded concurrency
        let mut handles = Vec::new();

        for file_id in affected_files {
            let analyzer = self.clone();
            let handle = tokio::spawn(async move {
                let _permit = analyzer.semaphore.acquire().await?;
                analyzer.analyze_file_coverage(&file_id).await
            });
            handles.push(handle);
        }

        let mut file_coverage = HashMap::new();
        for handle in handles {
            let (file_id, coverage) = handle.await??;
            file_coverage.insert(file_id, coverage);
        }

        let aggregate = self.calculate_aggregate_coverage(&file_coverage)?;
        let delta = self.calculate_delta_coverage(changeset, &file_coverage)?;

        Ok(CoverageUpdate {
            file_coverage,
            aggregate_coverage: aggregate,
            delta_coverage: delta,
        })
    }

    async fn compute_affected_files(&self, changeset: &ChangeSet) -> Result<Vec<FileId>> {
        let mut affected = HashSet::new();

        // Direct changes
        affected.extend(changeset.modified_files.iter().cloned());
        affected.extend(changeset.added_files.iter().cloned());

        // Transitive dependencies via call graph
        for file in &changeset.modified_files {
            let dependents = self.call_graph.get_dependents(&file.path.to_string_lossy());
            for dep in dependents {
                let path = PathBuf::from(dep);
                let hash = self.compute_file_hash(&path).await?;
                affected.insert(FileId { path, hash });
            }
        }

        Ok(affected.into_iter().collect())
    }

    async fn analyze_file_coverage(&self, file_id: &FileId) -> Result<(FileId, FileCoverage)> {
        // Check cache first
        if let Some(cached) = self.load_cached_coverage(file_id)? {
            return Ok((file_id.clone(), cached));
        }

        // Parse file and extract coverage data
        let ast = self.parse_file(&file_id.path).await?;
        let coverage = self.compute_coverage(&ast).await?;

        // Store in cache
        self.store_coverage(file_id, &coverage)?;

        Ok((file_id.clone(), coverage))
    }

    async fn compute_coverage(&self, ast: &AstNode) -> Result<FileCoverage> {
        // In production, would integrate with llvm-cov or similar
        // For now, return mock data
        let total_lines = ast
            .functions
            .iter()
            .map(|f| f.end_line - f.start_line + 1)
            .sum::<usize>()
            .max(100);

        let covered_lines = (0..total_lines)
            .filter(|i| i % 3 != 0) // Mock: 2/3 lines covered
            .collect::<Vec<_>>();

        Ok(FileCoverage {
            line_coverage: covered_lines.len() as f64 / total_lines as f64 * 100.0,
            branch_coverage: 75.0,   // Mock
            function_coverage: 80.0, // Mock
            covered_lines,
            total_lines,
        })
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "path_exists")]
    pub async fn compute_file_hash(&self, path: &Path) -> Result<[u8; 32]> {
        let content = tokio::fs::read(path).await?;
        let hash = blake3::hash(&content);
        Ok(*hash.as_bytes())
    }
}