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

// 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 Swift code elements
pub const ELEMENT_QUERY: &str = r#"
    ; Functions
    (function_declaration name: (simple_identifier) @func)

    ; Classes
    (class_declaration name: (type_identifier) @class)

    ; Protocols (interfaces)
    (protocol_declaration name: (type_identifier) @class)

    ; Imports
    (import_declaration) @import
"#;

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

    ; Method calls with navigation
    (call_expression
      (navigation_expression
        target: (_)
        suffix: (navigation_suffix
          suffix: (simple_identifier) @method.call)))

    ; Constructor calls
    (constructor_expression
      (user_type
        (type_identifier) @constructor.call))

    ; Async function calls
    (await_expression
      (call_expression
        (simple_identifier) @function.call))

    ; Async method calls
    (await_expression
      (call_expression
        (navigation_expression
          suffix: (navigation_suffix
            suffix: (simple_identifier) @method.call))))

    ; Static method calls (Type.method())
    (call_expression
      (navigation_expression
        target: (user_type)
        suffix: (navigation_suffix
          suffix: (simple_identifier) @scoped.call)))

    ; Closure calls
    (call_expression
      (navigation_expression) @function.call)
"#;

/// Extract function name for Swift-specific node kinds
pub fn extract_function_name_for_kind(
    _node: &tree_sitter::Node,
    _source: &str,
    kind: &str,
) -> Option<String> {
    match kind {
        "init_declaration" => Some("init".to_string()),
        "deinit_declaration" => Some("deinit".to_string()),
        _ => None,
    }
}