Merge pull request #32 from ProjectBabbage/feat/upgrade-script

Upgrade script

Author: QGarchery

makefile

@@ -17,12 +17,15 @@ build:
 	forge build --force --sizes
 
 deploy:
-	forge script script/DeployWeSplit.sol --private-key ${PRIVATE_KEY} --broadcast --etherscan-api-key ${POLYGONSCAN_API_KEY} --verify
+	forge script script/DeployWeSplit.sol --broadcast --etherscan-api-key ${POLYGONSCAN_API_KEY} --verify
+
+upgrade:
+	forge script script/UpgradeWeSplit.sol --broadcast --etherscan-api-key ${POLYGONSCAN_API_KEY} --verify -vvv
 
 verify:
 	forge verify-contract --chain 137 --watch --constructor-args ${ENCODED_CONSTRUCTOR_ARGS} ${ADDRESS} WeSplit ${POLYGONSCAN_API_KEY}
 
 gasprice:
 	cast gas-price --rpc-url ${FOUNDRY_ETH_RPC_URL}
 
-.PHONY: test storage-check build deploy verify gasprice
+.PHONY: test storage-check build deploy upgrade verify gasprice

script/DeployWeSplit.sol

@@ -2,14 +2,15 @@
 pragma solidity ^0.8.13;
 
 import "forge-std/Script.sol";
-import "../src/WeSplitProxy.sol";
-import "../src/WeSplit.sol";
+import "src/WeSplitProxy.sol";
+import "src/WeSplit.sol";
 
 contract DeployWeSplit is Script {
     bytes public constant emptyData = "";
 
     function run() public {
-        vm.startBroadcast();
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
+        vm.startBroadcast(deployerPrivateKey);
         WeSplit weSplitImplementation = new WeSplit();
         new WeSplitProxy(address(weSplitImplementation), emptyData);
         vm.stopBroadcast();

script/UpgradeWeSplit.sol

@@ -0,0 +1,17 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.13;
+
+import "forge-std/Script.sol";
+import "src/WeSplit.sol";
+
+contract UpgradeWeSplit is Script {
+    address constant internal weSplitProxy = 0x52decE2Fd883628eA46eBae183cd9D78a81Ef916;
+
+    function run() public {
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
+        vm.startBroadcast(deployerPrivateKey);
+        WeSplit newWeSplitImplementation = new WeSplit();
+        WeSplit(weSplitProxy).upgradeTo(address(newWeSplitImplementation));
+        vm.stopBroadcast();
+    }
+}

src/WeSplitProxy.sol

@@ -7,6 +7,7 @@ import "openzeppelin-contracts/proxy/ERC1967/ERC1967Proxy.sol";
 import "./WeSplitStructure.sol";
 
 /// @title WeSplit Proxy contract.
+/// @author Project Babbage.
 contract WeSplitProxy is ERC1967Proxy, WeSplitStructure {
     constructor(address _logic, bytes memory _data) ERC1967Proxy(_logic, _data) Ownable() {}
 }

test/TestArith.sol

@@ -1,11 +1,9 @@
 // SPDX-License-Identifier: MIT
 pragma solidity ^0.8.13;
 
-import "forge-std/Vm.sol";
-import "forge-std/console.sol";
 import "forge-std/Test.sol";
 
-import "../src/Arith.sol";
+import "src/Arith.sol";
 
 contract TestArith is Test {
     using Arith for uint256;

test/TestWeSplit.sol

@@ -4,11 +4,10 @@ pragma solidity ^0.8.13;
 import "../lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
 
 import "forge-std/Test.sol";
-import "forge-std/Vm.sol";
-import "forge-std/console.sol";
-import "../src/WeSplitProxy.sol";
-import "../src/WeSplit.sol";
-import "../src/Arith.sol";
+
+import "src/WeSplitProxy.sol";
+import "src/WeSplit.sol";
+import "src/Arith.sol";
 
 contract TestWeSplit is Test {
     using Arith for uint256;