Add deleverage function

Makefile

@@ -129,6 +129,9 @@ supply-script:
 repay-script:
 	@forge script script/Interactions.s.sol:Interactions ${NETWORK_ARGS} -vvvv --sig "aaveRepayAavePM()"
 
+deleverage-script:
+	@forge script script/Interactions.s.sol:Interactions ${NETWORK_ARGS} -vvvv --sig "aaveDeleverageAavePM()"
+
 closePosition-script:
 	@forge script script/Interactions.s.sol:Interactions ${NETWORK_ARGS} -vvvv --sig "aaveClosePositionAavePM()"
 
@@ -164,6 +167,7 @@ update-hft-anvil: anvil-network update-hft
 update-st-anvil: anvil-network update-st
 rebalance-anvil: anvil-network rebalance-script
 reinvest-anvil: anvil-network reinvest-script
+deleverage-anvil: anvil-network deleverage-script
 closePosition-anvil: anvil-network closePosition-script
 supply-anvil: anvil-network supply-script
 repay-anvil: anvil-network repay-script

script/Interactions.s.sol

@@ -85,6 +85,12 @@ contract Interactions is Script {
         vm.stopBroadcast();
     }
 
+    function aaveDeleverageAavePM() public {
+        vm.startBroadcast();
+        aavePM.deleverage();
+        vm.stopBroadcast();
+    }
+
     function aaveClosePositionAavePM() public {
         vm.startBroadcast();
         aavePM.aaveClosePosition(0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266);

src/AavePM.sol

@@ -293,6 +293,23 @@ contract AavePM is
         // TODO: Emit an event with suppliedCollateral, reinvestedDebt, reinvestedCollateral
     }
 
+    /// @notice // TODO: Add comment
+    function deleverage() public onlyRole(MANAGER_ROLE) {
+        // Update the Health Factor target to the maximum value.
+        uint16 previousHealthFactorTarget = s_healthFactorTarget;
+        if (previousHealthFactorTarget != type(uint16).max) {
+            s_healthFactorTarget = type(uint16).max;
+        }
+
+        // Rebalance position to repay all debt.
+        rebalance();
+
+        // Reset the Health Factor target to the previous value.
+        if (previousHealthFactorTarget != type(uint16).max) {
+            s_healthFactorTarget = previousHealthFactorTarget;
+        }
+    }
+
     /// @notice // TODO: Add comment
     function aaveSupplyFromContractBalance() public onlyRole(MANAGER_ROLE) returns (uint256 suppliedCollateral) {
         suppliedCollateral = _convertExistingBalanceToWstETHAndSupplyToAave();
@@ -421,15 +438,12 @@ contract AavePM is
 
     /// @notice // TODO: Add comment
     function aaveClosePosition(address _owner) public onlyRole(MANAGER_ROLE) checkOwner(_owner) {
-        // Update the Health Factor target to the maximum value.
-        if (!(s_healthFactorTarget == type(uint16).max)) {
-            emit HealthFactorTargetUpdated(s_healthFactorTarget, type(uint16).max);
-            s_healthFactorTarget = type(uint16).max;
+        // Check if position needs to be deleveraged before closing.
+        (,,,,, uint256 currentHealthFactor) = IPool(getContractAddress("aavePool")).getUserAccountData(address(this));
+        if (currentHealthFactor != type(uint256).max) {
+            deleverage();
         }
 
-        // Rebalance position to repay all debt.
-        rebalance();
-
         // Withdraw all wstETH to the owner.
         aaveWithdrawWstETH(getContractBalance("awstETH"), _owner);
 

src/interfaces/IAavePM.sol

@@ -91,6 +91,7 @@ interface IAavePM {
     // ================================================================
     function rebalance() external;
     function reinvest() external returns (uint256 reinvestedDebt);
+    function deleverage() external;
     function aaveSupplyFromContractBalance() external returns (uint256 suppliedCollateral);
     function aaveRepayUSDCFromContractBalance() external;