Compile return statements to stack IR

1 files changed, 8 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
index aa0a4795..9cd01151 100644
--- a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
+++ b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
@@ -98,6 +98,9 @@ public class StackCompiler {
             case TypedExpressionStatementNode expressionStatement ->
                 compileExpressionStatement(expressionStatement);
 
+            case TypedReturnNode return_ ->
+                compileReturn(return_);
+
             case TypedValNode val ->
                 compileVal(val);
         };
@@ -146,6 +149,11 @@ public class StackCompiler {
         return P.vector(new LocalLoadInstruction(node.name(), node.source()));
     }
 
+    private static PVector<Instruction> compileReturn(TypedReturnNode node) {
+        return compileExpression(node.expression())
+            .plus(new ReturnInstruction(node.source()));
+    }
+
     private static PVector<Instruction> compileStringLiteral(TypedStringLiteralNode node) {
         return P.vector(new PushStringInstruction(node.value(), node.source()));
     }