fixed: compilation

Author: brendanfh

compiler/src/astnodes.c

@@ -1970,6 +1970,7 @@ AstCall * create_implicit_for_expansion_call(Context *context, AstFor *fornode)
     body_code_block->token = fornode->token;
     body_code_block->type_node = context->builtins.code_type;
     body_code_block->code = (AstNode *) fornode->stmt;
+    body_code_block->enclosing_scope = context->checker.current_scope;
     ((AstBlock *) body_code_block->code)->rules = Block_Rule_Code_Block;
 
     bh_arr_new(context->ast_alloc, body_code_block->binding_symbols, bh_arr_length(fornode->indexing_variables));
@@ -1986,13 +1987,11 @@ AstCall * create_implicit_for_expansion_call(Context *context, AstFor *fornode)
 
     AstNumLit *flag_node = make_int_literal(context, flags);
     flag_node->type_node = context->builtins.for_expansion_flag_type;
-    // flag_node->type = type_build_from_ast(context, context->builtins.for_expansion_flag_type);
-    // assert(flag_node->type);
 
     // Arguments are: 
     //    Iterator
-    //    Code block with 2 inputs (value, index)
     //    Flags
+    //    Code block with 2 inputs (value, index)
     bh_arr_push(call->args.values, (AstTyped *) make_argument(context, (AstTyped *) fornode->iter));
     bh_arr_push(call->args.values, (AstTyped *) make_argument(context, (AstTyped *) flag_node));
     bh_arr_push(call->args.values, (AstTyped *) make_argument(context, (AstTyped *) body_code_block));

compiler/src/checker.c

@@ -3128,16 +3128,22 @@ CHECK_FUNC(expression, AstTyped** pexpr) {
             retval = check_insert_directive(context, (AstDirectiveInsert **) pexpr, 1);
             break;
 
-        case Ast_Kind_Code_Block:
-            expr->flags |= Ast_Flag_Comptime;
-            ((AstCodeBlock *) expr)->enclosing_scope = context->checker.current_scope;
+        case Ast_Kind_Code_Block: {
+            AstCodeBlock *block = (void *) expr;
+
             fill_in_type(context, expr);
-            bh_arr_each(CodeBlockBindingSymbol, sym, ((AstCodeBlock *) expr)->binding_symbols) {
+            block->flags |= Ast_Flag_Comptime;
+
+            if (!block->enclosing_scope)
+                block->enclosing_scope = context->checker.current_scope;
+
+            bh_arr_each(CodeBlockBindingSymbol, sym, block->binding_symbols) {
                 if (sym->type_node) {
                     CHECK(expression, (AstTyped **) &sym->type_node);
                 }
             }
             break;
+        }
 
         case Ast_Kind_Do_Block: {
             Scope* old_current_scope = context->checker.current_scope;

compiler/src/clone.c

@@ -118,6 +118,7 @@ static inline i32 ast_kind_to_size(AstNode* node) {
         case Ast_Kind_Union_Type: return sizeof(AstUnionType);
         case Ast_Kind_Union_Variant: return sizeof(AstUnionVariant);
         case Ast_Kind_Procedural_Expansion: return sizeof(AstProceduralExpansion);
+        case Ast_Kind_Code_Block: return sizeof(AstCodeBlock);
 
         default: break;
     }