Chaosnet support

This is a beta staker program for stakers willing to go the extra mile with
monitoring, share their logs with the dev team, and allow to more carefully
monitor the bootstrapping network. As the network matures, the beta program will
be ended.

Author: pdyraga

contracts/Chaosnet.sol

@@ -0,0 +1,78 @@
+pragma solidity 0.8.9;
+
+/// @title Chaosnet
+/// @notice This is a beta staker program for stakers willing to go the extra
+/// mile with monitoring, share their logs with the dev team, and allow to more
+/// carefully monitor the bootstrapping network. As the network matures, the
+/// beta program will be ended.
+contract Chaosnet {
+  /// @notice Indicates if the chaosnet is active. The chaosnet is active
+  /// after the contract deployment and can be ended with a call to
+  /// `deactivateChaosnet()`. Once deactivated chaosnet can not be activated
+  /// again.
+  bool public isChaosnetActive;
+
+  /// @notice Indicates if the given operator is a beta operator for chaosnet.
+  mapping(address => bool) public isBetaOperator;
+
+  /// @notice Address controlling chaosnet status and beta operator addresses.
+  address public chaosnetMaestro;
+
+  event BetaOperatorsAdded(address[] operators);
+
+  event ChaosnetMaestroRoleTransferred(
+    address oldChaosnetMaestro,
+    address newChaosnetMaestro
+  );
+
+  event ChaosnetDeactivated();
+
+  constructor() {
+    _transferChaosnetMaestro(msg.sender);
+    isChaosnetActive = true;
+  }
+
+  modifier onlyChaosnetMaestro() {
+    require(msg.sender == chaosnetMaestro, "Not the chaosnet maestro");
+    _;
+  }
+
+  /// @notice Adds beta operator to chaosnet. Can be called only by the
+  /// chaosnet maestro.
+  function addBetaOperators(address[] calldata operators)
+    public
+    onlyChaosnetMaestro
+  {
+    for (uint256 i = 0; i < operators.length; i++) {
+      isBetaOperator[operators[i]] = true;
+    }
+
+    emit BetaOperatorsAdded(operators);
+  }
+
+  /// @notice Deactivates the chaosnet. Can be called only by the chaosnet
+  /// maestro. Once deactivated chaosnet can not be activated again.
+  function deactivateChaosnet() public onlyChaosnetMaestro {
+    require(isChaosnetActive, "Chaosnet is not active");
+    isChaosnetActive = false;
+    emit ChaosnetDeactivated();
+  }
+
+  /// @notice Transfers the chaosnet maestro role to another non-zero address.
+  function transferChaosnetMaestroRole(address newChaosnetMaestro)
+    public
+    onlyChaosnetMaestro
+  {
+    require(
+      newChaosnetMaestro != address(0),
+      "New chaosnet maestro must not be zero address"
+    );
+    _transferChaosnetMaestro(newChaosnetMaestro);
+  }
+
+  function _transferChaosnetMaestro(address newChaosnetMaestro) internal {
+    address oldChaosnetMaestro = chaosnetMaestro;
+    chaosnetMaestro = newChaosnetMaestro;
+    emit ChaosnetMaestroRoleTransferred(oldChaosnetMaestro, newChaosnetMaestro);
+  }
+}

contracts/SortitionPool.sol

@@ -8,12 +8,19 @@ import "@openzeppelin/contracts/access/Ownable.sol";
 import "./RNG.sol";
 import "./SortitionTree.sol";
 import "./Rewards.sol";
+import "./Chaosnet.sol";
 
 /// @title Sortition Pool
 /// @notice A logarithmic data structure used to store the pool of eligible
 /// operators weighted by their stakes. It allows to select a group of operators
 /// based on the provided pseudo-random seed.
-contract SortitionPool is SortitionTree, Rewards, Ownable, IReceiveApproval {
+contract SortitionPool is
+  SortitionTree,
+  Rewards,
+  Ownable,
+  Chaosnet,
+  IReceiveApproval
+{
   using Branch for uint256;
   using Leaf for uint256;
   using Position for uint256;
@@ -98,7 +105,9 @@ contract SortitionPool is SortitionTree, Rewards, Ownable, IReceiveApproval {
   }
 
   /// @notice Inserts an operator to the pool. Reverts if the operator is
-  /// already present.
+  /// already present. Reverts if the operator is not eligible because of their
+  /// authorized stake. Reverts if the chaosnet is active and the operator is
+  /// not a beta operator.
   /// @dev Can be called only by the contract owner.
   /// @param operator Address of the inserted operator.
   /// @param authorizedStake Inserted operator's authorized stake for the application.
@@ -110,6 +119,10 @@ contract SortitionPool is SortitionTree, Rewards, Ownable, IReceiveApproval {
     uint256 weight = getWeight(authorizedStake);
     require(weight > 0, "Operator not eligible");
 
+    if (isChaosnetActive) {
+      require(isBetaOperator[operator], "Not beta operator for chaosnet");
+    }
+
     _insertOperator(operator, weight);
     uint32 id = getOperatorID(operator);
     Rewards.updateOperatorRewards(id, uint32(weight));