Deposit Cap (#5)

* feat: add events and checks

* chore: forge fmt

* chore: custom errors

* feat: claimInsurance, pro-rata withdraws

* chore: update test rpc

* uncomment lines

* added deposit cap

---------

Co-authored-by: sandybradley <[email protected]>

src/FoldCaptiveStaking.sol

@@ -41,6 +41,7 @@ contract FoldCaptiveStaking is Owned(msg.sender) {
     error ZeroLiquidity();
     error WithdrawFailed();
     error WithdrawProRata();
+    error DepositCapReached();
 
     /// @param _positionManager The Canonical UniswapV3 PositionManager
     /// @param _pool The FOLD Pool to Reward
@@ -134,6 +135,9 @@ contract FoldCaptiveStaking is Owned(msg.sender) {
     /// @dev For keeping track of positions fees
     uint256 public token1FeesPerLiquidity;
 
+    /// @dev The cap on deposits in the pool in liquidity, set to 0 if no cap
+    uint256 public depositCap;
+
     /*//////////////////////////////////////////////////////////////
                                   CHEF
     //////////////////////////////////////////////////////////////*/
@@ -199,6 +203,10 @@ contract FoldCaptiveStaking is Owned(msg.sender) {
         balances[msg.sender].amount += liquidity;
         liquidityUnderManagement += uint256(liquidity);
 
+        if (liquidityUnderManagement > depositCap && depositCap != 0) {
+            revert DepositCapReached();
+        }
+
         emit Deposit(msg.sender, amount0, amount1);
     }
 
@@ -327,6 +335,14 @@ contract FoldCaptiveStaking is Owned(msg.sender) {
         token1FeesPerLiquidity += amount1Collected;
     }
 
+    /*//////////////////////////////////////////////////////////////
+                             OWNER CONTROLS
+    //////////////////////////////////////////////////////////////*/
+    /// @param _newCap The new deposit cap, measured in liquidity
+    function setDepositCap(uint256 _newCap) public onlyOwner {
+        depositCap = _newCap;
+    }
+
     /// @notice Allows the owner to claim insurance in case of relay outage
     /// @param liquidity The amount of liquidity to claim
     function claimInsurance(uint128 liquidity) external onlyOwner {