Update c queries and test

nvim-treesitter · Dec 8, 2023 · 9d5cb59 · 9d5cb59
1 parent 2806d83
commit 9d5cb59
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 6 deletions.
diff --git a/queries/c/context.scm b/queries/c/context.scm
@@ -8,29 +8,33 @@
 ) @context
 
 (function_definition
-  body: (_ (_) @context.end)
+  body: (_) @context.end
 ) @context
 
 (for_statement
   (compound_statement) @context.end
 ) @context
 
 (if_statement
-  consequence: (_ (_) @context.end)
+  consequence: (_) @context.end
+) @context
+
+(else_clause
+  (_) @context.end
 ) @context
 
 (while_statement
-  body: (_ (_) @context.end)
+  body: (_) @context.end
 ) @context
 
 (do_statement
-  body: (_ (_) @context.end)
+  body: (_) @context.end
 ) @context
 
 (struct_specifier
-  body: (_ (_) @context.end)
+  body: (_) @context.end
 ) @context
 
 (enum_specifier
-  body: (_ (_) @context.end)
+  body: (_) @context.end
 ) @context
diff --git a/test/test.c b/test/test.c
@@ -72,5 +72,46 @@ int main(int arg1,
       // comment
     }
 
+  } else if (arg1 == 4) {
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+  } else {
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
+    // comment
   }
 }