Update FeeBankCharger.sol: Refactoring the Modifier

1inch · Aug 28, 2023 · eb117c6 · eb117c6
1 parent c90b5dd
commit eb117c6
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/contracts/FeeBankCharger.sol b/contracts/FeeBankCharger.sol
@@ -24,14 +24,21 @@ contract FeeBankCharger is IFeeBankCharger {
      * @dev Modifier to check if the sender is a feeBank contract.
      */
     modifier onlyFeeBank() {
-        if (msg.sender != address(feeBank)) revert OnlyFeeBankAccess();
+        checkFeeBank();
         _;
     }
 
     constructor(IERC20 token) {
         feeBank = new FeeBank(this, token, msg.sender);
     }
 
+    /**
+    * @dev Internal function to be called by the modifier 
+    */
+    function checkFeeBank() internal view {
+        if (msg.sender != address(feeBank)) revert OnlyFeeBankAccess();
+    }
+
     /**
      * @notice See {IFeeBankCharger-availableCredit}.
      */