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

Code refactoring

Browse Source
This commit is contained in:
Thrilok Kumar 2020-07-07 20:35:19 +05:30
parent 785a91b758
commit 7f5c9b5071
1 changed files with 67 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
contracts/connectors/synthetix.sol
View File

@ -2,7 +2,7 @@ pragma solidity ^0.6.0;
// import files from common directory // import files from common directory
import { Stores } from "../common/stores.sol"; import { Stores } from "../common/stores.sol";
// import { DSMath } from "../common/math.sol"; import { DSMath } from "../common/math.sol";
import { TokenInterface } from "../common/interfaces.sol"; import { TokenInterface } from "../common/interfaces.sol";
interface IStakingRewards { interface IStakingRewards {
@ -12,89 +12,105 @@ interface IStakingRewards {
function getReward() external; function getReward() external;
} }
contract Helper is Stores { contract SynthetixStakingHelper is DSMath, Stores {
function getSynthetixStakingAddr(address token) internal view returns(address){ function getSynthetixStakingAddr(address token) internal pure returns(address){
// SBTC
if (token == address(0x075b1bb99792c9E1041bA13afEf80C91a1e70fB3)){ if (token == address(0x075b1bb99792c9E1041bA13afEf80C91a1e70fB3)){
// SBTC
return 0x13C1542A468319688B89E323fe9A3Be3A90EBb27; return 0x13C1542A468319688B89E323fe9A3Be3A90EBb27;
// SUSD } else if (token == address(0xC25a3A3b969415c80451098fa907EC722572917F)){
}else if (token == address(0xC25a3A3b969415c80451098fa907EC722572917F)){ // SUSD
return 0xDCB6A51eA3CA5d3Fd898Fd6564757c7aAeC3ca92; return 0xDCB6A51eA3CA5d3Fd898Fd6564757c7aAeC3ca92;
}else{ } else {
revert("token-not-found"); revert("token-not-found");
} }
} }
function getSnxAddr() internal pure returns (address) {
return 0xC011a73ee8576Fb46F5E1c5751cA3B9Fe0af2a6F;
}
} }
contract SynthetixStakingRewardsProtocol is Helper { contract SynthetixStaking is SynthetixStakingHelper {
// Events // Events
event LogStake( event LogDeposit(
address stakeAddr, address token,
uint256 stakeAmt, uint256 amount,
uint getId uint getId,
uint setId
); );
event LogExit( event LogWithdraw(
address stakeAddr, address token,
uint256 stakeAmt, uint256 amount,
uint getId uint256 rewardAmt,
uint getId,
uint setId
); );
/** /**
* @dev Stake Token. * @dev Deposit Token.
* @param stakeAddr staking token address. * @param token staking token address.
* @param stakeAmt staking token amount. * @param amt staking token amount.
* @param getId Get token amount at this ID from `InstaMemory` Contract. * @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/ */
function stake( function deposit(
address stakeAddr, address token,
uint stakeAmt, uint amt,
uint getId uint getId,
uint setId
) external { ) external {
uint _stakeAmt = getUint(getId, stakeAmt); uint _amt = getUint(getId, amt);
IStakingRewards rewardPool = IStakingRewards(getSynthetixStakingAddr(stakeAddr)); IStakingRewards stakingContract = IStakingRewards(getSynthetixStakingAddr(token));
TokenInterface _stakeToken = TokenInterface(stakeAddr); TokenInterface _stakeToken = TokenInterface(token);
_stakeAmt = _stakeAmt == uint(-1) ? _stakeToken.balanceOf(address(this)) : _stakeAmt; _amt = _amt == uint(-1) ? _stakeToken.balanceOf(address(this)) : _amt;
_stakeToken.approve(address(rewardPool), _stakeAmt);
rewardPool.stake(_stakeAmt); _stakeToken.approve(address(stakingContract), _amt);
stakingContract.stake(_amt);
emit LogStake(stakeAddr, _stakeAmt, getId); emit LogDeposit(token, _amt, getId, setId);
bytes32 _eventCode = keccak256("LogStake(address,uint256, uint256)"); bytes32 _eventCode = keccak256("LogDeposit(address,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(stakeAddr, _stakeAmt, getId); bytes memory _eventParam = abi.encode(token, _amt, getId, setId);
emitEvent(_eventCode, _eventParam); emitEvent(_eventCode, _eventParam);
} }
/** /**
* @dev Exit Token. * @dev Withdraw Token.
* @param stakeAddr staking token address. * @param token staking token address.
* @param stakeAmt staking token amount. * @param amt staking token amount.
* @param getId Get token amount at this ID from `InstaMemory` Contract. * @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/ */
function exit( function withdraw(
address stakeAddr, address token,
uint stakeAmt, uint amt,
uint getId uint getId,
uint setId
) external { ) external {
uint _stakeAmt = getUint(getId, stakeAmt); uint _amt = getUint(getId, amt);
IStakingRewards rewardPool = IStakingRewards(getSynthetixStakingAddr(stakeAddr)); IStakingRewards stakingContract = IStakingRewards(getSynthetixStakingAddr(token));
TokenInterface snxToken = TokenInterface(getSnxAddr());
if(_stakeAmt == uint(-1)){ uint intialBal = snxToken.balanceOf(address(this));
rewardPool.exit(); if (_amt == uint(-1)) {
} stakingContract.exit();
else{ } else{
rewardPool.withdraw(_stakeAmt); stakingContract.withdraw(_amt);
rewardPool.getReward(); stakingContract.getReward();
} }
uint finalBal = snxToken.balanceOf(address(this));
emit LogExit(stakeAddr, _stakeAmt, getId); uint rewardAmt = sub(finalBal, intialBal);
bytes32 _eventCode = keccak256("LogExit(address, uint256, uint256)");
bytes memory _eventParam = abi.encode(stakeAddr, _stakeAmt, getId); setUint(setId, _amt);
emit LogWithdraw(token, _amt, rewardAmt, getId, setId);
bytes32 _eventCode = keccak256("LogWithdraw(address,uint256,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, rewardAmt, getId, setId);
emitEvent(_eventCode, _eventParam); emitEvent(_eventCode, _eventParam);
} }
} }
contract ConnectSynthetixStakingRewardsProtocol is SynthetixStakingRewardsProtocol { contract ConnectSynthetixStaking is SynthetixStaking {
string public name = "synthetix-staking-v1"; string public name = "synthetix-staking-v1";
} }