Add some templates

Author: aumetra

Cargo.lock

@@ -15,7 +15,6 @@ frunk = "0.4.3"
 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

@@ -10,10 +10,8 @@ use std::{
     io::{self, Write},
     path::PathBuf,
 };
-use strum::Display;
 
-#[derive(Clone, Copy, Default, Display, ValueEnum)]
-#[strum(serialize_all = "kebab-case")]
+#[derive(Clone, Copy, Default, ValueEnum)]
 pub enum Language {
     #[default]
     Rust,
@@ -25,7 +23,7 @@ pub enum Language {
 #[clap(about, author, version)]
 /// Official CosmWasm codegen tool
 struct Opts {
-    #[clap(default_value_t, long, short)]
+    #[clap(default_value_t, long, short, value_enum)]
     /// Programming language to generate code for
     language: Language,
 

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

@@ -2,7 +2,7 @@ use askama::Template;
 
 pub struct EnumVariantTemplate<'a> {
     pub name: &'a str,
-    pub types: Option<&'a [&'a str]>,
+    pub ty: TypeTemplate<'a>,
 }
 
 #[derive(Template)]
@@ -11,3 +11,23 @@ pub struct EnumTemplate<'a> {
     pub name: &'a str,
     pub variants: &'a [EnumVariantTemplate<'a>],
 }
+
+pub struct FieldTemplate<'a> {
+    pub name: &'a str,
+    pub ty: &'a str,
+}
+
+pub enum TypeTemplate<'a> {
+    Unit,
+    Tuple(&'a [&'a str]),
+    Named {
+        fields: &'a [FieldTemplate<'a>],
+    }
+}
+
+#[derive(Template)]
+#[template(escape = "none", path = "rust/struct.tpl.rs")]
+pub struct StructTemplate<'a> {
+    pub name: &'a str,
+    pub ty: TypeTemplate<'a>,
+}