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

// 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 Python code elements
pub const ELEMENT_QUERY: &str = r#"
    (function_definition name: (identifier) @func)
    (class_definition name: (identifier) @class)
    (import_statement) @import
    (import_from_statement) @import
    (aliased_import) @import
    (assignment left: (identifier) @class)
"#;

/// Tree-sitter query for extracting Python function calls
pub const CALL_QUERY: &str = r#"
    ; Function calls
    (call
      function: (identifier) @function.call)
    
    ; Method calls
    (call
      function: (attribute
        attribute: (identifier) @method.call))

    ; Decorator applications
    (decorator (identifier) @function.call)
    (decorator (attribute attribute: (identifier) @method.call))
"#;