Add enhanced for statement to Java AST

3 files changed, 87 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriter.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriter.java
index 38dde1f..4a21d88 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriter.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriter.java
@@ -395,6 +395,10 @@ public class JavaWriter implements AutoCloseable {
 
     void writeBlockStatement(JavaBlockStatement statement) throws IOException {
         switch (statement) {
+            case JavaEnhancedForStatement enhancedForStatement -> {
+                writeEnhancedForStatement(enhancedForStatement);
+            }
+
             case JavaExpressionStatement expressionStatement -> {
                 writeExpressionStatement(expressionStatement);
             }
@@ -409,6 +413,32 @@ public class JavaWriter implements AutoCloseable {
         }
     }
 
+    private void writeEnhancedForStatement(JavaEnhancedForStatement enhancedForStatement) throws IOException {
+        this.writer.write("for (");
+
+        if (enhancedForStatement.elementType().isPresent()) {
+            this.writeTypeRef(enhancedForStatement.elementType().get());
+        } else {
+            this.writer.write("var");
+        }
+
+        this.writer.write(" ");
+        this.writeIdentifier(enhancedForStatement.elementName());
+        this.writer.write(" : ");
+
+        this.writeTopLevelExpression(enhancedForStatement.elements());
+
+        this.writer.write(") {");
+        this.writer.indent();
+        this.writer.newLine();
+
+        this.writeBlock(enhancedForStatement.body());
+
+        this.writer.dedent();
+        this.writer.newLine();
+        this.writer.write("}");
+    }
+
     private void writeExpressionStatement(JavaExpressionStatement expressionStatement) throws IOException {
         writeTopLevelExpression(expressionStatement.expression());
         this.writer.write(";");
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBlockStatement.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBlockStatement.java
index 53e74f6..76098cb 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBlockStatement.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBlockStatement.java
@@ -5,7 +5,7 @@ package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
 // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlockStatement imports
 // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlockStatement imports
 
-public sealed interface JavaBlockStatement permits org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpressionStatement, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaLocalVariableDeclaration, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaReturn {
+public sealed interface JavaBlockStatement permits org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpressionStatement, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaLocalVariableDeclaration, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaReturn {
     public interface Builder {
         public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlockStatement build();
 
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaEnhancedForStatement.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaEnhancedForStatement.java
new file mode 100644
index 0000000..4d99c2b
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaEnhancedForStatement.java
@@ -0,0 +1,56 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement imports
+
+public record JavaEnhancedForStatement(java.util.Optional<org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaTypeRef> elementType, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIdentifier elementName, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression elements, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlock body) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlockStatement {
+    public static org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder arbitrary() {
+        return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(java.util.Optional.empty(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIdentifier.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlock.arbitrary().build());
+    }
+
+    public record Builder(java.util.Optional<org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaTypeRef> elementType, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIdentifier elementName, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression elements, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlock body) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlockStatement.Builder {
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement build() {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement(elementType, elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElementType(java.util.Optional<org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaTypeRef> elementType) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElementType(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaTypeRef elementType) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(java.util.Optional.of(elementType), elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElementName(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIdentifier elementName) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElementName(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIdentifier.Builder elementName) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName.build(), elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElements(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression elements) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withElements(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder elements) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements.build(), body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withBody(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlock body) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements, body);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder withBody(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBlock.Builder body) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder(elementType, elementName, elements, body.build());
+        }
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaEnhancedForStatement body
+}