107 lines
5.3 KiB
Solidity
107 lines
5.3 KiB
Solidity
// SPDX-License-Identifier: UNLICENSED
|
|
pragma solidity ^0.8.30;
|
|
|
|
import "forge-std/console2.sol";
|
|
import {CommonBase} from "../lib/forge-std/src/Base.sol";
|
|
import {Script} from "../lib/forge-std/src/Script.sol";
|
|
import {StdChains} from "../lib/forge-std/src/StdChains.sol";
|
|
import {StdCheatsSafe} from "../lib/forge-std/src/StdCheats.sol";
|
|
import {stdJson} from "../lib/forge-std/src/StdJson.sol";
|
|
import {StdUtils} from "../lib/forge-std/src/StdUtils.sol";
|
|
import {IERC3156FlashBorrower} from "../lib/openzeppelin-contracts/contracts/interfaces/IERC3156FlashBorrower.sol";
|
|
import {IERC20} from "../lib/openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
|
|
import {SafeERC20} from "../lib/openzeppelin-contracts/contracts/token/ERC20/utils/SafeERC20.sol";
|
|
import {IPartyInfo} from "../src/IPartyInfo.sol";
|
|
import {IPartyPlanner} from "../src/IPartyPlanner.sol";
|
|
import {IPartyPool} from "../src/IPartyPool.sol";
|
|
import {MockFlashBorrower} from "../test/MockFlashBorrower.sol";
|
|
|
|
|
|
contract ExercisePOC is Script {
|
|
// Successful on mainnet POC pool in blocks 23849608 and 23849609
|
|
// mint 0x59fd174ba3230aba2a29ee31d9c5ea24cfbfd9b6b732dee5f1ec2d5d3eae2209
|
|
// burn 0x9e4a722a91e04fd9a29010046d1f2122ec5398847ea0d045ca8cffc36dd51c31
|
|
// flash 0x4092966104f79c1406b6c384c3ed06d8d6ae9e458d03e5ef15bc0af09657e0d0
|
|
// swapMint 0x34b5a40babf18ece67696709a0b6d219ca5b1f70a3c6e62c5c5eb6e4e272ba27
|
|
// swap 0x557fec6da56d59502fd1300fb77972ffe8b860c8fbd863acb94eb9db87b684de
|
|
// collectProtocolFees 0xc14b99318b71ecb94909d5288191fae1bb3ad6168443c2ed26ebf5816952a624
|
|
// burnSwap 0xd711d357d46d3c861b592ca5a68655a4bb71d8895acd1c25823c3fa128d14b13
|
|
|
|
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 {
|
|
IPartyPool pool = IPartyPool(0xfD07d9B82A45C2Cc9eDEb80103441e14801F325d);
|
|
console2.log('Exercising pool at', address(pool));
|
|
vm.startBroadcast();
|
|
exercise(pool);
|
|
vm.stopBroadcast();
|
|
}
|
|
|
|
function exercise(IPartyPool pool) internal {
|
|
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++)
|
|
SafeERC20.forceApprove(tokens[i], address(pool), type(uint256).max);
|
|
|
|
// 1) Proportional mint (request some LP)
|
|
uint256 lpToMint = pool.totalSupply() / 20; // 5% 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); // approval not needed for burns
|
|
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);
|
|
require(WETH.balanceOf(address(borrower)) == 0, 'flash borrower retained WETH');
|
|
|
|
// 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, inputIndex, 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);
|
|
}
|
|
|
|
}
|