Handle both static and instance method calls in one node

1 files changed, 10 insertions, 5 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
index 8b65f35..c2380c7 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -65,11 +65,6 @@ public class NodeStructure {
             }
 
             @Override
-            public Stream<ExpressionNode> visit(StaticMethodCallNode staticMethodCall) {
-                return staticMethodCall.getArguments().stream();
-            }
-
-            @Override
             public Stream<? extends ExpressionNode> visit(ConstructorCallNode call) {
                 return call.getArguments().stream();
             }
@@ -85,6 +80,16 @@ public class NodeStructure {
             }
 
             @Override
+            public Stream<? extends Node> visit(InstanceReceiver receiver) {
+                return Stream.of(receiver.getExpression());
+            }
+
+            @Override
+            public Stream<? extends Node> visit(StaticReceiver receiver) {
+                return Stream.empty();
+            }
+
+            @Override
             public Stream<Node> visit(ReturnNode returnNode) {
                 return Stream.of(returnNode.getValue());
             }