Combine binary operations into single Java AST node

7 files changed, 72 insertions, 98 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javapreciselymatchers/JavaPreciselyMatchersGenerator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javapreciselymatchers/JavaPreciselyMatchersGenerator.java
index 03d6ae7..69e8e94 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javapreciselymatchers/JavaPreciselyMatchersGenerator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javapreciselymatchers/JavaPreciselyMatchersGenerator.java
@@ -133,7 +133,8 @@ public class JavaPreciselyMatchersGenerator implements Generator {
                 ),
                 new JavaBlock(List.of(
                     new JavaExpressionStatement(
-                        new JavaAssignment(
+                        new JavaBinaryOperation(
+                            JavaBinaryOperator.ASSIGN,
                             new JavaFieldAccess(JavaRef.THIS, submatchersIdentifier),
                             new JavaRef(submatchersIdentifier)
                         )
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 e512650..ec91f63 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
@@ -39,10 +39,15 @@ public class JavaWriter implements AutoCloseable {
         this.writer.close();
     }
 
-    private void writeAssignment(JavaAssignment assignment) throws IOException {
-        this.writeExpression(assignment.left());
-        this.writer.write(" = ");
-        this.writeExpression(assignment.right());
+    private void writeBinaryOperation(JavaBinaryOperation binaryOperation) throws IOException {
+        this.writeExpression(binaryOperation.left());
+        this.writer.write(" ");
+        this.writer.write(switch (binaryOperation.operator()) {
+            case ASSIGN -> "=";
+            case UNSIGNED_RIGHT_SHIFT -> ">>>";
+        });
+        this.writer.write(" ");
+        this.writeExpression(binaryOperation.right());
     }
 
     private void writeBooleanLiteral(JavaBooleanLiteral booleanLiteral) throws IOException {
@@ -226,8 +231,8 @@ public class JavaWriter implements AutoCloseable {
 
     void writeExpression(JavaExpression expression) throws IOException {
         switch (expression) {
-            case JavaAssignment assignment -> {
-                writeAssignment(assignment);
+            case JavaBinaryOperation binaryOperation -> {
+                writeBinaryOperation(binaryOperation);
             }
 
             case JavaBooleanLiteral booleanLiteral -> {
@@ -273,10 +278,6 @@ public class JavaWriter implements AutoCloseable {
             case JavaStringLiteral stringLiteral -> {
                 writeStringLiteral(stringLiteral);
             }
-
-            case JavaUnsignedRightShift unsignedRightShift -> {
-                writeUnsignedRightShift(unsignedRightShift);
-            }
         }
     }
 
@@ -571,12 +572,6 @@ public class JavaWriter implements AutoCloseable {
         }
     }
 
-    private void writeUnsignedRightShift(JavaUnsignedRightShift unsignedRightShift) throws IOException {
-        this.writeExpression(unsignedRightShift.left());
-        this.writer.write(" >>> ");
-        this.writeExpression(unsignedRightShift.right());
-    }
-
     private void writeVisibility(JavaVisibility visibility) throws IOException {
         this.writer.write(switch (visibility) {
             case PUBLIC -> "public";
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaAssignment.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaAssignment.java
deleted file mode 100644
index d9ca0c1..0000000
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaAssignment.java
+++ /dev/null
@@ -1,40 +0,0 @@
-// Generated by hobgoblin.
-
-package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
-
-// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment imports
-// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment imports
-
-public record JavaAssignment(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression {
-    public static org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder arbitrary() {
-        return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build());
-    }
-
-    public record Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder {
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment build() {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder left) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder(left.build(), right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder right) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder(left, right.build());
-        }
-
-        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder body
-        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment.Builder body
-    }
-
-    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment body
-    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment body
-}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperation.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperation.java
new file mode 100644
index 0000000..cb1c0ef
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperation.java
@@ -0,0 +1,44 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation imports
+
+public record JavaBinaryOperation(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator operator, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression {
+    public static org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder arbitrary() {
+        return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator.ASSIGN, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build());
+    }
+
+    public record Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator operator, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder {
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation build() {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation(operator, left, right);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder withOperator(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator operator) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(operator, left, right);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(operator, left, right);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder left) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(operator, left.build(), right);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(operator, left, right);
+        }
+
+        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder right) {
+            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder(operator, left, right.build());
+        }
+
+        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder body
+        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation.Builder body
+    }
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation body
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperator.java
new file mode 100644
index 0000000..4f646ba
--- /dev/null
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaBinaryOperator.java
@@ -0,0 +1,14 @@
+// Generated by hobgoblin.
+
+package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
+
+// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator imports
+// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator imports
+
+public enum JavaBinaryOperator {
+    ASSIGN,
+    UNSIGNED_RIGHT_SHIFT
+
+    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator body
+    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperator body
+}
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaExpression.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaExpression.java
index 7e9cdad..7652041 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaExpression.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaExpression.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.JavaExpression imports
 // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression imports
 
-public sealed interface JavaExpression permits org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaAssignment, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaFieldAccess, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIntegerLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaMethodCall, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaMethodRef, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaNewExpression, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaNullLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaRef, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStaticFieldAccess, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStaticMethodCall, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStringLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift {
+public sealed interface JavaExpression permits org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBinaryOperation, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaFieldAccess, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaIntegerLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaMethodCall, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaMethodRef, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaNewExpression, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaNullLiteral, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaRef, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStaticFieldAccess, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStaticMethodCall, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaStringLiteral {
     public interface Builder {
         public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression build();
 
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaUnsignedRightShift.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaUnsignedRightShift.java
deleted file mode 100644
index aced126..0000000
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/java/ast/JavaUnsignedRightShift.java
+++ /dev/null
@@ -1,40 +0,0 @@
-// Generated by hobgoblin.
-
-package org.zwobble.hobgoblin.compiler.output.lang.java.ast;
-
-// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift imports
-// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift imports
-
-public record JavaUnsignedRightShift(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression {
-    public static org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder arbitrary() {
-        return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaBooleanLiteral.arbitrary().build());
-    }
-
-    public record Builder(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left, org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) implements org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder {
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift build() {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression left) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder withLeft(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder left) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder(left.build(), right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression right) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder(left, right);
-        }
-
-        public org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder withRight(org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaExpression.Builder right) {
-            return new org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder(left, right.build());
-        }
-
-        // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder body
-        // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift.Builder body
-    }
-
-    // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift body
-    // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.java.ast.JavaUnsignedRightShift body
-}