fix: handle rebalance issue

own-protocol · Jan 18, 2025 · 162ae8c · 162ae8c
1 parent 3d1e0a4
commit 162ae8c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/interfaces/IAssetPool.sol b/src/interfaces/IAssetPool.sol
@@ -149,6 +149,8 @@ interface IAssetPool {
     error InsufficientLPLiquidity();
     /// @notice Thrown when rebalance parameters don't match requirements
     error RebalanceMismatch();
+    /// @notice Thrown when a user attempts to interact with an LP's rebalance
+    error InvalidCycleRequest();
 
     // --------------------------------------------------------------------------------
     //                                USER ACTIONS

diff --git a/src/protocol/AssetPool.sol b/src/protocol/AssetPool.sol
@@ -189,6 +189,8 @@ contract AssetPool is IAssetPool, Ownable, Pausable {
         cycleState = CycleState.ACTIVE;
         cycleTime = _cyclePeriod;
         rebalanceTime = _rebalancingPeriod;
+        nextRebalanceStartDate = block.timestamp + _cyclePeriod;
+        nextRebalanceEndDate = nextRebalanceStartDate + _rebalancingPeriod;
     }
 
     // --------------------------------------------------------------------------------
@@ -395,6 +397,17 @@ contract AssetPool is IAssetPool, Ownable, Pausable {
         }
     }
 
+    function updateCycle() external {
+        if (cycleState != CycleState.REBALANCING) revert InvalidCycleState();
+        if (cycleTotalDepositRequests[cycleIndex] > 0) revert InvalidCycleRequest();
+        if (cycleTotalRedemptionRequests[cycleIndex] > 0) revert InvalidCycleRequest();
+
+        cycleIndex++;
+        cycleState = CycleState.ACTIVE;
+        nextRebalanceStartDate = block.timestamp + cycleTime;
+        nextRebalanceEndDate = nextRebalanceStartDate + rebalanceTime;
+    }
+
     // --------------------------------------------------------------------------------
     //                            GOVERNANCE FUNCTIONS
     // --------------------------------------------------------------------------------