feat: add pause lib-ubiquity-pool logic

ubiquity · Sep 18, 2024 · b5dc054 · b5dc054
1 parent dd032cf
commit b5dc054
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 3 deletions.
diff --git a/packages/contracts/src/dollar/core/UbiquityPoolSecurityMonitor.sol b/packages/contracts/src/dollar/core/UbiquityPoolSecurityMonitor.sol
@@ -6,7 +6,7 @@ import {Initializable} from "@openzeppelin/contracts-upgradeable/proxy/utils/Ini
 import {UUPSUpgradeable} from "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
 import {AccessControlFacet} from "../facets/AccessControlFacet.sol";
 import {UbiquityPoolFacet} from "../facets/UbiquityPoolFacet.sol";
-
+import {LibUbiquityPool} from "../libraries/LibUbiquityPool.sol";
 import "../libraries/Constants.sol";
 import "forge-std/console.sol";
 
@@ -92,8 +92,10 @@ contract UbiquityPoolSecurityMonitor is Initializable, UUPSUpgradeable {
                 .div(liquidityVertex);
 
             if (liquidityDiffPercentage >= thresholdPercentage) {
-                // a) Pause the UbiquityDollarToken
-                // b) Pause LibUbiquityPool by disabling collateral
+                // TODO: Pause the UbiquityDollarToken
+
+                // Pause LibUbiquityPool by disabling collateral
+                _pauseLibUbiquityPool();
 
                 monitorPaused = true;
                 emit MonitorPaused(
@@ -103,4 +105,20 @@ contract UbiquityPoolSecurityMonitor is Initializable, UUPSUpgradeable {
             }
         }
     }
+
+    function _pauseLibUbiquityPool() internal {
+        address[] memory allCollaterals = ubiquityPoolFacet.allCollaterals();
+
+        for (uint256 i = 0; i < allCollaterals.length; i++) {
+            try
+                ubiquityPoolFacet.collateralInformation(allCollaterals[i])
+            returns (
+                LibUbiquityPool.CollateralInformation memory collateralInfo
+            ) {
+                ubiquityPoolFacet.toggleCollateral(collateralInfo.index);
+            } catch {
+                continue;
+            }
+        }
+    }
 }
diff --git a/...r/monitors/PoolLiquidityMonitorTest.t.sol → ...ollar/core/PoolLiquidityMonitorTest.t.sol b/...r/monitors/PoolLiquidityMonitorTest.t.sol → ...ollar/core/PoolLiquidityMonitorTest.t.sol
@@ -147,6 +147,8 @@ contract PoolLiquidityMonitorTest is DiamondTestSetup {
             address(ubiquityPoolFacet)
         );
 
+        accessControlFacet.grantRole(DEFAULT_ADMIN_ROLE, address(monitor));
+
         // stop being admin
         vm.stopPrank();
 
@@ -220,7 +222,11 @@ contract PoolLiquidityMonitorTest is DiamondTestSetup {
         vm.prank(defenderRelayer);
         monitor.checkLiquidityVertex();
 
+        vm.expectRevert("Invalid collateral");
+        ubiquityPoolFacet.collateralInformation(address(collateralToken));
+
         bool monitorPaused = monitor.monitorPaused();
+
         assertTrue(
             monitorPaused,
             "Monitor should be paused after liquidity drop"