AST - Added labels to while statements

2023-01-02 10:37:47 +01:00 · 2023-01-02 10:37:47 +01:00 · 2f44a81b6d
commit 2f44a81b6d
parent bf8eee7ecb
1 changed files with 16 additions and 2 deletions
--- a/src/ast.rs
+++ b/src/ast.rs
@ -27,6 +27,7 @@ pub enum StmtNode {
    },
    /// While loop statement.
    WhileStmt {
+        label: Option<Token>,
        condition: ExprNode,
        body: Box<StmtNode>,
    },
@ -124,8 +125,21 @@ impl AstDumper for StmtNode {
                ),
            },

-            Self::WhileStmt { condition, body } => {
-                format!("( while {} {} )", condition.dump(), body.dump())
+            Self::WhileStmt {
+                label,
+                condition,
+                body,
+            } => {
+                format!(
+                    "( {}while {} {} )",
+                    if let Some(label) = label {
+                        &format!("@{} ", label.lexeme)
+                    } else {
+                        ""
+                    },
+                    condition.dump(),
+                    body.dump()
+                )
            }

            Self::LoopControlStmt {