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

    string baseURI;
    string public baseExtension = ".json";
    uint256 public cost = 0.06 ether;
    uint256 public maxSupply = 5555;
    uint256 public discount = 0.01 ether;
    uint256 public maxMintAmount = 20;
    mapping(address=>bool) public whitelistedAdresses;
    bool public paused = true;
    bool public revealed;
    string public notRevealedUri;

    constructor(
        string memory _name,    
        string memory _symbol,
        string memory _initBaseURI,
        string memory _initNotRevealedUri
    ) ERC721(_name, _symbol) {
        setBaseURI(_initBaseURI);
        setNotRevealedURI(_initNotRevealedUri);
    }

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

    // public
    function mint(uint256 _mintAmount) public payable {
        require(!paused);
        uint256 supply = totalSupply();
        require(supply>=2);
        uint256 priceToPay = cost;
        if (msg.sender != owner()) {
            if(whitelistedAdresses[msg.sender] == true) {
                priceToPay -= discount;
            }
            require(msg.value >= priceToPay * _mintAmount,"not enough value");
        }
        require(_mintAmount > 0);
        require(_mintAmount <= maxMintAmount);
        require(supply + _mintAmount <= maxSupply);
        for (uint256 i = 1; i <= _mintAmount; i++) {
            _safeMint(msg.sender,supply+i);
        }
    }

    function mintUniqueAddress(address _user) public onlyOwner{
        uint256 supply = totalSupply();
        require(supply<2);
        _safeMint(_user,supply+1);
    }
    
    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) {
            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 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 setDiscount(uint256 _percent) public onlyOwner() {
        discount = _percent;
    }

    function whitelistUsers(address[] memory _users,bool whitelisted) public onlyOwner() {
        for(uint256 i = 0; i< _users.length;i++){
            whitelistedAdresses[_users[i]]=whitelisted;
        }
    }

    function withdraw() public payable onlyOwner() {
        (bool success, ) = payable(msg.sender).call{value: address(this).balance}("");
        require(success);
    }
}
function _baseURI() internal view virtual override returns (string memory) {
    return baseURI;
}

// internal