From e3a69a0bb022c78ef45f09402c000f45a7e291f3 Mon Sep 17 00:00:00 2001
From: SHAKOTN <shakotn@protonmail.com>
Date: Wed, 16 Aug 2023 14:30:42 +0300
Subject: [PATCH] [feat] - finish realisePnl function

---
 src/FluxStrategy.sol | 44 ++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/src/FluxStrategy.sol b/src/FluxStrategy.sol
index 7501a1c..689c819 100644
--- a/src/FluxStrategy.sol
+++ b/src/FluxStrategy.sol
@@ -315,7 +315,6 @@ contract FluxStrategy is BaseStrategy {
     /// @param _slippage ignore slippage as Flux is not AMM
     /// @return amount of assets denominated in 3crv that were divested
     function _divestAll(bool _slippage) internal override returns (uint256) {
-        uint256 threeCurveSnapshot = THREE_CRV.balanceOf(address(this));
         uint256 balance = _fToken.balanceOf(address(this));
         uint256 success = _fToken.redeem(balance);
         if (success != 0) revert FluxIntegrationErrors.RedeemFailed();
@@ -373,6 +372,47 @@ contract FluxStrategy is BaseStrategy {
         uint256 _excessDebt,
         uint256 _debtRatio
     ) internal override returns (uint256, uint256, uint256, uint256) {
-        return (0, 0, 0, 0);
+        uint256 profit;
+        uint256 loss;
+        uint256 debtRepayment;
+
+        uint256 debt = _gVault.getStrategyDebt();
+
+        (uint256 assets, uint256 balance, ) = _estimatedTotalAssets();
+        // Early revert
+        if (_excessDebt > assets) {
+            revert StrategyErrors.ExcessDebtGtThanAssets();
+        } else {
+            // If current assets are greater than debt, we have profit
+            if (assets > debt) {
+                profit = assets - debt;
+                uint256 profitToRepay = 0;
+                if (profit > profitThreshold) {
+                    profitToRepay =
+                        (profit * (PERCENTAGE_DECIMAL_FACTOR - _debtRatio)) /
+                        PERCENTAGE_DECIMAL_FACTOR;
+                }
+                if (profitToRepay + _excessDebt > balance) {
+                    balance += _divest(
+                        profitToRepay + _excessDebt - balance,
+                        true
+                    );
+                    debtRepayment = balance;
+                } else {
+                    debtRepayment = profitToRepay + _excessDebt;
+                }
+                // If current assets are less than debt, we have loss
+            } else if (assets < debt) {
+                loss = debt - assets;
+                // here for safety, but should really never be the case
+                //  that loss > _excessDebt
+                if (loss > _excessDebt) debtRepayment = 0;
+                else if (balance < _excessDebt - loss) {
+                    balance += _divest(_excessDebt - loss - balance, true);
+                    debtRepayment = balance;
+                } else debtRepayment = _excessDebt - loss;
+            }
+        }
+        return (profit, loss, debtRepayment, balance);
     }
 }