Allow returns without an explicit value.

src/wren_compiler.c

@@ -1828,7 +1828,6 @@ void namedSignature(Compiler* compiler, char* name, int* length)
   if (match(compiler, TOKEN_EQ))
   {
     // It's a setter.
-    // TODO: Allow setters with parameters? Like: foo.bar(1, 2) = "blah"
     name[(*length)++] = '=';
     name[(*length)++] = ' ';
 
@@ -2326,9 +2325,15 @@ void statement(Compiler* compiler)
   if (match(compiler, TOKEN_RETURN))
   {
     // Compile the return value.
-    // TODO: Implicitly return null if there is a newline or } after the
-    // "return".
-    expression(compiler);
+    if (peek(compiler) == TOKEN_LINE || peek(compiler) == TOKEN_RIGHT_BRACE)
+    {
+      // Implicitly return null if there is no value.
+      emit(compiler, CODE_NULL);
+    }
+    else
+    {
+      expression(compiler);
+    }
 
     emit(compiler, CODE_RETURN);
     return;

src/wren_core.c

@@ -185,7 +185,7 @@ DEF_NATIVE(fiber_create)
 
   ObjFiber* newFiber = wrenNewFiber(vm, AS_OBJ(args[1]));
 
-  // The compiler expect the first slot of a function to hold the receiver.
+  // The compiler expects the first slot of a function to hold the receiver.
   // Since a fiber's stack is invoked directly, it doesn't have one, so put it
   // in here.
   // TODO: Is there a cleaner solution?

test/return/in_function.wren

@@ -1 +1,6 @@
-var a = "ok"; IO.print(a) // expect: ok
+var f = fn {
+  return "ok"
+  IO.print("bad")
+}
+
+IO.print(f.call) // expect: ok

test/return/return_null_if_brace.wren

@@ -0,0 +1,6 @@
+var f = fn {
+  if (true) { return }
+  IO.print("bad")
+}
+
+IO.print(f.call) // expect: null

test/return/return_null_if_newline.wren

@@ -0,0 +1,6 @@
+var f = fn {
+  return
+  IO.print("bad")
+}
+
+IO.print(f.call) // expect: null