feat: Automatic consumer registration

contracts/babylon/benches/main.rs

@@ -44,6 +44,8 @@ pub fn setup_instance() -> Instance<MockApi, MockStorage, MockQuerier> {
     let msg = InstantiateMsg {
         network: babylon_bitcoin::chain_params::Network::Regtest,
         babylon_tag: "01020304".to_string(),
+        consumer_name: "TestConsumer".to_string(),
+        consumer_description: "Test Consumer Description".to_string(),
         btc_confirmation_depth: 10,
         checkpoint_finalization_timeout: 1,
         notify_cosmos_zone: false,
@@ -172,4 +174,4 @@ criterion_group!(
     config = make_config();
     targets = bench_btc_light_client
 );
-criterion_main!(btc_light_client);
+criterion_main!(btc_light_client);

contracts/babylon/src/contract.rs

@@ -34,6 +34,8 @@ pub fn instantiate(
         checkpoint_finalization_timeout: msg.checkpoint_finalization_timeout,
         notify_cosmos_zone: msg.notify_cosmos_zone,
         btc_staking: None, // Will be set in `reply` if `btc_staking_code_id` is provided
+        consumer_name: msg.consumer_name,
+        consumer_description: msg.consumer_description,
     };
     CONFIG.save(deps.storage, &cfg)?;
 
@@ -214,6 +216,8 @@ mod tests {
             btc_staking_code_id: None,
             btc_staking_msg: None,
             admin: None,
+            consumer_name: "MyConsumer".to_string(),
+            consumer_description: "This is a description of my consumer".to_string(),
         };
         let info = message_info(&deps.api.addr_make(CREATOR), &[]);
         let res = instantiate(deps.as_mut(), mock_env(), info, msg).unwrap();

contracts/babylon/src/error.rs

@@ -34,6 +34,8 @@ pub enum ContractError {
     Unauthorized {},
     #[error("The BTC staking contract is not set")]
     BtcStakingNotSet {},
+    #[error("Invalid configuration: {msg}")]
+    InvalidConfig { msg: String },
 }
 
 #[derive(Error, Debug, PartialEq)]
@@ -127,4 +129,4 @@ pub enum BabylonEpochChainError {
     EmptyTxKey {},
     #[error("The BTC header cannot be decoded")]
     BTCHeaderDecodeError {},
-}
+}

contracts/babylon/src/ibc.rs

@@ -3,14 +3,14 @@ use babylon_bindings::BabylonMsg;
 use babylon_proto::babylon::zoneconcierge::v1::{
     zoneconcierge_packet_data::Packet, BtcTimestamp, ZoneconciergePacketData,
 };
+use babylon_proto::babylon::btcstkconsumer::v1::ConsumerRegisterIbcPacket;
+
 use cosmwasm_std::{
-    DepsMut, Env, Event, Ibc3ChannelOpenResponse, IbcBasicResponse, IbcChannel, IbcChannelCloseMsg,
-    IbcChannelConnectMsg, IbcChannelOpenMsg, IbcChannelOpenResponse, IbcOrder, IbcPacketAckMsg,
-    IbcPacketReceiveMsg, IbcPacketTimeoutMsg, IbcReceiveResponse, IbcTimeout, Never, StdAck,
-    StdError, StdResult,
+    Binary, DepsMut, Env, Event, Ibc3ChannelOpenResponse, IbcBasicResponse, IbcChannel, IbcChannelCloseMsg, IbcChannelConnectMsg, IbcChannelOpenMsg, IbcChannelOpenResponse, IbcMsg, IbcOrder, IbcPacketAckMsg, IbcPacketReceiveMsg, IbcPacketTimeoutMsg, IbcReceiveResponse, IbcTimeout, Never, StdAck, StdError, StdResult
 };
 use cw_storage_plus::Item;
 use prost::Message;
+use crate::state::config::CONFIG;
 
 pub const IBC_VERSION: &str = "zoneconcierge-1";
 pub const IBC_ORDERING: IbcOrder = IbcOrder::Ordered;
@@ -57,7 +57,7 @@ pub fn ibc_channel_open(
 /// Second part of the 4-step handshake, i.e. ChannelOpenAck and ChannelOpenConfirm.
 pub fn ibc_channel_connect(
     deps: DepsMut,
-    _env: Env,
+    env: &Env,
     msg: IbcChannelConnectMsg,
 ) -> Result<IbcBasicResponse, ContractError> {
     // Ensure we have no channel yet
@@ -69,11 +69,45 @@ pub fn ibc_channel_connect(
     // Store the channel
     IBC_CHANNEL.save(deps.storage, channel)?;
 
+    // TODO: send IBC packet only on ChannelOpenConfirm?
+
+    // Load the config
+    let cfg = CONFIG.load(deps.storage)?;
+
+    // Check if consumer name or description is empty
+    if cfg.consumer_name.is_empty() || cfg.consumer_description.is_empty() {
+        return Err(ContractError::InvalidConfig {
+            msg: "Consumer name and description must not be empty".to_string(),
+        });
+    }
+
+    // Create the ConsumerRegisterIBCPacket
+    let consumer_register_packet = ConsumerRegisterIbcPacket {
+        consumer_name: cfg.consumer_name,
+        consumer_description: cfg.consumer_description,
+    };
+
+    // Create the ZoneconciergePacketData
+    let packet_data = ZoneconciergePacketData {
+        packet: Some(Packet::ConsumerRegister(consumer_register_packet)),
+    };
+
+    let packet_data_bytes = packet_data.encode_to_vec();
+
+    // Create the IBC packet message
+    let ibc_msg = IbcMsg::SendPacket {
+        channel_id: channel.endpoint.channel_id.clone(),
+        data: Binary::new(packet_data_bytes),
+        timeout: packet_timeout(env),
+    };
+
     let chan_id = &channel.endpoint.channel_id;
     Ok(IbcBasicResponse::new()
+        .add_message(ibc_msg)
         .add_attribute("action", "ibc_connect")
         .add_attribute("channel_id", chan_id)
-        .add_event(Event::new("ibc").add_attribute("channel", "connect")))
+        .add_event(Event::new("ibc").add_attribute("channel", "connect"))
+    )
 }
 
 /// This is invoked on the IBC Channel Close message
@@ -123,7 +157,10 @@ pub fn ibc_packet_receive(
             Packet::BtcTimestamp(btc_ts) => ibc_packet::handle_btc_timestamp(deps, caller, &btc_ts),
             Packet::BtcStaking(btc_staking) => {
                 ibc_packet::handle_btc_staking(deps, caller, &btc_staking)
-            }
+            },
+            Packet::ConsumerRegister(_) => {
+                return Err(StdError::generic_err("ConsumerRegister packet should not be received").into())
+            },
         }
     })()
     .or_else(|e| {
@@ -381,6 +418,8 @@ mod tests {
             btc_staking_code_id: None,
             btc_staking_msg: None,
             admin: None,
+            consumer_name: "TestConsumer".to_string(),
+            consumer_description: "Test Consumer Description".to_string(),
         };
         let info = message_info(&deps.api.addr_make(CREATOR), &[]);
         let res = instantiate(deps.as_mut(), mock_env(), info, msg).unwrap();

contracts/babylon/src/lib.rs

@@ -73,7 +73,7 @@ pub fn ibc_channel_connect(
     env: Env,
     msg: IbcChannelConnectMsg,
 ) -> Result<IbcBasicResponse, ContractError> {
-    ibc::ibc_channel_connect(deps, env, msg)
+    ibc::ibc_channel_connect(deps, &env, msg)
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]