Add upgrade contract for core

zkemail · Sep 5, 2024 · c10b443 · c10b443
1 parent 4a79b70
commit c10b443
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 3 deletions.
diff --git a/packages/contracts/script/UpgradeCore.s.sol b/packages/contracts/script/UpgradeCore.s.sol
@@ -0,0 +1,42 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+import {ERC1967Proxy} from "@openzeppelin/contracts/proxy/ERC1967/ERC1967Proxy.sol";
+import "forge-std/Script.sol";
+import "forge-std/console.sol";
+import "../src/EmailWalletCore.sol";
+
+contract Upgrade is Script {
+    function run() external {
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
+        if (deployerPrivateKey == 0) {
+            console.log("PRIVATE_KEY env var not set");
+            return;
+        }
+
+        address core = vm.envAddress("CORE");
+        if (core == address(0)) {
+            console.log("CORE env var not set. Deploy EmailWalletCore and set env var");
+            return;
+        }
+
+        vm.startBroadcast(deployerPrivateKey);
+
+        // FIXME Set new TokenRegistry implementation
+        // e.g. TokenRegistryV2 tokenRegistryImpl = new TokenRegistryV2();
+        EmailWalletCore coreImpl = new EmailWalletCore();
+
+        EmailWalletCore coreProxy = EmailWalletCore(payable(core));
+        coreProxy.upgradeTo(address(coreImpl));
+
+        // If you want to call some v2 function, refer to the following steps
+        //
+        // TokenRegistryV2 tokenRegistryV2 = TokenRegistryV2(address(tokenRegistry));
+        // address usdc = tokenRegistry.getTokenAddress(0, "USDC");
+
+        vm.stopBroadcast();
+
+        console.log("Core implementation deployed at: %s", address(coreImpl));
+        console.log("---- DONE ----");
+    }
+}
diff --git a/...acts/script/XX_UpgradeTokenRegistry.s.sol → ...ntracts/script/UpgradeTokenRegistry.s.sol b/...acts/script/XX_UpgradeTokenRegistry.s.sol → ...ntracts/script/UpgradeTokenRegistry.s.sol
@@ -6,9 +6,8 @@ import "forge-std/Script.sol";
 import "forge-std/console.sol";
 import "../src/utils/TokenRegistry.sol";
 
-contract Deploy is Script {
+contract Upgrade is Script {
     function run() external {
-
         uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
         if (deployerPrivateKey == 0) {
             console.log("PRIVATE_KEY env var not set");
@@ -34,7 +33,7 @@ contract Deploy is Script {
         //
         // TokenRegistryV2 tokenRegistryV2 = TokenRegistryV2(address(tokenRegistry));
         // address usdc = tokenRegistry.getTokenAddress(0, "USDC");
-                
+
         vm.stopBroadcast();
 
         console.log("TokenRegistry implementation deployed at: %s", address(tokenRegistryImpl));