Add SQL AST for CREATE TABLE statements

5 files changed, 232 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/SqlWriter.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/SqlWriter.java
new file mode 100644
index 0000000..20b28c3
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/SqlWriter.java
@@ -0,0 +1,82 @@
+package org.zwobble.hobgoblin.compiler.output.lang.sql;
+
+import org.zwobble.hobgoblin.compiler.output.CodeWriter;
+import org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition;
+import org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement;
+import org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier;
+import org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement;
+
+import java.io.IOException;
+import java.nio.file.Path;
+
+import static org.zwobble.hobgoblin.compiler.output.CodeWriter.writeWithSeparator;
+
+public class SqlWriter {
+    private static final String LINE_COMMENT_START = "--";
+
+    public interface Write {
+        void write(SqlWriter writer) throws IOException;
+    }
+
+    public static void writeTo(Path path, Write write) throws IOException {
+        CodeWriter.writeTo(
+            path,
+            LINE_COMMENT_START,
+            writer -> write.write(new SqlWriter(writer))
+        );
+    }
+
+    public static void writeTo(CodeWriter.CodeWriterFile file, Write write) throws IOException {
+        CodeWriter.writeTo(
+            file,
+            LINE_COMMENT_START,
+            writer -> write.write(new SqlWriter(writer))
+        );
+    }
+
+    private final CodeWriter writer;
+
+    public SqlWriter(CodeWriter writer) {
+        this.writer = writer;
+    }
+
+    void writeStatement(SqlStatement statement) throws IOException {
+        switch (statement) {
+            case SqlCreateTableStatement createTableStatement -> {
+                writeCreateTableStatement(createTableStatement);
+            }
+        }
+    }
+
+    private void writeCreateTableStatement(SqlCreateTableStatement createTableStatement) throws IOException {
+        this.writer.write("CREATE TABLE ");
+        this.writeIdentifier(createTableStatement.name());
+        this.writer.write(" (");
+        this.writer.newLine();
+        this.writer.indent();
+
+        writeWithSeparator(
+            createTableStatement.columns(),
+            column -> writeColumnDefinition(column),
+            () -> {
+                this.writer.write(",");
+                this.writer.newLine();
+            }
+        );
+
+        this.writer.dedent();
+        this.writer.newLine();
+        this.writer.write(");");
+    }
+
+    private void writeColumnDefinition(SqlColumnDefinition column) throws IOException {
+        this.writeIdentifier(column.name());
+        this.writer.write(" ");
+        this.writeIdentifier(column.typeName());
+    }
+
+    private void writeIdentifier(SqlIdentifier name) throws IOException {
+        // TODO: handle escaping
+        this.writer.write(name.value());
+    }
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlColumnDefinition.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlColumnDefinition.java
new file mode 100644
index 0000000..837da77
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlColumnDefinition.java
@@ -0,0 +1,46 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.sql.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition imports
+
+public record SqlColumnDefinition(
+    org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name,
+    org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier typeName
+) {
+    public static org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder arbitrary() {
+        return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.arbitrary().build());
+    }
+
+    public record Builder(
+        org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name,
+        org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier typeName
+    ) {
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition build() {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition(name, typeName);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder withName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder(name, typeName);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder withName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder name) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder(name.build(), typeName);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder withTypeName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier typeName) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder(name, typeName);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder withTypeName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder typeName) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder(name, typeName.build());
+        }
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition body
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlCreateTableStatement.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlCreateTableStatement.java
new file mode 100644
index 0000000..a5fec23
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlCreateTableStatement.java
@@ -0,0 +1,54 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.sql.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement imports
+
+public record SqlCreateTableStatement(
+    org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name,
+    java.util.List<org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition> columns
+) implements org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement {
+    public static org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder arbitrary() {
+        return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.arbitrary().build(), java.util.List.of());
+    }
+
+    public record Builder(
+        org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name,
+        java.util.List<org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition> columns
+    ) implements org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement.Builder {
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement build() {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement(name, columns);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder withName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier name) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(name, columns);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder withName(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder name) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(name.build(), columns);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder withColumns(java.util.List<org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition> columns) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(name, columns);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder addColumn(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition column) {
+            var columns = new java.util.ArrayList<org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition>(this.columns);
+            columns.add(column);
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(name, columns);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder addColumn(org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition.Builder column) {
+            var columns = new java.util.ArrayList<org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlColumnDefinition>(this.columns);
+            columns.add(column.build());
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder(name, columns);
+        }
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement body
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlIdentifier.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlIdentifier.java
new file mode 100644
index 0000000..9acc2e4
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlIdentifier.java
@@ -0,0 +1,32 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.sql.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier imports
+
+public record SqlIdentifier(
+    String value
+) {
+    public static org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder arbitrary() {
+        return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder("");
+    }
+
+    public record Builder(
+        String value
+    ) {
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier build() {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier(value);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder withValue(String value) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder(value);
+        }
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlIdentifier body
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlStatement.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlStatement.java
new file mode 100644
index 0000000..0f90dc1
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/sql/ast/SqlStatement.java
@@ -0,0 +1,18 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.sql.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement imports
+
+public sealed interface SqlStatement permits org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlCreateTableStatement {
+    public interface Builder {
+        public org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement build();
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.sql.ast.SqlStatement body
+}