diff options
Diffstat (limited to 'src/main/java/org/zwobble')
4 files changed, 115 insertions, 1 deletions
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriter.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriter.java index c3fa404..0630e07 100644 --- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriter.java +++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriter.java @@ -351,6 +351,10 @@ public class RustWriter implements AutoCloseable { this.writeIteratorLoopExpression(iteratorLoopExpression); } + case RustMatchExpression matchExpression -> { + this.writeMatchExpression(matchExpression); + } + case RustPath path -> { this.writePath(path); } @@ -414,6 +418,7 @@ public class RustWriter implements AutoCloseable { case RustIfExpression _ -> RustPrecedence.PRIMARY; case RustIntegerLiteral _ -> RustPrecedence.PRIMARY; case RustIteratorLoopExpression _ -> RustPrecedence.PRIMARY; + case RustMatchExpression _ -> RustPrecedence.PRIMARY; case RustPath _ -> RustPrecedence.PATHS; case RustPrefixExpression _ -> RustPrecedence.PREFIX; case RustRangeExpr _ -> RustPrecedence.RANGE; @@ -550,6 +555,27 @@ public class RustWriter implements AutoCloseable { this.writeBlockExpression(iteratorLoopExpression.body()); } + private void writeMatchExpression( + RustMatchExpression matchExpression + ) throws IOException { + this.writer.write("match "); + this.writeTopLevelExpression(matchExpression.scrutinee()); + this.writer.write(" {"); + this.writer.indent(); + this.writer.newLine(); + + for (var arm : matchExpression.arms()) { + this.writePattern(arm.pattern()); + this.writer.write(" => "); + this.writeTopLevelExpression(arm.body()); + this.writer.write(","); + this.writer.newLine(); + } + + this.writer.dedent(); + this.writer.write("}"); + } + private void writePrefixExpression(RustPrefixExpression prefixExpression) throws IOException { this.writer.write(switch (prefixExpression.operator()) { case BORROW -> "&"; diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustExpression.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustExpression.java index 6cfea75..8e55bc4 100644 --- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustExpression.java +++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustExpression.java @@ -5,7 +5,7 @@ package org.zwobble.hobgoblin.compiler.output.lang.rust.ast; // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression imports // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression imports -public sealed interface RustExpression permits org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustArrayRepeatExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBinaryExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBlockExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBoolLiteral, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustCallExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustFieldExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIfExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIntegerLiteral, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIteratorLoopExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPath, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPrefixExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustRangeExpr, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustStructExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustTryPropagationExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustTupleExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustVecRepeatExpression { +public sealed interface RustExpression permits org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustArrayRepeatExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBinaryExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBlockExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBoolLiteral, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustCallExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustFieldExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIfExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIntegerLiteral, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIteratorLoopExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPath, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPrefixExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustRangeExpr, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustStructExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustTryPropagationExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustTupleExpression, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustVecRepeatExpression { public interface Builder { public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression build(); diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchArm.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchArm.java new file mode 100644 index 0000000..a946ce1 --- /dev/null +++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchArm.java @@ -0,0 +1,40 @@ +// Generated by hobgoblin. + +package org.zwobble.hobgoblin.compiler.output.lang.rust.ast; + +// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm imports +// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm imports + +public record RustMatchArm(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPattern pattern, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression body) { + public static org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder arbitrary() { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustIdentifierPattern.arbitrary().build(), org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBlockExpression.arbitrary().build()); + } + + public record Builder(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPattern pattern, org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression body) { + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm build() { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm(pattern, body); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder withPattern(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPattern pattern) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder(pattern, body); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder withPattern(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustPattern.Builder pattern) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder(pattern.build(), body); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder withBody(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression body) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder(pattern, body); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder withBody(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression.Builder body) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder(pattern, body.build()); + } + + // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder body + // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder body + } + + // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm body + // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm body +} diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchExpression.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchExpression.java new file mode 100644 index 0000000..4770b86 --- /dev/null +++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/ast/RustMatchExpression.java @@ -0,0 +1,48 @@ +// Generated by hobgoblin. + +package org.zwobble.hobgoblin.compiler.output.lang.rust.ast; + +// Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression imports +// Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression imports + +public record RustMatchExpression(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression scrutinee, java.util.List<org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm> arms) implements org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression { + public static org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder arbitrary() { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustBlockExpression.arbitrary().build(), java.util.List.of()); + } + + public record Builder(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression scrutinee, java.util.List<org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm> arms) implements org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression.Builder { + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression build() { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression(scrutinee, arms); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder withScrutinee(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression scrutinee) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(scrutinee, arms); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder withScrutinee(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustExpression.Builder scrutinee) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(scrutinee.build(), arms); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder withArms(java.util.List<org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm> arms) { + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(scrutinee, arms); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder addArm(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm arm) { + var arms = new java.util.ArrayList<org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm>(this.arms); + arms.add(arm); + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(scrutinee, arms); + } + + public org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder addArm(org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm.Builder arm) { + var arms = new java.util.ArrayList<org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchArm>(this.arms); + arms.add(arm.build()); + return new org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder(scrutinee, arms); + } + + // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder body + // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression.Builder body + } + + // Custom area start: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression body + // Custom area end: org.zwobble.hobgoblin.compiler.output.lang.rust.ast.RustMatchExpression body +} |