文件 1 的 4: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 的 4:IERC721A.sol
pragma solidity ^0.8.4;
import '../IERC721A.sol';
文件 3 的 4:LosMuertosStaking.sol
pragma solidity >=0.8.0 <0.9.0;
import "@openzeppelin/contracts/access/Ownable.sol";
import "erc721a/contracts/interfaces/IERC721A.sol";
contract LosMuertosStaking is Ownable {
IERC721A public originalContract;
bool public stakingActive = true;
struct StakedInfo {
address owner;
uint256 stakedAt;
uint256 tokenId;
}
mapping(uint256 => StakedInfo) public tokenStakedInfo;
uint256 public stakedCount = 0;
constructor(address originalContract_) {
originalContract = IERC721A(originalContract_);
}
function stake(uint256[] memory tokenIds) external {
require(stakingActive, "Staking not active");
for (uint256 i = 0; i < tokenIds.length; i++) {
uint256 tokenId = tokenIds[i];
originalContract.transferFrom(msg.sender, address(this), tokenId);
tokenStakedInfo[tokenId] = StakedInfo(
msg.sender,
uint256(block.timestamp),
tokenId
);
}
stakedCount += tokenIds.length;
}
function unstake(uint256[] memory tokenIds) external {
for (uint256 i = 0; i < tokenIds.length; i++) {
uint256 tokenId = tokenIds[i];
StakedInfo memory info = tokenStakedInfo[tokenId];
require(info.owner == msg.sender, "Only owner can unstake");
delete tokenStakedInfo[tokenId];
originalContract.transferFrom(address(this), msg.sender, tokenId);
}
stakedCount -= tokenIds.length;
}
function setStakingActive(bool stakingActive_) external onlyOwner {
stakingActive = stakingActive_;
}
function setOriginalContract(address originalContract_) external onlyOwner {
originalContract = IERC721A(originalContract_);
}
function balanceOf(address owner_) public view returns (uint256) {
uint256 supply = originalContract.totalSupply();
uint256 count = 0;
for (uint256 tokenId = 1; tokenId <= supply; tokenId++) {
if (tokenStakedInfo[tokenId].owner == owner_) {
count++;
}
}
return count;
}
function snapshot() public view returns (StakedInfo[] memory) {
uint256 supply = originalContract.totalSupply();
StakedInfo[] memory currentState = new StakedInfo[](stakedCount);
uint256 count = 0;
for (uint256 tokenId = 1; tokenId <= supply; tokenId++) {
StakedInfo memory info = tokenStakedInfo[tokenId];
if (info.owner != address(0)) {
currentState[count] = info;
count++;
}
}
return currentState;
}
function walletOfOwner(address owner_)
public
view
returns (StakedInfo[] memory)
{
uint256 supply = originalContract.totalSupply();
uint256 balance = balanceOf(owner_);
StakedInfo[] memory tokens = new StakedInfo[](balance);
uint256 count = 0;
for (uint256 tokenId = 1; tokenId <= supply; tokenId++) {
StakedInfo memory token = tokenStakedInfo[tokenId];
if (token.owner == owner_) {
tokens[count] = token;
count++;
}
}
return tokens;
}
}
文件 4 的 4: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/LosMuertosStaking.sol": "LosMuertosStaking"
},
"evmVersion": "london",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": true,
"runs": 1000
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"originalContract_","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"},{"inputs":[{"internalType":"address","name":"owner_","type":"address"}],"name":"balanceOf","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"originalContract","outputs":[{"internalType":"contract IERC721A","name":"","type":"address"}],"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":[{"internalType":"address","name":"originalContract_","type":"address"}],"name":"setOriginalContract","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"bool","name":"stakingActive_","type":"bool"}],"name":"setStakingActive","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"snapshot","outputs":[{"components":[{"internalType":"address","name":"owner","type":"address"},{"internalType":"uint256","name":"stakedAt","type":"uint256"},{"internalType":"uint256","name":"tokenId","type":"uint256"}],"internalType":"struct LosMuertosStaking.StakedInfo[]","name":"","type":"tuple[]"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256[]","name":"tokenIds","type":"uint256[]"}],"name":"stake","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"stakedCount","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"stakingActive","outputs":[{"internalType":"bool","name":"","type":"bool"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"}],"name":"tokenStakedInfo","outputs":[{"internalType":"address","name":"owner","type":"address"},{"internalType":"uint256","name":"stakedAt","type":"uint256"},{"internalType":"uint256","name":"tokenId","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"},{"inputs":[{"internalType":"address","name":"owner_","type":"address"}],"name":"walletOfOwner","outputs":[{"components":[{"internalType":"address","name":"owner","type":"address"},{"internalType":"uint256","name":"stakedAt","type":"uint256"},{"internalType":"uint256","name":"tokenId","type":"uint256"}],"internalType":"struct LosMuertosStaking.StakedInfo[]","name":"","type":"tuple[]"}],"stateMutability":"view","type":"function"}]