add helper
This commit is contained in:
60
contracts/SmartRoute/helper/DODOCalleeHelper.sol
Normal file
60
contracts/SmartRoute/helper/DODOCalleeHelper.sol
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
/*
|
||||||
|
Copyright 2020 DODO ZOO.
|
||||||
|
SPDX-License-Identifier: Apache-2.0
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
pragma solidity 0.6.9;
|
||||||
|
pragma experimental ABIEncoderV2;
|
||||||
|
|
||||||
|
import {IDODOV2} from "../intf/IDODOV2.sol";
|
||||||
|
import {IERC20} from "../../intf/IERC20.sol";
|
||||||
|
import {IWETH} from "../../intf/IWETH.sol";
|
||||||
|
import {SafeERC20} from "../../lib/SafeERC20.sol";
|
||||||
|
import {ReentrancyGuard} from "../../lib/ReentrancyGuard.sol";
|
||||||
|
|
||||||
|
contract DODOCalleeHelper is ReentrancyGuard {
|
||||||
|
using SafeERC20 for IERC20;
|
||||||
|
address payable public _WETH_;
|
||||||
|
|
||||||
|
fallback() external payable {
|
||||||
|
require(msg.sender == _WETH_, "WE_DONT_SAVED_YOUR_ETH");
|
||||||
|
}
|
||||||
|
|
||||||
|
receive() external payable {
|
||||||
|
require(msg.sender == _WETH_, "WE_DONT_SAVED_YOUR_ETH");
|
||||||
|
}
|
||||||
|
|
||||||
|
constructor(address payable _weth) public {
|
||||||
|
_WETH_ = _weth;
|
||||||
|
}
|
||||||
|
|
||||||
|
function DVMSellShareCall(
|
||||||
|
address payable assetTo,
|
||||||
|
uint256,
|
||||||
|
uint256 baseAmount,
|
||||||
|
uint256 quoteAmount,
|
||||||
|
bytes calldata
|
||||||
|
) external preventReentrant {
|
||||||
|
address _baseToken = IDODOV2(msg.sender)._BASE_TOKEN_();
|
||||||
|
address _quoteToken = IDODOV2(msg.sender)._QUOTE_TOKEN_();
|
||||||
|
_withdraw(assetTo, _baseToken, baseAmount, _baseToken == _WETH_);
|
||||||
|
_withdraw(assetTo, _quoteToken, quoteAmount, _quoteToken == _WETH_);
|
||||||
|
}
|
||||||
|
|
||||||
|
function _withdraw(
|
||||||
|
address payable to,
|
||||||
|
address token,
|
||||||
|
uint256 amount,
|
||||||
|
bool isETH
|
||||||
|
) internal {
|
||||||
|
if (isETH) {
|
||||||
|
if (amount > 0) {
|
||||||
|
IWETH(_WETH_).withdraw(amount);
|
||||||
|
to.transfer(amount);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
SafeERC20.safeTransfer(IERC20(token), to, amount);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user