update init utility functions

0xJonaseb11 · Jun 3, 2024 · 85996dd · 85996dd
1 parent db5ae1e
commit 85996dd
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/smart_contract/contracts/Token.sol b/smart_contract/contracts/Token.sol
@@ -124,6 +124,38 @@ contract NFTMintDN404 is DN404, ERC20Permit, Ownable {
         return _symbol;
     }
 
+    // initialize tokenURI functionality - token metadata
+    function tokenURI(uint256 tokenId) public view override returns(string memory result) {
+        if (bytes(_baseURI).length != 0) {
+            result = string(abi.encodePacked(_baseURI, LibString.toString(tokenId));
+        }
+    }
+
+    // initialize allowlist mamagement functionality
+    function setAllowlist(bytes32 allowlistRoot_) public onlyOwner {
+        allowlistRoot = allowlistRoot_;
+    }
+
+    // initialize utility functions
+    function nftTotalSupply() public view returns(uint256) {
+        return _totalNFTSupply;
+    }
+
+    function nftBalanceOf(address owner) public view returns(uint256) {
+        return _balanceOfNFT(owner);
+    }
+
+    function previewNextTokenId() public view returns(uint256) {
+        return _nextTokenId;
+    }
+
+    function getURI() public view returns(string memory) {
+        return _baseURI;
+    }
+
+
+
+