Allow ETH as quote in DODOEthProxy
This commit is contained in:
@@ -10,45 +10,67 @@ pragma experimental ABIEncoderV2;
|
||||
|
||||
import {ReentrancyGuard} from "./lib/ReentrancyGuard.sol";
|
||||
import {SafeERC20} from "./lib/SafeERC20.sol";
|
||||
import {SafeMath} from "./lib/SafeMath.sol";
|
||||
import {IDODO} from "./intf/IDODO.sol";
|
||||
import {IERC20} from "./intf/IERC20.sol";
|
||||
import {IWETH} from "./intf/IWETH.sol";
|
||||
|
||||
|
||||
interface IDODOZoo {
|
||||
function getDODO(address baseToken, address quoteToken) external view returns (address);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @title DODO Eth Proxy
|
||||
* @author DODO Breeder
|
||||
*
|
||||
* @notice Handle ETH-WETH converting for users. Use it only when WETH is base token
|
||||
* @notice Handle ETH-WETH converting for users.
|
||||
*/
|
||||
contract DODOEthProxy is ReentrancyGuard {
|
||||
using SafeERC20 for IERC20;
|
||||
using SafeMath for uint256;
|
||||
|
||||
address public _DODO_ZOO_;
|
||||
address payable public _WETH_;
|
||||
|
||||
// ============ Events ============
|
||||
|
||||
event ProxySellEth(
|
||||
event ProxySellEthToToken(
|
||||
address indexed seller,
|
||||
address indexed quoteToken,
|
||||
uint256 payEth,
|
||||
uint256 receiveQuote
|
||||
uint256 receiveToken
|
||||
);
|
||||
|
||||
event ProxyBuyEth(
|
||||
event ProxyBuyEthWithToken(
|
||||
address indexed buyer,
|
||||
address indexed quoteToken,
|
||||
uint256 receiveEth,
|
||||
uint256 payQuote
|
||||
uint256 payToken
|
||||
);
|
||||
|
||||
event ProxyDepositEth(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
event ProxySellTokenToEth(
|
||||
address indexed seller,
|
||||
address indexed baseToken,
|
||||
uint256 payToken,
|
||||
uint256 receiveEth
|
||||
);
|
||||
|
||||
event ProxyWithdrawEth(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
event ProxyBuyTokenWithEth(
|
||||
address indexed buyer,
|
||||
address indexed baseToken,
|
||||
uint256 receiveToken,
|
||||
uint256 payEth
|
||||
);
|
||||
|
||||
event ProxyDepositEthAsBase(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
|
||||
event ProxyWithdrawEthAsBase(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
|
||||
event ProxyDepositEthAsQuote(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
|
||||
event ProxyWithdrawEthAsQuote(address indexed lp, address indexed DODO, uint256 ethAmount);
|
||||
|
||||
// ============ Functions ============
|
||||
|
||||
@@ -65,7 +87,7 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
require(msg.sender == _WETH_, "WE_SAVED_YOUR_ETH_:)");
|
||||
}
|
||||
|
||||
function sellEthTo(
|
||||
function sellEthToToken(
|
||||
address quoteTokenAddress,
|
||||
uint256 ethAmount,
|
||||
uint256 minReceiveTokenAmount
|
||||
@@ -77,11 +99,11 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
IWETH(_WETH_).approve(DODO, ethAmount);
|
||||
receiveTokenAmount = IDODO(DODO).sellBaseToken(ethAmount, minReceiveTokenAmount, "");
|
||||
_transferOut(quoteTokenAddress, msg.sender, receiveTokenAmount);
|
||||
emit ProxySellEth(msg.sender, quoteTokenAddress, ethAmount, receiveTokenAmount);
|
||||
emit ProxySellEthToToken(msg.sender, quoteTokenAddress, ethAmount, receiveTokenAmount);
|
||||
return receiveTokenAmount;
|
||||
}
|
||||
|
||||
function buyEthWith(
|
||||
function buyEthWithToken(
|
||||
address quoteTokenAddress,
|
||||
uint256 ethAmount,
|
||||
uint256 maxPayTokenAmount
|
||||
@@ -94,11 +116,48 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
IDODO(DODO).buyBaseToken(ethAmount, maxPayTokenAmount, "");
|
||||
IWETH(_WETH_).withdraw(ethAmount);
|
||||
msg.sender.transfer(ethAmount);
|
||||
emit ProxyBuyEth(msg.sender, quoteTokenAddress, ethAmount, payTokenAmount);
|
||||
emit ProxyBuyEthWithToken(msg.sender, quoteTokenAddress, ethAmount, payTokenAmount);
|
||||
return payTokenAmount;
|
||||
}
|
||||
|
||||
function depositEth(uint256 ethAmount, address quoteTokenAddress)
|
||||
function sellTokenToEth(
|
||||
address baseTokenAddress,
|
||||
uint256 tokenAmount,
|
||||
uint256 minReceiveEthAmount
|
||||
) external preventReentrant returns (uint256 receiveEthAmount) {
|
||||
address DODO = IDODOZoo(_DODO_ZOO_).getDODO(baseTokenAddress, _WETH_);
|
||||
require(DODO != address(0), "DODO_NOT_EXIST");
|
||||
IERC20(baseTokenAddress).approve(DODO, tokenAmount);
|
||||
_transferIn(baseTokenAddress, msg.sender, tokenAmount);
|
||||
receiveEthAmount = IDODO(DODO).sellBaseToken(tokenAmount, minReceiveEthAmount, "");
|
||||
IWETH(_WETH_).withdraw(receiveEthAmount);
|
||||
msg.sender.transfer(receiveEthAmount);
|
||||
emit ProxySellTokenToEth(msg.sender, baseTokenAddress, tokenAmount, receiveEthAmount);
|
||||
return receiveEthAmount;
|
||||
}
|
||||
|
||||
function buyTokenWithEth(
|
||||
address baseTokenAddress,
|
||||
uint256 tokenAmount,
|
||||
uint256 maxPayEthAmount
|
||||
) external payable preventReentrant returns (uint256 payEthAmount) {
|
||||
require(msg.value == maxPayEthAmount, "ETH_AMOUNT_NOT_MATCH");
|
||||
address DODO = IDODOZoo(_DODO_ZOO_).getDODO(baseTokenAddress, _WETH_);
|
||||
require(DODO != address(0), "DODO_NOT_EXIST");
|
||||
payEthAmount = IDODO(DODO).queryBuyBaseToken(tokenAmount);
|
||||
IWETH(_WETH_).deposit{value: payEthAmount}();
|
||||
IWETH(_WETH_).approve(DODO, payEthAmount);
|
||||
IDODO(DODO).buyBaseToken(tokenAmount, maxPayEthAmount, "");
|
||||
_transferOut(baseTokenAddress, msg.sender, tokenAmount);
|
||||
uint256 refund = maxPayEthAmount.sub(payEthAmount);
|
||||
if (refund > 0) {
|
||||
msg.sender.transfer(refund);
|
||||
}
|
||||
emit ProxyBuyTokenWithEth(msg.sender, baseTokenAddress, tokenAmount, payEthAmount);
|
||||
return payEthAmount;
|
||||
}
|
||||
|
||||
function depositEthAsBase(uint256 ethAmount, address quoteTokenAddress)
|
||||
external
|
||||
payable
|
||||
preventReentrant
|
||||
@@ -109,10 +168,10 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
IWETH(_WETH_).deposit{value: ethAmount}();
|
||||
IWETH(_WETH_).approve(DODO, ethAmount);
|
||||
IDODO(DODO).depositBaseTo(msg.sender, ethAmount);
|
||||
emit ProxyDepositEth(msg.sender, DODO, ethAmount);
|
||||
emit ProxyDepositEthAsBase(msg.sender, DODO, ethAmount);
|
||||
}
|
||||
|
||||
function withdrawEth(uint256 ethAmount, address quoteTokenAddress)
|
||||
function withdrawEthAsBase(uint256 ethAmount, address quoteTokenAddress)
|
||||
external
|
||||
preventReentrant
|
||||
returns (uint256 withdrawAmount)
|
||||
@@ -135,11 +194,11 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
uint256 wethAmount = IERC20(_WETH_).balanceOf(address(this));
|
||||
IWETH(_WETH_).withdraw(wethAmount);
|
||||
msg.sender.transfer(wethAmount);
|
||||
emit ProxyWithdrawEth(msg.sender, DODO, wethAmount);
|
||||
emit ProxyWithdrawEthAsBase(msg.sender, DODO, wethAmount);
|
||||
return wethAmount;
|
||||
}
|
||||
|
||||
function withdrawAllEth(address quoteTokenAddress)
|
||||
function withdrawAllEthAsBase(address quoteTokenAddress)
|
||||
external
|
||||
preventReentrant
|
||||
returns (uint256 withdrawAmount)
|
||||
@@ -158,7 +217,71 @@ contract DODOEthProxy is ReentrancyGuard {
|
||||
uint256 wethAmount = IERC20(_WETH_).balanceOf(address(this));
|
||||
IWETH(_WETH_).withdraw(wethAmount);
|
||||
msg.sender.transfer(wethAmount);
|
||||
emit ProxyWithdrawEth(msg.sender, DODO, wethAmount);
|
||||
emit ProxyWithdrawEthAsBase(msg.sender, DODO, wethAmount);
|
||||
return wethAmount;
|
||||
}
|
||||
|
||||
function depositEthAsQuote(uint256 ethAmount, address baseTokenAddress)
|
||||
external
|
||||
payable
|
||||
preventReentrant
|
||||
{
|
||||
require(msg.value == ethAmount, "ETH_AMOUNT_NOT_MATCH");
|
||||
address DODO = IDODOZoo(_DODO_ZOO_).getDODO(baseTokenAddress, _WETH_);
|
||||
require(DODO != address(0), "DODO_NOT_EXIST");
|
||||
IWETH(_WETH_).deposit{value: ethAmount}();
|
||||
IWETH(_WETH_).approve(DODO, ethAmount);
|
||||
IDODO(DODO).depositQuoteTo(msg.sender, ethAmount);
|
||||
emit ProxyDepositEthAsQuote(msg.sender, DODO, ethAmount);
|
||||
}
|
||||
|
||||
function withdrawEthAsQuote(uint256 ethAmount, address baseTokenAddress)
|
||||
external
|
||||
preventReentrant
|
||||
returns (uint256 withdrawAmount)
|
||||
{
|
||||
address DODO = IDODOZoo(_DODO_ZOO_).getDODO(baseTokenAddress, _WETH_);
|
||||
require(DODO != address(0), "DODO_NOT_EXIST");
|
||||
address ethLpToken = IDODO(DODO)._QUOTE_CAPITAL_TOKEN_();
|
||||
|
||||
// transfer all pool shares to proxy
|
||||
uint256 lpBalance = IERC20(ethLpToken).balanceOf(msg.sender);
|
||||
IERC20(ethLpToken).transferFrom(msg.sender, address(this), lpBalance);
|
||||
IDODO(DODO).withdrawQuote(ethAmount);
|
||||
|
||||
// transfer remain shares back to msg.sender
|
||||
lpBalance = IERC20(ethLpToken).balanceOf(address(this));
|
||||
IERC20(ethLpToken).transfer(msg.sender, lpBalance);
|
||||
|
||||
// because of withdraw penalty, withdrawAmount may not equal to ethAmount
|
||||
// query weth amount first and than transfer ETH to msg.sender
|
||||
uint256 wethAmount = IERC20(_WETH_).balanceOf(address(this));
|
||||
IWETH(_WETH_).withdraw(wethAmount);
|
||||
msg.sender.transfer(wethAmount);
|
||||
emit ProxyWithdrawEthAsQuote(msg.sender, DODO, wethAmount);
|
||||
return wethAmount;
|
||||
}
|
||||
|
||||
function withdrawAllEthAsQuote(address baseTokenAddress)
|
||||
external
|
||||
preventReentrant
|
||||
returns (uint256 withdrawAmount)
|
||||
{
|
||||
address DODO = IDODOZoo(_DODO_ZOO_).getDODO(baseTokenAddress, _WETH_);
|
||||
require(DODO != address(0), "DODO_NOT_EXIST");
|
||||
address ethLpToken = IDODO(DODO)._QUOTE_CAPITAL_TOKEN_();
|
||||
|
||||
// transfer all pool shares to proxy
|
||||
uint256 lpBalance = IERC20(ethLpToken).balanceOf(msg.sender);
|
||||
IERC20(ethLpToken).transferFrom(msg.sender, address(this), lpBalance);
|
||||
IDODO(DODO).withdrawAllQuote();
|
||||
|
||||
// because of withdraw penalty, withdrawAmount may not equal to ethAmount
|
||||
// query weth amount first and than transfer ETH to msg.sender
|
||||
uint256 wethAmount = IERC20(_WETH_).balanceOf(address(this));
|
||||
IWETH(_WETH_).withdraw(wethAmount);
|
||||
msg.sender.transfer(wethAmount);
|
||||
emit ProxyWithdrawEthAsQuote(msg.sender, DODO, wethAmount);
|
||||
return wethAmount;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user