feat: avante repo map rust crate (#628)
This commit is contained in:
11
crates/avante-repo-map/queries/tree-sitter-c-defs.scm
Normal file
11
crates/avante-repo-map/queries/tree-sitter-c-defs.scm
Normal file
@@ -0,0 +1,11 @@
|
||||
;; Capture extern functions, variables, public classes, and methods
|
||||
(function_definition
|
||||
(storage_class_specifier) @extern
|
||||
) @function
|
||||
(class_specifier
|
||||
(public) @class
|
||||
(function_definition) @method
|
||||
) @class
|
||||
(declaration
|
||||
(storage_class_specifier) @extern
|
||||
) @variable
|
||||
11
crates/avante-repo-map/queries/tree-sitter-cpp-defs.scm
Normal file
11
crates/avante-repo-map/queries/tree-sitter-cpp-defs.scm
Normal file
@@ -0,0 +1,11 @@
|
||||
;; Capture extern functions, variables, public classes, and methods
|
||||
(function_definition
|
||||
(storage_class_specifier) @extern
|
||||
) @function
|
||||
(class_specifier
|
||||
(public) @class
|
||||
(function_definition) @method
|
||||
) @class
|
||||
(declaration
|
||||
(storage_class_specifier) @extern
|
||||
) @variable
|
||||
18
crates/avante-repo-map/queries/tree-sitter-go-defs.scm
Normal file
18
crates/avante-repo-map/queries/tree-sitter-go-defs.scm
Normal file
@@ -0,0 +1,18 @@
|
||||
;; Capture top-level functions and struct definitions
|
||||
(var_declaration
|
||||
(var_spec) @variable
|
||||
)
|
||||
(const_declaration
|
||||
(const_spec) @variable
|
||||
)
|
||||
(function_declaration) @function
|
||||
(type_declaration
|
||||
(type_spec (struct_type)) @class
|
||||
)
|
||||
(type_declaration
|
||||
(type_spec
|
||||
(struct_type
|
||||
(field_declaration_list
|
||||
(field_declaration) @class_variable)))
|
||||
)
|
||||
(method_declaration) @method
|
||||
@@ -0,0 +1,23 @@
|
||||
;; Capture exported functions, arrow functions, variables, classes, and method definitions
|
||||
(export_statement
|
||||
declaration: (lexical_declaration
|
||||
(variable_declarator) @variable
|
||||
)
|
||||
)
|
||||
(export_statement
|
||||
declaration: (function_declaration) @function
|
||||
)
|
||||
(export_statement
|
||||
declaration: (class_declaration
|
||||
body: (class_body
|
||||
(field_definition) @class_variable
|
||||
)
|
||||
)
|
||||
)
|
||||
(export_statement
|
||||
declaration: (class_declaration
|
||||
body: (class_body
|
||||
(method_definition) @method
|
||||
)
|
||||
)
|
||||
)
|
||||
3
crates/avante-repo-map/queries/tree-sitter-lua-defs.scm
Normal file
3
crates/avante-repo-map/queries/tree-sitter-lua-defs.scm
Normal file
@@ -0,0 +1,3 @@
|
||||
;; Capture function and method definitions
|
||||
(variable_list) @variable
|
||||
(function_declaration) @function
|
||||
25
crates/avante-repo-map/queries/tree-sitter-python-defs.scm
Normal file
25
crates/avante-repo-map/queries/tree-sitter-python-defs.scm
Normal file
@@ -0,0 +1,25 @@
|
||||
;; Capture top-level functions, class, and method definitions
|
||||
(module
|
||||
(expression_statement
|
||||
(assignment) @assignment
|
||||
)
|
||||
)
|
||||
(module
|
||||
(function_definition) @function
|
||||
)
|
||||
(module
|
||||
(class_definition
|
||||
body: (block
|
||||
(expression_statement
|
||||
(assignment) @class_assignment
|
||||
)
|
||||
)
|
||||
)
|
||||
)
|
||||
(module
|
||||
(class_definition
|
||||
body: (block
|
||||
(function_definition) @method
|
||||
)
|
||||
)
|
||||
)
|
||||
16
crates/avante-repo-map/queries/tree-sitter-ruby-defs.scm
Normal file
16
crates/avante-repo-map/queries/tree-sitter-ruby-defs.scm
Normal file
@@ -0,0 +1,16 @@
|
||||
;; Capture top-level methods, class definitions, and methods within classes
|
||||
(program
|
||||
(class
|
||||
(body_statement
|
||||
(call) @class_call
|
||||
(assignment) @class_assignment
|
||||
(method) @method
|
||||
)
|
||||
) @class
|
||||
)
|
||||
(program
|
||||
(method) @function
|
||||
)
|
||||
(program
|
||||
(assignment) @assignment
|
||||
)
|
||||
20
crates/avante-repo-map/queries/tree-sitter-rust-defs.scm
Normal file
20
crates/avante-repo-map/queries/tree-sitter-rust-defs.scm
Normal file
@@ -0,0 +1,20 @@
|
||||
;; Capture public functions, structs, methods, and variable definitions
|
||||
(function_item) @function
|
||||
(impl_item
|
||||
body: (declaration_list
|
||||
(function_item) @method
|
||||
)
|
||||
)
|
||||
(struct_item) @class
|
||||
(struct_item
|
||||
body: (field_declaration_list
|
||||
(field_declaration) @class_variable
|
||||
)
|
||||
)
|
||||
(enum_item
|
||||
body: (enum_variant_list
|
||||
(enum_variant) @enum_item
|
||||
)
|
||||
)
|
||||
(const_item) @variable
|
||||
(static_item) @variable
|
||||
@@ -0,0 +1,33 @@
|
||||
;; Capture exported functions, arrow functions, variables, classes, and method definitions
|
||||
(export_statement
|
||||
declaration: (lexical_declaration
|
||||
(variable_declarator) @variable
|
||||
)
|
||||
)
|
||||
(export_statement
|
||||
declaration: (function_declaration) @function
|
||||
)
|
||||
(export_statement
|
||||
declaration: (class_declaration
|
||||
body: (class_body
|
||||
(public_field_definition) @class_variable
|
||||
)
|
||||
)
|
||||
)
|
||||
(interface_declaration
|
||||
body: (interface_body
|
||||
(property_signature) @class_variable
|
||||
)
|
||||
)
|
||||
(type_alias_declaration
|
||||
value: (object_type
|
||||
(property_signature) @class_variable
|
||||
)
|
||||
)
|
||||
(export_statement
|
||||
declaration: (class_declaration
|
||||
body: (class_body
|
||||
(method_definition) @method
|
||||
)
|
||||
)
|
||||
)
|
||||
Reference in New Issue
Block a user