Merge pull request #15 from enigmampc/contract-status

ContractStatus query
scrtlabs · May 25, 2021 · b76b4b4 · b76b4b4
2 parents f7585cd + dad03fc
commit b76b4b4
Show file tree

Hide file tree

Showing 4 changed files with 51 additions and 0 deletions.
diff --git a/schema/query_answer.json b/schema/query_answer.json
@@ -76,6 +76,25 @@
         }
       }
     },
+    {
+      "type": "object",
+      "required": [
+        "contract_status"
+      ],
+      "properties": {
+        "contract_status": {
+          "type": "object",
+          "required": [
+            "status"
+          ],
+          "properties": {
+            "status": {
+              "$ref": "#/definitions/ContractStatusLevel"
+            }
+          }
+        }
+      }
+    },
     {
       "type": "object",
       "required": [
@@ -271,6 +290,14 @@
         }
       }
     },
+    "ContractStatusLevel": {
+      "type": "string",
+      "enum": [
+        "normal_run",
+        "stop_all_but_redeems",
+        "stop_all"
+      ]
+    },
     "HumanAddr": {
       "type": "string"
     },

diff --git a/schema/query_msg.json b/schema/query_msg.json
@@ -24,6 +24,17 @@
         }
       }
     },
+    {
+      "type": "object",
+      "required": [
+        "contract_status"
+      ],
+      "properties": {
+        "contract_status": {
+          "type": "object"
+        }
+      }
+    },
     {
       "type": "object",
       "required": [

diff --git a/src/contract.rs b/src/contract.rs
@@ -228,6 +228,7 @@ pub fn query<S: Storage, A: Api, Q: Querier>(deps: &Extern<S, A, Q>, msg: QueryM
     match msg {
         QueryMsg::TokenInfo {} => query_token_info(&deps.storage),
         QueryMsg::TokenConfig {} => query_token_config(&deps.storage),
+        QueryMsg::ContractStatus {} => query_contract_status(&deps.storage),
         QueryMsg::ExchangeRate {} => query_exchange_rate(&deps.storage),
         QueryMsg::Minters { .. } => query_minters(deps),
         _ => authenticated_queries(deps, msg),
@@ -331,6 +332,14 @@ fn query_token_config<S: ReadonlyStorage>(storage: &S) -> QueryResult {
     })
 }
 
+fn query_contract_status<S: ReadonlyStorage>(storage: &S) -> QueryResult {
+    let config = ReadonlyConfig::from_storage(storage);
+
+    to_binary(&QueryAnswer::ContractStatus {
+        status: config.contract_status(),
+    })
+}
+
 pub fn query_transfers<S: Storage, A: Api, Q: Querier>(
     deps: &Extern<S, A, Q>,
     account: &HumanAddr,

diff --git a/src/msg.rs b/src/msg.rs
@@ -310,6 +310,7 @@ pub enum HandleAnswer {
 pub enum QueryMsg {
     TokenInfo {},
     TokenConfig {},
+    ContractStatus {},
     ExchangeRate {},
     Allowance {
         owner: HumanAddr,
@@ -370,6 +371,9 @@ pub enum QueryAnswer {
         mint_enabled: bool,
         burn_enabled: bool,
     },
+    ContractStatus {
+        status: ContractStatusLevel,
+    },
     ExchangeRate {
         rate: Uint128,
         denom: String,