Ensure lambdas wrap their env tables

1 files changed, 6 insertions, 5 deletions

diff --git a/src/parser.h b/src/parser.h
index 01f53b0..fa8858e 100755
--- a/src/parser.h
+++ b/src/parser.h
@@ -4,6 +4,11 @@
 #include "lexer.h"
 #include "hashtable.h"
 
+typedef struct Environment {
+    HashTable *table;
+    struct Environment *parent;
+} Environment;
+
 typedef enum ObjectType {
     OBJ_TYPE_NIL,
     OBJ_TYPE_TRUE,
@@ -38,6 +43,7 @@ typedef struct Object {
          struct {
              struct Object **params;
              struct Object **body;
+             Environment *env;
          };
 
          // OBJ_TYPE_IF
@@ -59,11 +65,6 @@ typedef struct Object {
     size_t col;
 } Object;
 
-typedef struct Environment {
-    HashTable *table;
-    struct Environment *parent;
-} Environment;
-
 typedef struct Parser {
     Token *tokens;
     size_t current;