From e4132ff564c59be72c6546752077a8f33bb4c00f Mon Sep 17 00:00:00 2001
From: Jonas Maier <jonas@x77.dev>
Date: Fri, 22 May 2026 21:31:19 +0200
Subject: syntax highlighting fixes

---
 src/lib.rs       | 1 +
 src/parse/mod.rs | 8 +-------
 2 files changed, 2 insertions(+), 7 deletions(-)

(limited to 'src')

diff --git a/src/lib.rs b/src/lib.rs
index 2a9a1ee..47dbdf5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -353,6 +353,7 @@ impl Session {
                 self.highlighter
                     .pretty_print(&buf, parser.highlights, &mut stdout)?;
                 cursor::fmove_cursor(cursor::Direction::Left, self.line.right_len(), &mut stdout)?;
+                stdout.flush()?;
             }
         }
 
diff --git a/src/parse/mod.rs b/src/parse/mod.rs
index 97a6e4a..1c1b184 100644
--- a/src/parse/mod.rs
+++ b/src/parse/mod.rs
@@ -1594,15 +1594,9 @@ impl Parse for If<PreExpansion> {
 
 impl Parse for While {
     fn parse(b: &mut Cursor<'_>) -> Result<Self> {
-        b.spaces();
-        if !b.buf.starts_with(b"while ") || b.buf.starts_with(b"while\t") {
-            return Err(ParseError::NotAWhile);
-        }
-        b.advance(6);
-        b.spaces();
+        b.consume_keyword(Keyword::While)?;
         let condition = Pipes::parse(b)?;
         let block = Block::parse(b)?;
-
         Ok(Self { condition, block })
     }
 }
-- 
cgit v1.2.3