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

force users to migrate or leave

This commit is contained in:
Ricardo Guilherme Schmidt 2023-06-26 12:26:02 -03:00
parent da5ea645b0
commit 6bf8d66221
No known key found for this signature in database
GPG Key ID: 3F95A3AD0B607030
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
contracts/StakeManager.sol
View File

@ -104,6 +104,16 @@ contract StakeManager is Ownable {
multiplierSupply += increasedMultiplier;
}
/**
* @notice leave without processing account
*/
function leave() external onlyVault {
Account memory account = accounts[msg.sender];
delete accounts[msg.sender];
multiplierSupply -= account.multiplier;
stakeSupply -= _account.mount;
}
/**
* @notice Release rewards for current epoch and increase epoch.
*/
@ -170,7 +180,8 @@ contract StakeManager is Ownable {
function processAccount(Account storage account, uint256 _limitEpoch) private {
processEpoch();
require(_limitEpoch <= currentEpoch, "Non-sese call");
require(address(migration) == address(0), "Contract ended, please migrate");
require(_limitEpoch <= currentEpoch, "Non-sense call");
uint256 userReward;
uint256 userEpoch = account.epoch;
for (Epoch memory iEpoch = epochs[userEpoch]; userEpoch < _limitEpoch; userEpoch++) {

5
contracts/StakeVault.sol
View File

@ -32,6 +32,11 @@ contract StakeVault is Ownable {
stakedToken.transferFrom(address(this), msg.sender, _amount);
}
function leave() external onlyOwner {
stakeManager.leave();
stakedToken.transferFrom(address(this), msg.sender, stakedToken.balanceOf(address(this)));
}
/**
* @notice Opt-in migration to a new StakeManager contract.
*/