microscript: syntax sugar for one-liner functions

1 files changed, 8 insertions, 4 deletions

diff --git a/uscript/parser.c b/uscript/parser.c
index dc721d5..f90e6d0 100644
--- a/uscript/parser.c
+++ b/uscript/parser.c
@@ -739,13 +739,17 @@ void fun_stat(struct parser *p, bool is_global)
                 expect(p, ')', "expected ')' after arguments");
         }
 
-        while (p->cur.kind != TOKEN_END && p->cur.kind != TOKEN_EOF)
-                stat(p);
+        if (consume(p, ':')) {
+                expr(p);
+                parser_add_byte(p, BC_RET);
+        } else {
+                while (p->cur.kind != TOKEN_END && p->cur.kind != TOKEN_EOF)
+                        stat(p);
+                expect(p, TOKEN_END, "unterminated function");
+        }
 
         end_function(p);
 
-        expect(p, TOKEN_END, "unterminated function");
-
         if (global_idx != -1) {
                 parser_add_byte(p, BC_SET_GLOBAL);
                 parser_add_byte(p, (global_idx >> 8) & 0xFF);