code-analyze 0.1.0

Analyze code structure and relationships using tree-sitter
Documentation 
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

// Copyright 2024 Block, Inc. (original code from https://github.com/block/goose)
// Copyright 2025 utapyngo (modifications)
// SPDX-License-Identifier: Apache-2.0

/// Tree-sitter query for extracting Go code elements
pub const ELEMENT_QUERY: &str = r#"
    (function_declaration name: (identifier) @func)
    (method_declaration name: (field_identifier) @func)
    (type_declaration (type_spec name: (type_identifier) @struct))
    (const_declaration (const_spec name: (identifier) @const))
    (import_declaration) @import
"#;

/// Tree-sitter query for extracting Go function calls and identifier references
pub const CALL_QUERY: &str = r#"
    ; Function calls
    (call_expression
      function: (identifier) @function.call)

    ; Method calls
    (call_expression
      function: (selector_expression
        field: (field_identifier) @method.call))

    ; Identifier references in various expression contexts
    (argument_list (identifier) @identifier.reference)
    (binary_expression left: (identifier) @identifier.reference)
    (binary_expression right: (identifier) @identifier.reference)
    (unary_expression operand: (identifier) @identifier.reference)
    (return_statement (expression_list (identifier) @identifier.reference))
    (assignment_statement right: (expression_list (identifier) @identifier.reference))
"#;

/// Tree-sitter query for extracting Go struct references and usage patterns
pub const REFERENCE_QUERY: &str = r#"
    ; Method receivers - pointer type
    (method_declaration
      receiver: (parameter_list
        (parameter_declaration
          type: (pointer_type (type_identifier) @method.receiver))))

    ; Method receivers - value type
    (method_declaration
      receiver: (parameter_list
        (parameter_declaration
          type: (type_identifier) @method.receiver)))

    ; Struct literals - simple
    (composite_literal
      type: (type_identifier) @struct.literal)

    ; Struct literals - qualified (package.Type)
    (composite_literal
      type: (qualified_type
        name: (type_identifier) @struct.literal))

    ; Field declarations in structs - simple type
    (field_declaration
      type: (type_identifier) @field.type)

    ; Field declarations - pointer type
    (field_declaration
      type: (pointer_type
        (type_identifier) @field.type))

    ; Field declarations - qualified type (package.Type)
    (field_declaration
      type: (qualified_type
        name: (type_identifier) @field.type))

    ; Field declarations - pointer to qualified type
    (field_declaration
      type: (pointer_type
        (qualified_type
          name: (type_identifier) @field.type)))
"#;

/// Find the method name for a method receiver node in Go
pub fn find_method_for_receiver(
    receiver_node: &tree_sitter::Node,
    source: &str,
    _ast_recursion_limit: Option<usize>,
) -> Option<String> {
    let mut current = *receiver_node;
    while let Some(parent) = current.parent() {
        if parent.kind() == "method_declaration" {
            for i in 0..parent.child_count() as u32 {
                if let Some(child) = parent.child(i)
                    && child.kind() == "field_identifier"
                {
                    return source.get(child.byte_range()).map(|s| s.to_string());
                }
            }
        }
        current = parent;
    }
    None
}