Add Patio code gen for loop instructions

1 files changed, 16 insertions, 6 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
index 0fa95c91..948bfd6a 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
@@ -346,14 +346,24 @@ public class PatioCodeGen {
                     (local, c) -> c.temporaryPopIntoLocal(local, localStore.source())
                 );
 
-            case LoopBreakInstruction loopBreakInstruction ->
-                throw new UnsupportedOperationException("TODO");
+            case LoopBreakInstruction loopBreak ->
+                context.addInstruction(new PatioLoopBreak(loopBreak.source()));
 
-            case LoopContinueInstruction loopContinueInstruction ->
-                throw new UnsupportedOperationException("TODO");
+            case LoopContinueInstruction loopContinue ->
+                context.addInstruction(new PatioLoopContinue(loopContinue.source()));
 
-            case LoopInstruction loopInstruction ->
-                throw new UnsupportedOperationException("TODO");
+            case LoopInstruction loop -> {
+                context = context.enterBlock();
+                context = generateInstructions(loop.body(), context);
+
+                var bodyInstructionsResult = context.exitBlock();
+                var bodyInstructions = bodyInstructionsResult.value();
+                context = bodyInstructionsResult.context();
+
+                context = context.addInstruction(new PatioLoop(bodyInstructions, loop.source()));
+
+                yield context;
+            }
 
             case NamespaceFieldStoreInstruction namespaceFieldStore -> {
                 var fieldValueResult = context.temporaryPop();