feat: remove updateRestriction function

cloudwalk · May 2, 2024 · c35a1e6 · c35a1e6
1 parent b31e270
commit c35a1e6
Show file tree

Hide file tree

Showing 4 changed files with 72 additions and 158 deletions.
diff --git a/contracts/base/ERC20Restrictable.sol b/contracts/base/ERC20Restrictable.sol
@@ -64,26 +64,6 @@ abstract contract ERC20Restrictable is ERC20Base, IERC20Restrictable {
         _purposeAssignments[account] = purposes;
     }
 
-    /**
-     * @inheritdoc IERC20Restrictable
-     */
-    function updateRestriction(address account, bytes32 purpose, uint256 balance) external onlyBlocklister {
-        if (purpose == bytes32(0)) {
-            revert ZeroPurpose();
-        }
-
-        uint256 oldBalance = _restrictedPurposeBalances[account][purpose];
-        _restrictedPurposeBalances[account][purpose] = balance;
-
-        if (oldBalance > balance) {
-            _totalRestrictedBalances[account] -= oldBalance - balance;
-        } else {
-            _totalRestrictedBalances[account] += balance - oldBalance;
-        }
-
-        emit UpdateRestriction(account, purpose, balance, oldBalance);
-    }
-
     /**
      * @inheritdoc IERC20Restrictable
      */

diff --git a/contracts/base/interfaces/IERC20Restrictable.sol b/contracts/base/interfaces/IERC20Restrictable.sol
@@ -42,15 +42,6 @@ interface IERC20Restrictable {
      */
     function assignedPurposes(address account) external view returns (bytes32[] memory);
 
-    /**
-     * @notice Updates the restriction balance for an account
-     *
-     * @param account The account to update restriction for
-     * @param purpose The restriction purpose
-     * @param balance The new restricted balance
-     */
-    function updateRestriction(address account, bytes32 purpose, uint256 balance) external;
-
     /**
      * @notice Increases the restriction balance for an account
      *