feat(ibc-union): migrate clients support (#3449)

cosmwasm/ibc-union/core/msg/src/msg.rs

@@ -36,6 +36,16 @@ pub enum ExecuteMsg {
     BatchAcks(MsgBatchAcks),
     PacketSend(MsgSendPacket),
     WriteAcknowledgement(MsgWriteAcknowledgement),
+    MigrateState(MsgMigrateState),
+}
+
+#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
+#[serde(deny_unknown_fields)]
+pub struct MsgMigrateState {
+    pub client_id: u32,
+    pub client_state: Bytes,
+    pub consensus_state: Bytes,
+    pub height: u64,
 }
 
 #[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]

cosmwasm/ibc-union/core/src/contract.rs

@@ -15,9 +15,9 @@ use ibc_union_msg::{
         ExecuteMsg, InitMsg, MsgBatchAcks, MsgBatchSend, MsgChannelCloseConfirm,
         MsgChannelCloseInit, MsgChannelOpenAck, MsgChannelOpenConfirm, MsgChannelOpenInit,
         MsgChannelOpenTry, MsgConnectionOpenAck, MsgConnectionOpenConfirm, MsgConnectionOpenInit,
-        MsgConnectionOpenTry, MsgCreateClient, MsgIntentPacketRecv, MsgPacketAcknowledgement,
-        MsgPacketRecv, MsgPacketTimeout, MsgRegisterClient, MsgSendPacket, MsgUpdateClient,
-        MsgWriteAcknowledgement,
+        MsgConnectionOpenTry, MsgCreateClient, MsgIntentPacketRecv, MsgMigrateState,
+        MsgPacketAcknowledgement, MsgPacketRecv, MsgPacketTimeout, MsgRegisterClient,
+        MsgSendPacket, MsgUpdateClient, MsgWriteAcknowledgement,
     },
     query::QueryMsg,
 };
@@ -391,7 +391,63 @@ pub fn execute(
             packets,
             acks.into_iter().map(Into::into).collect(),
         ),
+        ExecuteMsg::MigrateState(MsgMigrateState {
+            client_id,
+            client_state,
+            consensus_state,
+            height,
+        }) => migrate_state(
+            deps,
+            info.sender,
+            client_id,
+            client_state,
+            consensus_state,
+            height,
+        ),
+    }
+}
+
+fn migrate_state(
+    mut deps: DepsMut,
+    sender: Addr,
+    client_id: u32,
+    client_state: unionlabs::primitives::Bytes,
+    consensus_state: unionlabs::primitives::Bytes,
+    height: u64,
+) -> Result<Response, ContractError> {
+    let client_addr = CLIENT_IMPLS.load(deps.storage, client_id)?;
+
+    if client_addr != sender {
+        return Err(ContractError::UnauthorizedMigration {
+            client_id,
+            caller: sender,
+            client: client_addr,
+        });
     }
+
+    CLIENT_STATES.update(deps.storage, client_id, |s| {
+        let _ = s.ok_or(ContractError::CannotMigrateWithNoClientState { client_id })?;
+        Ok::<Binary, ContractError>(client_state.to_vec().into())
+    })?;
+
+    store_commit(
+        deps.branch(),
+        &ClientStatePath { client_id }.key(),
+        &commit(client_state),
+    )?;
+
+    CLIENT_CONSENSUS_STATES.update(deps.storage, (client_id, height), |s| {
+        let _ = s.ok_or(ContractError::CannotMigrateWithNoConsensusState { client_id, height })?;
+        Ok::<Binary, ContractError>(consensus_state.to_vec().into())
+    })?;
+
+    store_commit(
+        deps.branch(),
+        &ConsensusStatePath { client_id, height }.key(),
+        &commit(consensus_state),
+    )?;
+
+    Ok(Response::new())
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]

cosmwasm/ibc-union/core/src/lib.rs

@@ -98,6 +98,23 @@ pub enum ContractError {
     AcknowledgementMismatch { found: Bytes, expected: Bytes },
     #[error("{} the packet already exist", ContractErrorKind::from(self))]
     PacketCommitmentAlreadyExist,
+    #[error(
+        "{} caller {caller} don't have permission to migrate the client {client} with id {client_id}", ContractErrorKind::from(self)
+    )]
+    UnauthorizedMigration {
+        client_id: u32,
+        caller: Addr,
+        client: Addr,
+    },
+    #[error(
+        "{} cannot migrate the client {client_id} when there's no client state",
+        ContractErrorKind::from(self)
+    )]
+    CannotMigrateWithNoClientState { client_id: u32 },
+    #[error(
+        "{} cannot migrate the client {client_id} when there's no consensus state at height {height}", ContractErrorKind::from(self)
+    )]
+    CannotMigrateWithNoConsensusState { client_id: u32, height: u64 },
 }
 
 impl ContractErrorKind {