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

// Utility methods for Mermaid engine
// Included by deterministic_mermaid_engine.rs - shares parent module scope

impl DeterministicMermaidEngine {
    /// Filter graph to service modules only (heuristic)
    fn filter_to_services(
        &self,
        graph: &SimpleStableGraph<ModuleNode, EdgeType>,
    ) -> SimpleStableGraph<ModuleNode, EdgeType> {
        let mut service_graph = SimpleStableGraph::new();
        let mut node_mapping = BTreeMap::new();

        // Add nodes that look like services
        for idx in graph.node_indices() {
            let node = &graph[idx];
            if self.is_service_module(&node.name) {
                let new_idx = service_graph.add_node(node.clone());
                node_mapping.insert(idx, new_idx);
            }
        }

        // Add edges between service nodes
        for edge in graph.edge_references() {
            if let (Some(&source_idx), Some(&target_idx)) = (
                node_mapping.get(&edge.source()),
                node_mapping.get(&edge.target()),
            ) {
                service_graph.add_edge(source_idx, target_idx, edge.weight().clone());
            }
        }

        service_graph
    }

    /// Heuristic to determine if a module is a service
    fn is_service_module(&self, name: &str) -> bool {
        name.contains("service")
            || name.contains("handler")
            || name.contains("controller")
            || name.contains("api")
            || name.contains("engine")
    }

    /// Get Mermaid arrow style for edge type
    fn get_edge_arrow(&self, edge_type: &EdgeType) -> &'static str {
        match edge_type {
            EdgeType::Calls => "-->",
            EdgeType::Imports => "-.->",
            EdgeType::Inherits => "-->",
            EdgeType::Implements => "-.->",
            EdgeType::Uses => "---",
        }
    }

    /// Sanitize ID for Mermaid compatibility
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn sanitize_id(&self, id: &str) -> String {
        // Replace common multi-character patterns
        let sanitized = id.replace("::", "_").replace(['/', '.', '-', ' '], "_");

        // Replace any remaining non-alphanumeric characters with underscores
        let sanitized: String = sanitized
            .chars()
            .map(|c| {
                if c.is_ascii_alphanumeric() || c == '_' {
                    c
                } else {
                    '_'
                }
            })
            .collect();

        // Ensure it starts with a letter or underscore
        if sanitized.is_empty() {
            "_empty".to_string()
        } else if sanitized
            .chars()
            .next()
            .expect("sanitized is non-empty (checked above)")
            .is_numeric()
        {
            format!("_{sanitized}")
        } else {
            sanitized
        }
    }

    /// Escape label for Mermaid compatibility
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn escape_mermaid_label(&self, label: &str) -> String {
        // For maximum compatibility, use simple character replacements
        label
            .replace('&', " and ")
            .replace('"', "'")
            .replace('<', "(")
            .replace('>', ")")
            .replace('|', " - ")
            .replace('[', "(")
            .replace(']', ")")
            .replace('{', "(")
            .replace('}', ")")
            .replace('\n', " ")
    }
}