vacp2p/staking-reward-streamer
1
0
Fork 0
mirror of https://github.com/vacp2p/staking-reward-streamer.git synced 2026-01-08 20:48:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix unstake function

Browse Source
This commit is contained in:
Andrea Franz
2024-09-16 12:38:05 +02:00
parent e9ea8102e1
commit 90250c9ebc
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/RewardsStreamerMP.sol
View File

@@ -104,16 +104,32 @@ contract RewardsStreamerMP is ReentrancyGuard {
revert StakingManager__InsufficientBalance();
}
if (block.timestamp < user.lockUntil) {
revert StakingManager__TokensAreLocked();
}
updateRewardIndex();
updateGlobalMP();
updateUserMP(msg.sender);
uint256 userRewards = calculateUserRewards(msg.sender);
if (userRewards > 0) {
distributeRewards(msg.sender, userRewards);
}
uint256 previousStakedBalance = user.stakedBalance;
user.stakedBalance -= amount;
totalStaked -= amount;
uint256 amountRatio = (amount * SCALE_FACTOR) / previousStakedBalance;
uint256 mpToReduce = (user.userMP * amountRatio) / SCALE_FACTOR;
uint256 potentialMPToReduce = (user.userPotentialMP * amountRatio) / SCALE_FACTOR;
user.userMP -= mpToReduce;
user.userPotentialMP -= potentialMPToReduce;
totalMP -= mpToReduce;
potentialMP -= potentialMPToReduce;
bool success = stakingToken.transfer(msg.sender, amount);
if (!success) {
revert StakingManager__TransferFailed();