Merge pull request #84 from Layr-Labs/feat/add-setters-for-state-vars

feat: add setters for state variables
Layr-Labs · Oct 1, 2024 · 7343c04 · 7343c04
2 parents 20f4792 + 9e92a1e
commit 7343c04
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/contracts/src/IIncredibleSquaringTaskManager.sol b/contracts/src/IIncredibleSquaringTaskManager.sol
@@ -24,6 +24,10 @@ interface IIncredibleSquaringTaskManager {
         address indexed challenger
     );
 
+    event AggregatorUpdated(address indexed oldAggregator, address indexed newAggregator);
+
+    event GeneratorUpdated(address indexed oldGenerator, address indexed newGenerator);
+
     // STRUCTS
     struct Task {
         uint256 numberToBeSquared;

diff --git a/contracts/src/IncredibleSquaringTaskManager.sol b/contracts/src/IncredibleSquaringTaskManager.sol
@@ -46,7 +46,7 @@ contract IncredibleSquaringTaskManager is
     address public aggregator;
     address public generator;
 
-    /* MODIFIERS */
+
     modifier onlyAggregator() {
         require(msg.sender == aggregator, "Aggregator must be the caller");
         _;
@@ -74,8 +74,16 @@ contract IncredibleSquaringTaskManager is
     ) public initializer {
         _initializePauser(_pauserRegistry, UNPAUSE_ALL);
         _transferOwnership(initialOwner);
-        aggregator = _aggregator;
-        generator = _generator;
+        _setAggregator(_aggregator);
+        _setGenerator(_generator);
+    }
+
+    function setGenerator(address newGenerator) external onlyOwner {
+        _setGenerator(newGenerator);
+    }
+
+    function setAggregator(address newAggregator) external onlyOwner {
+        _setAggregator(newAggregator);
     }
 
     /* FUNCTIONS */
@@ -317,4 +325,18 @@ contract IncredibleSquaringTaskManager is
     function getTaskResponseWindowBlock() external view returns (uint32) {
         return TASK_RESPONSE_WINDOW_BLOCK;
     }
+
+    function _setGenerator(address newGenerator) internal {
+        address oldGenerator = generator;
+        generator = newGenerator;
+        emit GeneratorUpdated(oldGenerator, newGenerator);
+    }
+
+    function _setAggregator(address newAggregator) internal {
+        address oldAggregator = aggregator;
+        aggregator = newAggregator;
+        emit AggregatorUpdated(oldAggregator, newAggregator);
+    }
+
+
 }