feat: track total members and sets (#649)

* feat: track total members and sets

* test: track total members and sets

* fix: review changes

* refactor: review changes

* test: register multiple sets

* test: deregister multiple sets

* test: fuzz set creation

Author: 0xClandestine

src/contracts/core/AVSDirectory.sol

@@ -73,6 +73,8 @@ contract AVSDirectory is
             isOperatorSet[msg.sender][operatorSetIds[i]] = true;
             emit OperatorSetCreated(OperatorSet({avs: msg.sender, operatorSetId: operatorSetIds[i]}));
         }
+        operatorSetCount += operatorSetIds.length;
+        avsOperatorSetCount[msg.sender] += operatorSetIds.length;
     }
 
     /**
@@ -380,6 +382,8 @@ contract AVSDirectory is
                 "AVSDirectory._registerOperatorToOperatorSets: operator already registered to operator set"
             );
 
+            ++operatorSetMemberCount[avs][operatorSetIds[i]];
+
             // Mutate `isMember` to `true`.
             isMember[avs][operator][operatorSetIds[i]] = true;
 
@@ -403,6 +407,8 @@ contract AVSDirectory is
                 "AVSDirectory._deregisterOperatorFromOperatorSet: operator not registered for operator set"
             );
 
+            --operatorSetMemberCount[avs][operatorSetIds[i]];
+
             // Mutate `isMember` to `false`.
             isMember[avs][operator][operatorSetIds[i]] = false;
 

src/contracts/core/AVSDirectoryStorage.sol

@@ -34,18 +34,27 @@ abstract contract AVSDirectoryStorage is IAVSDirectory {
     /// @notice Mapping: avs => operator => OperatorAVSRegistrationStatus struct
     mapping(address => mapping(address => OperatorAVSRegistrationStatus)) public avsOperatorStatus;
 
-    /// @notice Mapping: operator => salt => whether the salt has been used
+    /// @notice Mapping: operator => salt => Whether the salt has been used or not.
     mapping(address => mapping(bytes32 => bool)) public operatorSaltIsSpent;
 
-    /// @notice Mapping: avs => whether it is a an operator set AVS
+    /// @notice Mapping: avs => Whether it is a an operator set AVS or not.
     mapping(address => bool) public isOperatorSetAVS;
 
-    /// @notice Mapping: avs => operatorSetId => whether the operatorSet exists
+    /// @notice Mapping: avs => operatorSetId => Whether or not an operator set is valid.
     mapping(address => mapping(uint32 => bool)) public isOperatorSet;
 
-    /// @notice Mapping: avs = operator => operatorSetId => whether the operator is a member of the operatorSet
+    /// @notice Mapping: avs = operator => operatorSetId => Whether or not an operator is a member of an operator set.
     mapping(address => mapping(address => mapping(uint32 => bool))) public isMember;
 
+    /// @notice Mapping: avs => operatorSetId => Total operators within the given operator set.
+    mapping(address => mapping(uint32 => uint256)) public operatorSetMemberCount;
+
+    /// @notice Mapping: avs => Total amount of operator sets created for a given AVS.
+    mapping(address => uint256) public avsOperatorSetCount;
+
+    /// @notice Total amount of operators sets created.
+    uint256 public operatorSetCount;
+
     constructor(IDelegationManager _delegation) {
         delegation = _delegation;
     }
@@ -55,5 +64,5 @@ abstract contract AVSDirectoryStorage is IAVSDirectory {
      * variables without shifting down storage in the inheritance chain.
      * See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
      */
-    uint256[43] private __gap;
+    uint256[40] private __gap;
 }