Read statement blocks

2 files changed, 53 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/NodeTypes.java b/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
index f7f8bf6..82090aa 100644
--- a/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
+++ b/src/main/java/org/zwobble/couscous/ast/NodeTypes.java
@@ -45,6 +45,7 @@ public class NodeTypes {
     public static final int VARIABLE_REFERENCE = 10029;
     public static final int WHILE = 10030;
     public static final int FOR_EACH = 10034;
+    public static final int STATEMENT_BLOCK = 10035;
 
     private NodeTypes() {
     }
@@ -85,6 +86,7 @@ public class NodeTypes {
         .put(VariableReferenceNode.class, VARIABLE_REFERENCE)
         .put(WhileNode.class, WHILE)
         .put(ForEachNode.class, FOR_EACH)
+        .put(StatementBlockNode.class, STATEMENT_BLOCK)
         .put(PrimitiveInstanceMethodCall.class, CsharpNodeTypes.PRIMITIVE_INSTANCE_METHOD_CALL)
         .put(PrimitiveStaticMethodCall.class, CsharpNodeTypes.PRIMITIVE_STATIC_METHOD_CALL)
         .build();
diff --git a/src/main/java/org/zwobble/couscous/ast/StatementBlockNode.java b/src/main/java/org/zwobble/couscous/ast/StatementBlockNode.java
new file mode 100644
index 0000000..0c24295
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/StatementBlockNode.java
@@ -0,0 +1,51 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+
+import java.util.List;
+
+public class StatementBlockNode implements StatementNode {
+    private final List<StatementNode> statements;
+
+    public StatementBlockNode(List<StatementNode> statements) {
+        this.statements = statements;
+    }
+
+    @Override
+    public int type() {
+        return NodeTypes.STATEMENT_BLOCK;
+    }
+
+    @Override
+    public Iterable<? extends Node> childNodes() {
+        return statements;
+    }
+
+    @Override
+    public StatementNode transformSubtree(NodeTransformer transformer) {
+        return new StatementBlockNode(transformer.transformStatements(statements));
+    }
+
+    @Override
+    public String toString() {
+        return "StatementBlockNode(" +
+            "statements=" + statements +
+            ')';
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        StatementBlockNode that = (StatementBlockNode) o;
+
+        return statements.equals(that.statements);
+
+    }
+
+    @Override
+    public int hashCode() {
+        return statements.hashCode();
+    }
+}