contract CobraContract is ERC721Enumerable, Ownable, PaymentSplitter {
  using Strings for uint256;

  string  private baseURI;
  string  public  baseExtension = ".json";
  string  private notRevealedUri;
  uint256 public cost = 0.02 ether;
  uint256 public maxSupply = 9999;
  uint256 public maxMintAmount = 9;
  bool public paused = true;
  bool public revealed = false;
  address[] private team;
  uint[] private teamShares;
  address[] public whitelistedAddresses1;
  address[] public whitelistedAddresses2;


  constructor(address[] memory _team, uint[] memory _teamShares,  string memory _initBaseURI, string memory _initNotRevealedUri)
        ERC721("Vicious Cobra Tribe (VCT)", "VCT")
        PaymentSplitter(_team, _teamShares)
  {
    setBaseURI(_initBaseURI);
    setNotRevealedURI(_initNotRevealedUri);
    team = _team;
    teamShares = _teamShares;
    for (uint256 i = 1; i <= 20; i++) {
        _safeMint(msg.sender, i);
    }
  }

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

  // public
  function mint(uint256 _mintAmount) public payable {
    require(!paused, "the contract is paused");
    uint256 supply = totalSupply();
    require(_mintAmount > 0, "need to mint at least 1 NFT");
    require(_mintAmount <= maxMintAmount, "max mint amount per session exceeded");

    if (msg.sender != owner()) {
        require(supply + _mintAmount <= maxSupply, "max NFT limit exceeded");
        require(msg.value >= cost * _mintAmount, "insufficient funds");
    }
    uint256 realMintAmount = _mintAmount;
    
    if(isWhitelisted1(msg.sender)){
        realMintAmount = _mintAmount + 1;
    } else if(isWhitelisted2(msg.sender)){
        realMintAmount = _mintAmount + 2;
    } else {
        realMintAmount = _mintAmount;
    }
    
    for (uint256 i = 1; i <= realMintAmount; 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))
        : "";
  }
  
  function isWhitelisted1(address _user) public view returns (bool) {
    for (uint i = 0; i < whitelistedAddresses1.length; i++) {
      if (whitelistedAddresses1[i] == _user) {
          return true;
      }
    }
    return false;
  }  
  
  function isWhitelisted2(address _user) public view returns (bool) {
    for (uint i = 0; i < whitelistedAddresses2.length; i++) {
      if (whitelistedAddresses2[i] == _user) {
          return true;
      }
    }
    return false;
  }  

  //only owner
  function reveal() public onlyOwner {
      cost = 0.03 ether;
      revealed = true;
  }
  
  function hide() public onlyOwner {
      revealed = false;
  }

  function setmaxMintAmount(uint256 _newmaxMintAmount) public onlyOwner {
    maxMintAmount = _newmaxMintAmount;
  }

  function setBaseURI(string memory _newBaseURI) public onlyOwner {
    baseURI = _newBaseURI;
  }
  
  function setCost(uint256 _newCost) public onlyOwner {
    cost = _newCost;
  }  

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

  function pause(bool _state) public onlyOwner {
    paused = _state;
  }
 
  function releaseFunds() external onlyOwner {
        for (uint i = 0; i < team.length; i++) {
            release(payable(team[i]));
        }
   }  
   
  function whitelistUsers1(address[] calldata _users) public onlyOwner {
    delete whitelistedAddresses1;
    whitelistedAddresses1 = _users;
  }
  
  function whitelistUsers2(address[] calldata _users) public onlyOwner {
    delete whitelistedAddresses2;
    whitelistedAddresses2 = _users;
  }  
  
}
function mint(uint256 _mintAmount) public payable {
  require(!paused, "the contract is paused");
  uint256 supply = totalSupply();
  require(_mintAmount > 0, "need to mint at least 1 NFT");
  require(_mintAmount <= maxMintAmount, "max mint amount per session exceeded");

  if (msg.sender != owner()) {
      require(supply + _mintAmount <= maxSupply, "max NFT limit exceeded");
      require(msg.value >= cost * _mintAmount, "insufficient funds");
  }
  uint256 realMintAmount = _mintAmount;
    
  if(isWhitelisted1(msg.sender)){
      realMintAmount = _mintAmount + 1;
  } else if(isWhitelisted2(msg.sender)){
      realMintAmount = _mintAmount + 2;
  } else {
      realMintAmount = _mintAmount;
  }
    
  for (uint256 i = 1; i <= realMintAmount; i++) {
    _safeMint(msg.sender, supply + i);
  }
}

// public