From ca91f4a57da06b2202e748ae5858f0a1973bba98 Mon Sep 17 00:00:00 2001
From: Nat Quayle Nelson <natquaylenelson@gmail.com>
Date: Thu, 7 Aug 2025 18:18:38 -0500
Subject: [PATCH] Stream.takeLinesAsStreamWhile

---
 src/kiss/Stream.hx | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/kiss/Stream.hx b/src/kiss/Stream.hx
index 9cf286d..a3e6913 100644
--- a/src/kiss/Stream.hx
+++ b/src/kiss/Stream.hx
@@ -426,6 +426,34 @@ class Stream {
         };
     }
 
+    // As long as lines keep passing a predicate, keep adding them to a Stream, and return it
+    public function takeLinesAsStreamWhile(predicate:String->Bool):Option<Stream> {
+        var lineNo = this.line;
+        var column = this.column;
+        var absoluteChar = this.absoluteChar;
+
+        var newContent = "";
+        while (true){
+            switch (peekLine()){
+                case Some(line) if(predicate(line)):
+                    newContent += line + "\n";
+                    takeLine();
+                default:
+                    break;
+            }
+        }
+        if(newContent.length == 0) return None;
+        // Remove trailing newline
+        newContent = newContent.substr(0, newContent.length - 1);
+
+        var s = Stream.fromString(newContent);
+        s.line = lineNo;
+        s.column = column;
+        s.file = this.file;
+        s.absoluteChar = absoluteChar;
+        return Some(s);
+    }
+
     public function expect(whatToExpect:String="something unspecified", f:Void->Option<String>):String {
         var position = position();
         switch (f()) {