From 0dbf155da52c5db1cab2688b04c87386a55c5d5b Mon Sep 17 00:00:00 2001
From: Tom Willemse
Date: Mon, 8 Feb 2021 17:42:18 -0800
Subject: Add return statement

---
 src/com/craftinginterpreters/lox/Parser.java | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/com/craftinginterpreters/lox/Parser.java')

diff --git a/src/com/craftinginterpreters/lox/Parser.java b/src/com/craftinginterpreters/lox/Parser.java
index c427bc4..528326c 100644
--- a/src/com/craftinginterpreters/lox/Parser.java
+++ b/src/com/craftinginterpreters/lox/Parser.java
@@ -52,6 +52,8 @@ class Parser {
             return ifStatement();
         if (match(PRINT))
             return printStatement();
+        if (match(RETURN))
+            return returnStatement();
         if (match(WHILE))
             return whileStatement();
         if (match(LEFT_BRACE))
@@ -120,6 +122,17 @@ class Parser {
         return new Stmt.Print(value);
     }
 
+    private Stmt returnStatement() {
+        Token keyword = previous();
+        Expr value = null;
+        if (!check(SEMICOLON)) {
+            value = expression();
+        }
+
+        consume(SEMICOLON, "Expect ';' after return value.");
+        return new Stmt.Return(keyword, value);
+    }
+
     private Stmt varDeclaration() {
         Token name = consume(IDENTIFIER, "Expect variable name.");
 
-- 
cgit v1.2.3-54-g00ecf