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

chore: add getters for user's balances

Browse Source
This commit is contained in:
Andrea Franz
2024-09-26 17:46:12 +02:00
parent 24c3469cd1
commit 982946ca5e
1 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/RewardsStreamerMP.sol
View File

@@ -194,16 +194,23 @@ contract RewardsStreamerMP is ReentrancyGuard {
}
function updateUserMP(address userAddress) internal {
UserInfo storage user = users[userAddress];
uint256 accruedMP = _getUserUnrealizedAccruedMP(userAddress);
UserInfo storage user = users[userAddress];
user.lastMPUpdateTime = block.timestamp;
user.userPotentialMP -= accruedMP;
user.userMP += accruedMP;
}
function _getUserUnrealizedAccruedMP(address userAddress) internal view returns (uint256) {
UserInfo storage user = users[userAddress];
if (user.userPotentialMP == 0 || user.stakedBalance == 0) {
user.lastMPUpdateTime = block.timestamp;
return;
return 0;
}
uint256 timeDiff = block.timestamp - user.lastMPUpdateTime;
if (timeDiff == 0) {
return;
return 0;
}
uint256 accruedMP = (timeDiff * user.stakedBalance * MP_RATE_PER_YEAR) / (365 days * SCALE_FACTOR);
@@ -212,10 +219,21 @@ contract RewardsStreamerMP is ReentrancyGuard {
accruedMP = user.userPotentialMP;
}
user.userPotentialMP -= accruedMP;
user.userMP += accruedMP;
return accruedMP;
}
user.lastMPUpdateTime = block.timestamp;
function getUserUnrealizedAccruedMP(address userAddress) external view returns (uint256) {
return _getUserUnrealizedAccruedMP(userAddress);
}
function getUserTotalMP(address userAddress) external view returns (uint256) {
UserInfo storage user = users[userAddress];
return user.userMP + _getUserUnrealizedAccruedMP(userAddress);
}
function getUserTotalBalance(address userAddress) external view returns (uint256) {
UserInfo storage user = users[userAddress];
return user.stakedBalance + user.userMP + _getUserUnrealizedAccruedMP(userAddress);
}
function calculateUserRewards(address userAddress) public view returns (uint256) {