From cea82209e5aa312f4a513308c148d3e944004cb4 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Thu, 17 Mar 2016 22:33:15 +0000
Subject: Improve support for generics

---
 .../java/org/zwobble/couscous/ast/visitors/NodeTransformer.java  | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/visitors')

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);
     }
-- 
cgit v1.2.3