diff options
Diffstat (limited to 'src/main/java')
| -rw-r--r-- | src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java index aaf49ba0..72404d42 100644 --- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java +++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java @@ -1779,15 +1779,19 @@ public class TypeChecker { }; for (var untypedMethodNode : untyped.methods()) { - context = context.enterBlock(P.map( - Map.entry( - "This", - Variable.defined( - Types.metaType(thisType), - VariableType.VAL - ) - ) - )); + var environment = environmentForTypeOrConstructor(typeOrConstructor); + // TODO: push This into environmentForTypeOrConstructor? + // Probably not (confusing for sum types), but might still + // be useful for method definitions. + environment = environment.add( + "This", + Variable.defined( + Types.metaType(thisType), + VariableType.VAL + ), + untyped.source() + ); + context = context.enterBlock(environment.build()); var typeCheckSignatureResult = typeCheckFunctionType( untypedMethodNode.signature(), |