harb/onchain/script/UpgradeOptimizer.sol

51 lines
2 KiB
Solidity
Raw Normal View History

// SPDX-License-Identifier: GPL-3.0-or-later
pragma solidity ^0.8.19;
import "../src/OptimizerV3.sol";
import { UUPSUpgradeable } from "@openzeppelin/proxy/utils/UUPSUpgradeable.sol";
import "forge-std/Script.sol";
/**
* @title UpgradeOptimizer
* @notice Upgrades an existing Optimizer UUPS proxy to OptimizerV3 implementation.
* @dev Usage:
* OPTIMIZER_PROXY=0x... forge script script/UpgradeOptimizer.sol \
* --rpc-url <RPC_URL> --broadcast
*
* The caller must be the proxy admin (the address that called initialize()).
*/
contract UpgradeOptimizer is Script {
function run() public {
address proxyAddress = vm.envAddress("OPTIMIZER_PROXY");
require(proxyAddress != address(0), "OPTIMIZER_PROXY env var required");
string memory seedPhrase = vm.readFile(".secret");
uint256 privateKey = vm.deriveKey(seedPhrase, 0);
vm.startBroadcast(privateKey);
address sender = vm.addr(privateKey);
console.log("\n=== Optimizer UUPS Upgrade ===");
console.log("Proxy address:", proxyAddress);
console.log("Admin (sender):", sender);
// Deploy new OptimizerV3 implementation
OptimizerV3 newImpl = new OptimizerV3();
console.log("New OptimizerV3 implementation:", address(newImpl));
// Upgrade proxy to new implementation (no reinitialize needed — storage layout compatible)
UUPSUpgradeable(proxyAddress).upgradeTo(address(newImpl));
console.log("Proxy upgraded to OptimizerV3");
// Verify upgrade by calling getLiquidityParams through the proxy
OptimizerV3 upgraded = OptimizerV3(proxyAddress);
(uint256 ci, uint256 as_, uint24 aw, uint256 dd) = upgraded.getLiquidityParams();
console.log("\n=== Post-Upgrade Verification ===");
console.log("capitalInefficiency:", ci);
console.log("anchorShare:", as_);
console.log("anchorWidth:", uint256(aw));
console.log("discoveryDepth:", dd);
vm.stopBroadcast();
}
}