From bc57c2bd62cd46208ecdb89eef2c561fc8a757d2 Mon Sep 17 00:00:00 2001
From: adairrr <32375605+adairrr@users.noreply.github.com>
Date: Tue, 10 Dec 2024 15:33:51 -0500
Subject: [PATCH] Add schema generation scripts

---
 .cargo/config.toml                    |  1 +
 contracts/account/examples/schema.rs  | 11 +++++++++++
 contracts/treasury/examples/schema.rs | 10 ++++++++++
 contracts/treasury/src/lib.rs         |  2 +-
 4 files changed, 23 insertions(+), 1 deletion(-)
 create mode 100644 contracts/account/examples/schema.rs
 create mode 100644 contracts/treasury/examples/schema.rs

diff --git a/.cargo/config.toml b/.cargo/config.toml
index ab0e7dc..4d85773 100644
--- a/.cargo/config.toml
+++ b/.cargo/config.toml
@@ -1,5 +1,6 @@
 [alias]
 wasm = "build --release --lib --target wasm32-unknown-unknown"
+schema = "run --example schema"
 
 [env]
 RUSTFLAGS = "-C link-arg=-s"
diff --git a/contracts/account/examples/schema.rs b/contracts/account/examples/schema.rs
new file mode 100644
index 0000000..416060a
--- /dev/null
+++ b/contracts/account/examples/schema.rs
@@ -0,0 +1,11 @@
+use account::msg::*;
+use cosmwasm_schema::write_api;
+
+fn main() {
+    write_api! {
+        instantiate: InstantiateMsg,
+        query: QueryMsg,
+        execute: ExecuteMsg,
+        migrate: MigrateMsg
+    };
+}
diff --git a/contracts/treasury/examples/schema.rs b/contracts/treasury/examples/schema.rs
new file mode 100644
index 0000000..9aa5200
--- /dev/null
+++ b/contracts/treasury/examples/schema.rs
@@ -0,0 +1,10 @@
+use treasury::msg::*;
+use cosmwasm_schema::write_api;
+
+fn main() {
+    write_api! {
+        instantiate: InstantiateMsg,
+        query: QueryMsg,
+        execute: ExecuteMsg,
+    };
+}
diff --git a/contracts/treasury/src/lib.rs b/contracts/treasury/src/lib.rs
index b7f80b0..2f24467 100644
--- a/contracts/treasury/src/lib.rs
+++ b/contracts/treasury/src/lib.rs
@@ -4,7 +4,7 @@ extern crate core;
 pub mod contract;
 mod error;
 mod execute;
-mod msg;
+pub mod msg;
 mod state;
 
 mod grant;