feat: improve ruby parsing (#1085)

2025-01-15 05:37:53 -05:00
parent 0df03dd213
commit 0a837a4583
2 changed files with 174 additions and 19 deletions
--- a/crates/avante-repo-map/queries/tree-sitter-ruby-defs.scm
+++ b/crates/avante-repo-map/queries/tree-sitter-ruby-defs.scm
@@ -1,16 +1,26 @@
 ;; Capture top-level methods, class definitions, and methods within classes
-(program
-  (class
-    (body_statement
-      (call) @class_call
-      (assignment) @class_assignment
-      (method) @method
-    )
-  ) @class
-)
+
+(class
+  (body_statement
+    (call)? @class_call
+    (assignment)? @class_assignment
+    (method)? @method
+  )
+) @class
+
 (program
  (method) @function
 )
 (program
  (assignment) @assignment
 )
+
+(module) @module
+
+(module
+  (body_statement
+    (call)? @class_call
+    (assignment)? @class_assignment
+    (method)? @method
+  )
+)