harb/onchain/script/DeployBase.sol

pragma solidity ^0.8.19;

import "../src/Kraiken.sol";

import { LiquidityManager } from "../src/LiquidityManager.sol";
import "../src/Optimizer.sol";
import "../src/Stake.sol";
import "../src/helpers/UniswapHelpers.sol";
import { IWETH9 } from "../src/interfaces/IWETH9.sol";
import { ERC1967Proxy } from "@openzeppelin/proxy/ERC1967/ERC1967Proxy.sol";
import "@uniswap-v3-core/interfaces/IUniswapV3Factory.sol";
import "@uniswap-v3-core/interfaces/IUniswapV3Pool.sol";
import "forge-std/Script.sol";

uint24 constant FEE = uint24(10_000);

/**
 * @title SeedSwapper
 * @notice One-shot helper deployed during DeployBase.run() to perform the initial seed buy.
 *         Executing a small buy before the protocol opens eliminates the cumulativeVolume==0
 *         front-run window: after the seed recenter, VWAP has a real anchor and the bootstrap
 *         path in LiquidityManager.recenter() is never reachable by external users.
 */
contract SeedSwapper {
    IWETH9 private immutable weth;
    IUniswapV3Pool private immutable pool;
    bool private immutable token0isWeth;

    constructor(address _weth, address _pool, bool _token0isWeth) {
        weth = IWETH9(_weth);
        pool = IUniswapV3Pool(_pool);
        token0isWeth = _token0isWeth;
    }

    /// @notice Wraps msg.value ETH to WETH and swaps it for KRK (buying KRK).
    ///         The KRK output is sent to `recipient`.  The fee generated by the swap
    ///         is captured in the LM's positions, so the subsequent recenter() call
    ///         will collect a non-zero ethFee and record VWAP.
    function executeSeedBuy(address recipient) external payable {
        weth.deposit{ value: msg.value }();

        // zeroForOne=true  when WETH is token0: sell token0(WETH) → token1(KRK)
        // zeroForOne=false when WETH is token1: sell token1(WETH) → token0(KRK)
        bool zeroForOne = token0isWeth;

        // Price limits: allow the swap to reach the extreme of the range.
        uint160 priceLimit = zeroForOne
            ? 4295128740 // TickMath.MIN_SQRT_RATIO + 1
            : 1461446703485210103287273052203988822378723970341; // TickMath.MAX_SQRT_RATIO - 1

        pool.swap(recipient, zeroForOne, int256(msg.value), priceLimit, "");
    }

    /// @notice Uniswap V3 callback: pay the WETH owed for the seed buy.
    function uniswapV3SwapCallback(int256 amount0Delta, int256 amount1Delta, bytes calldata) external {
        require(msg.sender == address(pool), "only pool");
        int256 wethDelta = token0isWeth ? amount0Delta : amount1Delta;
        if (wethDelta > 0) {
            weth.transfer(msg.sender, uint256(wethDelta));
        }
    }
}

