Add Python array nodes

3 files changed, 46 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
index 89711a2..0a7735d 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -46,7 +46,20 @@ public class PythonSerializer implements PythonNodeVisitor {
     public void visit(PythonVariableReferenceNode reference) {
         writer.writeIdentifier(reference.getName());
     }
-    
+
+    @Override
+    public void visit(PythonArrayNode array) {
+        writer.writeSymbol("[");
+        writer.writeWithSeparator(
+            array.getElements(),
+            element -> write(element),
+            () -> {
+                writer.writeSymbol(",");
+                writer.writeSpace();
+            });
+        writer.writeSymbol("]");
+    }
+
     @Override
     public void visit(PythonConditionalExpressionNode conditional) {
         writeParenthesised(conditional.getTrueValue(), conditional);
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonArrayNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonArrayNode.java
new file mode 100644
index 0000000..91588ea
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonArrayNode.java
@@ -0,0 +1,31 @@
+package org.zwobble.couscous.backends.python.ast;
+
+import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
+
+import java.util.List;
+
+public class PythonArrayNode implements PythonExpressionNode {
+    public static PythonExpressionNode pythonArray(List<PythonExpressionNode> elements) {
+        return new PythonArrayNode(elements);
+    }
+
+    private final List<PythonExpressionNode> elements;
+
+    public PythonArrayNode(List<PythonExpressionNode> elements) {
+        this.elements = elements;
+    }
+
+    public List<PythonExpressionNode> getElements() {
+        return elements;
+    }
+
+    @Override
+    public int precedence() {
+        return Integer.MAX_VALUE;
+    }
+
+    @Override
+    public void accept(PythonNodeVisitor visitor) {
+        visitor.visit(this);
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
index d43691d..bcc8604 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
@@ -7,6 +7,7 @@ public interface PythonNodeVisitor {
     void visit(PythonStringLiteralNode stringLiteral);
     void visit(PythonBooleanLiteralNode booleanLiteral);
     void visit(PythonVariableReferenceNode reference);
+    void visit(PythonArrayNode array);
     void visit(PythonConditionalExpressionNode conditional);
     void visit(PythonAttributeAccessNode attributeAccess);
     void visit(PythonCallNode call);