文件 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:LinkTokenInterface.sol
pragma solidity ^0.8.0;
interface LinkTokenInterface {
function allowance(address owner, address spender) external view returns (uint256 remaining);
function approve(address spender, uint256 value) external returns (bool success);
function balanceOf(address owner) external view returns (uint256 balance);
function decimals() external view returns (uint8 decimalPlaces);
function decreaseApproval(address spender, uint256 addedValue) external returns (bool success);
function increaseApproval(address spender, uint256 subtractedValue) external;
function name() external view returns (string memory tokenName);
function symbol() external view returns (string memory tokenSymbol);
function totalSupply() external view returns (uint256 totalTokensIssued);
function transfer(address to, uint256 value) external returns (bool success);
function transferAndCall(
address to,
uint256 value,
bytes calldata data
) external returns (bool success);
function transferFrom(
address from,
address to,
uint256 value
) external returns (bool success);
}
文件 3 的 7:MerkleProof.sol
pragma solidity ^0.8.0;
library MerkleProof {
function verify(
bytes32[] memory proof,
bytes32 root,
bytes32 leaf
) internal pure returns (bool) {
return processProof(proof, leaf) == root;
}
function processProof(bytes32[] memory proof, bytes32 leaf) internal pure returns (bytes32) {
bytes32 computedHash = leaf;
for (uint256 i = 0; i < proof.length; i++) {
bytes32 proofElement = proof[i];
if (computedHash <= proofElement) {
computedHash = _efficientHash(computedHash, proofElement);
} else {
computedHash = _efficientHash(proofElement, computedHash);
}
}
return computedHash;
}
function _efficientHash(bytes32 a, bytes32 b) private pure returns (bytes32 value) {
assembly {
mstore(0x00, a)
mstore(0x20, b)
value := keccak256(0x00, 0x40)
}
}
}
文件 4 的 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);
}
}
文件 5 的 7:VRFConsumerBaseV2.sol
pragma solidity ^0.8.0;
abstract contract VRFConsumerBaseV2 {
error OnlyCoordinatorCanFulfill(address have, address want);
address private immutable vrfCoordinator;
constructor(address _vrfCoordinator) {
vrfCoordinator = _vrfCoordinator;
}
function fulfillRandomWords(uint256 requestId, uint256[] memory randomWords) internal virtual;
function rawFulfillRandomWords(uint256 requestId, uint256[] memory randomWords) external {
if (msg.sender != vrfCoordinator) {
revert OnlyCoordinatorCanFulfill(msg.sender, vrfCoordinator);
}
fulfillRandomWords(requestId, randomWords);
}
}
文件 6 的 7:VRFCoordinatorV2Interface.sol
pragma solidity ^0.8.0;
interface VRFCoordinatorV2Interface {
function getRequestConfig()
external
view
returns (
uint16,
uint32,
bytes32[] memory
);
function requestRandomWords(
bytes32 keyHash,
uint64 subId,
uint16 minimumRequestConfirmations,
uint32 callbackGasLimit,
uint32 numWords
) external returns (uint256 requestId);
function createSubscription() external returns (uint64 subId);
function getSubscription(uint64 subId)
external
view
returns (
uint96 balance,
uint64 reqCount,
address owner,
address[] memory consumers
);
function requestSubscriptionOwnerTransfer(uint64 subId, address newOwner) external;
function acceptSubscriptionOwnerTransfer(uint64 subId) external;
function addConsumer(uint64 subId, address consumer) external;
function removeConsumer(uint64 subId, address consumer) external;
function cancelSubscription(uint64 subId, address to) external;
}
文件 7 的 7:WhitelistV1.sol
pragma solidity ^0.8.7;
import "@openzeppelin/contracts/access/Ownable.sol";
import "@openzeppelin/contracts/utils/cryptography/MerkleProof.sol";
import "@chainlink/contracts/src/v0.8/interfaces/LinkTokenInterface.sol";
import "@chainlink/contracts/src/v0.8/interfaces/VRFCoordinatorV2Interface.sol";
import "@chainlink/contracts/src/v0.8/VRFConsumerBaseV2.sol";
abstract contract NftContract {
function mint(
address to,
uint256 tokenId,
uint8 width,
uint8 height
) external virtual;
}
contract WhitelistV1 is VRFConsumerBaseV2, Ownable {
VRFCoordinatorV2Interface public COORDINATOR;
LinkTokenInterface public LINKTOKEN;
uint256[] private randomSeedList;
uint64 public subscriptionId;
bytes32 public keyHash;
uint256 public requestId;
struct Whitelist {
uint8 width;
uint8 height;
uint16 seedOffset;
uint16 currentSupply;
uint32 startTime;
uint32 endTime;
bytes32 merkleRoot;
}
mapping(uint256 => Whitelist) public whitelistList;
mapping(uint256 => uint16[]) private whitelistTokenList;
mapping(uint256 => mapping(address => bool)) public minted;
address immutable nftContractAddress;
event RandomWordsReceived(uint256 length);
event WhitelistCreated(
uint256 listId,
uint32 startTime,
uint32 endTime,
uint8 width,
uint8 height,
bytes32 merkleRoot
);
event WhitelistUpdated(uint256 listId, uint32 startTime, uint32 endTime);
event Mint(address to, uint256 tokenId, uint256 listId);
constructor(
address _nftContractAddress,
uint64 _subscriptionId,
address vrfCoordinator,
address linkTokenAddress,
bytes32 _keyHash
) VRFConsumerBaseV2(vrfCoordinator) {
nftContractAddress = _nftContractAddress;
subscriptionId = _subscriptionId;
COORDINATOR = VRFCoordinatorV2Interface(vrfCoordinator);
LINKTOKEN = LinkTokenInterface(linkTokenAddress);
keyHash = _keyHash;
}
modifier isActive(uint256 listId) {
Whitelist memory whitelist = whitelistList[listId];
require(whitelist.startTime <= block.timestamp, "Session not started");
require(whitelist.endTime >= block.timestamp, "Session ended");
_;
}
function setKeyHash(bytes32 _keyHash) external onlyOwner {
keyHash = _keyHash;
}
function generateRandomWords(uint32 numWords) external onlyOwner {
uint256[] memory randomWords = new uint256[](numWords);
uint256 random;
for (uint32 i = 0; i < numWords; i++) {
random = uint256(
keccak256(abi.encodePacked(random, block.difficulty, block.timestamp))
);
randomWords[i] = random;
}
fulfillRandomWords(0, randomWords);
}
function requestRandomWords(uint32 numWords, uint32 callbackGasLimit)
external
onlyOwner
{
requestId = COORDINATOR.requestRandomWords(
keyHash,
subscriptionId,
3,
callbackGasLimit,
numWords
);
}
function createWhitelist(
uint256 listId,
uint8 width,
uint8 height,
uint16 seedOffset,
uint32 startTime,
uint32 endTime,
bytes32 merkleRoot,
uint16[] calldata tokenList
) external onlyOwner {
require(
tokenList.length <= randomSeedList.length * 16,
"Insufficient random seeds"
);
require(whitelistList[listId].width == 0, "listId exists");
require(width > 0 && height > 0, "Incorrect width or height");
whitelistList[listId] = Whitelist(
width,
height,
uint16(listId + seedOffset),
0,
startTime,
endTime,
merkleRoot
);
whitelistTokenList[listId] = tokenList;
shuffleTokenList(listId);
emit WhitelistCreated(
listId,
startTime,
endTime,
width,
height,
merkleRoot
);
}
function updateWhitelistStartTime(uint256 listId, uint32 _startTime)
external
onlyOwner
{
whitelistList[listId].startTime = _startTime;
emit WhitelistUpdated(
listId,
whitelistList[listId].startTime,
whitelistList[listId].endTime
);
}
function updateWhitelistEndTime(uint256 listId, uint32 _endTime)
external
onlyOwner
{
whitelistList[listId].endTime = _endTime;
emit WhitelistUpdated(
listId,
whitelistList[listId].startTime,
whitelistList[listId].endTime
);
}
function mint(uint256 listId, bytes32[] calldata proof)
external
isActive(listId)
{
require(msg.sender == tx.origin, "Contract interaction not allowed");
require(verifyProof(listId, msg.sender, proof), "Invalid merkle proof");
require(!minted[listId][msg.sender], "Already minted");
minted[listId][msg.sender] = true;
uint256 tokenId = drawRandomTokenId(listId);
NftContract nftContract = NftContract(nftContractAddress);
Whitelist memory whitelist = whitelistList[listId];
nftContract.mint(msg.sender, tokenId, whitelist.width, whitelist.height);
emit Mint(msg.sender, tokenId, listId);
}
function getRandomSeedListLength() external view returns (uint256) {
return randomSeedList.length * 16;
}
function verifyProof(
uint256 listId,
address who,
bytes32[] calldata proof
) public view returns (bool) {
bytes32 leaf = keccak256(abi.encodePacked(who));
return MerkleProof.verify(proof, whitelistList[listId].merkleRoot, leaf);
}
function fulfillRandomWords(
uint256,
uint256[] memory randomWords
) internal override {
for (uint256 i = 0; i < randomWords.length; i++) {
randomSeedList.push(randomWords[i]);
}
emit RandomWordsReceived(randomWords.length);
}
function shuffleTokenList(uint256 listId) internal {
uint16[] storage tokenList = whitelistTokenList[listId];
for (uint16 i = 0; i < tokenList.length; i++) {
uint256 seedIndex = ((i + whitelistList[listId].seedOffset) / 16) %
randomSeedList.length;
uint256 random = uint16(randomSeedList[seedIndex] >> (i % 16)) %
tokenList.length;
(tokenList[i], tokenList[random]) = (tokenList[random], tokenList[i]);
}
}
function drawRandomTokenId(uint256 listId) internal returns (uint256) {
whitelistList[listId].currentSupply += 1;
return whitelistTokenList[listId][whitelistList[listId].currentSupply - 1];
}
}
{
"compilationTarget": {
"contracts/WhitelistV1.sol": "WhitelistV1"
},
"evmVersion": "london",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": true,
"runs": 200
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"_nftContractAddress","type":"address"},{"internalType":"uint64","name":"_subscriptionId","type":"uint64"},{"internalType":"address","name":"vrfCoordinator","type":"address"},{"internalType":"address","name":"linkTokenAddress","type":"address"},{"internalType":"bytes32","name":"_keyHash","type":"bytes32"}],"stateMutability":"nonpayable","type":"constructor"},{"inputs":[{"internalType":"address","name":"have","type":"address"},{"internalType":"address","name":"want","type":"address"}],"name":"OnlyCoordinatorCanFulfill","type":"error"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"to","type":"address"},{"indexed":false,"internalType":"uint256","name":"tokenId","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"listId","type":"uint256"}],"name":"Mint","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":"uint256","name":"length","type":"uint256"}],"name":"RandomWordsReceived","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"listId","type":"uint256"},{"indexed":false,"internalType":"uint32","name":"startTime","type":"uint32"},{"indexed":false,"internalType":"uint32","name":"endTime","type":"uint32"},{"indexed":false,"internalType":"uint8","name":"width","type":"uint8"},{"indexed":false,"internalType":"uint8","name":"height","type":"uint8"},{"indexed":false,"internalType":"bytes32","name":"merkleRoot","type":"bytes32"}],"name":"WhitelistCreated","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"listId","type":"uint256"},{"indexed":false,"internalType":"uint32","name":"startTime","type":"uint32"},{"indexed":false,"internalType":"uint32","name":"endTime","type":"uint32"}],"name":"WhitelistUpdated","type":"event"},{"inputs":[],"name":"COORDINATOR","outputs":[{"internalType":"contract VRFCoordinatorV2Interface","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"LINKTOKEN","outputs":[{"internalType":"contract LinkTokenInterface","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"listId","type":"uint256"},{"internalType":"uint8","name":"width","type":"uint8"},{"internalType":"uint8","name":"height","type":"uint8"},{"internalType":"uint16","name":"seedOffset","type":"uint16"},{"internalType":"uint32","name":"startTime","type":"uint32"},{"internalType":"uint32","name":"endTime","type":"uint32"},{"internalType":"bytes32","name":"merkleRoot","type":"bytes32"},{"internalType":"uint16[]","name":"tokenList","type":"uint16[]"}],"name":"createWhitelist","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint32","name":"numWords","type":"uint32"}],"name":"generateRandomWords","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"getRandomSeedListLength","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"keyHash","outputs":[{"internalType":"bytes32","name":"","type":"bytes32"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"listId","type":"uint256"},{"internalType":"bytes32[]","name":"proof","type":"bytes32[]"}],"name":"mint","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"},{"internalType":"address","name":"","type":"address"}],"name":"minted","outputs":[{"internalType":"bool","name":"","type":"bool"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"owner","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"requestId","type":"uint256"},{"internalType":"uint256[]","name":"randomWords","type":"uint256[]"}],"name":"rawFulfillRandomWords","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"renounceOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"requestId","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint32","name":"numWords","type":"uint32"},{"internalType":"uint32","name":"callbackGasLimit","type":"uint32"}],"name":"requestRandomWords","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"bytes32","name":"_keyHash","type":"bytes32"}],"name":"setKeyHash","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"subscriptionId","outputs":[{"internalType":"uint64","name":"","type":"uint64"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"newOwner","type":"address"}],"name":"transferOwnership","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"listId","type":"uint256"},{"internalType":"uint32","name":"_endTime","type":"uint32"}],"name":"updateWhitelistEndTime","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"listId","type":"uint256"},{"internalType":"uint32","name":"_startTime","type":"uint32"}],"name":"updateWhitelistStartTime","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"listId","type":"uint256"},{"internalType":"address","name":"who","type":"address"},{"internalType":"bytes32[]","name":"proof","type":"bytes32[]"}],"name":"verifyProof","outputs":[{"internalType":"bool","name":"","type":"bool"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"}],"name":"whitelistList","outputs":[{"internalType":"uint8","name":"width","type":"uint8"},{"internalType":"uint8","name":"height","type":"uint8"},{"internalType":"uint16","name":"seedOffset","type":"uint16"},{"internalType":"uint16","name":"currentSupply","type":"uint16"},{"internalType":"uint32","name":"startTime","type":"uint32"},{"internalType":"uint32","name":"endTime","type":"uint32"},{"internalType":"bytes32","name":"merkleRoot","type":"bytes32"}],"stateMutability":"view","type":"function"}]