harb/onchain/script/UpgradeOptimizer.sol

// SPDX-License-Identifier: GPL-3.0-or-later
pragma solidity ^0.8.19;

import "../src/OptimizerV3Push3.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 OptimizerV3Push3 implementation
        OptimizerV3Push3 newImpl = new OptimizerV3Push3();
        console.log("New OptimizerV3Push3 implementation:", address(newImpl));

        // Upgrade proxy to new implementation (no reinitialize needed — storage layout compatible)
        UUPSUpgradeable(proxyAddress).upgradeTo(address(newImpl));
        console.log("Proxy upgraded to OptimizerV3Push3");

        // Verify upgrade by calling isBullMarket through the proxy
        OptimizerV3Push3 upgraded = OptimizerV3Push3(proxyAddress);
        bool bull = upgraded.isBullMarket(0, 0);
        console.log("\n=== Post-Upgrade Verification ===");
        console.log("isBullMarket(0,0):", bull);

        vm.stopBroadcast();
    }
}