contract DeployBase is Script {
    using UniswapHelpers for IUniswapV3Pool;

    bool public token0isWeth;
    address public feeDest;
    address public weth;
    address public v3Factory;
    address public optimizer;

    // Seed amounts for VWAP bootstrap.
    // Kept small: deployer only needs this ETH on top of gas.
    // With very thin bootstrap positions, even 0.005 ETH moves the price >400 ticks.
    uint256 internal constant SEED_LM_ETH = 0.01 ether;
    uint256 internal constant SEED_SWAP_ETH = 0.005 ether;

    // Deployed contracts
    Kraiken public kraiken;
    Stake public stake;
    LiquidityManager public liquidityManager;
    IUniswapV3Pool public pool;

    function run() public {
        string memory seedPhrase = vm.readFile(".secret");
        uint256 privateKey = vm.deriveKey(seedPhrase, 0);
        vm.startBroadcast(privateKey);
        address sender = vm.addr(privateKey);

        console.log("Deploying from:", sender);

        // Deploy Kraiken token
        kraiken = new Kraiken("Kraiken", "KRK");
        console.log("Kraiken deployed at:", address(kraiken));

        // Determine token ordering
        token0isWeth = address(weth) < address(kraiken);
        console.log("token0isWeth:", token0isWeth);

        // Deploy Stake contract
        stake = new Stake(address(kraiken), feeDest);
        console.log("Stake deployed at:", address(stake));

        // Set staking pool in Kraiken
        kraiken.setStakingPool(address(stake));

        // Get or create Uniswap V3 pool
        IUniswapV3Factory factory = IUniswapV3Factory(v3Factory);
        address liquidityPool = factory.getPool(weth, address(kraiken), FEE);
        if (liquidityPool == address(0)) {
            liquidityPool = factory.createPool(weth, address(kraiken), FEE);
            console.log("Uniswap pool created at:", liquidityPool);
        } else {
            console.log("Using existing pool at:", liquidityPool);
        }
        pool = IUniswapV3Pool(liquidityPool);

        // Initialize pool at 1 cent price if not already initialized
        try pool.slot0() returns (uint160, int24, uint16, uint16, uint16, uint8, bool) {
            console.log("Pool already initialized");
        } catch {
            pool.initializePoolFor1Cent(token0isWeth);
            console.log("Pool initialized");
        }

        // Deploy Optimizer (if not already deployed)
        address optimizerAddress;
        if (optimizer == address(0)) {
            Optimizer optimizerImpl = new Optimizer();
            bytes memory params = abi.encodeWithSignature("initialize(address,address)", address(kraiken), address(stake));
            ERC1967Proxy proxy = new ERC1967Proxy(address(optimizerImpl), params);
            optimizerAddress = address(proxy);
            console.log("Optimizer deployed at:", optimizerAddress);
        } else {
            optimizerAddress = optimizer;
            console.log("Using existing optimizer at:", optimizerAddress);
        }

        // Deploy LiquidityManager
        liquidityManager = new LiquidityManager(v3Factory, weth, address(kraiken), optimizerAddress);
        console.log("LiquidityManager deployed at:", address(liquidityManager));

        // Set liquidity manager in Kraiken
        kraiken.setLiquidityManager(address(liquidityManager));

        // =====================================================================
        // VWAP Bootstrap -> seed trade during deployment
        //
        // The cumulativeVolume==0 path in recenter() records VWAP from whatever
        // price exists at the time of the first fee event.  An attacker who
        // front-runs deployment with a whale buy inflates that anchor.
        //
        // Fix: execute a small buy BEFORE handing control to users so that
        // cumulativeVolume>0 by the time the protocol is live.
        //
        // Deployer must have SEED_LM_ETH + SEED_SWAP_ETH available (≈0.015 ETH).
        // =====================================================================
        console.log("\nBootstrapping VWAP with seed trade...");

        // Step 1: Temporarily set deployer as feeDestination to call setRecenterAccess.
        liquidityManager.setFeeDestination(sender);
        liquidityManager.setRecenterAccess(sender);

        // Step 2: Fund LM and place initial bootstrap positions.
        (bool funded,) = address(liquidityManager).call{ value: SEED_LM_ETH }("");
        require(funded, "Failed to fund LM for seed bootstrap");
        liquidityManager.recenter();

        // Step 3: Seed buy -> generates a non-zero fee in the anchor position.
        SeedSwapper seedSwapper = new SeedSwapper(weth, address(pool), token0isWeth);
        seedSwapper.executeSeedBuy{ value: SEED_SWAP_ETH }(sender);

        // Step 4: Second recenter records VWAP (bootstrap path + ethFee > 0).
        liquidityManager.recenter();
        require(liquidityManager.cumulativeVolume() > 0, "VWAP bootstrap failed: cumulativeVolume is 0");
        console.log("VWAP bootstrapped -> cumulativeVolume:", liquidityManager.cumulativeVolume());

        // Step 5: Clean up -> revoke temporary access and set the real feeDestination.
        liquidityManager.revokeRecenterAccess();
        liquidityManager.setFeeDestination(feeDest);
        console.log("recenterAccess revoked, feeDestination set to", feeDest);

        console.log("\n=== Deployment Complete ===");
        console.log("Kraiken:", address(kraiken));
        console.log("Stake:", address(stake));
        console.log("Pool:", address(pool));
        console.log("LiquidityManager:", address(liquidityManager));
        console.log("Optimizer:", optimizerAddress);
        console.log("\nPost-deploy steps:");
        console.log("  1. Fund LiquidityManager with operational ETH (VWAP already bootstrapped)");
        console.log("  2. Set recenterAccess to txnBot: lm.setRecenterAccess(txnBot) from feeDestination");
        console.log("  3. txnBot can now call recenter()");

        vm.stopBroadcast();
    }
}