Improve support for generics

1 files changed, 9 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 38c5709..0998564 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -13,6 +13,8 @@ import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
 
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
+
 public class NodeTransformer {
     public static NodeTransformer replaceExpressions(Map<ExpressionNode, ExpressionNode> replacements) {
         return builder().
@@ -197,6 +199,13 @@ public class NodeTransformer {
         this.transformMethodName = transformMethodName;
     }
 
+    public MethodSignature transform(MethodSignature signature) {
+        return new MethodSignature(
+            transformMethodName(signature),
+            eagerMap(signature.getArguments(), this::transform),
+            transform(signature.getReturnType()));
+    }
+
     public Type transform(Type type) {
         return transformType.apply(type);
     }