add while loops

1 files changed, 28 insertions, 0 deletions

diff --git a/src/compile.zig b/src/compile.zig
index c0ba77d..4d39293 100644
--- a/src/compile.zig
+++ b/src/compile.zig
@@ -336,6 +336,34 @@ const CompileContext = struct {
                 const val = try self.compileExpr(assign_var.value);
                 try self.assignLocalVar(local, val);
             },
+            .@"while" => |@"while"| {
+                const curr = self.current_block;
+                const cond_block = try self.switchToNewBlock();
+                self.current_block = curr;
+                try self.addInstr(.{
+                    .loc = stmt.loc,
+                    .type = .{ .jump = .{ .to = cond_block } },
+                });
+                const do = try self.switchToNewBlock();
+                const after = try self.switchToNewBlock();
+                self.current_block = cond_block;
+                const cond = try self.compileExpr(@"while".cond);
+                try self.addInstr(.{
+                    .loc = stmt.loc,
+                    .type = .{ .branch = .{
+                        .cond = cond,
+                        .true = .{ .to = do },
+                        .false = .{ .to = after },
+                    } },
+                });
+                self.current_block = do;
+                try self.compileBlock(@"while".do);
+                try self.addInstr(.{
+                    .loc = stmt.loc,
+                    .type = .{ .jump = .{ .to = cond_block } },
+                });
+                self.current_block = after;
+            },
         }
     }