DecentHats is wearer of TopHat until end of transaction

decentdao · Jul 5, 2024 · 396e1e3 · 396e1e3
1 parent d7ae2de
commit 396e1e3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/contracts/DecentHats.sol b/contracts/DecentHats.sol
@@ -44,7 +44,7 @@ contract DecentHats {
         Hat[] calldata _hats
     ) public {
         uint256 topHatId = hats.mintTopHat(
-            msg.sender,
+            address(this),
             _topHatDetails,
             _topHatImageURI
         );
@@ -121,5 +121,7 @@ contract DecentHats {
                 ++i;
             }
         }
+
+        hats.transferHat(topHatId, address(this), msg.sender);
     }
 }
diff --git a/contracts/interfaces/hats/IHats.sol b/contracts/interfaces/hats/IHats.sol
@@ -37,4 +37,6 @@ interface IHats {
         uint256 _hatId,
         address _wearer
     ) external returns (bool success);
+
+    function transferHat(uint256 _hatId, address _from, address _to) external;
 }
diff --git a/contracts/mock/MockHats.sol b/contracts/mock/MockHats.sol
@@ -31,4 +31,6 @@ contract MockHats is IHats {
     function mintHat(uint256, address) external pure returns (bool success) {
         success = true;
     }
+
+    function transferHat(uint256, address, address) external {}
 }