Instadapp/dsa-connectors
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors.git synced 2024-07-29 22:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

support native token

Browse Source
This commit is contained in:
bhavik-m 2022-01-15 23:27:35 +05:30
parent e8407f72e8
commit 54a2b62e7b
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
contracts/mainnet/connectors/uniswap/v3_auto_router/helpers.sol
View File

@ -15,13 +15,19 @@ abstract contract Helpers is DSMath, Basic {
/** /**
* @dev UniswapV3 swapHelper * @dev UniswapV3 swapHelper
* @param swapData - Struct defined in interfaces.sol * @param swapData - Struct defined in interfaces.sol
* @param ethAmt - Eth to swap for .value()
*/ */
function _SwapHelper( function _SwapHelper(
SwapData memory swapData, SwapData memory swapData
uint ethAmt
) internal returns (uint buyAmt) { ) internal returns (uint buyAmt) {
TokenInterface buyToken = swapData.buyToken;
TokenInterface sellToken = address(swapData.sellToken) == ethAddr ? TokenInterface(wethAddr) : swapData.sellToken;
TokenInterface buyToken = address(swapData.buyToken) == ethAddr ? TokenInterface(wethAddr) : swapData.buyToken;
bool isEth = address(swapData.sellToken) == ethAddr;
convertEthToWeth(isEth, sellToken, swapData._sellAmt);
approve(TokenInterface(sellToken), V3_SWAP_ROUTER_ADDRESS, swapData._sellAmt);
(uint _buyDec, uint _sellDec) = getTokensDec(buyToken, swapData.sellToken); (uint _buyDec, uint _sellDec) = getTokensDec(buyToken, swapData.sellToken);
uint _sellAmt18 = convertTo18(_sellDec, swapData._sellAmt); uint _sellAmt18 = convertTo18(_sellDec, swapData._sellAmt);
uint _slippageAmt = convert18ToDec(_buyDec, wmul(swapData.unitAmt, _sellAmt18)); uint _slippageAmt = convert18ToDec(_buyDec, wmul(swapData.unitAmt, _sellAmt18));
@ -29,14 +35,16 @@ abstract contract Helpers is DSMath, Basic {
uint initalBal = getTokenBal(buyToken); uint initalBal = getTokenBal(buyToken);
// solium-disable-next-line security/no-call-value // solium-disable-next-line security/no-call-value
(bool success, ) = V3_SWAP_ROUTER_ADDRESS.call{value: ethAmt}(swapData.callData); (bool success, ) = V3_SWAP_ROUTER_ADDRESS.call(swapData.callData);
if (!success) revert("uniswapV3-swap-failed"); if (!success) revert("uniswapV3-swap-failed");
uint finalBal = getTokenBal(buyToken); uint finalBal = getTokenBal(buyToken);
buyAmt = sub(finalBal, initalBal); buyAmt = sub(finalBal, initalBal);
require(_slippageAmt <= buyAmt, "Too much slippage"); require(_slippageAmt <= buyAmt, "Too much slippage");
isEth = address(buyToken) == ethAddr;
convertWethToEth(isEth,buyToken,buyAmt);
} }
/** /**
@ -48,16 +56,8 @@ abstract contract Helpers is DSMath, Basic {
SwapData memory swapData, SwapData memory swapData,
uint setId uint setId
) internal returns (SwapData memory) { ) internal returns (SwapData memory) {
TokenInterface _sellAddr = swapData.sellToken;
swapData._buyAmt = _SwapHelper(swapData);
uint ethAmt;
if (address(_sellAddr) == ethAddr) {
ethAmt = swapData._sellAmt;
} else {
approve(TokenInterface(_sellAddr), V3_SWAP_ROUTER_ADDRESS, swapData._sellAmt);
}
swapData._buyAmt = _SwapHelper(swapData, ethAmt);
setUint(setId, swapData._buyAmt); setUint(setId, swapData._buyAmt);
return swapData; return swapData;