[Contracts] Redemption helper

contracts/src/Interfaces/IRedemptionHelper.sol

@@ -0,0 +1,45 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+
+interface IRedemptionHelper {
+    struct SimulationContext {
+        address troveManager;
+        address sortedTroves;
+        bool redeemable;
+        uint256 price;
+        uint256 proportion;
+        uint256 attemptedBold;
+        uint256 redeemedBold;
+        uint256 iterations;
+    }
+
+    struct Redeemed {
+        uint256 bold;
+        uint256 coll;
+    }
+
+    function simulateRedemption(uint256 _bold, uint256 _maxIterationsPerCollateral)
+        external
+        returns (SimulationContext[] memory branch, uint256 totalProportions);
+
+    // Find the maximal amount of BOLD that can be redeemed proportionally within
+    // a given iteration limit. This helps prevent the redeemer from overpaying on
+    // the redemption fee.
+    //
+    // Also returns the expected fee that will be paid (as a percentage), and the
+    // expected collateral amounts that will be paid out in exchange for the
+    // redeemed BOLD. The latter may be used to calculate the _minCollRedeemed
+    // parameter passed to redeemCollateral().
+    function truncateRedemption(uint256 _bold, uint256 _maxIterationsPerCollateral)
+        external
+        returns (uint256 truncatedBold, uint256 feePct, Redeemed[] memory redeemed);
+
+    // Wrapper around CollateralRegistry's redeemCollateral() that adds slippage
+    // protection in the form of a minimum acceptable collateral amounts parameter.
+    function redeemCollateral(
+        uint256 _bold,
+        uint256 _maxIterationsPerCollateral,
+        uint256 _maxFeePct,
+        uint256[] memory _minCollRedeemed
+    ) external;
+}

contracts/src/RedemptionHelper.sol

