// 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();
    }
}