Created
December 21, 2020 15:08
-
-
Save lychees/23514ee68fee18a20a83e0cbb4f25eae to your computer and use it in GitHub Desktop.
gogogo
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// Dependency file: contracts/interfaces/IUnisaveV2Pair.sol | |
// pragma solidity =0.6.12; | |
interface IUnisaveV2Pair { | |
event Approval(address indexed owner, address indexed spender, uint value); | |
event Transfer(address indexed from, address indexed to, uint value); | |
function name() external pure returns (string memory); | |
function symbol() external pure returns (string memory); | |
function decimals() external pure returns (uint8); | |
function totalSupply() external view returns (uint); | |
function balanceOf(address owner) external view returns (uint); | |
function allowance(address owner, address spender) external view returns (uint); | |
function approve(address spender, uint value) external returns (bool); | |
function transfer(address to, uint value) external returns (bool); | |
function transferFrom(address from, address to, uint value) external returns (bool); | |
function DOMAIN_SEPARATOR() external view returns (bytes32); | |
function PERMIT_TYPEHASH() external pure returns (bytes32); | |
function nonces(address owner) external view returns (uint); | |
function permit(address owner, address spender, uint value, uint deadline, uint8 v, bytes32 r, bytes32 s) external; | |
event Mint(address indexed sender, uint amount0, uint amount1); | |
event Burn(address indexed sender, uint amount0, uint amount1, address indexed to); | |
event Swap( | |
address indexed sender, | |
uint amount0In, | |
uint amount1In, | |
uint amount0Out, | |
uint amount1Out, | |
address indexed to | |
); | |
event Sync(uint112 reserve0, uint112 reserve1); | |
function MINIMUM_LIQUIDITY() external pure returns (uint); | |
function factory() external view returns (address); | |
function token0() external view returns (address); | |
function token1() external view returns (address); | |
function getReserves() external view returns (uint112 reserve0, uint112 reserve1, uint32 blockTimestampLast); | |
function getDeposited() external view returns (uint _deposited0, uint _deposited1); | |
function getDummy() external view returns (uint _dummy0, uint _dummy1); | |
function price0CumulativeLast() external view returns (uint); | |
function price1CumulativeLast() external view returns (uint); | |
function kLast() external view returns (uint); | |
function fee() external view returns (uint8); | |
function mint(address to) external returns (uint liquidity); | |
function burn(address to) external returns (uint amount0, uint amount1); | |
function swap(uint amount0Out, uint amount1Out, address to, bytes calldata data) external; | |
function skim(address to) external; | |
function sync() external; | |
function b0() external view returns (uint b); | |
function b1() external view returns (uint b); | |
function initialize(address, address) external; | |
} | |
// Dependency file: contracts/libraries/SafeMath.sol | |
// pragma solidity =0.6.12; | |
// a library for performing overflow-safe math, courtesy of DappHub (https://github.com/dapphub/ds-math) | |
library SafeMathUnisave { | |
function add(uint x, uint y) internal pure returns (uint z) { | |
require((z = x + y) >= x, 'ds-math-add-overflow'); | |
} | |
function sub(uint x, uint y) internal pure returns (uint z) { | |
require((z = x - y) <= x, 'ds-math-sub-underflow'); | |
} | |
function mul(uint x, uint y) internal pure returns (uint z) { | |
require(y == 0 || (z = x * y) / y == x, 'ds-math-mul-overflow'); | |
} | |
function div(uint256 a, uint256 b) internal pure returns (uint256) { | |
return div(a, b, "SafeMath: division by zero"); | |
} | |
function div(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) { | |
// Solidity only automatically asserts when dividing by 0 | |
require(b > 0, errorMessage); | |
uint256 c = a / b; | |
return c; | |
} | |
} | |
// Dependency file: contracts/libraries/UnisaveV2Library.sol | |
// pragma solidity >=0.5.0; | |
// import 'contracts/interfaces/IUnisaveV2Pair.sol'; | |
// import "contracts/libraries/SafeMath.sol"; | |
library UnisaveV2Library { | |
using SafeMathUnisave for uint; | |
// returns sorted token addresses, used to handle return values from pairs sorted in this order | |
function sortTokens(address tokenA, address tokenB) internal pure returns (address token0, address token1) { | |
require(tokenA != tokenB, 'UnisaveV2Library: IDENTICAL_ADDRESSES'); | |
(token0, token1) = tokenA < tokenB ? (tokenA, tokenB) : (tokenB, tokenA); | |
require(token0 != address(0), 'UnisaveV2Library: ZERO_ADDRESS'); | |
} | |
// calculates the CREATE2 address for a pair without making any external calls | |
function pairFor(address factory, address tokenA, address tokenB) internal pure returns (address pair) { | |
(address token0, address token1) = sortTokens(tokenA, tokenB); | |
pair = address(uint(keccak256(abi.encodePacked( | |
hex'ff', | |
factory, | |
keccak256(abi.encodePacked(token0, token1)), | |
hex'7b9686f044376bf1a04de2e9f1dbb6c9a4fe516f7cbd89b68be801982b62f913' // init code hash | |
)))); | |
} | |
// fetches and sorts the reserves and fee for a pair | |
function getReservesWithoutDummy(address factory, address tokenA, address tokenB) internal view returns (uint reserveA, uint reserveB, uint8 fee) { | |
(address token0,) = sortTokens(tokenA, tokenB); | |
IUnisaveV2Pair pair = IUnisaveV2Pair(pairFor(factory, tokenA, tokenB)); | |
(uint reserve0, uint reserve1,) = pair.getReserves(); | |
(uint dummy0, uint dummy1) = pair.getDummy(); | |
reserve0 -= dummy0; reserve1 -= dummy1; | |
(reserveA, reserveB) = tokenA == token0 ? (reserve0, reserve1) : (reserve1, reserve0); | |
fee = pair.fee(); | |
} | |
// fetches and sorts the reserves and fee for a pair | |
function getReserves(address factory, address tokenA, address tokenB) internal view returns (uint reserveA, uint reserveB, uint8 fee) { | |
(address token0,) = sortTokens(tokenA, tokenB); | |
(uint reserve0, uint reserve1,) = IUnisaveV2Pair(pairFor(factory, tokenA, tokenB)).getReserves(); | |
(reserveA, reserveB) = tokenA == token0 ? (reserve0, reserve1) : (reserve1, reserve0); | |
fee = IUnisaveV2Pair(pairFor(factory, tokenA, tokenB)).fee(); | |
} | |
// given some amount of an asset and pair reserves, returns an equivalent amount of the other asset | |
function quote(uint amountA, uint reserveA, uint reserveB) internal pure returns (uint amountB) { | |
require(amountA > 0, 'UnisaveV2Library: INSUFFICIENT_AMOUNT'); | |
require(reserveA > 0 && reserveB > 0, 'UnisaveV2Library: INSUFFICIENT_LIQUIDITY'); | |
amountB = amountA.mul(reserveB) / reserveA; | |
} | |
// given an input amount of an asset and pair reserves, returns the maximum output amount of the other asset | |
function getAmountOut(uint amountIn, uint reserveIn, uint reserveOut, uint8 fee) internal pure returns (uint amountOut) { | |
require(amountIn > 0, 'UnisaveV2Library: INSUFFICIENT_INPUT_AMOUNT'); | |
require(reserveIn > 0 && reserveOut > 0, 'UnisaveV2Library: INSUFFICIENT_LIQUIDITY'); | |
uint amountInWithFee = amountIn.mul(1000-fee); | |
uint numerator = amountInWithFee.mul(reserveOut); | |
uint denominator = reserveIn.mul(1000).add(amountInWithFee); | |
amountOut = numerator / denominator; | |
} | |
// given an output amount of an asset and pair reserves, returns a required input amount of the other asset | |
function getAmountIn(uint amountOut, uint reserveIn, uint reserveOut, uint8 fee) internal pure returns (uint amountIn) { | |
require(amountOut > 0, 'UnisaveV2Library: INSUFFICIENT_OUTPUT_AMOUNT'); | |
require(reserveIn > 0 && reserveOut > 0, 'UnisaveV2Library: INSUFFICIENT_LIQUIDITY'); | |
uint numerator = reserveIn.mul(amountOut).mul(1000); | |
uint denominator = reserveOut.sub(amountOut).mul(1000-fee); | |
amountIn = (numerator / denominator).add(1); | |
} | |
// performs chained getAmountOut calculations on any number of pairs | |
function getAmountsOut(address factory, uint amountIn, address[] memory path) internal view returns (uint[] memory amounts) { | |
require(path.length >= 2, 'UnisaveV2Library: INVALID_PATH'); | |
amounts = new uint[](path.length); | |
amounts[0] = amountIn; | |
for (uint i; i < path.length - 1; i++) { | |
(uint reserveIn, uint reserveOut, uint8 fee) = getReserves(factory, path[i], path[i + 1]); | |
amounts[i + 1] = getAmountOut(amounts[i], reserveIn, reserveOut, fee); | |
} | |
} | |
// performs chained getAmountIn calculations on any number of pairs | |
function getAmountsIn(address factory, uint amountOut, address[] memory path) internal view returns (uint[] memory amounts) { | |
require(path.length >= 2, 'UnisaveV2Library: INVALID_PATH'); | |
amounts = new uint[](path.length); | |
amounts[amounts.length - 1] = amountOut; | |
for (uint i = path.length - 1; i > 0; i--) { | |
(uint reserveIn, uint reserveOut, uint8 fee) = getReserves(factory, path[i - 1], path[i]); | |
amounts[i - 1] = getAmountIn(amounts[i], reserveIn, reserveOut, fee); | |
} | |
} | |
} | |
// Dependency file: contracts/libraries/TransferHelper.sol | |
// SPDX-License-Identifier: GPL-3.0-or-later | |
// pragma solidity >=0.6.0; | |
// helper methods for interacting with ERC20 tokens and sending ETH that do not consistently return true/false | |
library TransferHelper { | |
function safeApprove(address token, address to, uint value) internal { | |
// bytes4(keccak256(bytes('approve(address,uint256)'))); | |
(bool success, bytes memory data) = token.call(abi.encodeWithSelector(0x095ea7b3, to, value)); | |
require(success && (data.length == 0 || abi.decode(data, (bool))), 'TransferHelper: APPROVE_FAILED'); | |
} | |
function safeTransfer(address token, address to, uint value) internal { | |
// bytes4(keccak256(bytes('transfer(address,uint256)'))); | |
(bool success, bytes memory data) = token.call(abi.encodeWithSelector(0xa9059cbb, to, value)); | |
require(success && (data.length == 0 || abi.decode(data, (bool))), 'TransferHelper: TRANSFER_FAILED'); | |
} | |
function safeTransferFrom(address token, address from, address to, uint value) internal { | |
// bytes4(keccak256(bytes('transferFrom(address,address,uint256)'))); | |
(bool success, bytes memory data) = token.call(abi.encodeWithSelector(0x23b872dd, from, to, value)); | |
require(success && (data.length == 0 || abi.decode(data, (bool))), 'TransferHelper: TRANSFER_FROM_FAILED'); | |
} | |
function safeTransferETH(address to, uint value) internal { | |
(bool success,) = to.call{value:value}(new bytes(0)); | |
require(success, 'TransferHelper: ETH_TRANSFER_FAILED'); | |
} | |
} | |
// Dependency file: contracts/interfaces/IUnisaveV2Router01.sol | |
// pragma solidity >=0.6.2; | |
interface IUnisaveV2Router01 { | |
function factory() external pure returns (address); | |
function WETH() external pure returns (address); | |
function addLiquidity( | |
address tokenA, | |
address tokenB, | |
uint amountADesired, | |
uint amountBDesired, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline | |
) external returns (uint amountA, uint amountB, uint liquidity); | |
function addLiquidityETH( | |
address token, | |
uint amountTokenDesired, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) external payable returns (uint amountToken, uint amountETH, uint liquidity); | |
function removeLiquidity( | |
address tokenA, | |
address tokenB, | |
uint liquidity, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline | |
) external returns (uint amountA, uint amountB); | |
function removeLiquidityETH( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) external returns (uint amountToken, uint amountETH); | |
function removeLiquidityWithPermit( | |
address tokenA, | |
address tokenB, | |
uint liquidity, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external returns (uint amountA, uint amountB); | |
function removeLiquidityETHWithPermit( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external returns (uint amountToken, uint amountETH); | |
function swapExactTokensForTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external returns (uint[] memory amounts); | |
function swapTokensForExactTokens( | |
uint amountOut, | |
uint amountInMax, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external returns (uint[] memory amounts); | |
function swapExactETHForTokens(uint amountOutMin, address[] calldata path, address to, uint deadline) | |
external | |
payable | |
returns (uint[] memory amounts); | |
function swapTokensForExactETH(uint amountOut, uint amountInMax, address[] calldata path, address to, uint deadline) | |
external | |
returns (uint[] memory amounts); | |
function swapExactTokensForETH(uint amountIn, uint amountOutMin, address[] calldata path, address to, uint deadline) | |
external | |
returns (uint[] memory amounts); | |
function swapETHForExactTokens(uint amountOut, address[] calldata path, address to, uint deadline) | |
external | |
payable | |
returns (uint[] memory amounts); | |
function quote(uint amountA, uint reserveA, uint reserveB) external pure returns (uint amountB); | |
function getAmountOut(uint amountIn, uint reserveIn, uint reserveOut, uint8 fee) external pure returns (uint amountOut); | |
function getAmountIn(uint amountOut, uint reserveIn, uint reserveOut, uint8 fee) external pure returns (uint amountIn); | |
function getAmountsOut(uint amountIn, address[] calldata path) external view returns (uint[] memory amounts); | |
function getAmountsIn(uint amountOut, address[] calldata path) external view returns (uint[] memory amounts); | |
} | |
// Dependency file: contracts/interfaces/IUnisaveV2Router02.sol | |
// pragma solidity =0.6.12; | |
// import 'contracts/interfaces/IUnisaveV2Router01.sol'; | |
interface IUnisaveV2Router02 is IUnisaveV2Router01 { | |
function removeLiquidityETHSupportingFeeOnTransferTokens( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) external returns (uint amountETH); | |
function removeLiquidityETHWithPermitSupportingFeeOnTransferTokens( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external returns (uint amountETH); | |
function swapExactTokensForTokensSupportingFeeOnTransferTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external; | |
function swapExactETHForTokensSupportingFeeOnTransferTokens( | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external payable; | |
function swapExactTokensForETHSupportingFeeOnTransferTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external; | |
} | |
// Dependency file: contracts/interfaces/IUnisaveV2Factory.sol | |
// pragma solidity =0.6.12; | |
interface IUnisaveV2Factory { | |
event PairCreated(address indexed token0, address indexed token1, address pair, uint); | |
function feeTo() external view returns (address); | |
function feeToSetter() external view returns (address); | |
function getPair(address tokenA, address tokenB) external view returns (address pair); | |
function allPairs(uint) external view returns (address pair); | |
function allPairsLength() external view returns (uint); | |
function createPair(address tokenA, address tokenB) external returns (address pair); | |
function setFeeTo(address) external; | |
function setFeeToSetter(address) external; | |
} | |
// Dependency file: contracts/interfaces/IERC20.sol | |
// pragma solidity =0.6.12; | |
interface IERC20 { | |
event Approval(address indexed owner, address indexed spender, uint value); | |
event Transfer(address indexed from, address indexed to, uint value); | |
function name() external view returns (string memory); | |
function symbol() external view returns (string memory); | |
function decimals() external view returns (uint8); | |
function totalSupply() external view returns (uint); | |
function balanceOf(address owner) external view returns (uint); | |
function allowance(address owner, address spender) external view returns (uint); | |
function approve(address spender, uint value) external returns (bool); | |
function transfer(address to, uint value) external returns (bool); | |
function transferFrom(address from, address to, uint value) external returns (bool); | |
} | |
// Dependency file: contracts/interfaces/IWETH.sol | |
// pragma solidity =0.6.12; | |
interface IWETH { | |
function deposit() external payable; | |
function transfer(address to, uint value) external returns (bool); | |
function withdraw(uint) external; | |
} | |
// Root file: contracts/UnisaveV2Router02.sol | |
pragma solidity =0.6.12; | |
// import 'contracts/libraries/UnisaveV2Library.sol'; | |
// import 'contracts/libraries/SafeMath.sol'; | |
// import 'contracts/libraries/TransferHelper.sol'; | |
// import 'contracts/interfaces/IUnisaveV2Router02.sol'; | |
// import 'contracts/interfaces/IUnisaveV2Factory.sol'; | |
// import 'contracts/interfaces/IERC20.sol'; | |
// import 'contracts/interfaces/IWETH.sol'; | |
contract UnisaveV2Router02 is IUnisaveV2Router02 { | |
using SafeMathUnisave for uint; | |
address public immutable override factory; | |
address public immutable override WETH; | |
modifier ensure(uint deadline) { | |
require(deadline >= block.timestamp, 'UnisaveV2Router: EXPIRED'); | |
_; | |
} | |
constructor(address _factory, address _WETH) public { | |
factory = _factory; | |
WETH = _WETH; | |
} | |
receive() external payable { | |
assert(msg.sender == WETH); // only accept ETH via fallback from the WETH contract | |
} | |
// **** ADD LIQUIDITY **** | |
function _addLiquidity( | |
address tokenA, | |
address tokenB, | |
uint amountADesired, | |
uint amountBDesired, | |
uint amountAMin, | |
uint amountBMin | |
) internal virtual returns (uint amountA, uint amountB) { | |
// create the pair if it doesn't exist yet | |
if (IUnisaveV2Factory(factory).getPair(tokenA, tokenB) == address(0)) { | |
IUnisaveV2Factory(factory).createPair(tokenA, tokenB); | |
} | |
(uint reserveA, uint reserveB, ) = UnisaveV2Library.getReservesWithoutDummy(factory, tokenA, tokenB); | |
if (reserveA == 0 && reserveB == 0) { | |
(amountA, amountB) = (amountADesired, amountBDesired); | |
} else { | |
uint amountBOptimal = UnisaveV2Library.quote(amountADesired, reserveA, reserveB); | |
if (amountBOptimal <= amountBDesired) { | |
require(amountBOptimal >= amountBMin, 'UnisaveV2Router: INSUFFICIENT_B_AMOUNT'); | |
(amountA, amountB) = (amountADesired, amountBOptimal); | |
} else { | |
uint amountAOptimal = UnisaveV2Library.quote(amountBDesired, reserveB, reserveA); | |
assert(amountAOptimal <= amountADesired); | |
require(amountAOptimal >= amountAMin, 'UnisaveV2Router: INSUFFICIENT_A_AMOUNT'); | |
(amountA, amountB) = (amountAOptimal, amountBDesired); | |
} | |
} | |
} | |
function addLiquidity( | |
address tokenA, | |
address tokenB, | |
uint amountADesired, | |
uint amountBDesired, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline | |
) external virtual override ensure(deadline) returns (uint amountA, uint amountB, uint liquidity) { | |
(amountA, amountB) = _addLiquidity(tokenA, tokenB, amountADesired, amountBDesired, amountAMin, amountBMin); | |
address pair = UnisaveV2Library.pairFor(factory, tokenA, tokenB); | |
TransferHelper.safeTransferFrom(tokenA, msg.sender, pair, amountA); | |
TransferHelper.safeTransferFrom(tokenB, msg.sender, pair, amountB); | |
liquidity = IUnisaveV2Pair(pair).mint(to); | |
} | |
function addLiquidityETH( | |
address token, | |
uint amountTokenDesired, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) external virtual override payable ensure(deadline) returns (uint amountToken, uint amountETH, uint liquidity) { | |
(amountToken, amountETH) = _addLiquidity( | |
token, | |
WETH, | |
amountTokenDesired, | |
msg.value, | |
amountTokenMin, | |
amountETHMin | |
); | |
address pair = UnisaveV2Library.pairFor(factory, token, WETH); | |
TransferHelper.safeTransferFrom(token, msg.sender, pair, amountToken); | |
IWETH(WETH).deposit{value: amountETH}(); | |
assert(IWETH(WETH).transfer(pair, amountETH)); | |
liquidity = IUnisaveV2Pair(pair).mint(to); | |
// refund dust eth, if any | |
if (msg.value > amountETH) TransferHelper.safeTransferETH(msg.sender, msg.value - amountETH); | |
} | |
// **** REMOVE LIQUIDITY **** | |
function removeLiquidity( | |
address tokenA, | |
address tokenB, | |
uint liquidity, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline | |
) public virtual override ensure(deadline) returns (uint amountA, uint amountB) { | |
address pair = UnisaveV2Library.pairFor(factory, tokenA, tokenB); | |
IUnisaveV2Pair(pair).transferFrom(msg.sender, pair, liquidity); // send liquidity to pair | |
(uint amount0, uint amount1) = IUnisaveV2Pair(pair).burn(to); | |
(address token0,) = UnisaveV2Library.sortTokens(tokenA, tokenB); | |
(amountA, amountB) = tokenA == token0 ? (amount0, amount1) : (amount1, amount0); | |
require(amountA >= amountAMin, 'UnisaveV2Router: INSUFFICIENT_A_AMOUNT'); | |
require(amountB >= amountBMin, 'UnisaveV2Router: INSUFFICIENT_B_AMOUNT'); | |
} | |
function removeLiquidityETH( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) public virtual override ensure(deadline) returns (uint amountToken, uint amountETH) { | |
(amountToken, amountETH) = removeLiquidity( | |
token, | |
WETH, | |
liquidity, | |
amountTokenMin, | |
amountETHMin, | |
address(this), | |
deadline | |
); | |
TransferHelper.safeTransfer(token, to, amountToken); | |
IWETH(WETH).withdraw(amountETH); | |
TransferHelper.safeTransferETH(to, amountETH); | |
} | |
function removeLiquidityWithPermit( | |
address tokenA, | |
address tokenB, | |
uint liquidity, | |
uint amountAMin, | |
uint amountBMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external virtual override returns (uint amountA, uint amountB) { | |
address pair = UnisaveV2Library.pairFor(factory, tokenA, tokenB); | |
uint value = approveMax ? uint(-1) : liquidity; | |
IUnisaveV2Pair(pair).permit(msg.sender, address(this), value, deadline, v, r, s); | |
(amountA, amountB) = removeLiquidity(tokenA, tokenB, liquidity, amountAMin, amountBMin, to, deadline); | |
} | |
function removeLiquidityETHWithPermit( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external virtual override returns (uint amountToken, uint amountETH) { | |
address pair = UnisaveV2Library.pairFor(factory, token, WETH); | |
uint value = approveMax ? uint(-1) : liquidity; | |
IUnisaveV2Pair(pair).permit(msg.sender, address(this), value, deadline, v, r, s); | |
(amountToken, amountETH) = removeLiquidityETH(token, liquidity, amountTokenMin, amountETHMin, to, deadline); | |
} | |
// **** REMOVE LIQUIDITY (supporting fee-on-transfer tokens) **** | |
function removeLiquidityETHSupportingFeeOnTransferTokens( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline | |
) public virtual override ensure(deadline) returns (uint amountETH) { | |
(, amountETH) = removeLiquidity( | |
token, | |
WETH, | |
liquidity, | |
amountTokenMin, | |
amountETHMin, | |
address(this), | |
deadline | |
); | |
TransferHelper.safeTransfer(token, to, IERC20(token).balanceOf(address(this))); | |
IWETH(WETH).withdraw(amountETH); | |
TransferHelper.safeTransferETH(to, amountETH); | |
} | |
function removeLiquidityETHWithPermitSupportingFeeOnTransferTokens( | |
address token, | |
uint liquidity, | |
uint amountTokenMin, | |
uint amountETHMin, | |
address to, | |
uint deadline, | |
bool approveMax, uint8 v, bytes32 r, bytes32 s | |
) external virtual override returns (uint amountETH) { | |
address pair = UnisaveV2Library.pairFor(factory, token, WETH); | |
uint value = approveMax ? uint(-1) : liquidity; | |
IUnisaveV2Pair(pair).permit(msg.sender, address(this), value, deadline, v, r, s); | |
amountETH = removeLiquidityETHSupportingFeeOnTransferTokens( | |
token, liquidity, amountTokenMin, amountETHMin, to, deadline | |
); | |
} | |
// **** SWAP **** | |
// requires the initial amount to have already been sent to the first pair | |
function _swap(uint[] memory amounts, address[] memory path, address _to) internal virtual { | |
for (uint i; i < path.length - 1; i++) { | |
(address input, address output) = (path[i], path[i + 1]); | |
(address token0,) = UnisaveV2Library.sortTokens(input, output); | |
uint amountOut = amounts[i + 1]; | |
(uint amount0Out, uint amount1Out) = input == token0 ? (uint(0), amountOut) : (amountOut, uint(0)); | |
address to = i < path.length - 2 ? UnisaveV2Library.pairFor(factory, output, path[i + 2]) : _to; | |
IUnisaveV2Pair(UnisaveV2Library.pairFor(factory, input, output)).swap( | |
amount0Out, amount1Out, to, new bytes(0) | |
); | |
} | |
} | |
function swapExactTokensForTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external virtual override ensure(deadline) returns (uint[] memory amounts) { | |
amounts = UnisaveV2Library.getAmountsOut(factory, amountIn, path); | |
require(amounts[amounts.length - 1] >= amountOutMin, 'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT'); | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0] | |
); | |
_swap(amounts, path, to); | |
} | |
function swapTokensForExactTokens( | |
uint amountOut, | |
uint amountInMax, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external virtual override ensure(deadline) returns (uint[] memory amounts) { | |
amounts = UnisaveV2Library.getAmountsIn(factory, amountOut, path); | |
require(amounts[0] <= amountInMax, 'UnisaveV2Router: EXCESSIVE_INPUT_AMOUNT'); | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0] | |
); | |
_swap(amounts, path, to); | |
} | |
function swapExactETHForTokens(uint amountOutMin, address[] calldata path, address to, uint deadline) | |
external | |
virtual | |
override | |
payable | |
ensure(deadline) | |
returns (uint[] memory amounts) | |
{ | |
require(path[0] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
amounts = UnisaveV2Library.getAmountsOut(factory, msg.value, path); | |
require(amounts[amounts.length - 1] >= amountOutMin, 'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT'); | |
IWETH(WETH).deposit{value: amounts[0]}(); | |
assert(IWETH(WETH).transfer(UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0])); | |
_swap(amounts, path, to); | |
} | |
function swapTokensForExactETH(uint amountOut, uint amountInMax, address[] calldata path, address to, uint deadline) | |
external | |
virtual | |
override | |
ensure(deadline) | |
returns (uint[] memory amounts) | |
{ | |
require(path[path.length - 1] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
amounts = UnisaveV2Library.getAmountsIn(factory, amountOut, path); | |
require(amounts[0] <= amountInMax, 'UnisaveV2Router: EXCESSIVE_INPUT_AMOUNT'); | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0] | |
); | |
_swap(amounts, path, address(this)); | |
IWETH(WETH).withdraw(amounts[amounts.length - 1]); | |
TransferHelper.safeTransferETH(to, amounts[amounts.length - 1]); | |
} | |
function swapExactTokensForETH(uint amountIn, uint amountOutMin, address[] calldata path, address to, uint deadline) | |
external | |
virtual | |
override | |
ensure(deadline) | |
returns (uint[] memory amounts) | |
{ | |
require(path[path.length - 1] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
amounts = UnisaveV2Library.getAmountsOut(factory, amountIn, path); | |
require(amounts[amounts.length - 1] >= amountOutMin, 'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT'); | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0] | |
); | |
_swap(amounts, path, address(this)); | |
IWETH(WETH).withdraw(amounts[amounts.length - 1]); | |
TransferHelper.safeTransferETH(to, amounts[amounts.length - 1]); | |
} | |
function swapETHForExactTokens(uint amountOut, address[] calldata path, address to, uint deadline) | |
external | |
virtual | |
override | |
payable | |
ensure(deadline) | |
returns (uint[] memory amounts) | |
{ | |
require(path[0] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
amounts = UnisaveV2Library.getAmountsIn(factory, amountOut, path); | |
require(amounts[0] <= msg.value, 'UnisaveV2Router: EXCESSIVE_INPUT_AMOUNT'); | |
IWETH(WETH).deposit{value: amounts[0]}(); | |
assert(IWETH(WETH).transfer(UnisaveV2Library.pairFor(factory, path[0], path[1]), amounts[0])); | |
_swap(amounts, path, to); | |
// refund dust eth, if any | |
if (msg.value > amounts[0]) TransferHelper.safeTransferETH(msg.sender, msg.value - amounts[0]); | |
} | |
// **** SWAP (supporting fee-on-transfer tokens) **** | |
// requires the initial amount to have already been sent to the first pair | |
function _swapSupportingFeeOnTransferTokens(address[] memory path, address _to) internal virtual { | |
for (uint i; i < path.length - 1; i++) { | |
(address input, address output) = (path[i], path[i + 1]); | |
(address token0,) = UnisaveV2Library.sortTokens(input, output); | |
IUnisaveV2Pair pair = IUnisaveV2Pair(UnisaveV2Library.pairFor(factory, input, output)); | |
uint8 fee = pair.fee(); | |
(uint reserve0, uint reserve1,) = pair.getReserves(); | |
uint amount0Out; | |
uint amount1Out; | |
if (input == token0) { | |
(uint reserveInput, uint reserveOutput, uint amountInput) = (reserve0, reserve1, pair.b0().sub(reserve0)); | |
uint amountOutput = UnisaveV2Library.getAmountOut(amountInput, reserveInput, reserveOutput, fee); | |
(amount0Out, amount1Out) = (uint(0), amountOutput); | |
} else { | |
(uint reserveInput, uint reserveOutput, uint amountInput) = (reserve1, reserve0, pair.b1().sub(reserve1)); | |
uint amountOutput = UnisaveV2Library.getAmountOut(amountInput, reserveInput, reserveOutput, fee); | |
(amount0Out, amount1Out) = (amountOutput, uint(0)); | |
} | |
address to = i < path.length - 2 ? UnisaveV2Library.pairFor(factory, output, path[i + 2]) : _to; | |
pair.swap(amount0Out, amount1Out, to, new bytes(0)); | |
} | |
} | |
function swapExactTokensForTokensSupportingFeeOnTransferTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) external virtual override ensure(deadline) { | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amountIn | |
); | |
uint balanceBefore = IERC20(path[path.length - 1]).balanceOf(to); | |
_swapSupportingFeeOnTransferTokens(path, to); | |
require( | |
IERC20(path[path.length - 1]).balanceOf(to).sub(balanceBefore) >= amountOutMin, | |
'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT' | |
); | |
} | |
function swapExactETHForTokensSupportingFeeOnTransferTokens( | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) | |
external | |
virtual | |
override | |
payable | |
ensure(deadline) | |
{ | |
require(path[0] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
uint amountIn = msg.value; | |
IWETH(WETH).deposit{value: amountIn}(); | |
assert(IWETH(WETH).transfer(UnisaveV2Library.pairFor(factory, path[0], path[1]), amountIn)); | |
uint balanceBefore = IERC20(path[path.length - 1]).balanceOf(to); | |
_swapSupportingFeeOnTransferTokens(path, to); | |
require( | |
IERC20(path[path.length - 1]).balanceOf(to).sub(balanceBefore) >= amountOutMin, | |
'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT' | |
); | |
} | |
function swapExactTokensForETHSupportingFeeOnTransferTokens( | |
uint amountIn, | |
uint amountOutMin, | |
address[] calldata path, | |
address to, | |
uint deadline | |
) | |
external | |
virtual | |
override | |
ensure(deadline) | |
{ | |
require(path[path.length - 1] == WETH, 'UnisaveV2Router: INVALID_PATH'); | |
TransferHelper.safeTransferFrom( | |
path[0], msg.sender, UnisaveV2Library.pairFor(factory, path[0], path[1]), amountIn | |
); | |
_swapSupportingFeeOnTransferTokens(path, address(this)); | |
uint amountOut = IERC20(WETH).balanceOf(address(this)); | |
require(amountOut >= amountOutMin, 'UnisaveV2Router: INSUFFICIENT_OUTPUT_AMOUNT'); | |
IWETH(WETH).withdraw(amountOut); | |
TransferHelper.safeTransferETH(to, amountOut); | |
} | |
// **** LIBRARY FUNCTIONS **** | |
function quote(uint amountA, uint reserveA, uint reserveB) public pure virtual override returns (uint amountB) { | |
return UnisaveV2Library.quote(amountA, reserveA, reserveB); | |
} | |
function getAmountOut(uint amountIn, uint reserveIn, uint reserveOut, uint8 fee) | |
public | |
pure | |
virtual | |
override | |
returns (uint amountOut) | |
{ | |
return UnisaveV2Library.getAmountOut(amountIn, reserveIn, reserveOut, fee); | |
} | |
function getAmountIn(uint amountOut, uint reserveIn, uint reserveOut, uint8 fee) | |
public | |
pure | |
virtual | |
override | |
returns (uint amountIn) | |
{ | |
return UnisaveV2Library.getAmountIn(amountOut, reserveIn, reserveOut, fee); | |
} | |
function getAmountsOut(uint amountIn, address[] memory path) | |
public | |
view | |
virtual | |
override | |
returns (uint[] memory amounts) | |
{ | |
return UnisaveV2Library.getAmountsOut(factory, amountIn, path); | |
} | |
function getAmountsIn(uint amountOut, address[] memory path) | |
public | |
view | |
virtual | |
override | |
returns (uint[] memory amounts) | |
{ | |
return UnisaveV2Library.getAmountsIn(factory, amountOut, path); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment