Add action contracts for migrating live orbit chains to espresso

.gitmodules

@@ -4,3 +4,6 @@
 [submodule "lib/arbitrum-sdk"]
 	path = lib/arbitrum-sdk
 	url = https://github.com/OffchainLabs/arbitrum-sdk
+[submodule "lib/nitro-contracts.git"]
+	path = lib/nitro-contracts
+	url = https://github.com/EspressoSystems/nitro-contracts

contracts/child-chain/arbos-upgrade/DeployArbOSUpgradeAction.s.sol

@@ -0,0 +1,17 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.9;
+
+import "forge-std/Script.sol";
+import "./EspressoArbOSUpgrade.sol";
+
+contract DeployArbOSUpgradeAction is Script{
+    function run() external{
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");  
+        uint64 upgradeTimestamp = uint64(vm.envUint("UPGRADE_TIMESTAMP"));      
+        vm.startBroadcast(deployerPrivateKey);
+        EspressoArbOSUpgrade arbOSUpgrade = new EspressoArbOSUpgrade(
+            upgradeTimestamp
+        );
+        vm.stopBroadcast();
+    }
+}

contracts/child-chain/arbos-upgrade/EspressoArbOSUpgrade.sol

@@ -0,0 +1,16 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.16;
+
+import "forge-std/Script.sol";
+import "./UpgradeArbOSVersionAtTimestampAction.sol";
+
+contract EspressoArbOSUpgrade is UpgradeArbOSVersionAtTimestampAction, Script   {
+    constructor(
+        uint64 upgradeTimestamp
+    )
+        UpgradeArbOSVersionAtTimestampAction(
+            35,
+            upgradeTimestamp
+        )
+        {}
+}

contracts/parent-chain/contract-upgrades/example-env/.env-espresso-osp-migration.example

@@ -0,0 +1,6 @@
+export NEW_OSP_ENTRY="0x7B5F0B437EE68A22992DdD629AA22525Fc5dfa9A"
+export NEW_WASM_MODULE_ROOT="0x43683fe4e619e9ee7c0d959259d10ff2a38f4325a2b8936c47d9162704fe072a"
+export CURRENT_OSP_ENTRY="0x4838B106FCe9647Bdf1E7877BF73cE8B0BAD5f97"
+export CURRENT_WASM_MODULE_ROOT="0xbc1026ff45c20ea97e9e6057224a5668ea78d8f885c9b14fc849238e8ef5c5dc"
+export ROLLUP_ADDRESS="0xF2099c4783921f44Ac988B67e743DAeFd4A00efd"
+export PROXY_ADMIN="0x94e2090Fb86906C799A67263e6beAe528b81B3D6"

contracts/parent-chain/contract-upgrades/example-env/.env-schedule-arbos-upgrade.example

@@ -0,0 +1 @@
+export UPGRADE_TIMESTAMP = "1723664126"

contracts/parent-chain/espresso-migration/DeployEspressoOsp.s.sol

@@ -0,0 +1,25 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.9;
+
+import "forge-std/Script.sol";
+import "nitro-contracts/osp/OneStepProverMemory.sol";
+import "nitro-contracts/osp/OneStepProverMath.sol";
+import "nitro-contracts/osp/OneStepProverHostIo.sol";
+import "nitro-contracts/osp/OneStepProver0.sol";
+import "nitro-contracts/osp/OneStepProofEntry.sol";
+
+
+contract DeployEspressoOsp is Script{
+    function run() external{
+        address hotshotAddr = vm.envAddress("HOTSHOT_ADDRESS");
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
+        vm.startBroadcast(deployerPrivateKey);
+        OneStepProofEntry newOSP = new OneStepProofEntry(
+            new OneStepProver0(),
+            new OneStepProverMemory(),
+            new OneStepProverMath(),
+            new OneStepProverHostIo(hotshotAddr)
+        );
+        vm.stopBroadcast();
+    }
+}

contracts/parent-chain/espresso-migration/DeployEspressoOspMigrationAction.s.sol

@@ -0,0 +1,28 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.9;
+
+import "forge-std/Script.sol";
+import "./EspressoOspMigrationAction.sol";
+
+contract DeployEspressoOspMigrationAction is Script{
+    function run() external{
+        uint256 deployerPrivateKey = vm.envUint("PRIVATE_KEY");
+        address newOspEntry = vm.envAddress("NEW_OSP_ENTRY");
+        bytes32 newWasmModuleRoot = vm.envBytes32("NEW_WASM_MODULE_ROOT");
+        address currentOspEntry = vm.envAddress("CURRENT_OSP_ENTRY");
+        bytes32 currentWasmModuleRoot = vm.envBytes32("CURRENT_WASM_MODULE_ROOT");
+        address rollup = vm.envAddress("ROLLUP_ADDRESS");
+        address proxyAdmin = vm.envAddress("PROXY_ADMIN");
+
+        vm.startBroadcast(deployerPrivateKey);
+        OspMigrationAction migrationAction = new OspMigrationAction(
+            newOspEntry,
+            newWasmModuleRoot,
+            currentOspEntry,
+            currentWasmModuleRoot,
+            rollup,
+            proxyAdmin
+        );
+        vm.stopBroadcast();
+    }
+}

