Allow ETH as quote in DODOEthProxy

This commit is contained in:
Michael Zhou
2020-09-07 18:08:27 +08:00
parent b83b940c49
commit 4bebaa9057
4 changed files with 616 additions and 142 deletions

View File

@@ -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;
}