Implement return statements at root level

src/abstract_tree/mod.rs

@@ -60,7 +60,11 @@ impl AbstractTree for Root {
         let mut value = Value::none();
 
         for statement in &self.statements {
-            value = statement.run(source, context)?;
+            if let Statement::Return(inner_statement) = statement {
+                return inner_statement.run(source, context);
+            } else {
+                value = statement.run(source, context)?;
+            }
         }
 
         Ok(value)

tests/interpret.rs

@@ -476,4 +476,18 @@ mod blocks {
             Ok(Value::Integer(1))
         );
     }
+
+    #[test]
+    fn root_returns_like_block() {
+        assert_eq!(
+            interpret(
+                "
+                return 1
+                1 + 1
+                3
+                "
+            ),
+            Ok(Value::Integer(1))
+        );
+    }
 }