文件 1 的 7: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 的 7:DHStakingV2.sol
pragma solidity 0.8.7;
import "@openzeppelin/contracts/access/Ownable.sol";
import "@openzeppelin/contracts/token/ERC721/IERC721.sol";
import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
import "@openzeppelin/contracts/token/ERC721/IERC721Receiver.sol";
interface IShowBiz {
function mint(address to, uint256 amount) external;
}
contract DHStakingV2 is Ownable, IERC721Receiver {
using EnumerableSet for EnumerableSet.UintSet;
using EnumerableSet for EnumerableSet.AddressSet;
IShowBiz _showBiz = IShowBiz(0x136209a516D1C2660F045e70634c9d95D64325F9);
struct StakedToken {
uint tokenId;
uint stakedAt;
uint endsAt;
uint monthlyRewards;
uint months;
address owner;
}
struct PartnerContract {
bool active;
IERC721 instance;
uint[] availablePeriods;
uint[] monthlyRewards;
}
mapping(address => mapping(uint => StakedToken)) public tokenIdToStakedToken;
mapping(address => mapping(uint => uint)) public tokenIdToClaimedRewards;
mapping(address => mapping(address => EnumerableSet.UintSet)) addressToStakedTokensSet;
mapping(address => PartnerContract) public contracts;
uint public totalClaimedRewards;
EnumerableSet.AddressSet activeContracts;
event Stake(address contractAddress, uint contractTokenId, address owner, uint endsAt);
event Unstake(address contractAddress, uint contractTokenId, address owner);
event ClaimTokenRewards(address contractAddress, uint contractTokenId, address owner, uint rewards);
function onERC721Received(address _operator, address, uint256, bytes calldata) external view override returns(bytes4) {
require(_operator == address(this), "token must be staked over stake method");
return bytes4(keccak256("onERC721Received(address,address,uint256,bytes)"));
}
function stake(address contractAddress, uint tokenId, uint months) public {
PartnerContract storage _contract = contracts[contractAddress];
require(_contract.active, "token contract is not active");
require(months >= 1, "invalid minimum period");
uint endsAt = block.timestamp + months * 28 days;
uint monthlyRewards = 0;
for (uint i = 0; i < _contract.availablePeriods.length; i++) {
if (_contract.availablePeriods[i] == months) {
monthlyRewards = _contract.monthlyRewards[i];
}
}
require(monthlyRewards != 0, "invalid stake period");
tokenIdToStakedToken[contractAddress][tokenId] = StakedToken({
tokenId: tokenId,
stakedAt: block.timestamp,
endsAt: endsAt,
monthlyRewards: monthlyRewards,
months: months,
owner: msg.sender
});
tokenIdToClaimedRewards[contractAddress][tokenId] = 0;
_contract.instance.safeTransferFrom(msg.sender, address(this), tokenId);
addressToStakedTokensSet[contractAddress][msg.sender].add(tokenId);
emit Stake(contractAddress, tokenId, msg.sender, endsAt);
}
function renewStake(address contractAddress, uint[] memory tokenIds) public {
for (uint i = 0; i < tokenIds.length; i++) {
StakedToken storage stakedToken = tokenIdToStakedToken[contractAddress][tokenIds[i]];
require(stakedToken.owner != msg.sender, "caller does not have this token");
stakedToken.endsAt = block.timestamp + stakedToken.months * 28 days;
}
}
function stakeBatch(address contractAddress, uint[] calldata tokenIds, uint months) external {
for (uint i = 0; i < tokenIds.length; i++) {
stake(contractAddress, tokenIds[i], months);
}
}
function unclaimedRewards(address contractAddress, uint tokenId) public view returns (uint) {
StakedToken storage stakedToken = tokenIdToStakedToken[contractAddress][tokenId];
require(stakedToken.owner != address(0), "cannot query an unstaked token");
uint rewardLimit = (stakedToken.endsAt - stakedToken.stakedAt) / 7 days;
uint rewardsUntilNow = (block.timestamp - stakedToken.stakedAt) / 7 days;
return (rewardsUntilNow > rewardLimit ? rewardLimit : rewardsUntilNow)
* stakedToken.monthlyRewards / 4 - tokenIdToClaimedRewards[contractAddress][tokenId];
}
function claimTokenRewards(address contractAddress, uint tokenId) public {
StakedToken storage stakedToken = tokenIdToStakedToken[contractAddress][tokenId];
require(stakedToken.owner == msg.sender, "caller did not stake this token");
uint _unclaimedRewards = unclaimedRewards(contractAddress, tokenId);
if (_unclaimedRewards > 0) {
_showBiz.mint(msg.sender, _unclaimedRewards);
tokenIdToClaimedRewards[contractAddress][tokenId] += _unclaimedRewards;
totalClaimedRewards += _unclaimedRewards;
emit ClaimTokenRewards(contractAddress, tokenId, msg.sender, _unclaimedRewards);
}
}
function claimContractRewards(address contractAddress) public {
EnumerableSet.UintSet storage stakedTokens = addressToStakedTokensSet[contractAddress][msg.sender];
uint totalStakedTokens = stakedTokens.length();
for (uint i = 0; i < totalStakedTokens; i++) {
claimTokenRewards(contractAddress, stakedTokens.at(i));
}
}
function claimRewards() public {
for (uint i = 0; i < activeContracts.length(); i++) {
claimContractRewards(activeContracts.at(i));
}
}
function unstake(address contractAddress, uint tokenId) public {
PartnerContract storage _contract = contracts[contractAddress];
StakedToken storage stakedToken = tokenIdToStakedToken[contractAddress][tokenId];
require(stakedToken.owner == msg.sender, "caller not owns this token");
require(block.timestamp > stakedToken.endsAt, "staked period did not finish yet");
claimTokenRewards(contractAddress, tokenId);
_contract.instance.safeTransferFrom(address(this), msg.sender, stakedToken.tokenId);
addressToStakedTokensSet[contractAddress][msg.sender].remove(stakedToken.tokenId);
emit Unstake(contractAddress, stakedToken.tokenId, msg.sender);
delete tokenIdToStakedToken[contractAddress][tokenId];
}
function unstakeBatch(address contractAddress, uint[] calldata tokenIds) external {
for (uint i = 0; i < tokenIds.length; i++) {
unstake(contractAddress, tokenIds[i]);
}
}
function stakedTokensOfOwner(address contractAddress, address accountAddress) external view returns (uint[] memory tokenIds) {
EnumerableSet.UintSet storage stakedTokens = addressToStakedTokensSet[contractAddress][accountAddress];
tokenIds = new uint[](stakedTokens.length());
for (uint i = 0; i < stakedTokens.length(); i++) {
tokenIds[i] = stakedTokens.at(i);
}
return tokenIds;
}
function addContract(address contractAddress, uint[] memory availablePeriods, uint[] memory monthlyRewards) public onlyOwner {
contracts[contractAddress] = PartnerContract(
true,
IERC721(contractAddress),
availablePeriods,
monthlyRewards
);
activeContracts.add(contractAddress);
}
function updateContract(address contractAddress, bool active, uint[] memory availablePeriods, uint[] memory monthlyRewards) public onlyOwner {
require(activeContracts.contains(contractAddress), "contract not added");
contracts[contractAddress].active = active;
contracts[contractAddress].availablePeriods = availablePeriods;
contracts[contractAddress].monthlyRewards = monthlyRewards;
}
}
文件 3 的 7: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;
}
}
文件 4 的 7:IERC165.sol
pragma solidity ^0.8.0;
interface IERC165 {
function supportsInterface(bytes4 interfaceId) external view returns (bool);
}
文件 5 的 7:IERC721.sol
pragma solidity ^0.8.0;
import "../../utils/introspection/IERC165.sol";
interface IERC721 is IERC165 {
event Transfer(address indexed from, address indexed to, uint256 indexed tokenId);
event Approval(address indexed owner, address indexed approved, uint256 indexed tokenId);
event ApprovalForAll(address indexed owner, address indexed operator, bool approved);
function balanceOf(address owner) external view returns (uint256 balance);
function ownerOf(uint256 tokenId) external view returns (address owner);
function safeTransferFrom(
address from,
address to,
uint256 tokenId
) external;
function transferFrom(
address from,
address to,
uint256 tokenId
) external;
function approve(address to, uint256 tokenId) external;
function getApproved(uint256 tokenId) external view returns (address operator);
function setApprovalForAll(address operator, bool _approved) external;
function isApprovedForAll(address owner, address operator) external view returns (bool);
function safeTransferFrom(
address from,
address to,
uint256 tokenId,
bytes calldata data
) external;
}
文件 6 的 7:IERC721Receiver.sol
pragma solidity ^0.8.0;
interface IERC721Receiver {
function onERC721Received(
address operator,
address from,
uint256 tokenId,
bytes calldata data
) external returns (bytes4);
}
文件 7 的 7: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);
}
}
{
"compilationTarget": {
"contracts/DHStakingV2.sol": "DHStakingV2"
},
"evmVersion": "london",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": true,
"runs": 200
},
"remappings": []
}
[{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"contractAddress","type":"address"},{"indexed":false,"internalType":"uint256","name":"contractTokenId","type":"uint256"},{"indexed":false,"internalType":"address","name":"owner","type":"address"},{"indexed":false,"internalType":"uint256","name":"rewards","type":"uint256"}],"name":"ClaimTokenRewards","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":false,"internalType":"address","name":"contractAddress","type":"address"},{"indexed":false,"internalType":"uint256","name":"contractTokenId","type":"uint256"},{"indexed":false,"internalType":"address","name":"owner","type":"address"},{"indexed":false,"internalType":"uint256","name":"endsAt","type":"uint256"}],"name":"Stake","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"contractAddress","type":"address"},{"indexed":false,"internalType":"uint256","name":"contractTokenId","type":"uint256"},{"indexed":false,"internalType":"address","name":"owner","type":"address"}],"name":"Unstake","type":"event"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256[]","name":"availablePeriods","type":"uint256[]"},{"internalType":"uint256[]","name":"monthlyRewards","type":"uint256[]"}],"name":"addContract","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"}],"name":"claimContractRewards","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"claimRewards","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256","name":"tokenId","type":"uint256"}],"name":"claimTokenRewards","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"","type":"address"}],"name":"contracts","outputs":[{"internalType":"bool","name":"active","type":"bool"},{"internalType":"contract IERC721","name":"instance","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"_operator","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":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"}],"name":"renewStake","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"renounceOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256","name":"tokenId","type":"uint256"},{"internalType":"uint256","name":"months","type":"uint256"}],"name":"stake","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"},{"internalType":"uint256","name":"months","type":"uint256"}],"name":"stakeBatch","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"address","name":"accountAddress","type":"address"}],"name":"stakedTokensOfOwner","outputs":[{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"","type":"address"},{"internalType":"uint256","name":"","type":"uint256"}],"name":"tokenIdToClaimedRewards","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"","type":"address"},{"internalType":"uint256","name":"","type":"uint256"}],"name":"tokenIdToStakedToken","outputs":[{"internalType":"uint256","name":"tokenId","type":"uint256"},{"internalType":"uint256","name":"stakedAt","type":"uint256"},{"internalType":"uint256","name":"endsAt","type":"uint256"},{"internalType":"uint256","name":"monthlyRewards","type":"uint256"},{"internalType":"uint256","name":"months","type":"uint256"},{"internalType":"address","name":"owner","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"totalClaimedRewards","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"newOwner","type":"address"}],"name":"transferOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256","name":"tokenId","type":"uint256"}],"name":"unclaimedRewards","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256","name":"tokenId","type":"uint256"}],"name":"unstake","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"}],"name":"unstakeBatch","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"contractAddress","type":"address"},{"internalType":"bool","name":"active","type":"bool"},{"internalType":"uint256[]","name":"availablePeriods","type":"uint256[]"},{"internalType":"uint256[]","name":"monthlyRewards","type":"uint256[]"}],"name":"updateContract","outputs":[],"stateMutability":"nonpayable","type":"function"}]