Add struct codegen

Author: aumetra

packages/cw-schema-codegen/src/main.rs

@@ -54,7 +54,10 @@ fn main() -> anyhow::Result<()> {
         .init()?;
 
     let opts: Opts = Opts::parse();
-    info!("Generating code for {:?} from {:?}", opts.language, opts.file);
+    info!(
+        "Generating code for {:?} from {:?}",
+        opts.language, opts.file
+    );
 
     let schema = std::fs::read_to_string(&opts.file)?;
     let schema: cw_schema::Schema = serde_json::from_str(&schema)?;

packages/cw-schema-codegen/src/rust.rs

@@ -20,9 +20,7 @@ pub struct FieldTemplate<'a> {
 pub enum TypeTemplate<'a> {
     Unit,
     Tuple(&'a [&'a str]),
-    Named {
-        fields: &'a [FieldTemplate<'a>],
-    }
+    Named { fields: &'a [FieldTemplate<'a>] },
 }
 
 #[derive(Template)]

packages/cw-schema-codegen/templates/rust/struct.tpl.rs

@@ -0,0 +1,18 @@
+pub struct {{ name }}
+
+{% match ty %}
+    {% when TypeTemplate::Unit %}
+        ;
+    {% when TypeTemplate::Tuple with (types) %}
+        (
+            {% for ty in types %}
+                {{ ty }},
+            {% endfor %}
+        );
+    {% when TypeTemplate::Named with { fields } %}
+        {
+            {% for field in fields %}
+                {{ field.name }}: {{ field.ty }},
+            {% endfor %}
+        }
+{% endmatch %}

packages/cw-schema-codegen/tests/rust_tpl.rs

@@ -1,5 +1,7 @@
 use askama::Template;
-use cw_schema_codegen::rust::{EnumTemplate, EnumVariantTemplate, FieldTemplate, TypeTemplate};
+use cw_schema_codegen::rust::{
+    EnumTemplate, EnumVariantTemplate, FieldTemplate, StructTemplate, TypeTemplate,
+};
 
 #[test]
 fn simple_enum() {
@@ -62,3 +64,47 @@ fn empty_enum() {
     let rendered = tpl.render().unwrap();
     insta::assert_snapshot!(rendered);
 }
+
+#[test]
+fn empty_struct() {
+    let tpl = StructTemplate {
+        name: "Empty",
+        ty: TypeTemplate::Unit,
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}
+
+#[test]
+fn tuple_struct() {
+    let tpl = StructTemplate {
+        name: "Tuple",
+        ty: TypeTemplate::Tuple(&["u64", "String"]),
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}
+
+#[test]
+fn named_struct() {
+    let tpl = StructTemplate {
+        name: "Named",
+        ty: TypeTemplate::Named {
+            fields: &[
+                FieldTemplate {
+                    name: "a",
+                    ty: "u64",
+                },
+                FieldTemplate {
+                    name: "b",
+                    ty: "String",
+                },
+            ],
+        },
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}

packages/cw-schema-codegen/tests/snapshots/rust_tpl__empty_struct.snap

@@ -0,0 +1,8 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub struct Empty
+
+
+        ;

packages/cw-schema-codegen/tests/snapshots/rust_tpl__named_struct.snap

@@ -0,0 +1,14 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub struct Named
+
+
+        {
+            
+                a: u64,
+            
+                b: String,
+            
+        }

packages/cw-schema-codegen/tests/snapshots/rust_tpl__tuple_struct.snap

@@ -0,0 +1,14 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub struct Tuple
+
+
+        (
+            
+                u64,
+            
+                String,
+            
+        );