Implement BurnExtension contract

artgenexyz · Apr 30, 2023 · 931b3b8 · 931b3b8
1 parent a59cf87
commit 931b3b8
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/contracts/extensions/BurnExtension.sol b/contracts/extensions/BurnExtension.sol
@@ -5,5 +5,21 @@ pragma solidity ^0.8.0;
 import "@openzeppelin/contracts/token/ERC721/extensions/ERC721Enumerable.sol";
 
 contract BurnExtension is ERC721Enumerable {
-    constructor(string memory _name, string memory _symbol) ERC721Enumerable(_name, _symbol) {}
+    uint256 private _maxSupply;
+
+    constructor(string memory _name, string memory _symbol, uint256 maxSupply) ERC721Enumerable(_name, _symbol) {
+        _maxSupply = maxSupply;
+    }
+
+    function burn(uint256 tokenId) public {
+        require(_exists(tokenId), "Token does not exist");
+        require(ownerOf(tokenId) == _msgSender(), "Not token owner");
+
+        _burn(tokenId);
+        _maxSupply--;
+    }
+
+    function maxSupply() public view returns (uint256) {
+        return _maxSupply;
+    }
 }