-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #23 from iotexproject/fix-reward
Fix reward pool
- Loading branch information
Showing
10 changed files
with
502 additions
and
117 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,192 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity ^0.8.0; | ||
|
|
||
| import {IERC721} from "@openzeppelin/contracts/token/ERC721/IERC721.sol"; | ||
| import {ERC721Holder} from "@openzeppelin/contracts/token/ERC721/utils/ERC721Holder.sol"; | ||
| import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol"; | ||
| import {ReentrancyGuardUpgradeable} from "@openzeppelin/contracts-upgradeable/security/ReentrancyGuardUpgradeable.sol"; | ||
|
|
||
| import {IWeightedNFT} from "../interfaces/IWeightedNFT.sol"; | ||
|
|
||
| contract FixedRewardPool is OwnableUpgradeable, ReentrancyGuardUpgradeable, ERC721Holder { | ||
| event Deposit(address indexed user, uint256 tokenId, uint256 rewards, uint256 weight); | ||
| event Withdraw(address indexed user, uint256 tokenId, uint256 rewards, uint256 weight); | ||
| event EmergencyWithdraw(address indexed user, uint256 tokenId, uint256 weight); | ||
| event Poke(address indexed user, uint256 tokenId, uint256 rewards, uint256 originalWeight, uint256 newWeight); | ||
| event ClaimRewards(address indexed user, uint256 rewards); | ||
|
|
||
| struct UserInfo { | ||
| uint256 amount; // How many staked tokens the user has provided | ||
| uint256 rewardDebt; // Reward debt | ||
| } | ||
|
|
||
| // The precision factor | ||
| uint256 immutable PRECISION_FACTOR = 1e12; | ||
|
|
||
| // The weighted NFT contract | ||
| IWeightedNFT public weightNFT; | ||
| // Accrued token per share | ||
| uint256 public accTokenPerShare; | ||
| // The block number of the last pool update | ||
| uint256 public lastRewardBlock; | ||
| // Reward tokens created per block. | ||
| uint256 public rewardPerBlock; | ||
| // Total staked weight | ||
| uint256 public totalStakedWeight; | ||
| // Info of each user that stakes tokens (stakedToken) | ||
| mapping(address => UserInfo) public userInfo; | ||
|
|
||
| // Mapping from tokenId to staker | ||
| mapping(uint256 => address) public tokenStaker; | ||
| // Mapping from tokenId to weight | ||
| mapping(uint256 => uint256) public tokenWeight; | ||
|
|
||
| function initialize(address _nft, uint256 _startBlock, uint256 _rewardPerBlock) external initializer { | ||
| require(_startBlock >= block.number, "invalid start block"); | ||
| require(_rewardPerBlock > 0, "invalid reward per block"); | ||
|
|
||
| __Ownable_init(); | ||
| __ReentrancyGuard_init(); | ||
| weightNFT = IWeightedNFT(_nft); | ||
| lastRewardBlock = _startBlock; | ||
| rewardPerBlock = _rewardPerBlock; | ||
| } | ||
|
|
||
| function deposit(uint256 _tokenId) external nonReentrant { | ||
| UserInfo storage user = userInfo[msg.sender]; | ||
| _updatePool(); | ||
|
|
||
| uint256 _pending = 0; | ||
| if (user.amount > 0) { | ||
| _pending = (user.amount * accTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
| if (_pending > 0) { | ||
| (bool success, ) = msg.sender.call{value: _pending}(""); | ||
| require(success, "Failed to send reward"); | ||
| } | ||
| } | ||
|
|
||
| uint _amount = weightNFT.weight(_tokenId); | ||
| if (_amount > 0) { | ||
| user.amount = user.amount + _amount; | ||
| IERC721(weightNFT.nft()).safeTransferFrom(msg.sender, address(this), _tokenId); | ||
| tokenStaker[_tokenId] = msg.sender; | ||
| tokenWeight[_tokenId] = _amount; | ||
| totalStakedWeight = totalStakedWeight + _amount; | ||
| } | ||
|
|
||
| user.rewardDebt = (user.amount * accTokenPerShare) / PRECISION_FACTOR; | ||
|
|
||
| emit Deposit(msg.sender, _tokenId, _pending, _amount); | ||
| } | ||
|
|
||
| function withdraw(uint256 _tokenId) external nonReentrant { | ||
| require(tokenStaker[_tokenId] == msg.sender, "Invalid staker"); | ||
| UserInfo storage user = userInfo[msg.sender]; | ||
|
|
||
| _updatePool(); | ||
|
|
||
| uint256 _pending = (user.amount * accTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
|
|
||
| uint256 _amount = tokenWeight[_tokenId]; | ||
| user.amount = user.amount - _amount; | ||
| IERC721(weightNFT.nft()).safeTransferFrom(address(this), msg.sender, _tokenId); | ||
| tokenStaker[_tokenId] = address(0); | ||
| tokenWeight[_tokenId] = 0; | ||
| totalStakedWeight = totalStakedWeight - _amount; | ||
|
|
||
| if (_pending > 0) { | ||
| (bool success, ) = msg.sender.call{value: _pending}(""); | ||
| require(success, "Failed to send reward"); | ||
| } | ||
|
|
||
| user.rewardDebt = (user.amount * accTokenPerShare) / PRECISION_FACTOR; | ||
|
|
||
| emit Withdraw(msg.sender, _tokenId, _pending, _amount); | ||
| } | ||
|
|
||
| function emergencyWithdraw(uint256 _tokenId) external nonReentrant { | ||
| require(tokenStaker[_tokenId] == msg.sender, "Invalid staker"); | ||
| UserInfo storage user = userInfo[msg.sender]; | ||
|
|
||
| uint256 _amount = tokenWeight[_tokenId]; | ||
|
|
||
| user.amount = user.amount - _amount; | ||
| IERC721(weightNFT.nft()).safeTransferFrom(address(this), msg.sender, _tokenId); | ||
| tokenStaker[_tokenId] = address(0); | ||
| tokenWeight[_tokenId] = 0; | ||
| totalStakedWeight = totalStakedWeight - _amount; | ||
|
|
||
| emit EmergencyWithdraw(msg.sender, _tokenId, user.amount); | ||
| } | ||
|
|
||
| function poke(uint256 _tokenId) external nonReentrant { | ||
| address _staker = tokenStaker[_tokenId]; | ||
| require(_staker != address(0), "Invalid token"); | ||
|
|
||
| UserInfo storage user = userInfo[_staker]; | ||
|
|
||
| _updatePool(); | ||
|
|
||
| uint256 _pending = (user.amount * accTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
| if (_pending > 0) { | ||
| (bool success, ) = _staker.call{value: _pending}(""); | ||
| require(success, "Failed to send reward"); | ||
| } | ||
|
|
||
| uint256 _originalWeight = tokenWeight[_tokenId]; | ||
| uint256 _newWeight = weightNFT.weight(_tokenId); | ||
| if (_originalWeight != _newWeight) { | ||
| user.amount = user.amount - _originalWeight + _newWeight; | ||
| tokenWeight[_tokenId] = _newWeight; | ||
| totalStakedWeight = totalStakedWeight - _originalWeight + _newWeight; | ||
| } | ||
|
|
||
| user.rewardDebt = (user.amount * accTokenPerShare) / PRECISION_FACTOR; | ||
|
|
||
| emit Poke(_staker, _tokenId, _pending, _originalWeight, _newWeight); | ||
| } | ||
|
|
||
| function claimRewards(address _user) external nonReentrant { | ||
| UserInfo storage user = userInfo[_user]; | ||
|
|
||
| _updatePool(); | ||
|
|
||
| uint256 _pending = (user.amount * accTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
| if (_pending > 0) { | ||
| (bool success, ) = _user.call{value: _pending}(""); | ||
| require(success, "Failed to send reward"); | ||
| } | ||
|
|
||
| user.rewardDebt = (user.amount * accTokenPerShare) / PRECISION_FACTOR; | ||
| emit ClaimRewards(_user, _pending); | ||
| } | ||
|
|
||
| function pendingReward(address _user) external view returns (uint256) { | ||
| UserInfo storage user = userInfo[_user]; | ||
| uint256 _totalStakedWeight = totalStakedWeight; | ||
| if (block.number > lastRewardBlock && _totalStakedWeight != 0) { | ||
| uint256 rewards = (block.number - lastRewardBlock) * rewardPerBlock; | ||
| uint256 adjustedTokenPerShare = accTokenPerShare + (rewards * PRECISION_FACTOR) / _totalStakedWeight; | ||
| return (user.amount * adjustedTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
| } else { | ||
| return (user.amount * accTokenPerShare) / PRECISION_FACTOR - user.rewardDebt; | ||
| } | ||
| } | ||
|
|
||
| function _updatePool() internal { | ||
| if (block.number <= lastRewardBlock) { | ||
| return; | ||
| } | ||
|
|
||
| if (totalStakedWeight == 0) { | ||
| lastRewardBlock = block.number; | ||
| return; | ||
| } | ||
|
|
||
| uint256 rewards = (block.number - lastRewardBlock) * rewardPerBlock; | ||
| accTokenPerShare = accTokenPerShare + (rewards * PRECISION_FACTOR) / totalStakedWeight; | ||
| lastRewardBlock = block.number; | ||
| } | ||
|
|
||
| receive() external payable {} | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,38 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity ^0.8.0; | ||
|
|
||
| import {IERC721} from "@openzeppelin/contracts/token/ERC721/IERC721.sol"; | ||
| import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol"; | ||
|
|
||
| import {IWeightedNFT} from "../interfaces/IWeightedNFT.sol"; | ||
|
|
||
| contract OwnedWeightedNFT is Ownable, IWeightedNFT { | ||
| event WeightChanged(uint256 tokenId, uint256 weight); | ||
|
|
||
| uint256 public immutable DEFAULT_WEIGHT = 100; | ||
|
|
||
| address public immutable override nft; | ||
| mapping(uint256 => uint256) _weights; | ||
|
|
||
| constructor(address _nft, address _owner) { | ||
| nft = _nft; | ||
| transferOwnership(_owner); | ||
| } | ||
|
|
||
| function weight(uint256 tokenId) external view override returns (uint256) { | ||
| require(IERC721(nft).ownerOf(tokenId) != address(0), "token not minted"); | ||
|
|
||
| uint256 _weight = _weights[tokenId]; | ||
| if (_weight == 0) { | ||
| return DEFAULT_WEIGHT; | ||
| } | ||
| return _weights[tokenId]; | ||
| } | ||
|
|
||
| function setWeight(uint256 tokenId, uint256 _weight) external onlyOwner { | ||
| require(_weight > 0, "invalid weight"); | ||
|
|
||
| _weights[tokenId] = _weight; | ||
| emit WeightChanged(tokenId, _weight); | ||
| } | ||
| } |
Oops, something went wrong.