Add in the yield keyword

1 files changed, 7 insertions, 1 deletions

diff --git a/src/compiler.rs b/src/compiler.rs
index 92bf3fc..b7fdce1 100644
--- a/src/compiler.rs
+++ b/src/compiler.rs
@@ -473,7 +473,8 @@ impl Compiler {
         let mut function_block = Block::new(&name, &self.current_file, self.line());
 
         let block_id = self.blocks.len();
-        self.blocks.push(Rc::new(RefCell::new(Block::new(&name, &self.current_file, self.line()))));
+        let new_block = Block::new(&name, &self.current_file, self.line());
+        self.blocks.push(Rc::new(RefCell::new(new_block)));
 
         let _ret = push_frame!(self, function_block, {
             loop {
@@ -888,6 +889,11 @@ impl Compiler {
                 }
             }
 
+            (Token::Yield, ..) => {
+                self.eat();
+                block.add(Op::Yield, self.line());
+            }
+
             (Token::Identifier(name), Token::ColonEqual, ..) => {
                 self.eat();
                 self.eat();