1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use crate::features::ModuleIdentification;
use crate::{base::Handler, AbstractSdkError};
use abstract_std::ibc::ModuleIbcMsg;
use cosmwasm_std::{Addr, Deps, DepsMut, Env, MessageInfo, Response};

/// Trait for a contract to call itself on an IBC counterpart.
pub trait ModuleIbcEndpoint: Handler {
    /// Get the address of the ibc host associated with this module
    fn ibc_host(&self, deps: Deps) -> Result<Addr, Self::Error>;

    /// Handler for the `ExecuteMsg::ModuleIbc(ModuleIbcMsg)` variant.
    fn module_ibc(
        self,
        deps: DepsMut,
        env: Env,
        info: MessageInfo,
        msg: ModuleIbcMsg,
    ) -> Result<Response, Self::Error> {
        // Only an IBC host can call this endpoint
        let ibc_host = self.ibc_host(deps.as_ref())?;
        if info.sender.ne(&ibc_host) {
            return Err(AbstractSdkError::ModuleIbcNotCalledByHost {
                caller: info.sender,
                host_addr: ibc_host,
                module: self.info().0.to_string(),
            }
            .into());
        };

        // If there is no handler and this endpoint is called we need to error
        let handler =
            self.maybe_module_ibc_handler()
                .ok_or(AbstractSdkError::NoModuleIbcHandler(
                    self.module_id().to_string(),
                ))?;
        handler(deps, env, self, msg.src_module_info, msg.msg)
    }
}