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

minor refactoring

Browse Source
This commit is contained in:
Samyak Jain 2021-06-13 17:00:49 +05:30
parent d72c1480f9
commit f37a543b81
2 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
contracts/mainnet/connectors/uniswap_v3_erc20_staking/helpers.sol
View File

@ -14,4 +14,11 @@ abstract contract Helpers is DSMath, Basic {
TokenInterface constant internal rewardToken = TokenInterface(address(0)); // TODO
function getStakingContract(address stakingToken) internal view returns (address) {
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
return IStakingRewards(stakingRewardsInfo.stakingRewards);
}
}

15
contracts/mainnet/connectors/uniswap_v3_erc20_staking/main.sol
View File

@ -30,10 +30,7 @@ contract Main is Helpers, Events {
) external payable returns (string memory _eventName, bytes memory _eventParam) {
uint _amt = getUint(getId, amt);
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
TokenInterface stakingTokenContract = TokenInterface(stakingToken);
_amt = _amt == uint(-1) ? stakingTokenContract.balanceOf(address(this)) : _amt;
@ -64,10 +61,7 @@ contract Main is Helpers, Events {
) external payable returns (string memory _eventName, bytes memory _eventParam) {
uint _amt = getUint(getId, amt);
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
_amt = _amt == uint(-1) ? stakingContract.balanceOf(address(this)) : _amt;
uint intialBal = rewardToken.balanceOf(address(this));
@ -94,10 +88,7 @@ contract Main is Helpers, Events {
address stakingToken,
uint setId
) external payable returns (string memory _eventName, bytes memory _eventParam) {
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
uint intialBal = rewardToken.balanceOf(address(this));
stakingContract.getReward();