Upgrade gateway on Sepolia for westend (#1407)

* Log gateway codehash

* Upgrade Gateway on Westend-Sepolia

Author: yrong

contracts/scripts/DeployLocalGatewayLogic.sol

@@ -7,6 +7,7 @@ import {Gateway} from "../src//Gateway.sol";
 import {ParaID} from "../src//Types.sol";
 import {Script} from "forge-std/Script.sol";
 import {stdJson} from "forge-std/StdJson.sol";
+import {console} from "forge-std/console.sol";
 
 contract DeployLocalGatewayLogic is Script {
     using stdJson for string;
@@ -28,7 +29,7 @@ contract DeployLocalGatewayLogic is Script {
 
         AgentExecutor executor = new AgentExecutor();
 
-        new Gateway(
+        Gateway gatewayLogic = new Gateway(
             address(beefyClient),
             address(executor),
             bridgeHubParaID,
@@ -37,6 +38,10 @@ contract DeployLocalGatewayLogic is Script {
             maxDestinationFee
         );
 
+        console.log("Gateway contract address: %s", address(gatewayLogic));
+        console.log("Gateway contract codehash:");
+        console.logBytes32(address(gatewayLogic).codehash);
+
         vm.stopBroadcast();
     }
 }

contracts/scripts/upgrades/westend/DeployGateway20250311.sol

@@ -0,0 +1,47 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
+pragma solidity 0.8.28;
+
+import {AgentExecutor} from "../../../src/AgentExecutor.sol";
+import {Gateway20250311} from "../../../src/upgrades/westend/Gateway20250311.sol";
+import {ParaID} from "../../../src/Types.sol";
+import {Script} from "forge-std/Script.sol";
+import {stdJson} from "forge-std/StdJson.sol";
+import {console} from "forge-std/console.sol";
+
+contract DeployGateway20250311 is Script {
+    using stdJson for string;
+
+    function setUp() public {}
+
+    function run() public {
+        uint256 privateKey = vm.envUint("PRIVATE_KEY");
+        address deployer = vm.rememberKey(privateKey);
+        vm.startBroadcast(deployer);
+
+        address beefyClient = vm.envAddress("BEEFY_CLIENT_CONTRACT_ADDRESS");
+
+        ParaID bridgeHubParaID = ParaID.wrap(uint32(vm.envUint("BRIDGE_HUB_PARAID")));
+        bytes32 bridgeHubAgentID = vm.envBytes32("BRIDGE_HUB_AGENT_ID");
+
+        uint8 foreignTokenDecimals = uint8(vm.envUint("FOREIGN_TOKEN_DECIMALS"));
+        uint128 maxDestinationFee = uint128(vm.envUint("RESERVE_TRANSFER_MAX_DESTINATION_FEE"));
+
+        AgentExecutor executor = new AgentExecutor();
+
+        Gateway20250311 gatewayLogic = new Gateway20250311(
+            address(beefyClient),
+            address(executor),
+            bridgeHubParaID,
+            bridgeHubAgentID,
+            foreignTokenDecimals,
+            maxDestinationFee
+        );
+
+        console.log("Gateway contract address: %s", address(gatewayLogic));
+        console.log("Gateway contract codehash:");
+        console.logBytes32(address(gatewayLogic).codehash);
+
+        vm.stopBroadcast();
+    }
+}

contracts/src/upgrades/westend/Gateway20250311.sol

@@ -0,0 +1,34 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
+pragma solidity 0.8.28;
+
+import "../../Gateway.sol";
+
+// New `Gateway` logic contract with updated Token contract
+contract Gateway20250311 is Gateway {
+    constructor(
+        address beefyClient,
+        address agentExecutor,
+        ParaID bridgeHubParaID,
+        bytes32 bridgeHubAgentID,
+        uint8 foreignTokenDecimals,
+        uint128 destinationMaxTransferFee
+    )
+        Gateway(
+            beefyClient,
+            agentExecutor,
+            bridgeHubParaID,
+            bridgeHubAgentID,
+            foreignTokenDecimals,
+            destinationMaxTransferFee
+        )
+    {}
+
+    // Override parent initializer to prevent re-initialization of storage.
+    function initialize(bytes memory) external override {
+        // Ensure that arbitrary users cannot initialize storage in this logic contract.
+        if (ERC1967.load() == address(0)) {
+            revert Unauthorized();
+        }
+    }
+}