From b0d81e8b2354ffb0efba0f395b4aa4e114fd1579 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 1 Oct 2023 19:46:22 +0100
Subject: Check when expression is instance of sum type

---
 .../org/shedlang/compiler/typechecker/TypeChecker.java |  7 +++++++
 .../WhenExpressionIsNotInstanceOfSumTypeError.java     | 18 ++++++++++++++++++
 2 files changed, 25 insertions(+)
 create mode 100644 src/main/java/org/shedlang/compiler/typechecker/errors/WhenExpressionIsNotInstanceOfSumTypeError.java

(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 58396e80..3b4df564 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -1139,6 +1139,13 @@ public class TypeChecker {
         var typedExpressionResult = typeCheckExpression(untyped.expression(), context);
         var typedExpression = typedExpressionResult.value();
         context = typedExpressionResult.context();
+
+        if (!(typedExpression.type() instanceof SumType)) {
+            throw new WhenExpressionIsNotInstanceOfSumTypeError(
+                typedExpression.type(),
+                typedExpression.source()
+            );
+        }
         
         var typedCasesResult = context.mapWithContext(
             untyped.cases(),
diff --git a/src/main/java/org/shedlang/compiler/typechecker/errors/WhenExpressionIsNotInstanceOfSumTypeError.java b/src/main/java/org/shedlang/compiler/typechecker/errors/WhenExpressionIsNotInstanceOfSumTypeError.java
new file mode 100644
index 00000000..1583c8ae
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/typechecker/errors/WhenExpressionIsNotInstanceOfSumTypeError.java
@@ -0,0 +1,18 @@
+package org.shedlang.compiler.typechecker.errors;
+
+import org.shedlang.compiler.errors.SourceError;
+import org.shedlang.compiler.sources.Source;
+import org.shedlang.compiler.types.Type;
+
+public class WhenExpressionIsNotInstanceOfSumTypeError extends SourceError {
+    private final Type actualType;
+
+    public WhenExpressionIsNotInstanceOfSumTypeError(Type actualType, Source source) {
+        super("when expression must be an instance of a sum type, but was " + actualType.describe(), source);
+        this.actualType = actualType;
+    }
+
+    public Type actualType() {
+        return actualType;
+    }
+}
-- 
cgit v1.2.3