From daec88c2f81a0e56d3158a717047fb4d80cf4cf0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Emmanuel=20Beno=C3=AEt?= <tseeker@nocternity.net>
Date: Sun, 1 Jan 2023 18:40:05 +0100
Subject: [PATCH] Parser - While loops

---
 src/parser.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/parser.rs b/src/parser.rs
index 15b581d..24c92a7 100644
--- a/src/parser.rs
+++ b/src/parser.rs
@@ -80,6 +80,8 @@ impl Parser {
     /// statement := "print" expression ";"
     /// statement := declaration ";"
     /// statement := block
+    /// statement := if_statement
+    /// statement := while_statement
     /// ```
     fn parse_statement(&mut self) -> ParserResult<ast::StmtNode> {
         if self.expect(&[TokenType::Var]).is_some() {
@@ -88,6 +90,8 @@ impl Parser {
             self.parse_block()
         } else if self.expect(&[TokenType::If]).is_some() {
             self.parse_if_statement()
+        } else if self.expect(&[TokenType::While]).is_some() {
+            self.parse_while_statement()
         } else if self.expect(&[TokenType::Print]).is_some() {
             let expression = self.parse_expression()?;
             self.consume(&TokenType::Semicolon, "expected ';' after value")?;
@@ -153,6 +157,16 @@ impl Parser {
         })
     }
 
+    /// Parse the following rule:
+    /// ```
+    /// while := "while" condition statement
+    /// ```
+    fn parse_while_statement(&mut self) -> ParserResult<ast::StmtNode> {
+        let condition = self.parse_expression()?;
+        let body = Box::new(self.parse_statement()?);
+        Ok(ast::StmtNode::WhileStmt { condition, body })
+    }
+
     /// Parse the following rule:
     /// ```
     /// expression := assignment