Add support for reading vararg method invocations

1 files changed, 70 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ArrayNode.java b/src/main/java/org/zwobble/couscous/ast/ArrayNode.java
new file mode 100644
index 0000000..ac4c284
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/ArrayNode.java
@@ -0,0 +1,70 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.types.Type;
+import org.zwobble.couscous.types.Types;
+
+import java.util.List;
+
+public class ArrayNode implements ExpressionNode {
+    public static ArrayNode array(Type elementType, List<ExpressionNode> expressions) {
+        return new ArrayNode(elementType, expressions);
+    }
+
+    private final Type elementType;
+    private final List<ExpressionNode> elements;
+
+    public ArrayNode(Type elementType, List<ExpressionNode> elements) {
+        this.elementType = elementType;
+        this.elements = elements;
+    }
+
+    public List<ExpressionNode> getElements() {
+        return elements;
+    }
+
+    @Override
+    public Type getType() {
+        return Types.array(elementType);
+    }
+
+    @Override
+    public <T> T accept(ExpressionNodeMapper<T> visitor) {
+        return visitor.visit(this);
+    }
+
+    @Override
+    public ExpressionNode transform(NodeTransformer transformer) {
+        return new ArrayNode(
+            transformer.transform(elementType),
+            transformer.transformExpressions(elements));
+    }
+
+    @Override
+    public String toString() {
+        return "ArrayNode(" +
+            "elementType=" + elementType +
+            ", elements=" + elements +
+            ')';
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        ArrayNode arrayNode = (ArrayNode) o;
+
+        if (!elementType.equals(arrayNode.elementType)) return false;
+        return elements.equals(arrayNode.elements);
+
+    }
+
+    @Override
+    public int hashCode() {
+        int result = elementType.hashCode();
+        result = 31 * result + elements.hashCode();
+        return result;
+    }
+}