From c44f0e747f00ec71bd9c773bfea7d287e50ed77b Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 19 Jul 2026 21:41:55 +0100 Subject: Support sums in rust-transient-0 --- .../rusttransient0/RustTransient0Generator.java | 78 +++++++++++++++++++++- 1 file changed, 77 insertions(+), 1 deletion(-) (limited to 'src/main/java') diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/rusttransient0/RustTransient0Generator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/rusttransient0/RustTransient0Generator.java index f2b8254..fd7bd34 100644 --- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/rusttransient0/RustTransient0Generator.java +++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/rusttransient0/RustTransient0Generator.java @@ -406,7 +406,83 @@ public class RustTransient0Generator implements Generator { } case TypedSumDefinitionNode sumDefinition -> { - yield Stream.of(); + var encodeFunction = generateEncodeFunction( + sumDefinition.type(), + List.of( + new RustExpressionStatement(new RustMatchExpression( + RustPath.of(VALUE_NAME), + IntStream.range(0, sumDefinition.variants().size()) + .mapToObj(variantIndex -> { + var variant = sumDefinition.variants().get(variantIndex); + var variantType = variant.type().value(); + var variantPath = this.rustGenerator.generateRustTypeExpression(sumDefinition.type()) + .addSegment(RustPathSegment.of(this.rustGenerator.generateVariantName(variantType))); + + return new RustMatchArm( + new RustTupleStructPattern(variantPath, List.of(new RustIdentifierPattern(VALUE_NAME))), + new RustBlockExpression( + List.of( + new RustExpressionStatement(new RustBlockExpression( + generateEncode( + new RustPrefixExpression( + RustPrefixOperator.BORROW, + new RustIntegerLiteral(variantIndex, Optional.empty()) + ), + NativeTypes.INT_32 + ), + Optional.empty() + )), + new RustExpressionStatement(new RustBlockExpression( + generateEncode(RustPath.of(VALUE_NAME), variantType), + Optional.empty() + )) + ), + Optional.empty() + ) + ); + }) + .toList() + )) + ) + ); + + var decodeMatchArms = IntStream.range(0, sumDefinition.variants().size()) + .mapToObj(variantIndex -> { + var variant = sumDefinition.variants().get(variantIndex); + var variantType = variant.type().value(); + var variantPath = this.rustGenerator.generateRustTypeExpression(sumDefinition.type()) + .addSegment(RustPathSegment.of(this.rustGenerator.generateVariantName(variantType))); + + return new RustMatchArm( + new RustLiteralPattern(new RustIntegerLiteral(variantIndex, Optional.empty())), + new RustCallExpression(variantPath, List.of(generateDecode(variantType))) + ); + }) + .collect(toArrayList()); + + decodeMatchArms.add(new RustMatchArm( + new RustWildcardPattern(), + generateTodo() + )); + + var decodeFunction = generateDecodeFunction( + sumDefinition.type(), + new RustBlockExpression( + List.of(), + Optional.of(new RustMatchExpression( + new RustPrefixExpression( + RustPrefixOperator.BORROW, + generateDecode(NativeTypes.INT_32) + ), + decodeMatchArms + )) + ) + ); + + yield Stream.of( + encodeFunction, + decodeFunction + ); } }; } -- cgit v1.2.3