Add scope pointer to AST nodes

2 files changed, 4 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index 182a597..a16b0e6 100644
--- a/src/main.c
+++ b/src/main.c
@@ -86,7 +86,7 @@ analyzer_symbols(Analyzer *a, Node *node, Scope *scope) {
     if (!node) {
         return;
     }
-    assert(node);
+    node->scope = scope;
     switch (node->kind) {
         case NODE_FUN: {
             if (scope->depth != 0) {
@@ -104,6 +104,7 @@ analyzer_symbols(Analyzer *a, Node *node, Scope *scope) {
             }
 
             scope = scope_alloc(a, scope);
+            node->scope = scope;
             for (sz i = 0; i < array_size(node->func_params); i++) {
                 Node *param = node->func_params[i];
                 symmap_insert(&scope->symbols, param->param_name->value.str,
@@ -143,6 +144,7 @@ analyzer_symbols(Analyzer *a, Node *node, Scope *scope) {
             // case NODE_STRUCT_LIT:
         case NODE_BLOCK: {
             Scope *next = scope_alloc(a, scope);
+            node->scope = scope;
             for (sz i = 0; i < array_size(node->elements); i++) {
                 Node *expr = node->elements[i];
                 analyzer_symbols(a, expr, next);
diff --git a/src/parser.c b/src/parser.c
index 5928f7b..04692b3 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -192,6 +192,7 @@ typedef struct Node {
         };
     };
     bool is_ptr;
+    struct Scope* scope;
 } Node;
 
 //