90 lines
4.1 KiB
Solidity
90 lines
4.1 KiB
Solidity
// SPDX-License-Identifier: UNLICENSED
|
|
pragma solidity ^0.8.30;
|
|
|
|
import "forge-std/console2.sol";
|
|
import {Script} from "../lib/forge-std/src/Script.sol";
|
|
import {IERC3156FlashBorrower} from "../lib/openzeppelin-contracts/contracts/interfaces/IERC3156FlashBorrower.sol";
|
|
import {IERC20} from "../lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
|
|
import {IPartyInfo} from "../src/IPartyInfo.sol";
|
|
import {IPartyPool} from "../src/IPartyPool.sol";
|
|
import {MockFlashBorrower} from "../test/MockFlashBorrower.sol";
|
|
import {stdJson} from "forge-std/StdJson.sol";
|
|
import {IPartyPlanner} from "../src/IPartyPlanner.sol";
|
|
|
|
|
|
contract ExercisePOC is Script {
|
|
|
|
IPartyPlanner private immutable planner;
|
|
IPartyInfo private immutable info;
|
|
|
|
constructor() {
|
|
require(block.chainid==1, 'Not Ethereum');
|
|
string memory root = vm.projectRoot();
|
|
string memory path = string.concat(root, "/deployment/liqp-deployments.json");
|
|
string memory json = vm.readFile(path);
|
|
bytes memory partyPlannerRaw = stdJson.parseRaw(json, ".1.v1.PartyPlanner");
|
|
planner = IPartyPlanner(abi.decode(partyPlannerRaw, (address)));
|
|
bytes memory partyInfoRaw = stdJson.parseRaw(json, ".1.v1.PartyInfo");
|
|
info = IPartyInfo(abi.decode(partyInfoRaw, (address)));
|
|
}
|
|
|
|
function run() public {
|
|
exercise(planner.getAllPools(0,1)[0]);
|
|
}
|
|
|
|
function exercise(IPartyPool pool) internal {
|
|
console2.log('Exercising pool at', address(pool));
|
|
uint8 WETH_index = 3;
|
|
// gather tokens and denominators
|
|
IERC20[] memory tokens = pool.allTokens();
|
|
uint256 n = tokens.length;
|
|
IERC20 WETH = tokens[WETH_index];
|
|
require(address(WETH) == address(0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2), 'Expected WETH as the fourth token');
|
|
|
|
// approve all
|
|
for (uint256 i=0; i<n; i++)
|
|
tokens[i].approve(address(pool), type(uint256).max);
|
|
|
|
// 1) Proportional mint (request some LP)
|
|
uint256 lpToMint = pool.totalSupply() / 100; // 1% of the pool size
|
|
// payer = this contract, receiver = this contract
|
|
uint256 minted = pool.mint(msg.sender, msg.sender, lpToMint, 0);
|
|
|
|
// 2) Proportional burn (withdraw a small, non-even amount of LP)
|
|
pool.approve(address(pool), minted);
|
|
pool.burn(msg.sender, msg.sender, minted, 0, false);
|
|
|
|
// 3) Flash loan: borrow token 0 and immediately repay in callback
|
|
// deploy a temporary borrower that repays amount + fee back to the pool
|
|
MockFlashBorrower borrower = new MockFlashBorrower();
|
|
uint256 flashAmt = WETH.balanceOf(address(pool)); // flash the maximum
|
|
uint256 flashFee = info.flashFee(pool, address(WETH), flashAmt);
|
|
// send the fee amount to the flash borrower
|
|
WETH.transfer(address(borrower), flashFee);
|
|
// pass the pool address in data so borrower can repay back to this pool
|
|
bytes memory data = abi.encode(address(pool));
|
|
// call flashLoan (ignore success boolean/revert)
|
|
pool.flashLoan(IERC3156FlashBorrower(address(borrower)), address(WETH), flashAmt, data);
|
|
|
|
// 4) swapMint (single-token mint -> LP)
|
|
uint256 amountIn = WETH.balanceOf(address(pool)) / 100; // trade 1% of what's in the pool
|
|
require( WETH.balanceOf(msg.sender) >= amountIn, 'Insufficient WETH for swapMint');
|
|
(, uint256 lpMinted,) = pool.swapMint(msg.sender, msg.sender, WETH_index, amountIn, 0);
|
|
|
|
// 5) regular swap (token 0 -> last token)
|
|
require( WETH.balanceOf(msg.sender) >= amountIn, 'Insufficient WETH for swap');
|
|
WETH.approve(address(pool), amountIn);
|
|
uint256 inputIndex = WETH_index;
|
|
uint256 outputIndex = 0;
|
|
pool.swap(msg.sender, bytes4(0), msg.sender, WETH_index, outputIndex, amountIn, int128(0), 0, false, '');
|
|
|
|
// 6) Collect protocol fees now (after some swaps) so some will have been moved out
|
|
pool.collectProtocolFees();
|
|
|
|
// 7) Final swap-style operation: burnSwap (burn LP then swap to single asset)
|
|
// ensure we have some LP allowance
|
|
pool.burnSwap(msg.sender, msg.sender, lpMinted, WETH_index, 0, false);
|
|
}
|
|
|
|
}
|