Execute with command decoded

Snowfork · Feb 29, 2024 · 8ceb5ee · 8ceb5ee
1 parent 3d18d1a
commit 8ceb5ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/contracts/src/AgentExecutor.sol b/contracts/src/AgentExecutor.sol
@@ -19,8 +19,7 @@ contract AgentExecutor {
     /// @dev Execute a message which originated from the Polkadot side of the bridge. In other terms,
     /// the `data` parameter is constructed by the BridgeHub parachain.
     ///
-    function execute(bytes memory data) external {
-        (AgentExecuteCommand command, bytes memory params) = abi.decode(data, (AgentExecuteCommand, bytes));
+    function execute(AgentExecuteCommand command, bytes memory params) external {
         if (command == AgentExecuteCommand.TransferToken) {
             (address token, address recipient, uint128 amount) = abi.decode(params, (address, address, uint128));
             _transferToken(token, recipient, amount);

diff --git a/contracts/src/Gateway.sol b/contracts/src/Gateway.sol
@@ -282,7 +282,7 @@ contract Gateway is IGateway, IInitializable {
             }
         }
 
-        bytes memory call = abi.encodeCall(AgentExecutor.execute, params.payload);
+        bytes memory call = abi.encodeCall(AgentExecutor.execute, (command, commandParams));
 
         (bool success, bytes memory returndata) = Agent(payable(agent)).invoke(AGENT_EXECUTOR, call);
         if (!success) {