Fix depth resolution on recursive calls

1 files changed, 20 insertions, 17 deletions

diff --git a/src/bytecode/compiler.h b/src/bytecode/compiler.h
index 27ff312..124c345 100755
--- a/src/bytecode/compiler.h
+++ b/src/bytecode/compiler.h
@@ -608,26 +608,29 @@ parse_tree(Chunk *chunk, Compiler *compiler) {
             return;
         } break;
         case TOKEN_SYMBOL: {
-            size_t depth = compiler->scope_depth - 1;
-            ssize_t idx = -1;
-            do {
-                Scope *scope = &compiler->scopes[depth];
-                idx = find_local_index(scope, tok);
+            if (compiler->scope_depth > 1) {
+                size_t depth = compiler->scope_depth - 1;
+                ssize_t idx = -1;
+                do {
+                    Scope *scope = &compiler->scopes[depth];
+                    idx = find_local_index(scope, tok);
+                    if (idx >= 0) {
+                        break;
+                    }
+                    depth--;
+                } while (depth > 0);
+
                 if (idx >= 0) {
-                    break;
+                    emit_constant(chunk, tok, FIXNUM_VAL(depth));
+                    emit_constant(chunk, tok, FIXNUM_VAL(idx));
+                    add_code(chunk, OP_LOCAL, tok.line, tok.column);
+                    return;
                 }
-                depth--;
-            } while (depth > 0);
-
-            if (idx >= 0) {
-                emit_constant(chunk, tok, FIXNUM_VAL(depth));
-                emit_constant(chunk, tok, FIXNUM_VAL(idx));
-                add_code(chunk, OP_LOCAL, tok.line, tok.column);
-            } else {
-                Object obj = make_symbol(tok.value);
-                emit_constant(chunk, tok, obj);
-                add_code(chunk, OP_GET, tok.line, tok.column);
             }
+
+            Object obj = make_symbol(tok.value);
+            emit_constant(chunk, tok, obj);
+            add_code(chunk, OP_GET, tok.line, tok.column);
             return;
         } break;
         case TOKEN_NIL: {