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

WIP: custom revert when restaking

This commit is contained in:
r4bbit 2024-07-17 16:19:36 +02:00
parent 5c82bfceeb
commit bb364a970c
No known key found for this signature in database
GPG Key ID: E95F1E9447DC91A9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
contracts/StakeManager.sol
View File

@ -20,6 +20,7 @@ contract StakeManager is Ownable {
error StakeManager__InvalidMigration();
error StakeManager__AlreadyProcessedEpochs();
error StakeManager__InsufficientFunds();
error StakeManager__AlreadyStaked();
struct Account {
address rewardAddress;
@ -155,7 +156,9 @@ contract StakeManager is Ownable {
*/
function stake(uint256 _amount, uint256 _timeToIncrease) external onlyVault noPendingMigration finalizeEpoch {
Account storage account = accounts[msg.sender];
require(account.balance == 0); //cant add more stake
if (account.balance > 0) {
revert StakeManager__AlreadyStaked();
}
if (account.lockUntil == 0) {
// account not initialized