evm/script: add upgrade support to deploy script

Author: kcsongor

evm/script/DeployWormholeNtt.s.sol

@@ -1,15 +1,63 @@
 // SPDX-License-Identifier: Apache 2
 pragma solidity >=0.8.8 <0.9.0;
 
-import {Script} from "forge-std/Script.sol";
+import {Script, console} from "forge-std/Script.sol";
 import {DeployWormholeNttBase} from "./helpers/DeployWormholeNttBase.sol";
+import {INttManager} from "../src/interfaces/INttManager.sol";
+import {IWormholeTransceiver} from "../src/interfaces/IWormholeTransceiver.sol";
+import "../src/interfaces/IManagerBase.sol";
+import "openzeppelin-contracts/contracts/token/ERC20/ERC20.sol";
+import {NttManager} from "../src/NttManager/NttManager.sol";
+
+interface IWormhole {
+    function chainId() external view returns (uint16);
+}
 
 contract DeployWormholeNtt is Script, DeployWormholeNttBase {
-    function run() public {
+    function run(
+        address wormhole,
+        address token,
+        address wormholeRelayer,
+        address specialRelayer,
+        IManagerBase.Mode mode
+    ) public {
         vm.startBroadcast();
 
-        // Sanity check deployment parameters.
-        DeploymentParams memory params = _readEnvVariables();
+        console.log("Deploying Wormhole Ntt...");
+        IWormhole wh = IWormhole(wormhole);
+
+        (bool success, bytes memory queriedDecimals) =
+            token.staticcall(abi.encodeWithSignature("decimals()"));
+
+        if (!success) {
+            console.log("Failed to query token decimals");
+            vm.stopBroadcast();
+            return;
+        }
+
+        uint8 decimals = abi.decode(queriedDecimals, (uint8));
+
+        uint16 chainId = wh.chainId();
+
+        console.log("Chain ID: ", chainId);
+
+        uint256 scale =
+            decimals > TRIMMED_DECIMALS ? uint256(10 ** (decimals - TRIMMED_DECIMALS)) : 1;
+
+        DeploymentParams memory params = DeploymentParams({
+            token: token,
+            mode: mode,
+            wormholeChainId: chainId,
+            rateLimitDuration: 86400,
+            shouldSkipRatelimiter: false,
+            wormholeCoreBridge: wormhole,
+            wormholeRelayerAddr: wormholeRelayer,
+            specialRelayerAddr: specialRelayer,
+            consistencyLevel: 202,
+            gasLimit: 500000,
+            // the trimming will trim this number to uint64.max
+            outboundLimit: uint256(type(uint64).max) * scale
+        });
 
         // Deploy NttManager.
         address manager = deployNttManager(params);
@@ -24,4 +72,27 @@ contract DeployWormholeNtt is Script, DeployWormholeNttBase {
 
         vm.stopBroadcast();
     }
+
+    function upgrade(address manager) public {
+        vm.startBroadcast();
+
+        NttManager nttManager = NttManager(manager);
+
+        console.log("Upgrading manager...");
+
+        uint64 rateLimitDuration = nttManager.rateLimitDuration();
+        bool shouldSkipRatelimiter = rateLimitDuration == 0;
+
+        NttManager implementation = new NttManager(
+            nttManager.token(),
+            nttManager.mode(),
+            nttManager.chainId(),
+            nttManager.rateLimitDuration(),
+            shouldSkipRatelimiter
+        );
+
+        nttManager.upgrade(address(implementation));
+
+        vm.stopBroadcast();
+    }
 }

evm/script/helpers/DeployWormholeNttBase.sol

@@ -47,8 +47,7 @@ contract DeployWormholeNttBase is ParseNttConfig {
 
         nttManagerProxy.initialize();
 
-        console2.log("NttManager deployed at: ");
-        console2.logBytes32(toUniversalAddress(address(nttManagerProxy)));
+        console2.log("NttManager:", address(nttManagerProxy));
 
         return address(nttManagerProxy);
     }
@@ -72,8 +71,7 @@ contract DeployWormholeNttBase is ParseNttConfig {
 
         transceiverProxy.initialize();
 
-        console2.log("Wormhole Transceiver deployed at: ");
-        console2.logBytes32(toUniversalAddress(address(transceiverProxy)));
+        console2.log("WormholeTransceiver:", address(transceiverProxy));
 
         return address(transceiverProxy);
     }