codesynapse_core/ts_extract/
dart.rs

1use super::{add_contains_edge, add_node_if_missing, make_file_node, run_query_named};
2use crate::error::Result;
3use crate::extract::{make_id, ImportNode, LanguageExtractor};
4use crate::types::{Edge, ExtractionFragment, Node};
5use std::collections::HashMap;
6use std::path::Path;
7
8pub struct TsDartExtractor;
9
10impl TsDartExtractor {
11    pub fn extract(source: &[u8], path: &Path) -> Result<ExtractionFragment> {
12        let (file_id, _, file_node) = make_file_node(path);
13        let mut fragment = ExtractionFragment {
14            nodes: vec![file_node],
15            edges: vec![],
16        };
17
18        let lang = tree_sitter_dart::LANGUAGE.into();
19
20        let class_query = r#"
21            (class_declaration
22                name: (identifier) @class.name
23            )
24        "#;
25        if let Ok(mut captures) = run_query_named(source, &lang, class_query) {
26            for name in captures.remove("class.name").unwrap_or_default() {
27                let id = make_id(&[&file_id, &name]);
28                fragment.nodes.push(Node {
29                    id: id.clone(),
30                    label: name,
31                    file_type: "class".to_string(),
32                    source_file: path.to_string_lossy().to_string(),
33                    source_location: None,
34                    community: None,
35                    rationale: None,
36                    docstring: None,
37                    metadata: HashMap::new(),
38                });
39                add_contains_edge(&mut fragment, &file_id, id, path);
40            }
41        }
42
43        let method_query = r#"
44            (method_declaration
45                signature: (method_signature
46                    (function_signature
47                        name: (identifier) @method.name
48                    )
49                )
50            )
51        "#;
52        if let Ok(mut captures) = run_query_named(source, &lang, method_query) {
53            for name in captures.remove("method.name").unwrap_or_default() {
54                let label = format!("{}()", name);
55                let id = make_id(&[&file_id, &name]);
56                fragment.nodes.push(Node {
57                    id: id.clone(),
58                    label,
59                    file_type: "method".to_string(),
60                    source_file: path.to_string_lossy().to_string(),
61                    source_location: None,
62                    community: None,
63                    rationale: None,
64                    docstring: None,
65                    metadata: HashMap::new(),
66                });
67                add_contains_edge(&mut fragment, &file_id, id, path);
68            }
69        }
70
71        let import_query = r#"
72            (import_specification
73                uri: (uri
74                    (string_literal) @import.uri))
75        "#;
76        if let Ok(mut captures) = run_query_named(source, &lang, import_query) {
77            for uri in captures.remove("import.uri").unwrap_or_default() {
78                let module = uri.trim_matches('"').trim().trim_matches('\'');
79                let mod_id = make_id(&[&file_id, module]);
80                let mod_node = Node {
81                    id: mod_id.clone(),
82                    label: module.to_string(),
83                    file_type: "module".to_string(),
84                    source_file: path.to_string_lossy().to_string(),
85                    source_location: None,
86                    community: None,
87                    rationale: None,
88                    docstring: None,
89                    metadata: HashMap::new(),
90                };
91                add_node_if_missing(&mut fragment, mod_node);
92                fragment.edges.push(Edge {
93                    source: file_id.clone(),
94                    target: mod_id,
95                    relation: "imports".to_string(),
96                    confidence: "EXTRACTED".to_string(),
97                    source_file: Some(path.to_string_lossy().to_string()),
98                    weight: 1.0,
99                    context: None,
100                });
101            }
102        }
103
104        Ok(fragment)
105    }
106}
107
108impl LanguageExtractor for TsDartExtractor {
109    fn file_extensions(&self) -> Vec<&'static str> {
110        vec!["dart"]
111    }
112    fn extract(&self, source: &[u8], path: &Path) -> Result<ExtractionFragment> {
113        Self::extract(source, path)
114    }
115    fn resolve_imports(&self, _imports: &[ImportNode]) -> Vec<Edge> {
116        vec![]
117    }
118    fn collect_type_refs(&self, _fragment: &mut ExtractionFragment) {}
119}
codesynapse_core/ts_extract/dart.rs

codesynapse_core/ts_extract/
dart.rs
