contract TheDumbCorgiSociety is ERC721Enumerable, Ownable {
  using Strings for uint256;

  string baseURI = "ipfs://QmQ318BzSU5m3iSBGTejCbxkfuPm3zjV4dG1ikMRyrN7Bg/";
  string public baseExtension = ".json";
  uint256 public cost = 0.045 ether;
  uint256 public whitelistedCost = 0 ether;
  uint256 public maxSupply = 10000;
  uint256 public maxMintAmount = 5;
  uint16 public maxMintAmountPerWallet = 5;
  bool public paused = false;
  bool public revealed = false;
  string public notRevealedUri = "ipfs://QmaBgCiKUqjRFSpTWPAXESBTJqZidGx7XYdVHw7udbAydS/hidden.json";
  mapping(address => bool) whitelistedAddresses;


  constructor(
  ) ERC721("The Dumb Corgi Society", "TheDumbCorgiSociety") {

  }

  // internal
  function _baseURI() internal view virtual override returns (string memory) {
    return baseURI;
  }

  // public
  function mint(uint256 _mintAmount) public payable {
    uint256 supply = totalSupply();
    
    if (msg.sender != owner()) {
        uint256 ownerTokenCount = balanceOf(msg.sender);
        require(!paused);
        require(_mintAmount > 0);
        require(_mintAmount <= maxMintAmount);
        require(supply + _mintAmount <= maxSupply);
        require(ownerTokenCount + _mintAmount <= maxMintAmountPerWallet);

        //check user is whitelisted or not
        if(verifyUser(msg.sender)){
            require(msg.value >= whitelistedCost * _mintAmount);
        }else{
            require(msg.value >= cost * _mintAmount);
        }
    }

    for (uint256 i = 1; i <= _mintAmount; i++) {
      _safeMint(msg.sender, supply + i);
    }
  }

  function reserve(uint256 _amount) public onlyOwner{
      uint256 supply = totalSupply();
      for (uint256 i = 1; i <= _amount; i++) {
      _safeMint(msg.sender, supply + i);
    }
  }

  function walletOfOwner(address _owner)
    public
    view
    returns (uint256[] memory)
  {
    uint256 ownerTokenCount = balanceOf(_owner);
    uint256[] memory tokenIds = new uint256[](ownerTokenCount);
    for (uint256 i; i < ownerTokenCount; i++) {
      tokenIds[i] = tokenOfOwnerByIndex(_owner, i);
    }
    return tokenIds;
  }

  function tokenURI(uint256 tokenId)
    public
    view
    virtual
    override
    returns (string memory)
  {
    require(
      _exists(tokenId),
      "ERC721Metadata: URI query for nonexistent token"
    );
    
    if(revealed == false) {
        return notRevealedUri;
    }

    string memory currentBaseURI = _baseURI();
    return bytes(currentBaseURI).length > 0
        ? string(abi.encodePacked(currentBaseURI, tokenId.toString(), baseExtension))
        : "";
  }

  //only owner
  function reveal() public onlyOwner {
      revealed = true;
  }
  
  function setCost(uint256 _newCost) public onlyOwner {
    cost = _newCost;
  }

  function setSupply(uint256 _newSupply) public onlyOwner {
    maxSupply = _newSupply;
  }

  function setWhitelistPrice(uint256 _newCost) public onlyOwner {
    whitelistedCost = _newCost;
  }

  function setmaxMintAmount(uint256 _newmaxMintAmount) public onlyOwner {
    maxMintAmount = _newmaxMintAmount;
  }
  
  function setNotRevealedURI(string memory _notRevealedURI) public onlyOwner {
    notRevealedUri = _notRevealedURI;
  }

  function setBaseURI(string memory _newBaseURI) public onlyOwner {
    baseURI = _newBaseURI;
  }

  function setBaseExtension(string memory _newBaseExtension) public onlyOwner {
    baseExtension = _newBaseExtension;
  }

  function pause(bool _state) public onlyOwner {
    paused = _state;
  }
 
  function withdraw() public payable onlyOwner {
    payable(msg.sender).call{value: address(this).balance};
  }

//whitelisting
    function addUser(address _addressToWhitelist) public onlyOwner {
        whitelistedAddresses[_addressToWhitelist] = true;
    }

    function verifyUser(address _whitelistedAddress) public view returns(bool) {
    bool userIsWhitelisted = whitelistedAddresses[_whitelistedAddress];
    return userIsWhitelisted;
    }

    modifier isWhitelisted(address _address) {
    require(whitelistedAddresses[_address], "You need to be whitelisted");
    _;
    }
}
function reveal() public onlyOwner {
    revealed = true;
}

//only owner