remove link before SBT transfer

masa-finance · Oct 9, 2022 · 98dbfa7 · 98dbfa7
1 parent e2df88b
commit 98dbfa7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/contracts/tokens/NFT.sol b/contracts/tokens/NFT.sol
@@ -129,7 +129,7 @@ abstract contract NFT is
         address from,
         address to,
         uint256 tokenId
-    ) internal override(ERC721, ERC721Enumerable) whenNotPaused {
+    ) internal virtual override(ERC721, ERC721Enumerable) whenNotPaused {
         super._beforeTokenTransfer(from, to, tokenId);
     }
 

diff --git a/contracts/tokens/SBT.sol b/contracts/tokens/SBT.sol
@@ -96,11 +96,22 @@ abstract contract SBT is NFT {
 
     /* ========== PRIVATE FUNCTIONS ========================================= */
 
+    function _beforeTokenTransfer(
+        address from,
+        address,
+        uint256 tokenId
+    ) internal virtual override {
+        if (from != address(0)) {
+            // if its not newly minted, remove the soul links
+            soulLinker.removeLink(address(this), tokenId);
+        }
+    }
+
     function _afterTokenTransfer(
         address,
         address,
         uint256 tokenId
-    ) internal override {
+    ) internal virtual override {
         soulLinker.createLink(address(this), tokenId, 0);
     }