add to interface

Snowfork · Mar 31, 2024 · cac7b08 · cac7b08
1 parent 104cdbf
commit cac7b08
Show file tree

Hide file tree

Showing 5 changed files with 49 additions and 3 deletions.
diff --git a/contracts/src/Gateway.sol b/contracts/src/Gateway.sol
@@ -97,7 +97,7 @@ contract Gateway is IGateway, IInitializable {
     }
 
     constructor(
-        address beefyClient,
+        address beefyLightClient,
         address agentExecutor,
         ParaID bridgeHubParaID,
         bytes32 bridgeHubAgentID,
@@ -107,7 +107,7 @@ contract Gateway is IGateway, IInitializable {
             revert InvalidConstructorParams();
         }
 
-        BEEFY_CLIENT = beefyClient;
+        BEEFY_CLIENT = beefyLightClient;
         AGENT_EXECUTOR = agentExecutor;
         BRIDGE_HUB_PARA_ID_ENCODED = ScaleCodec.encodeU32(uint32(ParaID.unwrap(bridgeHubParaID)));
         BRIDGE_HUB_PARA_ID = bridgeHubParaID;
@@ -224,6 +224,10 @@ contract Gateway is IGateway, IInitializable {
     /**
      * Getters
      */
+    function beefyClient() external view returns (address) {
+        return BEEFY_CLIENT;
+    }
+
     function operatingMode() external view returns (OperatingMode) {
         return CoreStorage.layout().mode;
     }

diff --git a/contracts/src/interfaces/IGateway.sol b/contracts/src/interfaces/IGateway.sol
@@ -78,6 +78,9 @@ interface IGateway {
     /// @dev Emitted when a command is sent to register a new wrapped token on AssetHub
     event TokenRegistrationSent(address token);
 
+    /// @dev the address of the BEEFY light client used to verify messages.
+    function beefyClient() external view returns (address);
+
     /// @dev Check whether a token is registered
     function isTokenRegistered(address token) external view returns (bool);
 

diff --git a/contracts/test/Gateway.t.sol b/contracts/test/Gateway.t.sol
@@ -909,4 +909,8 @@ contract GatewayTest is Test {
         vm.expectRevert(Assets.InvalidDestinationFee.selector);
         IGateway(address(gateway)).sendToken{value: fee}(address(token), destPara, recipientAddress32, 0, 1);
     }
+
+    function testGatewayExposesBeefyCleint() public {
+        assertEq(IGateway(address(gateway)).beefyClient(), address(0));
+    }
 }
diff --git a/contracts/test/mocks/GatewayUpgradeMock.sol b/contracts/test/mocks/GatewayUpgradeMock.sol
@@ -32,6 +32,10 @@ contract GatewayUpgradeMock is IGateway, IInitializable {
         return address(0);
     }
 
+    function beefyClient() external pure returns (address) {
+        return address(0);
+    }
+
     function isTokenRegistered(address) external pure returns (bool) {
         return true;
     }