code_analyze_mcp/languages/
java.rs

1/// Tree-sitter query for extracting Java elements (methods and classes).
2pub const ELEMENT_QUERY: &str = r#"
3(method_declaration
4  name: (identifier) @method_name) @function
5(class_declaration
6  name: (identifier) @class_name) @class
7(interface_declaration
8  name: (identifier) @interface_name) @class
9(enum_declaration
10  name: (identifier) @enum_name) @class
11"#;
12
13/// Tree-sitter query for extracting function calls.
14pub const CALL_QUERY: &str = r#"
15(method_invocation
16  name: (identifier) @call)
17"#;
18
19/// Tree-sitter query for extracting type references.
20pub const REFERENCE_QUERY: &str = r#"
21(type_identifier) @type_ref
22"#;
23
24/// Tree-sitter query for extracting Java imports.
25pub const IMPORT_QUERY: &str = r#"
26(import_declaration) @import_path
27"#;
28
29use tree_sitter::Node;
30
31/// Extract inheritance information from a Java class node.
32pub fn extract_inheritance(node: &Node, source: &str) -> Vec<String> {
33    let mut inherits = Vec::new();
34
35    // Extract superclass (extends)
36    if let Some(superclass) = node.child_by_field_name("superclass") {
37        for i in 0..superclass.named_child_count() {
38            if let Some(child) = superclass.named_child(i as u32)
39                && child.kind() == "type_identifier"
40            {
41                let text = &source[child.start_byte()..child.end_byte()];
42                inherits.push(format!("extends {}", text));
43            }
44        }
45    }
46
47    // Extract interfaces (implements)
48    if let Some(interfaces) = node.child_by_field_name("interfaces") {
49        for i in 0..interfaces.named_child_count() {
50            if let Some(type_list) = interfaces.named_child(i as u32) {
51                for j in 0..type_list.named_child_count() {
52                    if let Some(type_node) = type_list.named_child(j as u32)
53                        && type_node.kind() == "type_identifier"
54                    {
55                        let text = &source[type_node.start_byte()..type_node.end_byte()];
56                        inherits.push(format!("implements {}", text));
57                    }
58                }
59            }
60        }
61    }
62
63    inherits
64}
code_analyze_mcp/languages/java.rs

code_analyze_mcp/languages/
java.rs
