Parse type-level params for function type

2 files changed, 30 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedFunctionTypeNode.java b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedFunctionTypeNode.java
index b4a465b3..c1b7ad3d 100644
--- a/src/main/java/org/shedlang/compiler/ast/untyped/UntypedFunctionTypeNode.java
+++ b/src/main/java/org/shedlang/compiler/ast/untyped/UntypedFunctionTypeNode.java
@@ -5,9 +5,11 @@ import org.shedlang.compiler.sources.Source;
 import org.shedlang.compiler.sources.Sources;
 import org.shedlang.compiler.util.P;
 
+import java.util.List;
 import java.util.Optional;
 
 public record UntypedFunctionTypeNode(
+    Optional<List<UntypedTypeLevelParamNode>> typeLevelParams,
     UntypedFunctionTypeParamsNode params,
     Optional<UntypedTypeLevelExpressionNode> effect,
     UntypedTypeLevelExpressionNode returnType,
@@ -15,6 +17,7 @@ public record UntypedFunctionTypeNode(
 ) implements UntypedTypeLevelExpressionNode {
     public static Builder builder() {
         return new Builder(
+            Optional.empty(),
             P.vector(),
             P.vector(),
             Optional.empty(),
@@ -23,6 +26,7 @@ public record UntypedFunctionTypeNode(
     }
 
     public record Builder(
+        Optional<List<UntypedTypeLevelParamNode>> typeLevelParams,
         PVector<UntypedTypeLevelExpressionNode> positionalParams,
         PVector<UntypedFunctionTypeNamedParamNode> namedParams,
         Optional<UntypedTypeLevelExpressionNode> effect,
@@ -30,6 +34,7 @@ public record UntypedFunctionTypeNode(
     ) {
         public UntypedFunctionTypeNode build() {
             return new UntypedFunctionTypeNode(
+                this.typeLevelParams,
                 new UntypedFunctionTypeParamsNode(this.positionalParams, this.namedParams, Sources.NULL),
                 this.effect,
                 this.returnType,
@@ -37,8 +42,19 @@ public record UntypedFunctionTypeNode(
             );
         }
 
+        public Builder withTypeLevelParams(List<UntypedTypeLevelParamNode> typeLevelParams) {
+            return new Builder(
+                Optional.of(typeLevelParams),
+                this.positionalParams,
+                this.namedParams,
+                this.effect,
+                this.returnType
+            );
+        }
+
         public Builder addPositionalParam(UntypedTypeLevelExpressionNode param) {
             return new Builder(
+                this.typeLevelParams,
                 this.positionalParams.plus(param),
                 this.namedParams,
                 this.effect,
@@ -48,6 +64,7 @@ public record UntypedFunctionTypeNode(
 
         public Builder addNamedParam(String name, UntypedTypeLevelExpressionNode type) {
             return new Builder(
+                this.typeLevelParams,
                 this.positionalParams,
                 this.namedParams.plus(new UntypedFunctionTypeNamedParamNode(name, type, Sources.NULL)),
                 this.effect,
@@ -57,6 +74,7 @@ public record UntypedFunctionTypeNode(
 
         public Builder withEffect(UntypedTypeLevelExpressionNode effect) {
             return new Builder(
+                this.typeLevelParams,
                 this.positionalParams,
                 this.namedParams,
                 Optional.of(effect),
@@ -66,6 +84,7 @@ public record UntypedFunctionTypeNode(
 
         public Builder withReturnType(UntypedTypeLevelExpressionNode returnType) {
             return new Builder(
+                this.typeLevelParams,
                 this.positionalParams,
                 this.namedParams,
                 this.effect,
diff --git a/src/main/java/org/shedlang/compiler/parser/Parser.java b/src/main/java/org/shedlang/compiler/parser/Parser.java
index 003d909c..8a55de2d 100644
--- a/src/main/java/org/shedlang/compiler/parser/Parser.java
+++ b/src/main/java/org/shedlang/compiler/parser/Parser.java
@@ -338,7 +338,13 @@ public class Parser {
         var end = tokens.endPosition();
         var source = new FileFragmentSource(start.to(end));
 
-        return new UntypedFunctionTypeNode(params, Optional.empty(), returnType, source);
+        return new UntypedFunctionTypeNode(
+            Optional.empty(),
+            params,
+            Optional.empty(),
+            returnType,
+            source
+        );
     }
 
     private static final Map<TokenType, OperationParser<OperatorPrecedence, UntypedExpressionNode>> OPERATION_PARSERS = Stream.of(
@@ -579,6 +585,7 @@ public class Parser {
         var source = new FileFragmentSource(start.to(end));
 
         return new UntypedFunctionTypeNode(
+            signature.typeLevelParams(),
             signature.params(),
             signature.effect(),
             signature.returnType(),
@@ -591,6 +598,8 @@ public class Parser {
     ) {
         var start = tokens.startPosition();
 
+        var typeLevelParams = parseTypeLevelParamsOptional(tokens);
+
         tokens.skip(TokenType.SYMBOL_PAREN_OPEN);
         var params = parseParamTypes(tokens);
         tokens.skip(TokenType.SYMBOL_PAREN_CLOSE);
@@ -605,6 +614,7 @@ public class Parser {
         var source = new FileFragmentSource(start.to(end));
 
         return new UntypedFunctionTypeNode(
+            typeLevelParams,
             params,
             effect,
             returnType,