From 96de72e5f761eb4bf07536f4dfe282e2e10a2081 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 27 Aug 2023 20:23:30 +0100 Subject: Add type-checking for while nodes --- .../shedlang/compiler/typechecker/TypeChecker.java | 25 ++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) (limited to 'src/main/java') diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java index 3eff92bf..651a4f79 100644 --- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java +++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java @@ -559,8 +559,8 @@ public class TypeChecker { case UntypedVarNode var -> typeCheckVar(var, context); - case UntypedWhileNode untypedWhileNode -> - throw new UnsupportedOperationException("TODO"); + case UntypedWhileNode while_ -> + typeCheckWhile(while_, context); }; } @@ -885,4 +885,25 @@ public class TypeChecker { } }; } + + private static TypeCheckerResult typeCheckWhile( + UntypedWhileNode untyped, + TypeCheckerContext context + ) { + var typedConditionResult = typeCheckExpression(untyped.condition(), context); + var typedCondition = typedConditionResult.value(); + context = typedConditionResult.context(); + + checkExpressionType(typedCondition, Types.BOOL, context); + + var typedBodyResult = typeCheckBlockNested(untyped.body(), context); + var typedBody = typedBodyResult.value(); + context = typedBodyResult.context(); + + return context.toResult(new TypedWhileNode( + typedCondition, + typedBody, + untyped.source() + )); + } } -- cgit v1.2.3