Add enum codegen

Author: aumetra

Cargo.lock

@@ -16,3 +16,6 @@ log = "0.4.22"
 serde_json = "1.0.128"
 simple_logger = "5.0.0"
 strum = { version = "0.26.3", default-features = false, features = ["derive"] }
+
+[dev-dependencies]
+insta = "1.40.0"

packages/cw-schema-codegen/Cargo.toml

@@ -0,0 +1 @@
+pub mod rust;

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

@@ -0,0 +1,13 @@
+use askama::Template;
+
+pub struct EnumVariantTemplate<'a> {
+    pub name: &'a str,
+    pub types: Option<&'a [&'a str]>,
+}
+
+#[derive(Template)]
+#[template(escape = "none", path = "rust/enum.tpl.rs")]
+pub struct EnumTemplate<'a> {
+    pub name: &'a str,
+    pub variants: &'a [EnumVariantTemplate<'a>],
+}

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

@@ -0,0 +1,5 @@
+pub enum {{ name }} {
+    {% for variant in variants %}
+        {{ variant.name }} {% if let Some(types) = variant.types %}({% for ty in types %} {{ ty }}, {% endfor %}) {% endif %},
+    {% endfor %}
+}

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

@@ -0,0 +1,53 @@
+use askama::Template;
+use cw_schema_codegen::rust::{EnumTemplate, EnumVariantTemplate};
+
+#[test]
+fn simple_enum() {
+    let tpl = EnumTemplate {
+        name: "Simple",
+        variants: &[
+            EnumVariantTemplate {
+                name: "One",
+                types: None,
+            },
+            EnumVariantTemplate {
+                name: "Two",
+                types: None,
+            },
+        ],
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}
+
+#[test]
+fn complex_enum() {
+    let tpl = EnumTemplate {
+        name: "Complex",
+        variants: &[
+            EnumVariantTemplate {
+                name: "One",
+                types: Some(&["u64"]),
+            },
+            EnumVariantTemplate {
+                name: "Two",
+                types: Some(&["String", "u64"]),
+            },
+        ],
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}
+
+#[test]
+fn empty_enum() {
+    let tpl = EnumTemplate {
+        name: "Empty",
+        variants: &[],
+    };
+
+    let rendered = tpl.render().unwrap();
+    insta::assert_snapshot!(rendered);
+}

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

@@ -0,0 +1,11 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub enum Complex {
+    
+        One ( u64, ) ,
+    
+        Two ( String,  u64, ) ,
+    
+}

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

@@ -0,0 +1,7 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub enum Empty {
+    
+}

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

@@ -0,0 +1,11 @@
+---
+source: packages/cw-schema-codegen/tests/rust_tpl.rs
+expression: rendered
+---
+pub enum Simple {
+    
+        One ,
+    
+        Two ,
+    
+}