Sudo refactoring

contracts/consumer/virtual-staking/src/contract.rs

@@ -632,7 +632,7 @@ pub fn sudo(
     msg: SudoMsg,
 ) -> Result<Response<VirtualStakeCustomMsg>, ContractError> {
     match msg {
-        SudoMsg::Rebalance {} => VirtualStakingContract::new().handle_epoch(deps, env),
+        SudoMsg::HandleEpoch {} => VirtualStakingContract::new().handle_epoch(deps, env),
         SudoMsg::ValsetUpdate {
             additions,
             removals,

packages/apis/src/virtual_staking_api.rs

@@ -28,7 +28,7 @@ pub trait VirtualStakingApi {
     ) -> Result<Response, Self::Error>;
 
     /// Burns stake. This is called when the user's collateral is slashed and, as part of slashing
-    /// propagation, the native staking contract needs to burn / discount the indicated slashing amount.
+    /// propagation, the virtual staking contract needs to burn / discount the indicated slashing amount.
     /// Undelegates evenly from all `validators`.
     #[msg(exec)]
     fn burn(
@@ -41,12 +41,12 @@ pub trait VirtualStakingApi {
 
 #[cw_serde]
 pub enum SudoMsg {
-    /// SudoMsg::Rebalance{} should be called once per epoch by the sdk (in EndBlock).
+    /// SudoMsg::HandleEpoch{} should be called once per epoch by the sdk (in EndBlock).
     /// It allows the virtual staking contract to bond or unbond any pending requests, as well
     /// as to perform a rebalance if needed (over the max cap).
     ///
     /// It should also withdraw all pending rewards here, and send them to the converter contract.
-    Rebalance {},
+    HandleEpoch {},
     /// SudoMsg::ValsetUpdate{} should be called every time there's a validator set update:
     ///  - Addition of a new validator to the active validator set.
     ///  - Temporary removal of a validator from the active set. (i.e. `unbonded` state).

packages/bindings/src/lib.rs

@@ -1,13 +1,11 @@
 mod msg;
 mod query;
-mod sudo;
 
 pub use msg::{VirtualStakeCustomMsg, VirtualStakeMsg};
 pub use query::{
     BondStatusResponse, SlashRatioResponse, TokenQuerier, VirtualStakeCustomQuery,
     VirtualStakeQuery,
 };
-pub use sudo::SudoMsg;
 
 // This is a signal, such that any contract that imports these helpers
 // will only run on blockchains that support virtual_staking feature