modify cosmwasm gateway contracts to invoke a configured hook address…

… when calling the hyperlane mailbox
skip-mev · Oct 29, 2024 · 639ed27 · 639ed27
1 parent 92eebe5
commit 639ed27
Show file tree

Hide file tree

Showing 16 changed files with 631 additions and 14 deletions.
diff --git a/cosmwasm/contracts/fast-transfer-gateway/src/contract.rs b/cosmwasm/contracts/fast-transfer-gateway/src/contract.rs
@@ -42,6 +42,7 @@ pub fn instantiate(
         token_denom: msg.token_denom,
         address_prefix: msg.address_prefix,
         mailbox_addr: msg.mailbox_addr,
+        hook_addr: msg.hook_addr,
     };
 
     CONFIG.save(deps.storage, &config)?;

diff --git a/cosmwasm/contracts/fast-transfer-gateway/src/execute.rs b/cosmwasm/contracts/fast-transfer-gateway/src/execute.rs
@@ -4,7 +4,7 @@ use cosmwasm_std::{
 };
 use cw_ownable::assert_owner;
 use go_fast::{gateway::Config, FastTransferOrder};
-use hyperlane::mailbox::{get_default_hook, DispatchMsg, ExecuteMsg as MailboxExecuteMsg};
+use hyperlane::mailbox::{DispatchMsg, ExecuteMsg as MailboxExecuteMsg};
 
 use crate::{
     error::{ContractError, ContractResponse},
@@ -114,8 +114,6 @@ pub fn initiate_settlement(
 
     let remote_contract_address = remote_contract_address.unwrap();
 
-    let default_hook = get_default_hook(deps.as_ref(), config.mailbox_addr.clone())?;
-
     let settle_orders_message = SettleOrdersMessage {
         repayment_address,
         order_ids,
@@ -127,7 +125,7 @@ pub fn initiate_settlement(
             dest_domain: source_domain,
             recipient_addr: remote_contract_address.clone(),
             msg_body: settle_orders_message.encode(),
-            hook: Some(default_hook),
+            hook: Some(config.hook_addr.clone()),
             metadata: None,
         }))?,
         funds: info.funds,
@@ -169,8 +167,6 @@ pub fn initiate_timeout(
 
     let remote_contract_address = remote_contract_address.unwrap();
 
-    let default_hook = get_default_hook(deps.as_ref(), config.mailbox_addr.clone())?;
-
     let timeout_orders_message = TimeoutOrdersMessage { order_ids };
 
     let msg = WasmMsg::Execute {
@@ -179,7 +175,7 @@ pub fn initiate_timeout(
             dest_domain: source_domain,
             recipient_addr: remote_contract_address.clone(),
             msg_body: timeout_orders_message.encode(),
-            hook: Some(default_hook),
+            hook: Some(config.hook_addr.clone()),
             metadata: None,
         }))?,
         funds: info.funds,

diff --git a/cosmwasm/contracts/fast-transfer-gateway/src/query.rs b/cosmwasm/contracts/fast-transfer-gateway/src/query.rs
@@ -1,6 +1,6 @@
 use cosmwasm_std::{Addr, Coin, Deps, HexBinary, Order, StdError, StdResult};
 use go_fast::gateway::{Config, OrderFill, RemoteDomain};
-use hyperlane::mailbox::{get_default_hook, quote_dispatch, DispatchMsg};
+use hyperlane::mailbox::{quote_dispatch, DispatchMsg};
 
 use crate::{
     helpers::encode_settle_order_data,
@@ -64,13 +64,11 @@ pub fn quote_initiate_settlement(
 
     let remote_contract_address = remote_contract_address.unwrap();
 
-    let default_hook = get_default_hook(deps, config.mailbox_addr.clone())?;
-
     let dispatch_msg = DispatchMsg {
         dest_domain: source_domain,
         recipient_addr: remote_contract_address.clone(),
         msg_body: encode_settle_order_data(repayment_address, order_ids),
-        hook: Some(default_hook),
+        hook: Some(config.hook_addr.clone()),
         metadata: None,
     };
 

diff --git a/cosmwasm/contracts/fast-transfer-gateway/tests/common/mod.rs b/cosmwasm/contracts/fast-transfer-gateway/tests/common/mod.rs
@@ -29,6 +29,7 @@ pub fn default_instantiate() -> (OwnedDeps<MemoryStorage, MockApi, MockQuerier>,
                 token_denom: "uusdc".to_string(),
                 address_prefix: "osmo".to_string(),
                 mailbox_addr: "mailbox_contract_address".into(),
+                hook_addr: "hook_contract_address".into(),
             },
         )
         .unwrap();

diff --git a/cosmwasm/contracts/fast-transfer-gateway/tests/test_initiate_settlement.rs b/cosmwasm/contracts/fast-transfer-gateway/tests/test_initiate_settlement.rs
@@ -57,7 +57,7 @@ fn test_initiate_settlement() {
                     )
                     .unwrap(),
                     hook: Some(
-                        "osmo12pvc4v625ewl34uqqgm3ezw76durxlky5j4guz8kvhal7em3e5wqz7cnla".into()
+                        "hook_contract_address".into()
                     ),
                     metadata: None
                 }))
@@ -129,7 +129,7 @@ fn test_initiate_settlement_multiple_orders() {
                     )
                     .unwrap(),
                     hook: Some(
-                        "osmo12pvc4v625ewl34uqqgm3ezw76durxlky5j4guz8kvhal7em3e5wqz7cnla".into()
+                        "hook_contract_address".into()
                     ),
                     metadata: None
                 }))

diff --git a/cosmwasm/contracts/fast-transfer-gateway/tests/test_initiate_timeout.rs b/cosmwasm/contracts/fast-transfer-gateway/tests/test_initiate_timeout.rs
@@ -81,7 +81,7 @@ fn test_initiate_timeout() {
                     )
                     .unwrap(),
                     hook: Some(
-                        "osmo12pvc4v625ewl34uqqgm3ezw76durxlky5j4guz8kvhal7em3e5wqz7cnla".into()
+                        "hook_contract_address".into()
                     ),
                     metadata: None
                 }))

diff --git a/cosmwasm/contracts/fast-transfer-gateway/tests/test_instantiate.rs b/cosmwasm/contracts/fast-transfer-gateway/tests/test_instantiate.rs
@@ -20,6 +20,7 @@ fn test_instantiate() {
         token_denom: "uusdc".to_string(),
         address_prefix: "osmo".to_string(),
         mailbox_addr: "mailbox_contract_address".into(),
+        hook_addr: "hook_contract_address".into(),
         local_domain: 1,
     };
 
@@ -31,6 +32,7 @@ fn test_instantiate() {
     assert_eq!(config.token_denom, instantiate_msg.token_denom);
     assert_eq!(config.address_prefix, instantiate_msg.address_prefix);
     assert_eq!(config.mailbox_addr, instantiate_msg.mailbox_addr);
+    assert_eq!(config.hook_addr, instantiate_msg.hook_addr);
 
     let local_domain = LOCAL_DOMAIN.load(deps.as_ref().storage).unwrap();
     assert_eq!(local_domain, 1);