Type check expression statements

1 files changed, 20 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index 11f902b1..f707dac0 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -138,6 +138,26 @@ public class TypeChecker {
         };
     }
 
+    private static TypedFunctionStatementNode typeCheckExpressionStatement(
+        UntypedExpressionStatementNode untyped,
+        TypeCheckerContext context
+    ) {
+        return new TypedExpressionStatementNode(
+            typeCheckExpression(untyped.expression(), context),
+            untyped.source()
+        );
+    }
+
+    public static TypedFunctionStatementNode typeCheckFunctionStatement(
+        UntypedExpressionStatementNode untyped,
+        TypeCheckerContext context
+    ) {
+        return switch (untyped) {
+            case UntypedExpressionStatementNode expressionStatement ->
+                typeCheckExpressionStatement(expressionStatement, context);
+        };
+    }
+
     private static TypedExpressionNode typeCheckIntLiteral(
         UntypedIntLiteralNode untyped
     ) {