encode and decode u32

Author: blaine-arcjet

cmd/gravity/src/codegen/exports.rs

@@ -161,13 +161,15 @@ mod tests {
 
         // Verify function body
         assert!(generated.contains("arg0 := value"));
+        assert!(generated.contains("result0 := api.EncodeU32(arg0)"));
         assert!(
-            generated.contains("i.module.ExportedFunction(\"add_number\").Call(ctx, uint64(arg0))")
+            generated
+                .contains("i.module.ExportedFunction(\"add_number\").Call(ctx, uint64(result0))")
         );
-        assert!(generated.contains("if err0 != nil {"));
-        assert!(generated.contains("panic(err0)"));
-        assert!(generated.contains("results0 := raw0[0]"));
-        assert!(generated.contains("result1 := uint32(results0)"));
-        assert!(generated.contains("return result1"));
+        assert!(generated.contains("if err1 != nil {"));
+        assert!(generated.contains("panic(err1)"));
+        assert!(generated.contains("results1 := raw1[0]"));
+        assert!(generated.contains("result2 := api.DecodeU32(results1)"));
+        assert!(generated.contains("return result2"));
     }
 }

cmd/gravity/src/codegen/func.rs

@@ -310,17 +310,22 @@ impl Bindgen for Func<'_> {
                 results.push(Operand::SingleValue(value))
             }
             Instruction::I32FromU32 => {
-                // It seems like this isn't needed because Wazero works with Go's uint32 type
+                let tmp = self.tmp();
+                let result = &format!("result{tmp}");
                 let operand = &operands[0];
-                results.push(operand.clone());
+                quote_in! { self.body =>
+                    $['\r']
+                    $result := $(&self.go_imports.wazero_api_encode_u32)($operand)
+                };
+                results.push(Operand::SingleValue(result.into()));
             }
             Instruction::U32FromI32 => {
                 let tmp = self.tmp();
                 let result = &format!("result{tmp}");
                 let operand = &operands[0];
                 quote_in! { self.body =>
                     $['\r']
-                    $result := uint32($operand)
+                    $result := $(&self.go_imports.wazero_api_decode_u32)($operand)
                 };
                 results.push(Operand::SingleValue(result.into()));
             }

cmd/gravity/src/go/imports.rs

@@ -14,6 +14,7 @@ pub struct GoImports {
     pub wazero_api_decode_i32: Import,
     pub wazero_api_encode_i32: Import,
     pub wazero_api_decode_u32: Import,
+    pub wazero_api_encode_u32: Import,
 }
 
 impl Default for GoImports {
@@ -61,6 +62,10 @@ impl GoImports {
                 "github.com/tetratelabs/wazero/api",
                 "DecodeU32",
             ),
+            wazero_api_encode_u32: genco::lang::go::import(
+                "github.com/tetratelabs/wazero/api",
+                "EncodeU32",
+            ),
         }
     }
 }