From 1389f27e4ad44ddd59204a0f8f4b59a95e7f05d8 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 15 Jul 2016 20:59:54 +0100
Subject: Rename transform to transformSubTree, update usages accordingly

---
 src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
index 7fae204..30adff9 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
@@ -199,7 +199,7 @@ public class JavaReader {
             Maps.uniqueIndex(freeVariables, variable -> variable.freeVariable),
             freeVariable -> captureAccess(className, freeVariable));
         NodeTransformer transformer = NodeTransformer.replaceExpressions(replacements);
-        return eagerMap(body, statement -> statement.transform(transformer));
+        return eagerFlatMap(body, transformer::transformStatement);
     }
 
     private ConstructorNode buildConstructor(
@@ -321,7 +321,7 @@ public class JavaReader {
                 thisReference(anonymousClass.getType().get()),
                 thisReference(className)
             ));
-            classNode = classNode.transform(nodeTransformer);
+            classNode = nodeTransformer.transformClass(classNode);
         }
         return classWithCapture(scope, classNode);
     }
-- 
cgit v1.2.3