feat: add maxShares so we can rebalance

yieldnest · Nov 7, 2024 · bcfdf09 · bcfdf09
1 parent 9ae9f52
commit bcfdf09
Showing 1 changed file with 20 additions and 6 deletions.
diff --git a/src/ynEIGEN/withdrawalsProcessor.sol b/src/ynEIGEN/withdrawalsProcessor.sol
@@ -124,15 +124,29 @@ contract WithdrawalsProcessor is Ownable {
                 uint256 _withdrawnShares;
                 address _node = address(_nodes[j]);
                 uint256 _nodeShares = _strategy.shares(_node);
-                if (_nodeShares > _pendingWithdrawalRequestsInShares) {
-                    _withdrawnShares =
+                // if (_nodeShares > _pendingWithdrawalRequestsInShares) {
+                //     _withdrawnShares =
+                //         (_nodeShares - _pendingWithdrawalRequestsInShares) < MIN_DELTA
+                //             ? _nodeShares
+                //             : _pendingWithdrawalRequestsInShares;
+                //     _pendingWithdrawalRequestsInShares = 0;
+                // } else if (_nodeShares > _minNodeShares) {
+                //     _withdrawnShares = _nodeShares;
+                //     _pendingWithdrawalRequestsInShares -= _nodeShares;
+                // }
+
+                if (_nodeShares > _minNodeShares) {
+                    _nodeShares = (_nodeShares > _maxNodesShares) ? _maxNodesShares : _nodeShares;
+                    if (_nodeShares > _pendingWithdrawalRequestsInShares) {
+                        _withdrawnShares =
                         (_nodeShares - _pendingWithdrawalRequestsInShares) < MIN_DELTA
                             ? _nodeShares
                             : _pendingWithdrawalRequestsInShares;
-                    _pendingWithdrawalRequestsInShares = 0;
-                } else if (_nodeShares > _minNodeShares) {
-                    _withdrawnShares = _nodeShares;
-                    _pendingWithdrawalRequestsInShares -= _nodeShares;
+                        _pendingWithdrawalRequestsInShares = 0;
+                    } else {
+                        _withdrawnShares = _nodeShares;
+                        _pendingWithdrawalRequestsInShares -= _nodeShares;
+                    }
                 }
 
                 if (_withdrawnShares > 0) {