nologies/cycles-quartz
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Impl get_epoch_state query

Browse source
This commit is contained in:
hu55a1n1 2024-01-10 07:26:41 -08:00
parent 976c1cacf0
commit a52ba76265
2 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
bisenzone-cw-mvp/contracts/cw-tee-mtcs/src/contract.rs
View file

@ -144,6 +144,7 @@ pub mod execute {
pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
match msg {
QueryMsg::GetSgxState {} => to_json_binary(&query::get_sgx_state(deps)?),
QueryMsg::GetEpochState {} => to_json_binary(&query::get_epoch_state(deps)?),
QueryMsg::GetRequests {} => to_json_binary(&query::get_requests(deps)?),
}
}
@ -151,8 +152,8 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
pub mod query {
use cosmwasm_std::{Deps, StdResult};
use crate::msg::query::{GetRequestsResponse, GetSgxStateResponse};
use crate::state::{SgxState, REQUESTS, SGX_STATE};
use crate::msg::query::{GetEpochStateResponse, GetRequestsResponse, GetSgxStateResponse};
use crate::state::{EpochState, SgxState, EPOCH_STATE, REQUESTS, SGX_STATE};
pub fn get_sgx_state(deps: Deps) -> StdResult<GetSgxStateResponse> {
let SgxState {
@ -166,6 +167,11 @@ pub mod query {
})
}
pub fn get_epoch_state(deps: Deps) -> StdResult<GetEpochStateResponse> {
let EpochState { epoch_key } = EPOCH_STATE.load(deps.storage)?;
Ok(GetEpochStateResponse { epoch_key })
}
pub fn get_requests(deps: Deps) -> StdResult<GetRequestsResponse> {
Ok(GetRequestsResponse {
requests: REQUESTS.load(deps.storage)?,

9
bisenzone-cw-mvp/contracts/cw-tee-mtcs/src/msg.rs
View file

@ -38,6 +38,8 @@ pub mod execute {
pub enum QueryMsg {
#[returns(query::GetSgxStateResponse)]
GetSgxState {},
#[returns(query::GetEpochStateResponse)]
GetEpochState {},
#[returns(query::GetRequestsResponse)]
GetRequests {},
}
@ -45,7 +47,7 @@ pub enum QueryMsg {
pub mod query {
use super::*;
use crate::state::{RawMrenclave, RawNonce, Request};
use crate::state::{RawMrenclave, RawNonce, RawPublicKey, Request};
#[cw_serde]
pub struct GetSgxStateResponse {
@ -53,6 +55,11 @@ pub mod query {
pub key_manager_mrenclave: RawMrenclave,
}
#[cw_serde]
pub struct GetEpochStateResponse {
pub epoch_key: RawPublicKey,
}
#[cw_serde]
pub struct GetRequestsResponse {
pub requests: Vec<(RawNonce, Request)>,