Created
October 13, 2021 10:40
-
-
Save mbvissers/a2145e511e6b16d9acb92641b51e9f88 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// SPDX-License-Identifier: MIT | |
pragma solidity ^0.8.2; | |
import "@openzeppelin/contracts/token/ERC721/ERC721.sol"; | |
import "@openzeppelin/contracts/access/Ownable.sol"; | |
import "@openzeppelin/contracts/utils/Counters.sol"; | |
import "@openzeppelin/contracts/token/ERC721/extensions/ERC721Enumerable.sol"; | |
contract TestToken is ERC721, ERC721Enumerable, Ownable { | |
using Counters for Counters.Counter; | |
Counters.Counter private _tokenIdCounter; | |
uint256 public constant MAX_TOKENS = 100; | |
uint256 public constant MAX_TOKENS_PER_SALE = 5; | |
string private baseURI; | |
bool public isSaleActive = true; | |
uint256 private price = 0.02 ether; | |
constructor() ERC721("Test Token", "TT") {} | |
function flipSaleStatus() public onlyOwner() { | |
isSaleActive = !isSaleActive; | |
} | |
function setPrice(uint256 _price) public onlyOwner() { | |
price = _price; | |
} | |
function setBaseURI(string memory _newBaseURI) public onlyOwner() { | |
baseURI = _newBaseURI; | |
} | |
function _baseURI() internal view override returns (string memory) { | |
return baseURI; | |
} | |
function safeMint(uint256 _amount) public payable { | |
require(isSaleActive, "Sale is currently not active"); | |
require(MAX_TOKENS >= _amount + _tokenIdCounter.current(), "Not enough tokens left to buy."); | |
require(_amount > 0 && _amount < MAX_TOKENS_PER_SALE + 1, "Amount of tokens exceeds amount of tokens you can purchase in a single purchase."); | |
require(msg.value >= price * _amount, "Amount of ether sent not correct."); | |
for(uint256 i = 0; i < _amount; i++){ | |
_safeMint(msg.sender, _tokenIdCounter.current()); | |
_tokenIdCounter.increment(); | |
} | |
} | |
function reserveTokens(uint256 _amount) public onlyOwner() { | |
require(MAX_TOKENS >= _amount + _tokenIdCounter.current(), "Not enough tokens left to buy."); | |
for (uint i = 0; i < _amount; i++) { | |
_safeMint(msg.sender, _tokenIdCounter.current()); | |
_tokenIdCounter.increment(); | |
} | |
} | |
// The following functions are overrides required by Solidity for ERC721Enumerable. | |
function _beforeTokenTransfer(address from, address to, uint256 tokenId) | |
internal | |
override(ERC721, ERC721Enumerable) | |
{ | |
super._beforeTokenTransfer(from, to, tokenId); | |
} | |
function supportsInterface(bytes4 interfaceId) | |
public | |
view | |
override(ERC721, ERC721Enumerable) | |
returns (bool) | |
{ | |
return super.supportsInterface(interfaceId); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment