Generate Encode and Decode types

2 files changed, 59 insertions, 11 deletions

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 9d959bd..49a35f6 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
@@ -71,6 +71,8 @@ public class RustTransient0Generator implements Generator {
         var rustModule = new RustModule(
             rustModuleName,
             List.of(
+                generateEncodeType(),
+                generateDecodeType(),
                 generateEncodeBoolFunction(),
                 generateDecodeBoolFunction(),
                 generateEncodeInt8Function(),
@@ -87,6 +89,47 @@ public class RustTransient0Generator implements Generator {
         this.rustGenerator.write(rustModule);
     }
 
+    private RustItem generateEncodeType() {
+        var valueType = RustIdentifier.of("T");
+        var writeType = RustIdentifier.of("TWrite");
+        return new RustTypeAlias(
+            Optional.of(RustVisibility.PUB),
+            RustIdentifier.of("Encode"),
+            List.of(
+                new RustTypeParam(valueType, List.of()),
+                new RustTypeParam(writeType, List.of(RustTypes.IO_WRITE))
+            ),
+            new RustFunctionPointerType(
+                List.of(
+                    new RustSharedReferenceType(RustPath.of(valueType)),
+                    new RustMutableReferenceType(RustPath.of(writeType)),
+                    encoderSharedValuesType()
+                ),
+                RustTypes.ioResult(RustTypes.UNIT)
+            )
+        );
+    }
+
+    private RustItem generateDecodeType() {
+        var valueType = RustIdentifier.of("T");
+        var readType = RustIdentifier.of("TRead");
+        return new RustTypeAlias(
+            Optional.of(RustVisibility.PUB),
+            RustIdentifier.of("Decode"),
+            List.of(
+                new RustTypeParam(readType, List.of(RustTypes.IO_READ)),
+                new RustTypeParam(valueType, List.of())
+            ),
+            new RustFunctionPointerType(
+                List.of(
+                    new RustMutableReferenceType(RustPath.of(readType)),
+                    decoderSharedValuesType()
+                ),
+                RustTypes.ioResult(RustPath.of(valueType))
+            )
+        );
+    }
+
     private RustItem generateEncodeBoolFunction() {
         return generateEncodeFunction(
             NativeTypes.BOOL,
@@ -548,12 +591,13 @@ public class RustTransient0Generator implements Generator {
         return List.of(
             new RustFunctionParam(VALUE_NAME, new RustSharedReferenceType(rustType)),
             new RustFunctionParam(WRITER_NAME, new RustMutableReferenceType(new RustImplTraitType(RustPath.of("std", "io", "Write")))),
-            new RustFunctionParam(
-                SHARED_VALUES_NAME,
-                new RustMutableReferenceType(
-                    RustTypes.hashMap(RustTypes.USIZE, RustPath.primitive("i64"))
-                )
-            )
+            new RustFunctionParam(SHARED_VALUES_NAME, encoderSharedValuesType())
+        );
+    }
+
+    private static RustMutableReferenceType encoderSharedValuesType() {
+        return new RustMutableReferenceType(
+            RustTypes.hashMap(RustTypes.USIZE, RustPath.primitive("i64"))
         );
     }
 
@@ -578,11 +622,13 @@ public class RustTransient0Generator implements Generator {
     private static List<RustFunctionParam> generateDecodeParams() {
         return List.of(
             new RustFunctionParam(READER_NAME, new RustMutableReferenceType(new RustImplTraitType(RustPath.of("std", "io", "Read")))),
-            new RustFunctionParam(
-                SHARED_VALUES_NAME, new RustMutableReferenceType(
-                RustTypes.vec(RustTypes.arc(RustTypes.dyn(RustTypes.ANY, RustTypes.SYNC, RustTypes.SEND)))
-            )
-            )
+            new RustFunctionParam(SHARED_VALUES_NAME, decoderSharedValuesType())
+        );
+    }
+
+    private static RustMutableReferenceType decoderSharedValuesType() {
+        return new RustMutableReferenceType(
+            RustTypes.vec(RustTypes.arc(RustTypes.dyn(RustTypes.ANY, RustTypes.SYNC, RustTypes.SEND)))
         );
     }
 
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustTypes.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustTypes.java
index 3c5ab6c..b3d65d4 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustTypes.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustTypes.java
@@ -36,8 +36,10 @@ public class RustTypes {
     public static RustPath IO_ERROR = RustPath.global("std", "io", "Error");
     public static RustPath IO_ERROR_KIND = RustPath.global("std", "io", "ErrorKind");
     public static RustPath IO_ERROR_KIND_INVALID_DATA = IO_ERROR_KIND.addSegment(RustPathSegment.of("InvalidData"));
+    public static RustPath IO_READ = RustPath.global("std", "io", "Read");
     public static RustPath IO_RESULT = RustPath.global("std", "io", "Result");
     public static RustPath IO_RESULT_OK = IO_RESULT.addSegment(RustPathSegment.of("Ok"));
+    public static RustPath IO_WRITE = RustPath.global("std", "io", "Write");
 
     public static RustPath ioResult(RustType okType) {
         return IO_RESULT.withArgs(List.of(okType));