Check number of type-level arguments

2 files changed, 34 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
index c51758bb..d9a52038 100644
--- a/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
+++ b/src/main/java/org/shedlang/compiler/typechecker/TypeChecker.java
@@ -385,6 +385,7 @@ public class TypeChecker {
                     var typeLevelArgs = typeCheckTypeLevelArgs(
                         untypedTypeLevelArgNodes,
                         typeConstructor.params(),
+                        untyped.source(),
                         contextForReceiver
                     );
                     specificReceiverType = Types.metaType(
@@ -397,6 +398,7 @@ public class TypeChecker {
                     var typeLevelArgs = typeCheckTypeLevelArgs(
                         untypedTypeLevelArgNodes,
                         initialReceiverType.typeLevelParams().get(),
+                        untyped.source(),
                         contextForReceiver
                     );
                     specificReceiverType = initialReceiverType.applyTypeLevelArgs(typeLevelArgs);
@@ -1883,11 +1885,19 @@ public class TypeChecker {
     private static ArrayList<TypeLevelValue> typeCheckTypeLevelArgs(
         List<UntypedTypeLevelExpressionNode> untyped,
         List<TypeLevelParam> typeLevelParams,
+        Source source,
         TypeCheckerContext context
     ) {
+        if (untyped.size() != typeLevelParams.size()) {
+            throw new WrongNumberOfTypeLevelArgsError(
+                typeLevelParams.size(),
+                untyped.size(),
+                source
+            );
+        }
+
         var typeLevelArgs = new ArrayList<TypeLevelValue>();
 
-        // TODO: check length
         // TODO: check arg is appropriate for param
 
         for (var paramIndex = 0; paramIndex < typeLevelParams.size(); paramIndex++) {
diff --git a/src/main/java/org/shedlang/compiler/typechecker/errors/WrongNumberOfTypeLevelArgsError.java b/src/main/java/org/shedlang/compiler/typechecker/errors/WrongNumberOfTypeLevelArgsError.java
new file mode 100644
index 00000000..9af86c37
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/typechecker/errors/WrongNumberOfTypeLevelArgsError.java
@@ -0,0 +1,23 @@
+package org.shedlang.compiler.typechecker.errors;
+
+import org.shedlang.compiler.errors.SourceError;
+import org.shedlang.compiler.sources.Source;
+
+public class WrongNumberOfTypeLevelArgsError extends SourceError {
+    private final int expected;
+    private final int actual;
+
+    public WrongNumberOfTypeLevelArgsError(int expected, int actual, Source source) {
+        super("Expected " + expected + " type-level arguments but got " + actual, source);
+        this.expected = expected;
+        this.actual = actual;
+    }
+
+    public int expected() {
+        return expected;
+    }
+
+    public int actual() {
+        return actual;
+    }
+}