文件 1 的 5:Context.sol
pragma solidity ^0.8.0;
abstract contract Context {
function _msgSender() internal view virtual returns (address) {
return msg.sender;
}
function _msgData() internal view virtual returns (bytes calldata) {
return msg.data;
}
}
文件 2 的 5:EnumerableSet.sol
pragma solidity ^0.8.0;
library EnumerableSet {
struct Set {
bytes32[] _values;
mapping(bytes32 => uint256) _indexes;
}
function _add(Set storage set, bytes32 value) private returns (bool) {
if (!_contains(set, value)) {
set._values.push(value);
set._indexes[value] = set._values.length;
return true;
} else {
return false;
}
}
function _remove(Set storage set, bytes32 value) private returns (bool) {
uint256 valueIndex = set._indexes[value];
if (valueIndex != 0) {
uint256 toDeleteIndex = valueIndex - 1;
uint256 lastIndex = set._values.length - 1;
if (lastIndex != toDeleteIndex) {
bytes32 lastvalue = set._values[lastIndex];
set._values[toDeleteIndex] = lastvalue;
set._indexes[lastvalue] = valueIndex;
}
set._values.pop();
delete set._indexes[value];
return true;
} else {
return false;
}
}
function _contains(Set storage set, bytes32 value) private view returns (bool) {
return set._indexes[value] != 0;
}
function _length(Set storage set) private view returns (uint256) {
return set._values.length;
}
function _at(Set storage set, uint256 index) private view returns (bytes32) {
return set._values[index];
}
function _values(Set storage set) private view returns (bytes32[] memory) {
return set._values;
}
struct Bytes32Set {
Set _inner;
}
function add(Bytes32Set storage set, bytes32 value) internal returns (bool) {
return _add(set._inner, value);
}
function remove(Bytes32Set storage set, bytes32 value) internal returns (bool) {
return _remove(set._inner, value);
}
function contains(Bytes32Set storage set, bytes32 value) internal view returns (bool) {
return _contains(set._inner, value);
}
function length(Bytes32Set storage set) internal view returns (uint256) {
return _length(set._inner);
}
function at(Bytes32Set storage set, uint256 index) internal view returns (bytes32) {
return _at(set._inner, index);
}
function values(Bytes32Set storage set) internal view returns (bytes32[] memory) {
return _values(set._inner);
}
struct AddressSet {
Set _inner;
}
function add(AddressSet storage set, address value) internal returns (bool) {
return _add(set._inner, bytes32(uint256(uint160(value))));
}
function remove(AddressSet storage set, address value) internal returns (bool) {
return _remove(set._inner, bytes32(uint256(uint160(value))));
}
function contains(AddressSet storage set, address value) internal view returns (bool) {
return _contains(set._inner, bytes32(uint256(uint160(value))));
}
function length(AddressSet storage set) internal view returns (uint256) {
return _length(set._inner);
}
function at(AddressSet storage set, uint256 index) internal view returns (address) {
return address(uint160(uint256(_at(set._inner, index))));
}
function values(AddressSet storage set) internal view returns (address[] memory) {
bytes32[] memory store = _values(set._inner);
address[] memory result;
assembly {
result := store
}
return result;
}
struct UintSet {
Set _inner;
}
function add(UintSet storage set, uint256 value) internal returns (bool) {
return _add(set._inner, bytes32(value));
}
function remove(UintSet storage set, uint256 value) internal returns (bool) {
return _remove(set._inner, bytes32(value));
}
function contains(UintSet storage set, uint256 value) internal view returns (bool) {
return _contains(set._inner, bytes32(value));
}
function length(UintSet storage set) internal view returns (uint256) {
return _length(set._inner);
}
function at(UintSet storage set, uint256 index) internal view returns (uint256) {
return uint256(_at(set._inner, index));
}
function values(UintSet storage set) internal view returns (uint256[] memory) {
bytes32[] memory store = _values(set._inner);
uint256[] memory result;
assembly {
result := store
}
return result;
}
}
文件 3 的 5:IERC721Receiver.sol
pragma solidity ^0.8.0;
interface IERC721Receiver {
function onERC721Received(
address operator,
address from,
uint256 tokenId,
bytes calldata data
) external returns (bytes4);
}
文件 4 的 5:Ownable.sol
pragma solidity ^0.8.0;
import "../utils/Context.sol";
abstract contract Ownable is Context {
address private _owner;
event OwnershipTransferred(address indexed previousOwner, address indexed newOwner);
constructor() {
_transferOwnership(_msgSender());
}
function owner() public view virtual returns (address) {
return _owner;
}
modifier onlyOwner() {
require(owner() == _msgSender(), "Ownable: caller is not the owner");
_;
}
function renounceOwnership() public virtual onlyOwner {
_transferOwnership(address(0));
}
function transferOwnership(address newOwner) public virtual onlyOwner {
require(newOwner != address(0), "Ownable: new owner is the zero address");
_transferOwnership(newOwner);
}
function _transferOwnership(address newOwner) internal virtual {
address oldOwner = _owner;
_owner = newOwner;
emit OwnershipTransferred(oldOwner, newOwner);
}
}
文件 5 的 5:StakingPool.sol
pragma solidity >=0.4.22 <0.9.0;
import "@openzeppelin/contracts/token/ERC721/IERC721Receiver.sol";
import "@openzeppelin/contracts/access/Ownable.sol";
import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
interface TKONFT{
function ownerOf(uint256 tokenId) external returns (address);
function safeTransferFrom(
address from,
address to,
uint256 tokenId,
bytes memory _data
) external;
}
interface UtilityToken{
function transfer(address recipient, uint256 amount) external returns (bool);
}
contract StakingPool is IERC721Receiver, Ownable{
using EnumerableSet for EnumerableSet.UintSet;
event StakeStarted(address indexed user, uint256 indexed tokenId);
event StakeStopped(address indexed user, uint256 indexed tokenId);
event UtilityAddrSet(address from, address addr);
event StartStakingTimeChanged(uint256 newTime, uint256 currentTime);
event EndStakingTimeChanged(uint256 newTime, uint256 currentTime);
event GetReward(address indexed user, uint256 amount);
UtilityToken private _utilityToken;
TKONFT private _tentacle;
struct StakedInfo {
uint256 lastUpdate;
}
mapping(uint256 => StakedInfo) private tokenInfo;
mapping(address => EnumerableSet.UintSet) private stakedTko;
address private _tkoContract;
uint256 public rewardUnit = 333;
uint256 private _tTotal = 26000000;
uint256 private _nowTotal = 0;
uint256 public startTimeOfStaking;
uint256 public endTimeOfStaking;
modifier masterContract() {
require(
msg.sender == _tkoContract,
"Master Contract can only call Staking Contract"
);
_;
}
modifier enableStaking() {
require(startTimeOfStaking!=0 && endTimeOfStaking!=0, 'Start and end time of staking not set');
require(
endTimeOfStaking >= startTimeOfStaking,
"Staking Mechanism is not started yet"
);
_;
}
constructor(address _tkoAddr) {
_tkoContract = _tkoAddr;
startTimeOfStaking=0;
endTimeOfStaking=0;
_tentacle = TKONFT(_tkoAddr);
}
function setUtilitytoken(address payable _addr) external onlyOwner {
_utilityToken = UtilityToken(_addr);
emit UtilityAddrSet(address(this), _addr);
}
function changeRewardUnit(uint256 _rewardUnit) external onlyOwner {
rewardUnit = _rewardUnit;
}
function setStartStakingTime(uint256 _timeStamp) external onlyOwner {
startTimeOfStaking = _timeStamp;
emit StartStakingTimeChanged(startTimeOfStaking, block.timestamp);
}
function setEndStakingTime(uint256 _timeStamp) external onlyOwner {
endTimeOfStaking = _timeStamp;
emit EndStakingTimeChanged(endTimeOfStaking, block.timestamp);
}
function startStakings(address _user, uint256[] memory _tokenIds)
external
{
require(_tokenIds.length>=4, "4 NFTs min to stake");
for (uint256 i = 0; i < _tokenIds.length; i++) {
require(!stakedTko[_user].contains(_tokenIds[i]), "Already staked");
require(_tentacle.ownerOf(_tokenIds[i]) == msg.sender, "Staking: owner not matched");
}
for (uint256 i = 0; i < _tokenIds.length; i++) {
tokenInfo[_tokenIds[i]].lastUpdate = block.timestamp;
stakedTko[_user].add(_tokenIds[i]);
_tentacle.safeTransferFrom(_user, address(this), _tokenIds[i], "");
emit StakeStarted(_user, _tokenIds[i]);
}
}
function stopStaking(address _user, uint256 _tokenId)
external
enableStaking
masterContract
{
require(stakedTko[_user].contains(_tokenId), "You're not the owner");
require(block.timestamp >= startTimeOfStaking, "Didn't start staking");
uint256 interval;
if(block.timestamp<endTimeOfStaking){
interval = block.timestamp - tokenInfo[_tokenId].lastUpdate;
}else{
interval = endTimeOfStaking - tokenInfo[_tokenId].lastUpdate -1;
}
uint256 day=interval / 86400;
uint256 reward = rewardUnit * day * 10**9;
if(reward>0){
_utilityToken.transfer(_user, reward);
}
delete tokenInfo[_tokenId];
stakedTko[_user].remove(_tokenId);
emit StakeStopped(_user, _tokenId);
}
function stakedTokensOf(address _user)
public
view
returns (uint256[] memory)
{
uint256[] memory tokens = new uint256[](stakedTko[_user].length());
for (uint256 i = 0; i < stakedTko[_user].length(); i++) {
tokens[i] = stakedTko[_user].at(i);
}
return tokens;
}
function getClaimableToken(address _user) public view enableStaking returns (uint256) {
uint256[] memory tokens = stakedTokensOf(_user);
uint256 totalAmount = 0;
for (uint256 i = 0; i < tokens.length; i++) {
uint256 interval;
if(block.timestamp<endTimeOfStaking){
interval = block.timestamp - tokenInfo[tokens[i]].lastUpdate;
}else{
interval = endTimeOfStaking - tokenInfo[tokens[i]].lastUpdate -1;
}
uint256 day=interval / 86400;
uint256 reward = rewardUnit * day * 10**9;
totalAmount += reward;
}
return totalAmount;
}
function getReward() external enableStaking {
uint256 reward = getClaimableToken(msg.sender);
require(reward>0, "reward must be greater than zero");
_utilityToken.transfer(msg.sender, reward);
emit GetReward(msg.sender, reward);
for (uint256 i = 0; i < stakedTko[msg.sender].length(); i++) {
uint256 tokenId = stakedTko[msg.sender].at(i);
tokenInfo[tokenId].lastUpdate = block.timestamp;
}
}
function onERC721Received(
address,
address,
uint256,
bytes calldata
) external view override returns (bytes4) {
require(
_tkoContract == msg.sender,
"You can stake only Tko"
);
return this.onERC721Received.selector;
}
}
{
"compilationTarget": {
"contracts/StakingPool.sol": "StakingPool"
},
"evmVersion": "istanbul",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs",
"useLiteralContent": true
},
"optimizer": {
"enabled": true,
"runs": 500
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"_tkoAddr","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"newTime","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"currentTime","type":"uint256"}],"name":"EndStakingTimeChanged","type":"event"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"address","name":"user","type":"address"},{"indexed":false,"internalType":"uint256","name":"amount","type":"uint256"}],"name":"GetReward","type":"event"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"address","name":"previousOwner","type":"address"},{"indexed":true,"internalType":"address","name":"newOwner","type":"address"}],"name":"OwnershipTransferred","type":"event"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"address","name":"user","type":"address"},{"indexed":true,"internalType":"uint256","name":"tokenId","type":"uint256"}],"name":"StakeStarted","type":"event"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"address","name":"user","type":"address"},{"indexed":true,"internalType":"uint256","name":"tokenId","type":"uint256"}],"name":"StakeStopped","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"newTime","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"currentTime","type":"uint256"}],"name":"StartStakingTimeChanged","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"from","type":"address"},{"indexed":false,"internalType":"address","name":"addr","type":"address"}],"name":"UtilityAddrSet","type":"event"},{"inputs":[{"internalType":"uint256","name":"_rewardUnit","type":"uint256"}],"name":"changeRewardUnit","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"endTimeOfStaking","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"_user","type":"address"}],"name":"getClaimableToken","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"getReward","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"","type":"address"},{"internalType":"address","name":"","type":"address"},{"internalType":"uint256","name":"","type":"uint256"},{"internalType":"bytes","name":"","type":"bytes"}],"name":"onERC721Received","outputs":[{"internalType":"bytes4","name":"","type":"bytes4"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"owner","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"renounceOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"rewardUnit","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"_timeStamp","type":"uint256"}],"name":"setEndStakingTime","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"_timeStamp","type":"uint256"}],"name":"setStartStakingTime","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address payable","name":"_addr","type":"address"}],"name":"setUtilitytoken","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"_user","type":"address"}],"name":"stakedTokensOf","outputs":[{"internalType":"uint256[]","name":"","type":"uint256[]"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"_user","type":"address"},{"internalType":"uint256[]","name":"_tokenIds","type":"uint256[]"}],"name":"startStakings","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"startTimeOfStaking","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"_user","type":"address"},{"internalType":"uint256","name":"_tokenId","type":"uint256"}],"name":"stopStaking","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"newOwner","type":"address"}],"name":"transferOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"}]