Handle both static and instance method calls in one node

1 files changed, 59 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java b/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java
new file mode 100644
index 0000000..926af5d
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java
@@ -0,0 +1,59 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.NodeMapper;
+
+import java.util.function.Function;
+
+public class InstanceReceiver implements Receiver {
+    public static Receiver instanceReceiver(ExpressionNode expression) {
+        return new InstanceReceiver(expression);
+    }
+
+    private final ExpressionNode expression;
+
+    public InstanceReceiver(ExpressionNode expression) {
+        this.expression = expression;
+    }
+
+    public ExpressionNode getExpression() {
+        return expression;
+    }
+
+    @Override
+    public Receiver replaceExpressions(Function<ExpressionNode, ExpressionNode> replace) {
+        return new InstanceReceiver(replace.apply(expression));
+    }
+
+    @Override
+    public <T> T accept(Mapper<T> visitor) {
+        return visitor.visit(expression);
+    }
+
+    @Override
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
+    }
+
+    @Override
+    public String toString() {
+        return "InstanceReceiver(" +
+            "expression=" + expression +
+            ')';
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        InstanceReceiver that = (InstanceReceiver) o;
+
+        return expression.equals(that.expression);
+
+    }
+
+    @Override
+    public int hashCode() {
+        return expression.hashCode();
+    }
+}