Add `peek_n` to `package text/scanner`

1 files changed, 15 insertions, 0 deletions

diff --git a/core/text/scanner/scanner.odin b/core/text/scanner/scanner.odin
index 30fa961a8..a3c44e909 100644
--- a/core/text/scanner/scanner.odin
+++ b/core/text/scanner/scanner.odin
@@ -160,6 +160,21 @@ peek :: proc(s: ^Scanner) -> rune {
 	return s.ch;
 }
 
+peek_n :: proc(s: ^Scanner, n: int) -> rune {
+	assert(n >= 0);
+	if n == 0 {
+		return peek(s);
+	}
+
+	prev_s := s^;
+	for in 0..<n {
+		next(s);
+	}
+	ch := peek(s);
+	s^ = prev_s;
+	return ch;
+}
+
 
 error :: proc(s: ^Scanner, msg: string) {
 	s.error_count += 1;