refactor: upgrade script

alt-research · Apr 6, 2024 · 0c85569 · 0c85569
1 parent 3aba153
commit 0c85569
Showing 1 changed file with 6 additions and 21 deletions.
diff --git a/contracts/script/MachServiceManagerUpgrader.s.sol b/contracts/script/MachServiceManagerUpgrader.s.sol
@@ -4,32 +4,17 @@ pragma solidity ^0.8.12;
 import "forge-std/Script.sol";
 import "@openzeppelin/contracts/proxy/transparent/ProxyAdmin.sol";
 import {MachServiceManager} from "../src/core/MachServiceManager.sol";
-import {IStakeRegistry} from "eigenlayer-middleware/interfaces/IStakeRegistry.sol";
-import {IRegistryCoordinator} from "eigenlayer-middleware/interfaces/IRegistryCoordinator.sol";
-import {IAVSDirectory} from "eigenlayer-core/contracts/interfaces/IAVSDirectory.sol";
 
 contract MachServiceManagerUpgrader is Script {
     function run() external {
-        address proxyAdminAddr = vm.envAddress("PROXY_ADMIN");
         address machServiceManager = vm.envAddress("SERVICE_MANAGER");
-        address avsDirectory = vm.envAddress("AVS_DIRECTORY");
-        address registryCoordinator = vm.envAddress("REGISTRY_COORDINATOR");
-        address stakeRegistry = vm.envAddress("STAKE_REGISTRY");
-        uint256 chainId = vm.envUint("CHAIN_ID");
-
-        ProxyAdmin machAVSProxyAdmin = ProxyAdmin(proxyAdminAddr);
-
-        // 1. deploy new implementation contract
-        MachServiceManager machServiceManagerImplementation = new MachServiceManager(
-            IAVSDirectory(avsDirectory),
-            IRegistryCoordinator(registryCoordinator),
-            IStakeRegistry(stakeRegistry),
-            chainId
-        );
-
-        // 2. call upgrade
+        address newImplAddress = vm.envAddress("NEW_IMPL_ADDRESS");
+        address proxyAdmin = vm.envAddress("PROXY_ADMIN");
+        ProxyAdmin machAVSProxyAdmin = ProxyAdmin(proxyAdmin);
+        vm.startBroadcast();
         machAVSProxyAdmin.upgrade(
-            TransparentUpgradeableProxy(payable(machServiceManager)), address(machServiceManagerImplementation)
+            TransparentUpgradeableProxy(payable(address(machServiceManager))), address(newImplAddress)
         );
+        vm.stopBroadcast();
     }
 }