Parse while loops

1 files changed, 21 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index 9bf95612..a5a48756 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -315,6 +315,9 @@ public class Parser {
             case KEYWORD_VAR ->
                 parseVar(tokens);
 
+            case KEYWORD_WHILE ->
+                parseWhile(tokens);
+
             default ->
                 parseExpressionStatement(tokens);
         };
@@ -601,4 +604,22 @@ public class Parser {
 
         return new UntypedVarNode(target, expression, source);
     }
+
+    private static UntypedFunctionStatementNode parseWhile(
+        TokenIterator<TokenType> tokens
+    ) {
+        var start = tokens.position();
+
+        tokens.skip(TokenType.KEYWORD_WHILE);
+        tokens.skip(TokenType.SYMBOL_PAREN_OPEN);
+        var condition = parseTopLevelExpression(tokens);
+        tokens.skip(TokenType.SYMBOL_PAREN_CLOSE);
+
+        var body = parseBlock(tokens);
+
+        var end = tokens.position();
+        var source = start.to(end);
+
+        return new UntypedWhileNode(condition, body, source);
+    }
 }