feat(protocol): allow owner to update recipient in TokenUnlock (#18184)

Co-authored-by: David <david@taiko.xyz>
taikoxyz · Sep 28, 2024 · 773ae1b · 773ae1b
1 parent 02a71dd
commit 773ae1b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/packages/protocol/contracts/layer1/team/tokenunlock/TokenUnlock.sol b/packages/protocol/contracts/layer1/team/tokenunlock/TokenUnlock.sol
@@ -66,6 +66,11 @@ contract TokenUnlock is EssentialContract {
         _;
     }
 
+    modifier onlyRecipientOrOwner() {
+        if (msg.sender != recipient && msg.sender != owner()) revert PERMISSION_DENIED();
+        _;
+    }
+
     /// @notice Initializes the contract.
     /// @param _owner The contract owner address.
     /// @param _rollupAddressManager The rollup address manager.
@@ -151,7 +156,7 @@ contract TokenUnlock is EssentialContract {
         IERC20(resolve(LibStrings.B_TAIKO_TOKEN, false)).safeTransfer(recipient, amount);
     }
 
-    function changeRecipient(address _newRecipient) external onlyRecipient {
+    function changeRecipient(address _newRecipient) external onlyRecipientOrOwner {
         if (_newRecipient == address(0) || _newRecipient == recipient) {
             revert INVALID_PARAM();
         }

diff --git a/packages/protocol/script/layer1/tokenunlock/exercise1.data.json b/packages/protocol/script/layer1/tokenunlock/exercise1.data.json
@@ -501,7 +501,7 @@
   },
   {
     "proxy": "0x2Da30e14dE6a5fcE16a5Ea72199De76cebDC876C",
-    "recipient": "0x9c0c06ddbb72f70820d7e55b6b77db175d400fba",
+    "recipient": "0xf7a0cd60e3b6ab4523a5f54e9a410258895b138d",
     "vestAmount": 4013867
   },
   {