From 19fb6b2d97a970f94854cd17a58639b72d35c052 Mon Sep 17 00:00:00 2001
From: Bad Diode <bd@badd10de.dev>
Date: Tue, 26 Oct 2021 10:00:49 +0200
Subject: Ensure the last statement is returned to stdout

---
 src/bytecode/vm.h | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/bytecode/vm.h')

diff --git a/src/bytecode/vm.h b/src/bytecode/vm.h
index d654e2c..0ce6dec 100755
--- a/src/bytecode/vm.h
+++ b/src/bytecode/vm.h
@@ -263,6 +263,11 @@ vm_interpret(VM *vm) {
             } break;
             case OP_RETURN: {
                 if (frame->rp == NULL) {
+                    Object ret = array_pop(vm->stack);
+                    if (!IS_NIL(ret)) {
+                        object_display(ret);
+                        printf("\n");
+                    }
                     return;
                 }
                 vm->pc = frame->rp;
-- 
cgit v1.2.1