add approveAssetToken (#148)

plumenetwork · Jan 5, 2025 · 72a150f · 72a150f
1 parent 557dab1
commit 72a150f
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/nest/src/AggregateToken.sol b/nest/src/AggregateToken.sol
@@ -285,6 +285,21 @@ contract AggregateToken is ComponentToken, IAggregateToken, ERC1155Holder {
         IERC20(componentToken.asset()).approve(address(componentToken), amount);
     }
 
+    /**
+     * @notice Approve the given spender to spend the given amount of any asset token
+     * @dev Only the manager can call this function
+     * @param assetToken The ERC20 token to approve
+     * @param spender The address that will spend the token
+     * @param amount Amount of tokens to approve
+     */
+    function approveAssetToken(
+        IERC20 assetToken,
+        address spender,
+        uint256 amount
+    ) external nonReentrant onlyRole(MANAGER_ROLE) {
+        SafeERC20.forceApprove(assetToken, spender, amount);
+    }
+
     /**
      * @notice Add a ComponentToken to the component token list
      * @dev Only the owner can call this function, and there is no way to remove a ComponentToken later