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

added event and function overloading for withdraw and claimReward

Browse Source
This commit is contained in:
Lecky Lao 2020-08-20 05:37:44 +10:00
parent be23d7bba8
commit b4feb0467d
1 changed files with 86 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
contracts/connectors/curve_gauge.sol
View File

@ -62,12 +62,25 @@ contract CurveGauge is GaugeHelper {
uint setIdReward
);
event LogWithdraw(
string indexed gaugePoolName,
uint amount,
uint getId,
uint setId,
uint setIdCRVReward
);
event LogClaimedReward(
string indexed gaugePoolName,
uint setId,
uint setIdReward
);
event LogClaimedReward(
string indexed gaugePoolName,
uint setId
);
struct Balances{
uint intialCRVBal;
uint intialRewardBal;
@ -149,7 +162,6 @@ contract CurveGauge is GaugeHelper {
setUint(setIdReward, balances.rewardAmt);
emitLogWithdraw(gaugePoolName, _amt, getId, setId, setIdCRVReward, setIdReward);
}
function emitLogWithdraw(string memory gaugePoolName, uint _amt, uint getId, uint setId, uint setIdCRVReward, uint setIdReward) internal {
@ -195,6 +207,79 @@ contract CurveGauge is GaugeHelper {
bytes memory _eventParam = abi.encode(gaugePoolName, setId, setIdReward);
emitEvent(_eventCode, _eventParam);
}
/**
* @dev Withdraw LP Token.
* @param gaugePoolName gauge pool name.
* @param amt LP token amount.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
* @param setIdCRVReward Set CRV token reward amount at this ID in `InstaMemory` Contract.
*/
function withdraw(
string calldata gaugePoolName,
uint amt,
uint getId,
uint setId,
uint setIdCRVReward
) external payable {
uint _amt = getUint(getId, amt);
address curveGaugeAddr = CurveGaugeMapping(getCurveGaugeMappingAddr())
.gaugeMapping(bytes32(stringToBytes32(gaugePoolName)));
require(curveGaugeAddr != address(0), "wrong-gauge-pool-name");
IGauge gauge = IGauge(curveGaugeAddr);
TokenInterface crv_token = TokenInterface(address(gauge.crv_token()));
Balances memory balances;
_amt = _amt == uint(-1) ? TokenInterface(address(gauge.lp_token())).balanceOf(address(this)) : _amt;
balances.intialCRVBal = crv_token.balanceOf(address(this));
IMintor(getCurveMintorAddr()).mint(curveGaugeAddr);
gauge.withdraw(_amt);
balances.finalCRVBal = crv_token.balanceOf(address(this));
balances.crvRewardAmt = sub(balances.finalCRVBal, balances.intialCRVBal);
setUint(setId, _amt);
setUint(setIdCRVReward, balances.crvRewardAmt);
emitLogWithdraw(gaugePoolName, _amt, getId, setId, setIdCRVReward);
}
function emitLogWithdraw(string memory gaugePoolName, uint _amt, uint getId, uint setId, uint setIdCRVReward) internal {
emit LogWithdraw(gaugePoolName, _amt, getId, setId, setIdCRVReward);
bytes32 _eventCodeWithdraw = keccak256("LogWithdraw(string,uint256,uint256,uint256,uint256)");
bytes memory _eventParamWithdraw = abi.encode(gaugePoolName, _amt, getId, setId, setIdCRVReward);
emitEvent(_eventCodeWithdraw, _eventParamWithdraw);
}
/**
* @dev Claim Reward.
* @param gaugePoolName gauge pool name.
* @param setId Set CRV reward amount at this ID in `InstaMemory` Contract.
*/
function claimReward(
string calldata gaugePoolName,
uint setId
) external payable {
CurveGaugeMapping curveGaugeMapping = CurveGaugeMapping(getCurveGaugeMappingAddr());
address curveGaugeAddr = curveGaugeMapping.gaugeMapping(bytes32(stringToBytes32(gaugePoolName)));
require(curveGaugeAddr != address(0), "wrong-gauge-pool-name");
IMintor mintor = IMintor(getCurveMintorAddr());
IGauge gauge = IGauge(curveGaugeAddr);
TokenInterface crv_token = TokenInterface(address(gauge.crv_token()));
Balances memory balances;
balances.intialCRVBal = crv_token.balanceOf(address(this));
mintor.mint(curveGaugeAddr);
balances.finalCRVBal = crv_token.balanceOf(address(this));
balances.crvRewardAmt = sub(balances.finalCRVBal, balances.intialCRVBal);
setUint(setId, balances.crvRewardAmt);
emit LogClaimedReward(gaugePoolName, setId);
bytes32 _eventCode = keccak256("LogClaimedReward(string,uint256,uint256)");
bytes memory _eventParam = abi.encode(gaugePoolName, setId);
emitEvent(_eventCode, _eventParam);
}
}
contract ConnectCurveGauge is CurveGauge {