From 1cca01d385e68aefe2131319c53c4811101f15f7 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 10 Jan 2016 16:12:32 +0000
Subject: Handle both static and instance method calls in one node

---
 .../org/zwobble/couscous/ast/structure/NodeStructure.java | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

(limited to 'src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java')

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
@@ -64,11 +64,6 @@ public class NodeStructure {
                     methodCall.getArguments().stream());
             }
 
-            @Override
-            public Stream<ExpressionNode> visit(StaticMethodCallNode staticMethodCall) {
-                return staticMethodCall.getArguments().stream();
-            }
-
             @Override
             public Stream<? extends ExpressionNode> visit(ConstructorCallNode call) {
                 return call.getArguments().stream();
@@ -84,6 +79,16 @@ public class NodeStructure {
                 return Stream.of(typeCoercion.getExpression());
             }
 
+            @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());
-- 
cgit v1.2.3