Fixed an environment related issue that causes class inheritance not …

…function
theSalted · Jul 22, 2024 · 9896b1e · 9896b1e
1 parent 182f3a3
commit 9896b1e
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/Sources/Lox/Interpreter.swift b/Sources/Lox/Interpreter.swift
@@ -126,7 +126,20 @@ public final class Interpreter: StatementVisitor, ExpressionVisitor {
 
         // Class assembly
         let `class` = LoxClass(stmt.name.lexeme, superclass: superclass, methods: methods)
-        do { try environment.assign(name: stmt.name, value: `class`) }
+        do {
+            try environment.assign(name: stmt.name, value: `class`)
+
+            if superclass != nil {
+                guard let enclosing = environment.enclosing else {
+                    throw InterpreterError.runtime( /*This should never happen*/
+                        message: "Something went wrong while declaring your class with proper environment",
+                        onLine: stmt.name.line,
+                        locationDescription: nil)
+                }
+
+                environment = enclosing
+            }
+        }
         catch { return .failure(
             error as? InterpreterError ??
             InterpreterError.runtime( /*This should never happen*/