From 5cda2705b321cbea53249e181921195326c5fd55 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 28 Nov 2015 19:43:34 +0000
Subject: Simplify generation of anonymous classes

---
 src/main/java/org/zwobble/couscous/ast/MethodNode.java | 5 +++++
 1 file changed, 5 insertions(+)

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

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