C style for loop AST gen

parser/src/ast.rs

@@ -108,7 +108,7 @@ pub enum Statement {
     While(StatementBranch),
     CStyleFor {
         init: Box<Statement>,
-        condition: Box<Expression>,
+        condition: Expression,
         update: Box<Statement>,
         body: Box<Statement>,
     },

parser/src/lib.rs

@@ -333,6 +333,13 @@ impl From<pest::iterators::Pair<'_, Rule>> for Statement {
 
             Rule::while_stmt => Statement::While(pair.into()),
 
+            Rule::c_for_stmt => Statement::CStyleFor {
+                init: Box::new(Statement::from(inner.next().unwrap())),
+                condition: inner.next().unwrap().into(),
+                update: Box::new(Statement::from(inner.next().unwrap())),
+                body: Box::new(Statement::from(inner.next().unwrap())),
+            },
+
             Rule::assign_statement => Statement::VarAssign(VarAssignStmt {
                 target: Expression::from(inner.next().unwrap()),
                 value: Expression::from(inner.next().unwrap()),