From d6b4e028462f24861b8386dbd919f9ef977d9f01 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edvard=20Th=C3=B6rnros?= <edvard.thornros@gmail.com>
Date: Mon, 8 Mar 2021 23:11:32 +0100
Subject: return is implicit void

---
 src/compiler.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/compiler.rs b/src/compiler.rs
index 4c59d86..bdc5901 100644
--- a/src/compiler.rs
+++ b/src/compiler.rs
@@ -1752,7 +1752,13 @@ impl Compiler {
 
             (Token::Ret, ..) => {
                 self.eat();
-                self.expression(block);
+                if self.peek() == Token::Newline {
+                    self.eat();
+                    let nil = self.add_constant(Value::Nil);
+                    add_op(self, block, Op::Constant(nil));
+                } else {
+                    self.expression(block);
+                }
                 add_op(self, block, Op::Return);
             }
 
-- 
cgit v1.2.1