Simplify generation of anonymous classes

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index 3e97a36..5763b11 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -4,6 +4,7 @@ import com.google.common.collect.ImmutableList;
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 
 import java.util.List;
+import java.util.function.Function;
 
 public class MethodNode implements CallableNode {
     public static Builder staticMethod(String name) {
@@ -115,6 +116,10 @@ public class MethodNode implements CallableNode {
         return body;
     }
 
+    public MethodNode mapBody(Function<List<StatementNode>, List<StatementNode>> function) {
+        return new MethodNode(annotations, isStatic, name, arguments, function.apply(body));
+    }
+
     @Override
     public <T> T accept(NodeMapper<T> visitor) {
         return visitor.visit(this);