diff --git a/common/cosmwasm-smart-contracts/coconut-dkg/src/msg.rs b/common/cosmwasm-smart-contracts/coconut-dkg/src/msg.rs index dd99dde2c0..dbff18b590 100644 --- a/common/cosmwasm-smart-contracts/coconut-dkg/src/msg.rs +++ b/common/cosmwasm-smart-contracts/coconut-dkg/src/msg.rs @@ -36,12 +36,6 @@ pub enum ExecuteMsg { }, AdvanceEpochState {}, - - // DEBUG ONLY TXs. THEY SHALL BE REMOVED BEFORE FINALISING THE CODE - // only exists for debugging purposes on local network to reset the entire state of the contract - DebugUnsafeResetAll { - init_msg: InstantiateMsg, - }, } #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema)] diff --git a/contracts/coconut-dkg/src/lib.rs b/contracts/coconut-dkg/src/lib.rs index 7787776306..4a7a7407d3 100644 --- a/contracts/coconut-dkg/src/lib.rs +++ b/contracts/coconut-dkg/src/lib.rs @@ -89,53 +89,10 @@ pub fn execute( deps, info, owner, ) } - ExecuteMsg::DebugUnsafeResetAll { init_msg } => { - reset_contract_state(deps, env, info, init_msg) - } ExecuteMsg::AdvanceEpochState {} => advance_epoch_state(deps, info), } } -fn reset_contract_state( - mut deps: DepsMut<'_>, - env: Env, - info: MessageInfo, - init_msg: InstantiateMsg, -) -> Result { - ADMIN.assert_admin(deps.as_ref(), &info.sender)?; - // this resets the epoch - instantiate(deps.branch(), env, info, init_msg)?; - - // clear all dealings, public keys, etc - let current = dealers::storage::current_dealers() - .keys(deps.storage, None, None, cosmwasm_std::Order::Ascending) - .collect::, _>>()?; - let past = dealers::storage::past_dealers() - .keys(deps.storage, None, None, cosmwasm_std::Order::Ascending) - .collect::, _>>()?; - let dealings = dealings::storage::DEALINGS_BYTES[0] - .keys(deps.storage, None, None, cosmwasm_std::Order::Ascending) - .collect::, _>>()?; - - for dealer in current { - dealers::storage::current_dealers().remove(deps.storage, &dealer)?; - } - - for dealer in past { - dealers::storage::past_dealers().remove(deps.storage, &dealer)?; - } - - for addr in dealings { - for map in dealings::storage::DEALINGS_BYTES { - map.remove(deps.storage, &addr); - } - } - - dealers::storage::NODE_INDEX_COUNTER.save(deps.storage, &0u64)?; - - Ok(Response::default()) -} - #[entry_point] pub fn query(deps: Deps<'_>, _env: Env, msg: QueryMsg) -> Result { let response = match msg {