mirror of
https://github.com/Instadapp/dsa-connectors.git
synced 2024-07-29 22:37:00 +00:00
minor refactoring
This commit is contained in:
parent
d72c1480f9
commit
f37a543b81
|
@ -14,4 +14,11 @@ abstract contract Helpers is DSMath, Basic {
|
||||||
|
|
||||||
TokenInterface constant internal rewardToken = TokenInterface(address(0)); // TODO
|
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);
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
|
@ -30,10 +30,7 @@ contract Main is Helpers, Events {
|
||||||
) external payable returns (string memory _eventName, bytes memory _eventParam) {
|
) external payable returns (string memory _eventName, bytes memory _eventParam) {
|
||||||
uint _amt = getUint(getId, amt);
|
uint _amt = getUint(getId, amt);
|
||||||
|
|
||||||
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
|
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
|
||||||
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
|
|
||||||
|
|
||||||
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
|
|
||||||
TokenInterface stakingTokenContract = TokenInterface(stakingToken);
|
TokenInterface stakingTokenContract = TokenInterface(stakingToken);
|
||||||
|
|
||||||
_amt = _amt == uint(-1) ? stakingTokenContract.balanceOf(address(this)) : _amt;
|
_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) {
|
) external payable returns (string memory _eventName, bytes memory _eventParam) {
|
||||||
uint _amt = getUint(getId, amt);
|
uint _amt = getUint(getId, amt);
|
||||||
|
|
||||||
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
|
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
|
||||||
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
|
|
||||||
|
|
||||||
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
|
|
||||||
|
|
||||||
_amt = _amt == uint(-1) ? stakingContract.balanceOf(address(this)) : _amt;
|
_amt = _amt == uint(-1) ? stakingContract.balanceOf(address(this)) : _amt;
|
||||||
uint intialBal = rewardToken.balanceOf(address(this));
|
uint intialBal = rewardToken.balanceOf(address(this));
|
||||||
|
@ -94,10 +88,7 @@ contract Main is Helpers, Events {
|
||||||
address stakingToken,
|
address stakingToken,
|
||||||
uint setId
|
uint setId
|
||||||
) external payable returns (string memory _eventName, bytes memory _eventParam) {
|
) external payable returns (string memory _eventName, bytes memory _eventParam) {
|
||||||
IStakingRewardsFactory.StakingRewardsInfo memory stakingRewardsInfo =
|
IStakingRewards stakingContract = IStakingRewards(getStakingContract(stakingToken));
|
||||||
stakingRewardsFactory.stakingRewardsInfoByStakingToken(stakingToken);
|
|
||||||
|
|
||||||
IStakingRewards stakingContract = IStakingRewards(stakingRewardsInfo.stakingRewards);
|
|
||||||
|
|
||||||
uint intialBal = rewardToken.balanceOf(address(this));
|
uint intialBal = rewardToken.balanceOf(address(this));
|
||||||
stakingContract.getReward();
|
stakingContract.getReward();
|
||||||
|
|
Loading…
Reference in New Issue
Block a user