From 53742779e390d8e0363ac7b253f9a3cf73fc498d Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sat, 15 Oct 2022 02:29:32 +0200
Subject: [PATCH] highlights(rust): distinguish between "for" in loops and
 impl_item

Fixes #3641
---
 queries/rust/highlights.scm        |  6 ++++--
 tests/query/highlights/rust/for.rs | 15 +++++++++++++++
 2 files changed, 19 insertions(+), 2 deletions(-)
 create mode 100644 tests/query/highlights/rust/for.rs

diff --git a/queries/rust/highlights.scm b/queries/rust/highlights.scm
index db5e39828..c08912bea 100644
--- a/queries/rust/highlights.scm
+++ b/queries/rust/highlights.scm
@@ -211,13 +211,15 @@
 [
   "break"
   "continue"
-  "for"
   "in"
   "loop"
   "while"
 ] @repeat
 
-
+(impl_item
+  "for" @keyword)
+(for_expression
+  "for" @repeat)
 
 ;;; Operators & Punctuation
 
diff --git a/tests/query/highlights/rust/for.rs b/tests/query/highlights/rust/for.rs
new file mode 100644
index 000000000..27f072847
--- /dev/null
+++ b/tests/query/highlights/rust/for.rs
@@ -0,0 +1,15 @@
+struct Foo;
+
+// Issue https://github.com/nvim-treesitter/nvim-treesitter/issues/3641
+// Distinguish between for in loop or impl_item
+impl Drop for Foo {
+    //     ^ @keyword
+   fn drop(&mut self) {}
+}
+
+fn main() {
+    for i in 0..128 {
+    // <- @repeat
+        println!("{i}");
+    }
+}