From 81c613a8365bc414219559cd251a3ff3947aa423 Mon Sep 17 00:00:00 2001
From: Andrew Morris <voltrevo@gmail.com>
Date: Wed, 31 May 2023 14:57:21 +1000
Subject: [PATCH] Assemble generator functions

---
 valuescript_compiler/src/assembler.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/valuescript_compiler/src/assembler.rs b/valuescript_compiler/src/assembler.rs
index 64e67ee..059300b 100644
--- a/valuescript_compiler/src/assembler.rs
+++ b/valuescript_compiler/src/assembler.rs
@@ -68,7 +68,10 @@ impl Assembler {
   }
 
   fn function(&mut self, function: &Function) {
-    self.output.push(ValueType::Function as u8);
+    self.output.push(match function.is_generator {
+      false => ValueType::Function,
+      true => ValueType::GeneratorFunction,
+    } as u8);
 
     self.fn_data = Default::default();
 
@@ -421,6 +424,7 @@ enum ValueType {
   Class = 0x11,
   Lazy = 0x12,
   BigInt = 0x13,
+  GeneratorFunction = 0x14,
 }
 
 #[derive(Hash, PartialEq, Eq, Clone)]