From 4c54643a8df8cefac4afc573ec753278af8a3301 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Emmanuel=20Beno=C3=AEt?= <tseeker@nocternity.net>
Date: Mon, 2 Jan 2023 12:10:37 +0100
Subject: [PATCH] Parser - Use new loop AST node

---
 src/parser.rs | 23 ++++++++---------------
 1 file changed, 8 insertions(+), 15 deletions(-)

diff --git a/src/parser.rs b/src/parser.rs
index c6a8332..7460a8c 100644
--- a/src/parser.rs
+++ b/src/parser.rs
@@ -250,10 +250,11 @@ impl Parser {
             self.loop_state.pop();
             result?
         });
-        Ok(ast::StmtNode::WhileStmt {
+        Ok(ast::StmtNode::LoopStmt {
             label,
             condition,
             body,
+            after_body: None,
         })
     }
 
@@ -309,22 +310,14 @@ impl Parser {
             self.loop_state.pop();
             result?
         };
-        let body_with_incr = if let Some(incr) = increment {
-            let incr_stmt = Box::new(ast::StmtNode::Expression(incr));
-            let body_block = if let ast::StmtNode::Block(mut body_block) = body_stmt {
-                body_block.push(incr_stmt);
-                body_block
-            } else {
-                vec![Box::new(body_stmt), incr_stmt]
-            };
-            ast::StmtNode::Block(body_block)
-        } else {
-            body_stmt
-        };
-        let while_stmt = ast::StmtNode::WhileStmt {
+        let while_stmt = ast::StmtNode::LoopStmt {
             label,
             condition,
-            body: Box::new(body_with_incr),
+            body: Box::new(body_stmt),
+            after_body: match increment {
+                Some(incr) => Some(Box::new(ast::StmtNode::Expression(incr))),
+                None => None,
+            },
         };
         if let Some(init_stmt) = initializer {
             Ok(ast::StmtNode::Block(vec![