pmat 3.16.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

// Import parsing and dependency resolution methods for DependencyGraphBuilder
// Extracted from builder.rs

impl DependencyGraphBuilder {
    /// Resolve dependencies for a file
    /// Complexity: 9 (import parsing + edge creation)
    fn resolve_file_dependencies(&mut self, node_id: NodeId, path: &Path) -> Result<()> {
        let content = fs::read_to_string(path)?;

        let imports = match path.extension().and_then(|s| s.to_str()) {
            Some("rs") => self.parse_rust_imports(&content)?,
            Some("py") => self.parse_python_imports(&content)?,
            Some("ts") | Some("tsx") | Some("js") | Some("jsx") => {
                self.parse_typescript_imports(&content)?
            }
            _ => vec![],
        };

        // Create edges for each import
        for import in imports {
            if let Some(target_node) = self.resolve_import_to_node(&import) {
                let edge = EdgeData::Import {
                    weight: 1.0,
                    visibility: Visibility::Public,
                };
                self.graph.add_edge(node_id, target_node, edge);
            }
        }

        Ok(())
    }

    /// Parse Rust imports
    /// Complexity: 6
    fn parse_rust_imports(&self, content: &str) -> Result<Vec<String>> {
        let mut imports = Vec::new();

        for line in content.lines() {
            let trimmed = line.trim();
            if trimmed.starts_with("use ") && trimmed.ends_with(';') {
                let import = trimmed
                    .strip_prefix("use ")
                    .and_then(|s| s.strip_suffix(';'))
                    .unwrap_or("")
                    .trim();
                imports.push(import.to_string());
            }
        }

        Ok(imports)
    }

    /// Parse Python imports
    /// Complexity: 5
    fn parse_python_imports(&self, content: &str) -> Result<Vec<String>> {
        let mut imports = Vec::new();

        for line in content.lines() {
            let trimmed = line.trim();
            if trimmed.starts_with("import ") || trimmed.starts_with("from ") {
                imports.push(trimmed.to_string());
            }
        }

        Ok(imports)
    }

    /// Parse TypeScript/JavaScript imports
    /// Complexity: 6
    fn parse_typescript_imports(&self, content: &str) -> Result<Vec<String>> {
        let mut imports = Vec::new();

        for line in content.lines() {
            let trimmed = line.trim();
            if trimmed.starts_with("import ") {
                if let Some(end) = trimmed.rfind(" from ") {
                    let module = trimmed
                        .get(end + 6..)
                        .unwrap_or_default()
                        .trim_matches(|c| c == '\'' || c == '"' || c == ';');
                    imports.push(module.to_string());
                }
            } else if trimmed.starts_with("const ") && trimmed.contains(" = require(") {
                if let Some(start) = trimmed.find("require('") {
                    if let Some(end) = trimmed.get(start + 9..).unwrap_or_default().find('\'') {
                        let module = trimmed.get(start + 9..start + 9 + end).unwrap_or_default();
                        imports.push(module.to_string());
                    }
                }
            }
        }

        Ok(imports)
    }
}