Updated NFT

GeneralMagicio · Aug 21, 2024 · 02eb0f9 · 02eb0f9
1 parent e846ad4
commit 02eb0f9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/script/QAccProjectNFT.s.sol b/script/QAccProjectNFT.s.sol
@@ -12,7 +12,9 @@ contract QAccProjectNFTScript is Script {
     function run() public {
         vm.startBroadcast();
 
-        nft = new QAccProjectNFT("QAccProjectNFT", "QAPNFT");
+        address deployer = tx.origin;
+
+        nft = new QAccProjectNFT("QAccProjectNFT", "QAPNFT", deployer, "BaseURI");
 
         vm.stopBroadcast();
     }

diff --git a/src/QAccProjectNFT.sol b/src/QAccProjectNFT.sol
@@ -7,9 +7,14 @@ import "@openzeppelin/contracts/access/Ownable.sol";
 
 contract QAccProjectNFT is ERC721, Ownable {
     uint256 private _nextTokenId;
+    string private _baseTokenURI;
 
-    constructor(string memory name, string memory symbol) ERC721(name, symbol) Ownable(msg.sender) {
+    constructor(string memory name, string memory symbol, address initialOwner, string memory baseURI)
+        ERC721(name, symbol)
+        Ownable(initialOwner)
+    {
         _nextTokenId = 1; // Start token IDs at 1
+        _setBaseURI(baseURI);
     }
 
     function mint(address to) external onlyOwner {
@@ -20,4 +25,17 @@ contract QAccProjectNFT is ERC721, Ownable {
     function currentTokenId() external view returns (uint256) {
         return _nextTokenId - 1;
     }
+
+    // set base URI
+    function setBaseURI(string memory baseURI) external onlyOwner {
+        _setBaseURI(baseURI);
+    }
+
+    function _setBaseURI(string memory baseURI) internal {
+        _baseTokenURI = baseURI;
+    }
+
+    function _baseURI() internal view override returns (string memory) {
+        return _baseTokenURI;
+    }
 }