add tests for rust context parsing, and update rust embedding query

Co-authored-by: maxbrunsfeld <max@zed.dev>
2023-07-13 16:58:42 -04:00
parent 0a0e40fb24
commit 623cb9833c
3 changed files with 178 additions and 46 deletions
@@ -1,22 +1,50 @@
 (
    (line_comment)* @context
    .
-    [
-        (enum_item
-            name: (_) @name) @item
-        (struct_item
-            name: (_) @name) @item
-        (impl_item
-            trait: (_)? @name
-            "for"? @name
-            type: (_) @name) @item
-        (trait_item
-            name: (_) @name) @item
-        (function_item
-            name: (_) @name) @item
-        (macro_definition
-            name: (_) @name) @item
-        (function_signature_item
-            name: (_) @name) @item
-    ]
+    (enum_item
+        name: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (struct_item
+        name: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (impl_item
+        trait: (_)? @name
+        "for"? @name
+        type: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (trait_item
+        name: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (function_item
+        name: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (macro_definition
+        name: (_) @name) @item
+)
+
+(
+    (line_comment)* @context
+    .
+    (function_signature_item
+        name: (_) @name) @item
 )