Add MethodTypeParameter

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
index 710ef99..afa1681 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -5,6 +5,7 @@ import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Iterables;
 import org.zwobble.couscous.ast.*;
+import org.zwobble.couscous.types.MethodTypeParameter;
 import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.types.Type;
 
@@ -207,6 +208,7 @@ public class NodeTransformer {
     public MethodSignature transform(MethodSignature signature) {
         return new MethodSignature(
             transformMethodName(signature),
+            eagerMap(signature.getTypeParameters(), this::transform),
             eagerMap(signature.getArguments(), this::transform),
             transform(signature.getReturnType()));
     }
@@ -219,6 +221,10 @@ public class NodeTransformer {
         return (ScalarType)transformType.apply(type);
     }
 
+    public MethodTypeParameter transform(MethodTypeParameter type) {
+        return (MethodTypeParameter)transformType.apply(type);
+    }
+
     public String transformMethodName(MethodSignature signature) {
         return transformMethodName.apply(signature);
     }