Add isOpen boolean gate

IndexCoop · Dec 20, 2023 · bdf2861 · bdf2861
1 parent a8d67aa
commit bdf2861
Show file tree

Hide file tree

Showing 3 changed files with 556 additions and 495 deletions.
diff --git a/contracts/adapters/OptimisticAuctionRebalanceExtension.sol b/contracts/adapters/OptimisticAuctionRebalanceExtension.sol
@@ -108,6 +108,7 @@ contract OptimisticAuctionRebalanceExtension is  AuctionRebalanceExtension, Asse
     ProductSettings public productSettings; // Mapping of set token to ProductSettings
     mapping(bytes32 => bytes32) public assertionIds; // Maps proposal hashes to assertionIds.
     mapping(bytes32 => Proposal) public proposedProduct; // Maps assertionIds to a Proposal.
+    bool public isOpen; // Bool indicating whether the extension is open for proposing rebalances.
 
     // Keys for assertion claim data.
     bytes public constant PROPOSAL_HASH_KEY = "proposalHash";
@@ -124,6 +125,13 @@ contract OptimisticAuctionRebalanceExtension is  AuctionRebalanceExtension, Asse
 
     }
 
+    /* ============ Modifier ============ */
+
+    modifier onlyIfOpen() {
+        require(isOpen, "Must be open for rebalancing");
+        _;
+    }
+
     /* ============ External Functions ============ */
 
     /**
@@ -154,6 +162,16 @@ contract OptimisticAuctionRebalanceExtension is  AuctionRebalanceExtension, Asse
         _updateUseAssetAllowlist(_useAssetAllowlist);
     }
 
+    /**
+    * ONLY OPERATOR: Toggle isOpen on and off. When false the extension is closed for proposing rebalances.
+    * when true it is open.
+    *
+    * @param _isOpen           Bool indicating whether the extension is open for proposing rebalances.
+    */
+   function updateIsOpen(bool _isOpen) external onlyOperator {
+       isOpen = _isOpen;
+    }
+
 
 
     /**
@@ -198,6 +216,7 @@ contract OptimisticAuctionRebalanceExtension is  AuctionRebalanceExtension, Asse
     )
         external
         onlyAllowedAssets(_newComponents)
+        onlyIfOpen()
     {
         bytes32 proposalHash = keccak256(abi.encode(
             setToken,