implement return expressions

1 files changed, 2 insertions, 1 deletions

diff --git a/src/Lexer.zig b/src/Lexer.zig
index 64457bf..9be99aa 100644
--- a/src/Lexer.zig
+++ b/src/Lexer.zig
@@ -29,6 +29,7 @@ pub const Token = struct {
         @"else",
         @"while",
         proc,
+        @"return",
     };
 };
 
@@ -144,7 +145,7 @@ fn identifierOrKeyword(self: *Self) Token {
         }
         const value = self.source[self.start..self.pos];
         return self.create(switch (std.meta.stringToEnum(Token.Type, value) orelse .invalid) {
-            .@"if", .@"else", .@"while", .proc => |t| t,
+            .@"if", .@"else", .@"while", .proc, .@"return" => |t| t,
             else => .identifier,
         });
     }