From 07c978cb207e1558dd63b2b5c73fffe00216c20d Mon Sep 17 00:00:00 2001
From: Sergei Zharinov <zharinov@users.noreply.github.com>
Date: Sat, 13 Dec 2025 09:36:16 -0300
Subject: [PATCH] fix: Capture name offset

---
 crates/plotnik-lib/src/parser/grammar.rs                    | 1 +
 crates/plotnik-lib/src/parser/tests/grammar/trivia_tests.rs | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/crates/plotnik-lib/src/parser/grammar.rs b/crates/plotnik-lib/src/parser/grammar.rs
index b2a0085e..7c68a3d5 100644
--- a/crates/plotnik-lib/src/parser/grammar.rs
+++ b/crates/plotnik-lib/src/parser/grammar.rs
@@ -706,6 +706,7 @@ impl Parser<'_> {
     fn try_parse_capture(&mut self, checkpoint: Checkpoint) {
         if self.peek() == SyntaxKind::At {
             self.start_node_at(checkpoint, SyntaxKind::Capture);
+            self.drain_trivia();
             self.parse_capture_suffix();
             self.finish_node();
         }
diff --git a/crates/plotnik-lib/src/parser/tests/grammar/trivia_tests.rs b/crates/plotnik-lib/src/parser/tests/grammar/trivia_tests.rs
index 886def5f..f38080e0 100644
--- a/crates/plotnik-lib/src/parser/tests/grammar/trivia_tests.rs
+++ b/crates/plotnik-lib/src/parser/tests/grammar/trivia_tests.rs
@@ -17,9 +17,9 @@ fn whitespace_preserved() {
             ParenOpen "("
             Id "identifier"
             ParenClose ")"
+          Whitespace "  "
           At "@"
           Id "name"
-      Whitespace "  "
       Newline "\n"
     "#);
 }