Defer generation of classes for lambdas

1 files changed, 11 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index 0d37130..ce27694 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -9,6 +9,7 @@ import org.zwobble.couscous.util.ExtraIterables;
 import java.util.List;
 import java.util.Set;
 
+import static org.zwobble.couscous.util.ExtraLists.concat;
 import static org.zwobble.couscous.util.ExtraLists.eagerFlatMap;
 import static org.zwobble.couscous.util.ExtraLists.list;
 
@@ -138,6 +139,15 @@ public class ClassNode implements TypeNode {
 
     @Override
     public TypeNode stripInnerTypes() {
+        return withInnerTypes(list());
+    }
+
+    @Override
+    public TypeNode addInnerTypes(List<TypeNode> types) {
+        return withInnerTypes(concat(innerTypes, types));
+    }
+
+    private TypeNode withInnerTypes(List<TypeNode> types) {
         return new ClassNode(
             name,
             typeParameters,
@@ -146,7 +156,7 @@ public class ClassNode implements TypeNode {
             staticConstructor,
             constructor,
             methods,
-            list()
+            types
         );
     }