fix: fixed an issue that prevented capturing variables (when creating…

… a closure) declared inside a module-level loop
ArgonLang · Sep 20, 2023 · 9d74218 · 9d74218
1 parent 4bf908a
commit 9d74218
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/argon/lang/translation_unit.cpp b/argon/lang/translation_unit.cpp
@@ -49,9 +49,12 @@ bool TranslationUnit::IsFreeVar(String *id) const {
     // in turn then this is a free variable
     SymbolT *sym;
 
-    for (TranslationUnit *tu = this->prev; tu != nullptr && tu->symt->type == SymbolType::FUNC; tu = tu->prev) {
+    for (TranslationUnit *tu = this->prev; tu != nullptr; tu = tu->prev) {
         if ((sym = SymbolLookup(tu->symt, id)) != nullptr) {
-            if (sym->declared || sym->free) {
+            // WARNING: sym->nested must be greater than 0,
+            // otherwise this is a global variable.
+            if (sym->type == SymbolType::VARIABLE &&
+                sym->nested > 0 && (sym->declared || sym->free)) {
                 Release(sym);
 
                 return true;