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(StakeManager): inherit UUPSUpgradable, replace constructor code with disable initializer and set owner 0.

Browse Source
This commit is contained in:
Ricardo Guilherme Schmidt
2024-11-29 12:13:28 -03:00
parent 3eff0b4dc2
commit fa40f9026d
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/RewardsStreamerMP.sol
View File

@@ -4,11 +4,12 @@ pragma solidity ^0.8.26;
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import { ReentrancyGuardTransient } from "@openzeppelin/contracts/utils/ReentrancyGuardTransient.sol";
import { Initializable } from "@openzeppelin/contracts/proxy/utils/Initializable.sol";
import { UUPSUpgradeable } from "@openzeppelin/contracts/proxy/utils/UUPSUpgradeable.sol";
import { IStakeManager } from "./interfaces/IStakeManager.sol";
import { TrustedCodehashAccess } from "./TrustedCodehashAccess.sol";
// Rewards Streamer with Multiplier Points
contract RewardsStreamerMP is Initializable, IStakeManager, TrustedCodehashAccess, ReentrancyGuardTransient {
contract RewardsStreamerMP is UUPSUpgradeable, Initializable, IStakeManager, TrustedCodehashAccess, ReentrancyGuardTransient {
error StakingManager__AmountCannotBeZero();
error StakingManager__TransferFailed();
error StakingManager__InsufficientBalance();
@@ -54,10 +55,9 @@ contract RewardsStreamerMP is Initializable, IStakeManager, TrustedCodehashAcces
_;
}
constructor(address _owner, address _stakingToken, address _rewardToken) TrustedCodehashAccess(_owner) {
STAKING_TOKEN = IERC20(_stakingToken);
REWARD_TOKEN = IERC20(_rewardToken);
lastMPUpdatedTime = block.timestamp;
constructor() TrustedCodehashAccess(msg.sender) {
_transferOwnership(address(0));
_disableInitializers();
}
function initialize(address _owner, address _stakingToken, address _rewardToken) public initializer {
@@ -71,6 +71,10 @@ contract RewardsStreamerMP is Initializable, IStakeManager, TrustedCodehashAcces
lastMPUpdatedTime = block.timestamp;
}
function _authorizeUpgrade(address) internal view override {
_checkOwner();
}
function stake(uint256 amount, uint256 lockPeriod) external onlyTrustedCodehash onlyNotEmergencyMode nonReentrant {
if (amount == 0) {
revert StakingManager__AmountCannotBeZero();