contracts/parent-chain/espresso-migration/EspressoOspMigrationAction.sol

@@ -0,0 +1,102 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.9;
+
+import "forge-std/Script.sol";
+import "@arbitrum/nitro-contracts/src/challenge/IChallengeManager.sol";
+import "@arbitrum/nitro-contracts/src/rollup/IRollupCore.sol";
+import "@arbitrum/nitro-contracts/src/rollup/IRollupAdmin.sol";
+
+import "@openzeppelin/contracts/proxy/transparent/ProxyAdmin.sol";
+import "@openzeppelin/contracts/utils/Address.sol";
+
+contract OspMigrationAction{ 
+
+    error IncorrectWasmModuleRoot(bytes32 incorrectAddr);
+
+    error AddressIsNotContract(address incorrectAddr);
+
+    error ChallengeManagerUpdated(address newChallengeManagerAddr);
+
+    error OspNotUpgraded(address oldOspAddress);
+
+    error WasmModuleRootNotUpdated(bytes32 oldWasmModuleRoot);
+
+    address public immutable newOspEntry;
+    bytes32 public immutable newWasmModuleRoot;
+    bytes32 public immutable currentWasmModuleRoot;
+    address public immutable currentOspEntry;
+    address public immutable rollup;
+    address public immutable proxyAdmin;
+
+
+    constructor(
+        address _newOspEntry,
+        bytes32 _newWasmModuleRoot,
+        address _currentOspEntry,
+        bytes32 _currentWasmModuleRoot,
+        address _rollup,
+        address _proxyAdmin
+    ){
+        if(_newWasmModuleRoot == bytes32(0)){
+            revert IncorrectWasmModuleRoot(_newWasmModuleRoot);
+        }
+
+        if(_currentWasmModuleRoot == bytes32(0)){
+            revert IncorrectWasmModuleRoot(_currentWasmModuleRoot);
+        }
+
+        if(!Address.isContract(_newOspEntry)){
+            revert AddressIsNotContract(_newOspEntry);
+        }
+
+        if(!Address.isContract(_currentOspEntry)){
+            revert AddressIsNotContract(_currentOspEntry);
+        }
+
+        if(!Address.isContract(_proxyAdmin)){
+            revert AddressIsNotContract(_proxyAdmin);
+        }
+
+        if(!Address.isContract(_rollup)){
+            revert AddressIsNotContract(_rollup);
+        }
+        newOspEntry = _newOspEntry;
+        newWasmModuleRoot = _newWasmModuleRoot;
+        currentOspEntry = _currentOspEntry;
+        currentWasmModuleRoot = _currentWasmModuleRoot;
+        rollup = _rollup;
+        proxyAdmin = _proxyAdmin;
+    }
+
+    function perform() external{
+        //Handle assertions in the perform function as we shouldn't be storing local state for delegated calls.
+
+        // set the new challenge manager impl
+        TransparentUpgradeableProxy challengeManager =
+            TransparentUpgradeableProxy(payable(address(IRollupCore(rollup).challengeManager())));
+        address chalManImpl = ProxyAdmin(proxyAdmin).getProxyImplementation(challengeManager);
+        ProxyAdmin(proxyAdmin).upgradeAndCall(
+            challengeManager,
+            chalManImpl, // Use the rollups current challenge manager as we only need to upgrade the OSP
+            abi.encodeWithSelector(IChallengeManager.postUpgradeInit.selector, IOneStepProofEntry(newOspEntry), currentWasmModuleRoot, IOneStepProofEntry(currentOspEntry))
+        );
+        address postUpgradeChalManAddr = ProxyAdmin(proxyAdmin).getProxyImplementation(challengeManager);
+
+        if(postUpgradeChalManAddr != chalManImpl){
+            revert ChallengeManagerUpdated(postUpgradeChalManAddr);
+        }
+        IOneStepProofEntry newOsp = IChallengeManager(address(challengeManager)).osp();
+
+        if(newOsp != IOneStepProofEntry(newOspEntry) ){
+            revert OspNotUpgraded(address(newOsp));
+        }
+
+        IRollupAdmin(rollup).setWasmModuleRoot(newWasmModuleRoot);
+
+        bytes32 postUpgradeWasmModuleRoot = IRollupCore(rollup).wasmModuleRoot();
+
+        if(postUpgradeWasmModuleRoot != newWasmModuleRoot){
+            revert WasmModuleRootNotUpdated(postUpgradeWasmModuleRoot);
+        }
+    }
+}