Parse type level params for sum types

1 files changed, 9 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index 9fb8a93b..753a6d75 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -968,6 +968,8 @@ public class Parser {
 
         var name = tokens.nextValue(TokenType.IDENTIFIER);
 
+        var typeLevelParams = parseTypeLevelParamsOptional(tokens);
+
         tokens.skip(TokenType.SYMBOL_EQUALS);
 
         var variants = parseOneOrMoreNoTrailingSeparator(
@@ -980,7 +982,13 @@ public class Parser {
         var end = tokens.position();
         var source = start.to(end);
 
-        return new UntypedSumTypeDefinitionNode(name, visibility, variants, source);
+        return new UntypedSumTypeDefinitionNode(
+            name,
+            visibility,
+            typeLevelParams,
+            variants,
+            source
+        );
     }
 
     static UntypedExpressionNode parseTopLevelExpression(TokenIterator<TokenType> tokens) {