logos-co
/
staking
mirror of https://github.com/logos-co/staking.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Only vaults that Re previosly allowed

This commit is contained in:
Ricardo Guilherme Schmidt 2023-06-23 15:03:52 -03:00
parent d960512915
commit ff3a7b09ea
No known key found for this signature in database
GPG Key ID: 3F95A3AD0B607030
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
contracts/StakeManager.sol
View File

@ -4,7 +4,7 @@ pragma solidity 0.8.19;
import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
contract StakeManager is ERC20 {
contract StakeManager is Controlled {
struct Account {
uint256 lockUntil;
@ -27,12 +27,18 @@ contract StakeManager is ERC20 {
mapping (address => Account) accounts;
mapping (uint256 => Epoch) epoch;
mapping (bytes32 => bool) isVault;
ERC20 stakedToken;
uint256 currentEpoch;
uint256 pendingReward;
uint256 public totalSupply;
modifier onlyVault {
require(isVault[msg.sender.codehash], "Not a vault")
_;
}
constructor() {
epoch[0].startTime = now();
}
@ -135,6 +141,14 @@ contract StakeManager is ERC20 {
pendingReward -= userReward;
stakedToken.transfer(_vault, userReward);
}
/**
* @notice Enables a contract class to interact with staking functions
* @param _codehash bytecode hash of contract
*/
function setVault(bytes32 _codehash) external onlyController {
isVault[_codehash] = true;
}
function checkMaxMultiplier(uint256 _increasedMultiplier, uint256 _currentMp) private view returns(uint256 _maxToIncrease) {
uint256 newMp = _increasedMultiplier + _currentMp;