From 12a7a2cdf9cc8a01d79a24dec91e0e1bb305f5ee Mon Sep 17 00:00:00 2001
From: zoruka <zo.fmendes@gmail.com>
Date: Thu, 8 Dec 2022 10:10:57 -0300
Subject: [PATCH] refactor: remove upgradeTokenBuild and fix burn requirement

---
 contracts/FleekERC721.sol | 16 +---------------
 scripts/upgrade.js        |  2 +-
 2 files changed, 2 insertions(+), 16 deletions(-)

diff --git a/contracts/FleekERC721.sol b/contracts/FleekERC721.sol
index 2381164..a69fa22 100644
--- a/contracts/FleekERC721.sol
+++ b/contracts/FleekERC721.sol
@@ -84,16 +84,6 @@ contract FleekERC721 is ERC721, FleekAccessControl {
         return tokenId;
     }
 
-    function upgradeTokenBuild(
-        uint256 tokenId,
-        string memory commit,
-        string memory repository,
-        string memory author
-    ) public payable requireTokenOwner(tokenId) {
-        _requireMinted(tokenId);
-        setTokenBuild(tokenId, commit, repository, author);
-    }
-
     function tokenURI(
         uint256 tokenId
     ) public view virtual override returns (string memory) {
@@ -234,11 +224,7 @@ contract FleekERC721 is ERC721, FleekAccessControl {
 
     function burn(
         uint256 tokenId
-    ) public virtual requireTokenController(tokenId) {
-        require(
-            ownerOf(tokenId) == msg.sender,
-            "FleekERC721: must be token owner"
-        );
+    ) public virtual requireTokenOwner(tokenId) {
         super._burn(tokenId);
 
         if (bytes(_apps[tokenId].external_url).length != 0) {
diff --git a/scripts/upgrade.js b/scripts/upgrade.js
index 3fe695e..9dd6ca6 100644
--- a/scripts/upgrade.js
+++ b/scripts/upgrade.js
@@ -15,7 +15,7 @@ const params = [
     contractAddress
   );
 
-  const transaction = await contract.upgradeTokenBuild(...params);
+  const transaction = await contract.setTokenBuild(...params);
 
   console.log('Response: ', transaction);
 })();