1 files changed, 107 insertions, 0 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
new file mode 100644
index 0000000..e5a13e5
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -0,0 +1,107 @@
+package org.zwobble.couscous.backends.python;
+
+import org.zwobble.couscous.backends.python.ast.PythonBlock;
+import org.zwobble.couscous.backends.python.ast.PythonClassNode;
+import org.zwobble.couscous.backends.python.ast.PythonFunctionDefinitionNode;
+import org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode;
+import org.zwobble.couscous.backends.python.ast.PythonModuleNode;
+import org.zwobble.couscous.backends.python.ast.PythonNode;
+import org.zwobble.couscous.backends.python.ast.PythonPassNode;
+import org.zwobble.couscous.backends.python.ast.PythonReturnNode;
+import org.zwobble.couscous.backends.python.ast.PythonStringLiteralNode;
+import org.zwobble.couscous.backends.python.ast.PythonWriter;
+import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
+
+import lombok.val;
+
+public class PythonSerializer implements PythonNodeVisitor<Void> {
+    public static String serialize(PythonNode node) {
+        val writer = new PythonWriter();
+        val serializer = new PythonSerializer(writer);
+        serializer.write(node);
+        return writer.asString();
+    }
+    
+    private PythonWriter writer;
+    
+    private PythonSerializer(PythonWriter writer) {
+        this.writer = writer;
+    }
+    
+    public void write(PythonNode node) {
+        node.accept(this);
+    }
+    
+    @Override
+    public Void visit(PythonIntegerLiteralNode integerLiteral) {
+        writer.writeInteger(integerLiteral.getValue());
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonStringLiteralNode stringLiteral) {
+        writer.writeStringLiteral(stringLiteral.getValue());
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonReturnNode pythonReturn) {
+        writer.startStatement();
+        writer.writeKeyword("return");
+        writer.writeSpace();
+        write(pythonReturn.getValue());
+        writer.endStatement();
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonPassNode pass) {
+        writer.startStatement();
+        writer.writeKeyword("pass");
+        writer.endStatement();
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonClassNode pythonClass) {
+        writer.startStatement();
+        writer.writeKeyword("class");
+        writer.writeSpace();
+        writer.writeIdentifier(pythonClass.getName());
+        writer.writeSymbol("(");
+        writer.writeIdentifier("object");
+        writer.writeSymbol(")");
+        writeBlock(pythonClass.getBody());
+        writer.endStatement();
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonFunctionDefinitionNode functionDefinition) {
+        writer.startStatement();
+        writer.writeKeyword("def");
+        writer.writeSpace();
+        writer.writeIdentifier(functionDefinition.getName());
+        writer.writeSymbol("(");
+        writer.writeSymbol(")");
+        writeBlock(functionDefinition.getBody());
+        writer.endStatement();
+        return null;
+    }
+
+    @Override
+    public Void visit(PythonModuleNode module) {
+        for (val statement : module.getStatements()) {
+            write(statement);
+        }
+        return null;
+    }
+
+    private void writeBlock(PythonBlock body) {
+        writer.startBlock();
+        for (val statement : body) {
+            write(statement);
+        }
+        writer.endBlock();
+    }
+}