@@ -0,0 +1,165 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity 0.8.24;
+
+import {IERC20} from "openzeppelin-contracts/contracts/token/ERC20/IERC20.sol";
+import {SafeERC20} from "openzeppelin-contracts/contracts/token/ERC20/utils/SafeERC20.sol";
+import {Math} from "openzeppelin-contracts/contracts/utils/math/Math.sol";
+
+import {_100pct, DECIMAL_PRECISION} from "./Dependencies/Constants.sol";
+import {IAddressesRegistry} from "./Interfaces/IAddressesRegistry.sol";
+import {IBoldToken} from "./Interfaces/IBoldToken.sol";
+import {ICollateralRegistry} from "./Interfaces/ICollateralRegistry.sol";
+import {IRedemptionHelper} from "./Interfaces/IRedemptionHelper.sol";
+import {ISortedTroves} from "./Interfaces/ISortedTroves.sol";
+import {ITroveManager} from "./Interfaces/ITroveManager.sol";
+import {LatestTroveData} from "./Types/LatestTroveData.sol";
+
+contract RedemptionHelper is IRedemptionHelper {
+    using SafeERC20 for IERC20;
+
+    struct RedemptionContext {
+        IERC20 collToken;
+        uint256 collBalanceBefore;
+        uint256 collRedeemed;
+    }
+
+    uint256 public immutable numBranches;
+    ICollateralRegistry public immutable collateralRegistry;
+    IBoldToken public immutable boldToken;
+    IAddressesRegistry[] public addresses; // only used off-chain, so we don't care about storage cost
+
+    constructor(ICollateralRegistry _collateralRegistry, IAddressesRegistry[] memory _addresses) {
+        require(_addresses.length == _collateralRegistry.totalCollaterals(), "Wrong number of registries");
+        numBranches = _addresses.length;
+        collateralRegistry = _collateralRegistry;
+        boldToken = _collateralRegistry.boldToken();
+
+        for (uint256 i = 0; i < _addresses.length; ++i) {
+            require(_collateralRegistry.getTroveManager(i) == _addresses[i].troveManager(), "TroveManager mismatch");
+            addresses.push(_addresses[i]);
+        }
+
+        boldToken.approve(address(_collateralRegistry), type(uint256).max);
+    }
+
+    // Meant to be called off-chain
+    // Not a view because price fetching has side-effects
+    function simulateRedemption(uint256 _bold, uint256 _maxIterationsPerCollateral)
+        public
+        returns (SimulationContext[] memory branch, uint256 totalProportions)
+    {
+        branch = new SimulationContext[](numBranches);
+
+        // First priority: proportional to unbacked debt
+        for (uint256 i = 0; i < numBranches; ++i) {
+            branch[i].troveManager = address(addresses[i].troveManager());
+            branch[i].sortedTroves = address(addresses[i].sortedTroves());
+            (branch[i].proportion, branch[i].price, branch[i].redeemable) =
+                ITroveManager(branch[i].troveManager).getUnbackedPortionPriceAndRedeemability();
+            if (branch[i].redeemable) totalProportions += branch[i].proportion;
+        }
+
+        // CS-BOLD-013: truncate redemption if it would exceed total unbacked debt
+        if (0 < totalProportions && totalProportions < _bold) _bold = totalProportions;
+
+        // Fallback: proportional to total debt
+        if (totalProportions == 0) {
+            for (uint256 i = 0; i < numBranches; ++i) {
+                branch[i].proportion = ITroveManager(branch[i].troveManager).getEntireBranchDebt();
+                if (branch[i].redeemable) totalProportions += branch[i].proportion;
+            }
+        }
+
+        if (totalProportions == 0) return (branch, totalProportions);
+
+        for (uint256 i = 0; i < numBranches; ++i) {
+            if (!branch[i].redeemable) continue;
+
+            branch[i].attemptedBold = _bold * branch[i].proportion / totalProportions;
+            if (branch[i].attemptedBold == 0) continue;
+
+            uint256 lastZombieTroveId = ITroveManager(branch[i].troveManager).lastZombieTroveId();
+            uint256 lastTroveId = ISortedTroves(branch[i].sortedTroves).getLast();
+
+            (uint256 troveId, uint256 nextTroveId) = lastZombieTroveId != 0
+                ? (lastZombieTroveId, lastTroveId)
+                : (lastTroveId, ISortedTroves(branch[i].sortedTroves).getPrev(lastTroveId));
+
+            for (
+                branch[i].iterations = 0;
+                branch[i].iterations < _maxIterationsPerCollateral || _maxIterationsPerCollateral == 0;
+                ++branch[i].iterations
+            ) {
+                if (branch[i].redeemedBold == branch[i].attemptedBold || troveId == 0) break;
+
+                LatestTroveData memory trove = ITroveManager(branch[i].troveManager).getLatestTroveData(troveId);
+                if (trove.entireColl * branch[i].price / trove.entireDebt >= _100pct) {
+                    branch[i].redeemedBold +=
+                        Math.min(branch[i].attemptedBold - branch[i].redeemedBold, trove.entireDebt);
+                }
+
+                troveId = nextTroveId;
+                nextTroveId = ISortedTroves(branch[i].sortedTroves).getPrev(nextTroveId);
+            }
+        }
+    }
+
+    // Meant to be called off-chain
+    // Not a view because price fetching has side-effects
+    function truncateRedemption(uint256 _bold, uint256 _maxIterationsPerCollateral)
+        external
+        returns (uint256 truncatedBold, uint256 feePct, Redeemed[] memory redeemed)
+    {
+        (SimulationContext[] memory branch, uint256 totalProportions) =
+            simulateRedemption(_bold, _maxIterationsPerCollateral);
+
+        if (totalProportions == 0) return (0, 0, redeemed);
+
+        truncatedBold = _bold;
+        for (uint256 i = 0; i < numBranches; ++i) {
+            if (branch[i].redeemable && branch[i].proportion > 0) {
+                uint256 extrapolatedBold = branch[i].redeemedBold * totalProportions / branch[i].proportion;
+                if (extrapolatedBold < truncatedBold) truncatedBold = extrapolatedBold;
+            }
+        }
+
+        feePct = collateralRegistry.getRedemptionRateForRedeemedAmount(truncatedBold);
+        redeemed = new Redeemed[](numBranches);
+
+        for (uint256 i = 0; i < numBranches; ++i) {
+            (uint256 redemptionPrice,) = addresses[i].priceFeed().fetchRedemptionPrice();
+            redeemed[i].bold = truncatedBold * branch[i].proportion / totalProportions;
+            redeemed[i].coll = redeemed[i].bold * (DECIMAL_PRECISION - feePct) / redemptionPrice;
+        }
+    }
+
+    function redeemCollateral(
+        uint256 _bold,
+        uint256 _maxIterationsPerCollateral,
+        uint256 _maxFeePct,
+        uint256[] memory _minCollRedeemed
+    ) external {
+        require(_minCollRedeemed.length == numBranches, "Wrong _minCollRedeemed length");
+
+        RedemptionContext[] memory branch = new RedemptionContext[](numBranches);
+
+        for (uint256 i = 0; i < numBranches; ++i) {
+            branch[i].collToken = collateralRegistry.getToken(i);
+            branch[i].collBalanceBefore = branch[i].collToken.balanceOf(address(this));
+        }
+
+        uint256 boldBalanceBefore = boldToken.balanceOf(address(this));
+
+        boldToken.transferFrom(msg.sender, address(this), _bold);
+        collateralRegistry.redeemCollateral(_bold, _maxIterationsPerCollateral, _maxFeePct);
+
+        for (uint256 i = 0; i < numBranches; ++i) {
+            branch[i].collRedeemed = branch[i].collToken.balanceOf(address(this)) - branch[i].collBalanceBefore;
+            require(branch[i].collRedeemed >= _minCollRedeemed[i], "Insufficient collateral redeemed");
+            if (branch[i].collRedeemed > 0) branch[i].collToken.safeTransfer(msg.sender, branch[i].collRedeemed);
+        }
+
+        uint256 boldRemaining = boldToken.balanceOf(address(this)) - boldBalanceBefore;
+        if (boldRemaining > 0) boldToken.transfer(msg.sender, boldRemaining);
+    }
+}