feat: computeRebalanceAmount public function

AngleProtocol · Oct 8, 2024 · 7b0d455 · 7b0d455
1 parent 3eb3da7
commit 7b0d455
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/contracts/helpers/BaseHarvester.sol b/contracts/helpers/BaseHarvester.sol
@@ -109,6 +109,20 @@ abstract contract BaseHarvester is IHarvester, AccessControl {
         _setMaxSlippage(newMaxSlippage);
     }
 
+    /*//////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+                                                        VIEW FUNCTIONS
+    //////////////////////////////////////////////////////////////////////////////////////////////////////////////////*/
+
+    /**
+     * @notice Compute the amount needed to rebalance the Transmuter
+     * @param yieldBearingAsset address of the yield bearing asset
+     * @return increase whether the exposure should be increased
+     * @return amount amount to be rebalanced
+     */
+    function computeRebalanceAmount(address yieldBearingAsset) external view returns (uint8 increase, uint256 amount) {
+        return _computeRebalanceAmount(yieldBearingAsset, yieldBearingData[yieldBearingAsset]);
+    }
+
     /*//////////////////////////////////////////////////////////////////////////////////////////////////////////////////
                                                         INTERNAL FUNCTIONS
     //////////////////////////////////////////////////////////////////////////////////////////////////////////////////*/