From 1babfdf8daa45ffef3d624db6c717f846d295440 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 29 Aug 2016 16:23:58 +0100
Subject: Allow transformation of type declarations to return multiple types

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

(limited to 'src/main')

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 2db12c6..b2255d2 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -25,7 +25,7 @@ public class NodeTransformer {
     }
 
     public static List<TypeNode> apply(NodeTransformer transformer, List<TypeNode> types) {
-        return eagerMap(types, transformer::transformTypeDeclaration);
+        return eagerFlatMap(types, transformer::transformTypeDeclaration);
     }
 
     public static NodeTransformer replaceExpressions(Map<ExpressionNode, ExpressionNode> replacements) {
@@ -178,8 +178,8 @@ public class NodeTransformer {
         return field.transformSubtree(this);
     }
 
-    public TypeNode transformTypeDeclaration(TypeNode typeNode) {
-        return typeNode.transformSubtree(this);
+    public List<TypeNode> transformTypeDeclaration(TypeNode typeNode) {
+        return list(typeNode.transformSubtree(this));
     }
 
     public ClassNode transformClass(ClassNode classNode) {
-- 
cgit v1.2.3