add loop and conditional textobjects for c, rust, and python

meain · Dec 22, 2023 · 220ceae · 220ceae
1 parent 9a9edd4
commit 220ceae
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/treesit-queries/c/textobjects.scm b/treesit-queries/c/textobjects.scm
@@ -19,3 +19,15 @@
 (comment) @comment.inner
 
 (comment)+ @comment.outer
+
+(for_statement
+ body: (_) @loop.inner) @loop.outer
+
+(while_statement
+ body: (_) @loop.inner) @loop.outer
+
+(do_statement
+ body: (_) @loop.inner) @loop.outer
+
+(if_statement
+ consequence: (_) @conditional.inner) @conditional.outer
diff --git a/treesit-queries/python/textobjects.scm b/treesit-queries/python/textobjects.scm
@@ -21,3 +21,12 @@
    name: (identifier) @_name
    body: (block)? @test.inner) @test.outer
  (#match "^test_" @_name))
+
+(for_statement
+ body: (_) @loop.inner) @loop.outer
+
+(while_statement
+ body: (_) @loop.inner) @loop.outer
+
+(if_statement
+ consequence: (_) @conditional.inner) @conditional.outer
diff --git a/treesit-queries/rust/textobjects.scm b/treesit-queries/rust/textobjects.scm
@@ -56,3 +56,15 @@
  (function_item
    body: (_) @test.inner) @test.outer
  (#equal @_test_attribute "test"))
+
+(loop_expression
+  body: (_) @loop.inner) @loop.outer
+
+(while_expression
+  body: (_) @loop.inner) @loop.outer
+
+(for_expression
+  body: (_) @loop.inner) @loop.outer
+
+(if_expression
+  consequence: (_) @conditional.inner) @conditional.outer