Add transform to Receiver interface

1 files changed, 1 insertions, 11 deletions

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 4f7f819..683b3b9 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -191,17 +191,7 @@ public class NodeTransformer {
     }
 
     public Receiver transformReceiver(Receiver receiver) {
-        return receiver.accept(new Receiver.Mapper<Receiver>() {
-            @Override
-            public Receiver visit(ExpressionNode receiver) {
-                return new InstanceReceiver(receiver.transform(NodeTransformer.this));
-            }
-
-            @Override
-            public Receiver visit(TypeName receiver) {
-                return new StaticReceiver(transform(receiver));
-            }
-        });
+        return receiver.transform(this);
     }
 
     public VariableDeclaration transform(VariableDeclaration declaration) {