feat: move queries to the messages

hadronlabs-org · Dec 14, 2023 · 2514fbc · 2514fbc
1 parent ac15cc0
commit 2514fbc
Show file tree

Hide file tree

Showing 8 changed files with 32 additions and 39 deletions.
diff --git a/contracts/core/src/bin/lido-core-schema.rs b/contracts/core/src/bin/lido-core-schema.rs
@@ -1,8 +1,5 @@
 use cosmwasm_schema::write_api;
-use lido_staking_base::{
-    msg::core::{ExecuteMsg, InstantiateMsg, MigrateMsg},
-    state::core::QueryMsg,
-};
+use lido_staking_base::msg::core::{ExecuteMsg, InstantiateMsg, MigrateMsg, QueryMsg};
 
 fn main() {
     write_api! {

diff --git a/contracts/core/src/contract.rs b/contracts/core/src/contract.rs
@@ -4,9 +4,9 @@ use cosmwasm_std::{
     DepsMut, Env, MessageInfo, Response, StdResult, Uint128, WasmMsg,
 };
 use cw2::set_contract_version;
-use lido_staking_base::msg::core::{ExecuteMsg, InstantiateMsg};
+use lido_staking_base::msg::core::{ExecuteMsg, InstantiateMsg, QueryMsg};
 use lido_staking_base::msg::token::ExecuteMsg as TokenExecuteMsg;
-use lido_staking_base::state::core::{QueryMsg, CONFIG};
+use lido_staking_base::state::core::CONFIG;
 use neutron_sdk::{
     bindings::{msg::NeutronMsg, query::NeutronQuery},
     NeutronResult,

diff --git a/contracts/token/src/bin/lido-token-schema.rs b/contracts/token/src/bin/lido-token-schema.rs
@@ -1,9 +1,6 @@
 use cosmwasm_schema::write_api;
 
-use lido_staking_base::{
-    msg::token::{ExecuteMsg, InstantiateMsg, MigrateMsg},
-    state::token::QueryMsg,
-};
+use lido_staking_base::msg::token::{ExecuteMsg, InstantiateMsg, MigrateMsg, QueryMsg};
 
 fn main() {
     write_api! {

diff --git a/contracts/token/src/lib.rs b/contracts/token/src/lib.rs
@@ -4,8 +4,8 @@ use cosmwasm_std::{
 };
 
 use lido_staking_base::{
-    msg::token::{ExecuteMsg, InstantiateMsg, MigrateMsg},
-    state::token::{ConfigResponse, QueryMsg, CORE_ADDRESS, DENOM},
+    msg::token::{ConfigResponse, ExecuteMsg, InstantiateMsg, MigrateMsg, QueryMsg},
+    state::token::{CORE_ADDRESS, DENOM},
 };
 use neutron_sdk::{
     bindings::{msg::NeutronMsg, query::NeutronQuery},

diff --git a/packages/base/src/msg/core.rs b/packages/base/src/msg/core.rs
@@ -1,5 +1,5 @@
-use cosmwasm_schema::cw_serde;
-use cosmwasm_std::Uint128;
+use cosmwasm_schema::{cw_serde, QueryResponses};
+use cosmwasm_std::{Decimal256, Uint128};
 
 use crate::state::core::Config;
 
@@ -11,6 +11,15 @@ pub struct InstantiateMsg {
     pub owner: String,
 }
 
+#[cw_serde]
+#[derive(QueryResponses)]
+pub enum QueryMsg {
+    #[returns(Config)]
+    Config {},
+    #[returns(Decimal256)]
+    ExchangeRate {},
+}
+
 #[cw_serde]
 pub enum ExecuteMsg {
     Bond {},

diff --git a/packages/base/src/msg/token.rs b/packages/base/src/msg/token.rs
@@ -1,6 +1,19 @@
-use cosmwasm_schema::cw_serde;
+use cosmwasm_schema::{cw_serde, QueryResponses};
 use cosmwasm_std::Uint128;
 
+#[cosmwasm_schema::cw_serde]
+#[derive(QueryResponses)]
+pub enum QueryMsg {
+    #[returns(ConfigResponse)]
+    Config {},
+}
+
+#[cosmwasm_schema::cw_serde]
+pub struct ConfigResponse {
+    pub core_address: String,
+    pub denom: String,
+}
+
 #[cw_serde]
 pub enum ExecuteMsg {
     Mint { amount: Uint128, receiver: String },

diff --git a/packages/base/src/state/core.rs b/packages/base/src/state/core.rs
@@ -1,5 +1,4 @@
-use cosmwasm_schema::{cw_serde, QueryResponses};
-use cosmwasm_std::Decimal256;
+use cosmwasm_schema::cw_serde;
 use cw_storage_plus::Item;
 
 #[cw_serde]
@@ -10,13 +9,4 @@ pub struct Config {
     pub owner: String,
 }
 
-#[cw_serde]
-#[derive(QueryResponses)]
-pub enum QueryMsg {
-    #[returns(Config)]
-    Config {},
-    #[returns(Decimal256)]
-    ExchangeRate {},
-}
-
 pub const CONFIG: Item<Config> = Item::new("config");
diff --git a/packages/base/src/state/token.rs b/packages/base/src/state/token.rs
@@ -1,18 +1,5 @@
 use cosmwasm_std::Addr;
 use cw_storage_plus::Item;
 
-#[cosmwasm_schema::cw_serde]
-#[derive(cosmwasm_schema::QueryResponses)]
-pub enum QueryMsg {
-    #[returns(ConfigResponse)]
-    Config {},
-}
-
-#[cosmwasm_schema::cw_serde]
-pub struct ConfigResponse {
-    pub core_address: String,
-    pub denom: String,
-}
-
 pub const CORE_ADDRESS: Item<Addr> = Item::new("core");
 pub const DENOM: Item<String> = Item::new("denom");