文件 1 的 3:KitsuMintuRaffleContract.sol
import "./libs/SafeMath.sol";
import "./libs/SignedSafeMath.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;
}
}
pragma solidity ^0.8.0;
interface IERC721Receiver {
function onERC721Received(
address operator,
address from,
uint256 tokenId,
bytes calldata data
) external returns (bytes4);
}
pragma solidity ^0.8.0;
abstract contract Ownable is Context {
address private _owner;
event OwnershipTransferred(address indexed previousOwner, address indexed newOwner);
constructor() {
_setOwner(_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 {
_setOwner(address(0));
}
function transferOwnership(address newOwner) public virtual onlyOwner {
require(newOwner != address(0), "Ownable: new owner is the zero address");
_setOwner(newOwner);
}
function _setOwner(address newOwner) private {
address oldOwner = _owner;
_owner = newOwner;
emit OwnershipTransferred(oldOwner, newOwner);
}
}
pragma solidity ^0.8.9;
interface IERC721Kitsu {
function safeTransferFrom(address from, address to, uint256 tokenId) external;
function sendFromVault(address payable dst, uint256 amount) external;
function availableVaultLiquidity() external view returns(uint256);
function totalSupply() external view returns (uint256);
}
pragma solidity >=0.7.0 <0.9.0;
contract KitsuMintuRaffleContract is Ownable, IERC721Receiver {
using SafeMath for uint256;
using SignedSafeMath for int256;
struct StakedToken {
uint256 id;
address user;
uint256 timestamp;
uint256[] tokenIds;
uint256 rewards;
uint256 totalRewards;
}
mapping(address => StakedToken) public stakes;
mapping(uint256 => address) public participants;
uint256 public participantsCount;
uint256 public lockedRewardAmount;
address public token;
event Stake(address user, uint256[] tokenIds, uint256 id);
event Unstake(address user, uint256[] tokenIds, uint256 rewards, uint256 id);
event Raffle(address[] winners, uint256[] rewards);
constructor(address _tokenAddress) {
require(_tokenAddress != address(0));
token = _tokenAddress;
}
function totalSupply() public view returns (uint256) {
return IERC721Kitsu(token).totalSupply();
}
function availableVaultLiquidity() external view returns(uint256) {
return IERC721Kitsu(token).availableVaultLiquidity() - lockedRewardAmount;
}
function stake(uint256[] calldata _tokenIds) external {
uint256 participantId = stakes[msg.sender].id == 0 ? participantsCount++ : stakes[msg.sender].id;
participants[participantId] = msg.sender;
stakes[msg.sender].user = msg.sender;
stakes[msg.sender].timestamp = block.timestamp;
for (uint256 i = 0; i < _tokenIds.length; i++) {
IERC721Kitsu(token).safeTransferFrom(msg.sender, address(this), _tokenIds[i]);
stakes[msg.sender].tokenIds.push(_tokenIds[i]);
}
emit Stake(msg.sender, stakes[msg.sender].tokenIds, participantId);
}
function raffle(address[] calldata _winners, uint256[] calldata _rewards) external onlyOwner {
require(_winners.length == _rewards.length, "Incorect raffle");
for (uint256 i = 0; i < _winners.length; i++) {
stakes[_winners[i]].rewards += _rewards[i];
stakes[_winners[i]].totalRewards += _rewards[i];
lockedRewardAmount += _rewards[i];
}
require(lockedRewardAmount < IERC721Kitsu(token).availableVaultLiquidity(), "Amount of total rewards exceed the vault liquidity.");
emit Raffle(_winners, _rewards);
}
function unstake(uint256[] calldata _tokenIds) external {
require(stakes[msg.sender].user == msg.sender, "Any stake");
stakes[msg.sender].timestamp = 0;
for (uint256 i = 0; i < _tokenIds.length; i++) {
uint256 index = indexOfArray(stakes[msg.sender].tokenIds, _tokenIds[i]);
require(index < stakes[msg.sender].tokenIds.length, "Not staked");
IERC721Kitsu(token).safeTransferFrom(address(this), msg.sender, stakes[msg.sender].tokenIds[i]);
removeElementFromArray(stakes[msg.sender].tokenIds, index);
}
emit Unstake(msg.sender, stakes[msg.sender].tokenIds, stakes[msg.sender].rewards, stakes[msg.sender].id);
}
function claim() external {
require(stakes[msg.sender].user == msg.sender, "Any stake");
if (stakes[msg.sender].rewards != 0) {
IERC721Kitsu(token).sendFromVault(payable(msg.sender), stakes[msg.sender].rewards);
lockedRewardAmount -= stakes[msg.sender].rewards;
}
stakes[msg.sender].rewards = 0;
}
function getStakes(int256 page, int256 pageSize) external view returns (StakedToken[] memory) {
uint256 poolLength = participantsCount;
int256 queryStartPoolIndex = int256(poolLength).sub(pageSize.mul(page.add(1))).add(pageSize);
require(queryStartPoolIndex >= 0, "Out of bounds");
int256 queryEndPoolIndex = queryStartPoolIndex.sub(pageSize);
if (queryEndPoolIndex < 0) {
queryEndPoolIndex = 0;
}
int256 currentPoolIndex = queryStartPoolIndex;
require(uint256(currentPoolIndex) <= poolLength, "Out of bounds");
StakedToken[] memory results = new StakedToken[](uint256(currentPoolIndex - queryEndPoolIndex));
uint256 index = 0;
for (currentPoolIndex; currentPoolIndex > queryEndPoolIndex; currentPoolIndex--) {
address user = participants[uint256(currentPoolIndex).sub(1)];
results[index] = stakes[user];
index++;
}
return results;
}
function getStake(address _user) external view returns (StakedToken memory) {
return stakes[_user];
}
function onERC721Received(
address ,
address ,
uint256 ,
bytes calldata
) external pure returns (bytes4) {
return IERC721Receiver.onERC721Received.selector;
}
function indexOfArray(uint[] storage _array, uint _element) internal view returns (uint) {
for (uint i = 0; i < _array.length; i++) {
if (_array[i] == _element) {
return i;
}
}
return _array.length;
}
function removeElementFromArray(uint[] storage _array, uint _index) internal {
require(_index < _array.length);
for (uint i = _index; i < _array.length - 1; i++) {
_array[i] = _array[i + 1];
}
_array.pop();
}
}
文件 2 的 3:SafeMath.sol
pragma solidity ^0.8.0;
library SafeMath {
function tryAdd(uint256 a, uint256 b) internal pure returns (bool, uint256) {
unchecked {
uint256 c = a + b;
if (c < a) return (false, 0);
return (true, c);
}
}
function trySub(uint256 a, uint256 b) internal pure returns (bool, uint256) {
unchecked {
if (b > a) return (false, 0);
return (true, a - b);
}
}
function tryMul(uint256 a, uint256 b) internal pure returns (bool, uint256) {
unchecked {
if (a == 0) return (true, 0);
uint256 c = a * b;
if (c / a != b) return (false, 0);
return (true, c);
}
}
function tryDiv(uint256 a, uint256 b) internal pure returns (bool, uint256) {
unchecked {
if (b == 0) return (false, 0);
return (true, a / b);
}
}
function tryMod(uint256 a, uint256 b) internal pure returns (bool, uint256) {
unchecked {
if (b == 0) return (false, 0);
return (true, a % b);
}
}
function add(uint256 a, uint256 b) internal pure returns (uint256) {
return a + b;
}
function sub(uint256 a, uint256 b) internal pure returns (uint256) {
return a - b;
}
function mul(uint256 a, uint256 b) internal pure returns (uint256) {
return a * b;
}
function div(uint256 a, uint256 b) internal pure returns (uint256) {
return a / b;
}
function mod(uint256 a, uint256 b) internal pure returns (uint256) {
return a % b;
}
function sub(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
unchecked {
require(b <= a, errorMessage);
return a - b;
}
}
function div(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
unchecked {
require(b > 0, errorMessage);
return a / b;
}
}
function mod(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
unchecked {
require(b > 0, errorMessage);
return a % b;
}
}
}
文件 3 的 3:SignedSafeMath.sol
pragma solidity ^0.8.0;
library SignedSafeMath {
function mul(int256 a, int256 b) internal pure returns (int256) {
return a * b;
}
function div(int256 a, int256 b) internal pure returns (int256) {
return a / b;
}
function sub(int256 a, int256 b) internal pure returns (int256) {
return a - b;
}
function add(int256 a, int256 b) internal pure returns (int256) {
return a + b;
}
}
{
"compilationTarget": {
"project:/contracts/KitsuMintuRaffleContract.sol": "KitsuMintuRaffleContract"
},
"evmVersion": "london",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": false,
"runs": 200
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"_tokenAddress","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"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":"winners","type":"address[]"},{"indexed":false,"internalType":"uint256[]","name":"rewards","type":"uint256[]"}],"name":"Raffle","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"user","type":"address"},{"indexed":false,"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"},{"indexed":false,"internalType":"uint256","name":"id","type":"uint256"}],"name":"Stake","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"user","type":"address"},{"indexed":false,"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"},{"indexed":false,"internalType":"uint256","name":"rewards","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"id","type":"uint256"}],"name":"Unstake","type":"event"},{"inputs":[],"name":"availableVaultLiquidity","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"claim","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"_user","type":"address"}],"name":"getStake","outputs":[{"components":[{"internalType":"uint256","name":"id","type":"uint256"},{"internalType":"address","name":"user","type":"address"},{"internalType":"uint256","name":"timestamp","type":"uint256"},{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"},{"internalType":"uint256","name":"rewards","type":"uint256"},{"internalType":"uint256","name":"totalRewards","type":"uint256"}],"internalType":"struct KitsuMintuRaffleContract.StakedToken","name":"","type":"tuple"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"int256","name":"page","type":"int256"},{"internalType":"int256","name":"pageSize","type":"int256"}],"name":"getStakes","outputs":[{"components":[{"internalType":"uint256","name":"id","type":"uint256"},{"internalType":"address","name":"user","type":"address"},{"internalType":"uint256","name":"timestamp","type":"uint256"},{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"},{"internalType":"uint256","name":"rewards","type":"uint256"},{"internalType":"uint256","name":"totalRewards","type":"uint256"}],"internalType":"struct KitsuMintuRaffleContract.StakedToken[]","name":"","type":"tuple[]"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"lockedRewardAmount","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","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":"pure","type":"function"},{"inputs":[],"name":"owner","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"}],"name":"participants","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"participantsCount","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address[]","name":"_winners","type":"address[]"},{"internalType":"uint256[]","name":"_rewards","type":"uint256[]"}],"name":"raffle","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"renounceOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256[]","name":"_tokenIds","type":"uint256[]"}],"name":"stake","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"","type":"address"}],"name":"stakes","outputs":[{"internalType":"uint256","name":"id","type":"uint256"},{"internalType":"address","name":"user","type":"address"},{"internalType":"uint256","name":"timestamp","type":"uint256"},{"internalType":"uint256","name":"rewards","type":"uint256"},{"internalType":"uint256","name":"totalRewards","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"token","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"totalSupply","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":"uint256[]","name":"_tokenIds","type":"uint256[]"}],"name":"unstake","outputs":[],"stateMutability":"nonpayable","type":"function"}]