Merge pull request #284 from fleekxyz/feat/remove-transfer

feat: disable transfers
fleekxyz · Jul 7, 2023 · 83f7ee2 · 83f7ee2
2 parents c900272 + ba5958a
commit 83f7ee2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/contracts/contracts/FleekApps.sol b/contracts/contracts/FleekApps.sol
@@ -52,6 +52,14 @@ contract FleekApps is Initializable, ERC721Upgradeable {
         ));
     }
 
+    /**
+     * @dev Override of transfer of ERC721.
+     * Transfer is disabled for NFA tokens.
+     */
+    function _transfer(address from, address to, uint256 tokenId) internal virtual override {
+        revert TransferIsDisabled();
+    }
+
     function _baseURI() internal view virtual override returns (string memory) {
         return "data:application/json;base64,";
     }

diff --git a/contracts/contracts/FleekERC721.sol b/contracts/contracts/FleekERC721.sol
@@ -16,6 +16,7 @@ import "./IERCX.sol";
 error MustBeTokenOwner(uint256 tokenId);
 error MustBeTokenVerifier(uint256 tokenId);
 error ThereIsNoTokenMinted();
+error TransferIsDisabled();
 
 contract FleekERC721 is
     IERCX,
@@ -250,6 +251,14 @@ contract FleekERC721 is
         super._beforeTokenTransfer(from, to, tokenId, batchSize);
     }
 
+    /**
+     * @dev Override of transfer of ERC721.
+     * Transfer is disabled for NFA tokens.
+     */
+    function _transfer(address from, address to, uint256 tokenId) internal virtual override whenNotPaused {
+        revert TransferIsDisabled();
+    }
+
     /**
      * @dev A baseURI internal function implementation to be called in the `tokenURI` function.
      */