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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use cosmwasm_schema::{cw_serde, QueryResponses};
use cosmwasm_std::Addr;

use crate::{
    ownable::{OwnableMsg, OwnableQueryMsg},
    router::{RouterMsg, RouterQuery},
    Order,
};

use super::{HookQueryMsg, PostDispatchMsg};

#[cw_serde]
pub struct InstantiateMsg {
    pub owner: String,
}

#[cw_serde]
pub struct RegisterCustomHookMsg {
    pub dest_domain: u32,
    pub recipient: String,
    pub hook: String,
}

#[cw_serde]
pub struct ClearCustomHookMsg {
    pub dest_domain: u32,
    pub recipient: String,
}

#[cw_serde]
pub enum ExecuteMsg {
    Ownable(OwnableMsg),
    PostDispatch(PostDispatchMsg),
    Router(RouterMsg<Addr>),

    RegisterCustomHook(RegisterCustomHookMsg),
    RegisterCustomHooks(Vec<RegisterCustomHookMsg>),

    ClearCustomHook(ClearCustomHookMsg),
    ClearCustomHooks(Vec<ClearCustomHookMsg>),
}

#[cw_serde]
#[derive(QueryResponses)]
#[query_responses(nested)]
pub enum QueryMsg {
    Ownable(OwnableQueryMsg),
    Router(RouterQuery<Addr>),
    Hook(HookQueryMsg),
    CustomRoutingHook(CustomRoutingHookQueryMsg),
}

#[cw_serde]
#[derive(QueryResponses)]
pub enum CustomRoutingHookQueryMsg {
    #[returns(CustomHookResponse)]
    CustomHook { dest_domain: u32, recipient: String },

    #[returns(CustomHooksResponse)]
    CustomHooks {
        dest_domain: u32,
        offset: Option<String>,
        limit: Option<u32>,
        order: Option<Order>,
    },
}

#[cw_serde]
pub struct CustomHookResponse {
    pub dest_domain: u32,
    pub recipient: String,
    pub hook: String,
}

#[cw_serde]
pub struct CustomHooksResponse {
    pub custom_hooks: Vec<CustomHookResponse>,
}

#[cfg(test)]
mod test {
    use cosmwasm_std::HexBinary;

    use super::*;
    use crate::{
        hook::{ExpectedHookQueryMsg, PostDispatchMsg, QuoteDispatchMsg},
        msg_checker,
    };

    #[test]
    fn test_hook_interface() {
        let _checked: ExecuteMsg = msg_checker(
            PostDispatchMsg {
                metadata: HexBinary::default(),
                message: HexBinary::default(),
            }
            .wrap(),
        );

        let _checked: QueryMsg = msg_checker(ExpectedHookQueryMsg::Hook(HookQueryMsg::Mailbox {}));
        let _checked: QueryMsg = msg_checker(
            QuoteDispatchMsg {
                metadata: HexBinary::default(),
                message: HexBinary::default(),
            }
            .request(),
        );
    }
}