Bothead Collection

pragma solidity ^0.8.0;

import "https://github.com/OpenZeppelin/openzeppelin-contracts/blob/release-v4.3/contracts/token/ERC721/extensions/ERC721Enumerable.sol";
import "https://github.com/OpenZeppelin/openzeppelin-contracts/blob/release-v4.3/contracts/access/Ownable.sol";

contract BotHead is ERC721Enumerable, Ownable {
    address benefitAddr = 0x1Aa03735CB96F4C1b73eDDBf422dBA0882ba8882;
    
    string baseURI;

    uint256 public constant MAX_BOTHEADS = 10000;
    uint256 public price = 80000000000000000;  //0.08 ETH
    uint256 public prePrice = 75000000000000000;  //0.075 ETH
    
    bool public publicSale = false;
    bool public preSale = false;
    
    mapping (address => uint256) public preMintAddrs;
    uint256 public preMintNums;
    uint256 public ownerMintedNum;
    
    constructor(string memory name, string memory symbol) ERC721(name, symbol) {
    }
    
    function getPreMintAddrNums() public view returns (uint256){
        return preMintAddrs[_msgSender()];
    }
    
    function getPrice() public view returns (uint256){
        return price;
    }
    
    function getPrePrice() public view returns (uint256){
        return prePrice;
    }
    
    function getSoldNum() public view returns (uint256){
        return totalSupply();
    }
    
    function withdraw() public onlyOwner {
        uint balance = address(this).balance;
        payable(benefitAddr).transfer(balance);
    }

    function flipSaleState() public onlyOwner {
        publicSale = !publicSale;
    }
    
    function flipPreSaleState() public onlyOwner {
        preSale = !preSale;
    }
    
    function changePrice(uint256 newPrice) public onlyOwner {
        price = newPrice;
    }
    
    function changePrePrice(uint256 newPrice) public onlyOwner {
        prePrice = newPrice;
    }
    
    function setBaseURI(string memory _uri) public onlyOwner {
        baseURI = _uri;
    }
    
    function _baseURI() internal view virtual override returns (string memory) {
        return baseURI;
    }
    
    function ownerMint(uint nums) public onlyOwner{
        require(ownerMintedNum + nums <= 300, "onwer mint limit reached.");
        require(totalSupply() + nums < MAX_BOTHEADS, "heroes sold out.");
        
        for(uint i = 0; i < nums; i++){
            uint id = totalSupply();
            if(id < MAX_BOTHEADS){
                _safeMint(0xcdc114094a76727CB95304bA257F4C578e5Bd018, id);
            }
        }
        ownerMintedNum = ownerMintedNum + nums;
    }
    
    function setPreMintAddr(address[] memory addr) public onlyOwner{
        require(preSale, "Pre Sale not started.");

        uint addrLength = addr.length;

        for(uint i = 0; i < addrLength; i++){
            preMintAddrs[addr[i]] = preMintAddrs[addr[i]] + 5;
        }
    }
    
    function preMint(uint nums) external payable {
        require(preSale, "Pre Sale not started.");
        require(preMintAddrs[_msgSender()] >= nums, "can not pre mint more than you own.");
        require(preMintNums < 800, "ony 800 address can pre mint.");
        require(totalSupply() < MAX_BOTHEADS, "sold out.");
        require(totalSupply() + nums <= MAX_BOTHEADS, "not enough left.");
        
        uint salePrice = prePrice * nums;
        require(msg.value >= salePrice, "not enough funds to purchase.");
        
        for(uint i = 0; i < nums; i++){
            uint id = totalSupply();
            if(id < MAX_BOTHEADS){
                _safeMint(_msgSender(), id);
            }
        }
        
        preMintAddrs[_msgSender()] = preMintAddrs[_msgSender()] - nums;
        preMintNums = preMintNums + 1;
    }
    
    function mint(uint nums) external payable {
        require(publicSale, "Sale not started.");
        require(nums <= 20, "nums not more than 20.");
        require(totalSupply() < MAX_BOTHEADS, "sold out.");
        require(totalSupply() + nums <= MAX_BOTHEADS, "not enough left.");
        
        uint salePrice = price * nums;
        require(msg.value >= salePrice, "not enough funds to purchase.");
        
        for(uint i = 0; i < nums; i++){
            uint id = totalSupply();
            if(id < MAX_BOTHEADS){
                _safeMint(_msgSender(), id);
            }
        }
    }
}

{
  "compilationTarget": {
    "contracts/bothead1.sol": "BotHead"
  },
  "evmVersion": "london",
  "libraries": {},
  "metadata": {
    "bytecodeHash": "ipfs"
  },
  "optimizer": {
    "enabled": false,
    "runs": 200
  },
  "remappings": []
}