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

Minor changes in mapping contract

Browse Source
This commit is contained in:
Thrilok Kumar 2020-07-14 22:55:45 +05:30
parent 9ce77c1336
commit 18eedb8833
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
contracts/mapping/staking.sol
View File

@ -45,11 +45,12 @@ contract Helpers is BytesHelper {
address public constant instaIndex = 0x2971AdFa57b20E5a416aE5a708A8655A9c74f723; address public constant instaIndex = 0x2971AdFa57b20E5a416aE5a708A8655A9c74f723;
uint public version = 1; uint public version = 1;
mapping (bytes32 => StakingData) public stakingMapping;
struct StakingData { struct StakingData {
address stakingPool; address stakingPool;
address stakingToken; address stakingToken;
} }
mapping (bytes32 => StakingData) public stakingMapping;
event LogAddStakingMapping(string stakingName, bytes32 stakingType, address stakingAddress, address stakingToken); event LogAddStakingMapping(string stakingName, bytes32 stakingType, address stakingAddress, address stakingToken);
event LogRemoveStakingMapping(string stakingName, bytes32 stakingType, address stakingAddress, address stakingToken); event LogRemoveStakingMapping(string stakingName, bytes32 stakingType, address stakingAddress, address stakingToken);
@ -60,13 +61,14 @@ contract Helpers is BytesHelper {
IndexInterface(instaIndex).master() == msg.sender, "not-Chief"); IndexInterface(instaIndex).master() == msg.sender, "not-Chief");
_; _;
} }
function addStakingMapping(string memory stakingName, address stakingAddress, address stakingToken) public isChief { function addStakingMapping(string memory stakingName, address stakingAddress, address stakingToken) public isChief {
require(stakingAddress != address(0), "StakingPool-not-vaild"); require(stakingAddress != address(0), "stakingAddress-not-vaild");
require(stakingToken != address(0), "StakingToken-not-vaild"); require(stakingToken != address(0), "stakingToken-not-vaild");
require(bytes(stakingName).length < 32, "Lenght-exceeds"); // TODO - test this. require(bytes(stakingName).length <= 32, "Length-exceeds");
bytes32 stakeType = stringToBytes32(stakingName); bytes32 stakeType = stringToBytes32(stakingName);
require(stakingMapping[stakeType].stakingPool == address(0), "StakingPool-Already-Added"); require(stakingMapping[stakeType].stakingPool == address(0), "StakingPool-already-added");
require(stakingMapping[stakeType].stakingToken == address(0), "StakingPool-Already-Added"); require(stakingMapping[stakeType].stakingToken == address(0), "StakingToken-already-added");
stakingMapping[stakeType] = StakingData( stakingMapping[stakeType] = StakingData(
stakingAddress, stakingAddress,
@ -76,12 +78,11 @@ contract Helpers is BytesHelper {
} }
function removeStakingMapping(string memory stakingName, address stakingAddress) public isChief { function removeStakingMapping(string memory stakingName, address stakingAddress) public isChief {
require(stakingAddress != address(0), "StakingPool-not-vaild"); require(stakingAddress != address(0), "stakingAddress-not-vaild");
require(bytes(stakingName).length < 32, "Lenght-exceeds"); // TODO - test this.
bytes32 stakeType = stringToBytes32(stakingName); bytes32 stakeType = stringToBytes32(stakingName);
require(stakingMapping[stakeType].stakingPool != address(0), "StakingPool-Already-Added"); require(stakingMapping[stakeType].stakingPool != address(0), "StakingPool-not-added-yet");
require(stakingMapping[stakeType].stakingToken != address(0), "StakingPool-Already-Added"); require(stakingMapping[stakeType].stakingToken != address(0), "StakingToken-not-added-yet");
require(stakingMapping[stakeType].stakingPool == stakingAddress, "Not-same-staking-pool"); require(stakingMapping[stakeType].stakingPool == stakingAddress, "different-staking-pool");
emit LogRemoveStakingMapping(stakingName, stakeType, stakingAddress, stakingMapping[stakeType].stakingToken); emit LogRemoveStakingMapping(stakingName, stakeType, stakingAddress, stakingMapping[stakeType].stakingToken);
delete stakingMapping[stakeType]; delete stakingMapping[stakeType];