Fixed stack too deep error

This commit is contained in:
Thrilok Kumar 2020-07-06 16:18:22 +05:30
parent 3d6d1e21e2
commit 668d0457b1

View File

@ -171,15 +171,13 @@ contract LiquidityHelpers is UniswapHelpers {
address tokenB, address tokenB,
uint _amt, uint _amt,
uint unitAmt, uint unitAmt,
uint slippage, uint slippage
uint deadline
) internal returns (uint _amtA, uint _amtB, uint _liquidity) { ) internal returns (uint _amtA, uint _amtB, uint _liquidity) {
IUniswapV2Router02 router = IUniswapV2Router02(getUniswapAddr()); IUniswapV2Router02 router = IUniswapV2Router02(getUniswapAddr());
(TokenInterface _tokenA, TokenInterface _tokenB) = changeEthAddress(tokenA, tokenB); (TokenInterface _tokenA, TokenInterface _tokenB) = changeEthAddress(tokenA, tokenB);
_amtA = _amt == uint(-1) ? getTokenBalace(tokenA) : _amt; _amtA = _amt == uint(-1) ? getTokenBalace(tokenA) : _amt;
uint _amtA18 = convertTo18(_tokenA.decimals(), _amtA); _amtB = convert18ToDec(_tokenB.decimals(), wmul(unitAmt, convertTo18(_tokenA.decimals(), _amtA)));
_amtB = convert18ToDec(_tokenB.decimals(), wmul(unitAmt, _amtA18));
convertEthToWeth(_tokenA, _amtA); convertEthToWeth(_tokenA, _amtA);
convertEthToWeth(_tokenB, _amtB); convertEthToWeth(_tokenB, _amtB);
@ -194,7 +192,7 @@ contract LiquidityHelpers is UniswapHelpers {
getMinAmount(_tokenA, _amtA, slippage), getMinAmount(_tokenA, _amtA, slippage),
getMinAmount(_tokenB, _amtB, slippage), getMinAmount(_tokenB, _amtB, slippage),
address(this), address(this),
now + deadline // TODO - deadline? now + 1
); );
} }
@ -203,34 +201,47 @@ contract LiquidityHelpers is UniswapHelpers {
address tokenB, address tokenB,
uint _amt, uint _amt,
uint unitAmtA, uint unitAmtA,
uint unitAmtB, uint unitAmtB
uint deadline
) internal returns (uint _amtA, uint _amtB, uint _uniAmt) { ) internal returns (uint _amtA, uint _amtB, uint _uniAmt) {
IUniswapV2Router02 router = IUniswapV2Router02(getUniswapAddr()); IUniswapV2Router02 router;
(TokenInterface _tokenA, TokenInterface _tokenB) = changeEthAddress(tokenA, tokenB); TokenInterface _tokenA;
address exchangeAddr = IUniswapV2Factory(router.factory()).getPair(address(_tokenA), address(_tokenB)); TokenInterface _tokenB;
require(exchangeAddr != address(0), "pair-not-found."); (router, _tokenA, _tokenB, _uniAmt) = _getRemoveLiquidityData(
tokenA,
TokenInterface uniToken = TokenInterface(exchangeAddr); tokenB,
_uniAmt = _amt == uint(-1) ? uniToken.balanceOf(address(this)) : _amt; _amt
uniToken.approve(address(router), _uniAmt); );
{
uint minAmtA = convert18ToDec(_tokenA.decimals(), wmul(unitAmtA, _uniAmt)); uint minAmtA = convert18ToDec(_tokenA.decimals(), wmul(unitAmtA, _uniAmt));
uint minAmtB = convert18ToDec(_tokenB.decimals(), wmul(unitAmtB, _uniAmt)); uint minAmtB = convert18ToDec(_tokenB.decimals(), wmul(unitAmtB, _uniAmt));
(_amtA, _amtB) = router.removeLiquidity(
(_amtA, _amtB) = router.removeLiquidity(
address(_tokenA), address(_tokenA),
address(_tokenB), address(_tokenB),
_amt, _amt,
minAmtA, minAmtA,
minAmtB, minAmtB,
address(this), address(this),
now + deadline // TODO - deadline? now + 1
); );
}
convertWethToEth(_tokenA, _amtA); convertWethToEth(_tokenA, _amtA);
convertWethToEth(_tokenB, _amtB); convertWethToEth(_tokenB, _amtB);
} }
function _getRemoveLiquidityData(
address tokenA,
address tokenB,
uint _amt
) internal returns (IUniswapV2Router02 router, TokenInterface _tokenA, TokenInterface _tokenB, uint _uniAmt) {
router = IUniswapV2Router02(getUniswapAddr());
(_tokenA, _tokenB) = changeEthAddress(tokenA, tokenB);
address exchangeAddr = IUniswapV2Factory(router.factory()).getPair(address(_tokenA), address(_tokenB));
require(exchangeAddr != address(0), "pair-not-found.");
TokenInterface uniToken = TokenInterface(exchangeAddr);
_uniAmt = _amt == uint(-1) ? uniToken.balanceOf(address(this)) : _amt;
uniToken.approve(address(router), _uniAmt);
}
} }
contract UniswapLiquidity is LiquidityHelpers { contract UniswapLiquidity is LiquidityHelpers {
@ -323,7 +334,6 @@ contract UniswapLiquidity is LiquidityHelpers {
uint amtA, uint amtA,
uint unitAmt, uint unitAmt,
uint slippage, uint slippage,
uint deadline,
uint getId, uint getId,
uint setId uint setId
) external payable { ) external payable {
@ -334,8 +344,7 @@ contract UniswapLiquidity is LiquidityHelpers {
tokenB, tokenB,
_amt, _amt,
unitAmt, unitAmt,
slippage, slippage
deadline
); );
setUint(setId, _uniAmt); setUint(setId, _uniAmt);
emitDeposit(tokenA, tokenB, _amtA, _amtB, _uniAmt, getId, setId); emitDeposit(tokenA, tokenB, _amtA, _amtB, _uniAmt, getId, setId);
@ -347,7 +356,6 @@ contract UniswapLiquidity is LiquidityHelpers {
uint uinAmt, uint uinAmt,
uint unitAmtA, uint unitAmtA,
uint unitAmtB, uint unitAmtB,
uint deadline,
uint getId, uint getId,
uint[] calldata setIds uint[] calldata setIds
) external payable { ) external payable {
@ -358,8 +366,7 @@ contract UniswapLiquidity is LiquidityHelpers {
tokenB, tokenB,
_amt, _amt,
unitAmtA, unitAmtA,
unitAmtB, unitAmtB
deadline
); );
setUint(setIds[0], _amtA); setUint(setIds[0], _amtA);