Read for-each as sugared statements

2 files changed, 81 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ForEachNode.java b/src/main/java/org/zwobble/couscous/ast/ForEachNode.java
new file mode 100644
index 0000000..e30f130
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/ForEachNode.java
@@ -0,0 +1,79 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.util.ExtraIterables;
+
+import java.util.List;
+
+public class ForEachNode implements StatementNode {
+    private final VariableDeclaration target;
+    private final ExpressionNode iterable;
+    private final List<StatementNode> statements;
+
+    public ForEachNode(VariableDeclaration target, ExpressionNode iterable, List<StatementNode> statements) {
+        this.target = target;
+        this.iterable = iterable;
+        this.statements = statements;
+    }
+
+    public VariableDeclaration getTarget() {
+        return target;
+    }
+
+    public ExpressionNode getIterable() {
+        return iterable;
+    }
+
+    public List<StatementNode> getStatements() {
+        return statements;
+    }
+
+    @Override
+    public int type() {
+        return NodeTypes.FOR_EACH;
+    }
+
+    @Override
+    public Iterable<? extends Node> childNodes() {
+        return ExtraIterables.of(iterable);
+    }
+
+    @Override
+    public StatementNode transformSubtree(NodeTransformer transformer) {
+        return new ForEachNode(
+            transformer.transform(target),
+            transformer.transformExpression(iterable),
+            transformer.transformStatements(statements)
+        );
+    }
+
+    @Override
+    public String toString() {
+        return "ForEachNode(" +
+            "target=" + target +
+            ", iterable=" + iterable +
+            ", statements=" + statements +
+            ')';
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        ForEachNode that = (ForEachNode) o;
+
+        if (!target.equals(that.target)) return false;
+        if (!iterable.equals(that.iterable)) return false;
+        return statements.equals(that.statements);
+
+    }
+
+    @Override
+    public int hashCode() {
+        int result = target.hashCode();
+        result = 31 * result + iterable.hashCode();
+        result = 31 * result + statements.hashCode();
+        return result;
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/ast/NodeTypes.java b/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
index 9a9a244..f7f8bf6 100644
--- a/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
+++ b/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
@@ -44,6 +44,7 @@ public class NodeTypes {
     public static final int TYPE_COERCION = 10028;
     public static final int VARIABLE_REFERENCE = 10029;
     public static final int WHILE = 10030;
+    public static final int FOR_EACH = 10034;
 
     private NodeTypes() {
     }
@@ -83,6 +84,7 @@ public class NodeTypes {
         .put(TypeCoercionNode.class, TYPE_COERCION)
         .put(VariableReferenceNode.class, VARIABLE_REFERENCE)
         .put(WhileNode.class, WHILE)
+        .put(ForEachNode.class, FOR_EACH)
         .put(PrimitiveInstanceMethodCall.class, CsharpNodeTypes.PRIMITIVE_INSTANCE_METHOD_CALL)
         .put(PrimitiveStaticMethodCall.class, CsharpNodeTypes.PRIMITIVE_STATIC_METHOD_CALL)
         .build();