// SPDX-License-Identifier: MIT
pragma solidity ^0.8.19;

import {Script, console} from "forge-std/Script.sol";
import {DODOPMMIntegration} from "../../contracts/dex/DODOPMMIntegration.sol";
import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";

/**
 * @title AddLiquidityPMMPoolsChain138
 * @notice Add liquidity to the three DODO PMM pools on Chain 138 (cUSDT/cUSDC, cUSDT/USDT, cUSDC/USDC).
 * @dev Env: PRIVATE_KEY, RPC_URL_138, DODO_PMM_INTEGRATION_ADDRESS (or DODO_PMM_INTEGRATION),
 *      POOL_CUSDTCUSDC, POOL_CUSDTUSDT, POOL_CUSDCUSDC,
 *      ADD_LIQUIDITY_BASE_AMOUNT, ADD_LIQUIDITY_QUOTE_AMOUNT (e.g. 1000000e6 for 1M units, 6 decimals).
 *      Optional: ADD_LIQUIDITY_CUSDTCUSDC_BASE, ADD_LIQUIDITY_CUSDTCUSDC_QUOTE, etc. for per-pool amounts.
 */
contract AddLiquidityPMMPoolsChain138 is Script {
    function run() external {
        uint256 pk = vm.envUint("PRIVATE_KEY");
        address integrationAddr = vm.envAddress("DODO_PMM_INTEGRATION");
        if (integrationAddr == address(0)) integrationAddr = vm.envAddress("DODO_PMM_INTEGRATION_ADDRESS");
        require(integrationAddr != address(0), "DODO_PMM_INTEGRATION not set");

        address poolCusdtCusdc = vm.envOr("POOL_CUSDTCUSDC", address(0));
        address poolCusdtUsdt = vm.envOr("POOL_CUSDTUSDT", address(0));
        address poolCusdcUsdc = vm.envOr("POOL_CUSDCUSDC", address(0));

        uint256 defaultBase = vm.envOr("ADD_LIQUIDITY_BASE_AMOUNT", uint256(0));
        uint256 defaultQuote = vm.envOr("ADD_LIQUIDITY_QUOTE_AMOUNT", uint256(0));

        DODOPMMIntegration integration = DODOPMMIntegration(integrationAddr);
        address cusdt = integration.compliantUSDT();
        address cusdc = integration.compliantUSDC();
        address usdt = integration.officialUSDT();
        address usdc = integration.officialUSDC();

        vm.startBroadcast(pk);

        if (poolCusdtCusdc != address(0) && (defaultBase > 0 || defaultQuote > 0)) {
            uint256 b = vm.envOr("ADD_LIQUIDITY_CUSDTCUSDC_BASE", defaultBase);
            uint256 q = vm.envOr("ADD_LIQUIDITY_CUSDTCUSDC_QUOTE", defaultQuote);
            if (b > 0 && q > 0) {
                _approveAndAdd(integration, cusdt, cusdc, poolCusdtCusdc, b, q);
                console.log("Added liquidity to cUSDT/cUSDC pool:", poolCusdtCusdc);
            }
        }
        if (poolCusdtUsdt != address(0) && (defaultBase > 0 || defaultQuote > 0)) {
            uint256 b = vm.envOr("ADD_LIQUIDITY_CUSDTUSDT_BASE", defaultBase);
            uint256 q = vm.envOr("ADD_LIQUIDITY_CUSDTUSDT_QUOTE", defaultQuote);
            if (b > 0 && q > 0) {
                _approveAndAdd(integration, cusdt, usdt, poolCusdtUsdt, b, q);
                console.log("Added liquidity to cUSDT/USDT pool:", poolCusdtUsdt);
            }
        }
        if (poolCusdcUsdc != address(0) && (defaultBase > 0 || defaultQuote > 0)) {
            uint256 b = vm.envOr("ADD_LIQUIDITY_CUSDCUSDC_BASE", defaultBase);
            uint256 q = vm.envOr("ADD_LIQUIDITY_CUSDCUSDC_QUOTE", defaultQuote);
            if (b > 0 && q > 0) {
                _approveAndAdd(integration, cusdc, usdc, poolCusdcUsdc, b, q);
                console.log("Added liquidity to cUSDC/USDC pool:", poolCusdcUsdc);
            }
        }

        vm.stopBroadcast();
    }

    function _approveAndAdd(
        DODOPMMIntegration integration,
        address baseToken,
        address quoteToken,
        address pool,
        uint256 baseAmount,
        uint256 quoteAmount
    ) internal {
        IERC20(baseToken).approve(address(integration), type(uint256).max);
        IERC20(quoteToken).approve(address(integration), type(uint256).max);
        integration.addLiquidity(pool, baseAmount, quoteAmount);
    }
}