Merge pull request #16 from symbioticfi/new-delegator-support

Support new delegator type
symbioticfi · Jan 7, 2025 · a002fa9 · a002fa9
2 parents a880d2f + 6833457
commit a002fa9
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 237 deletions.
diff --git a/src/extensions/operators/ApprovalRegisterOperators.sol b/src/extensions/operators/ApprovalRegisterOperators.sol
diff --git a/src/extensions/operators/ForcePauseApprovalRegisterOperators.sol b/src/extensions/operators/ForcePauseApprovalRegisterOperators.sol
diff --git a/src/interfaces/extensions/operators/IApprovalRegisterOperators.sol b/src/interfaces/extensions/operators/IApprovalRegisterOperators.sol
diff --git a/src/managers/VaultManager.sol b/src/managers/VaultManager.sol
@@ -65,7 +65,8 @@ abstract contract VaultManager is NetworkStorage, SlashingWindowStorage, Capture
     enum DelegatorType {
         FULL_RESTAKE,
         NETWORK_RESTAKE,
-        OPERATOR_SPECIFIC
+        OPERATOR_SPECIFIC,
+        OPERATOR_NETWORK_SPECIFIC
     }
 
     // keccak256(abi.encode(uint256(keccak256("symbiotic.storage.VaultManager")) - 1)) & ~bytes32(uint256(0xff))
@@ -732,9 +733,12 @@ abstract contract VaultManager is NetworkStorage, SlashingWindowStorage, Capture
 
     function _validateOperatorVault(address operator, address vault) internal view {
         address delegator = IVault(vault).delegator();
+        uint64 delegatorType = IEntity(delegator).TYPE();
         if (
-            IEntity(delegator).TYPE() != uint64(DelegatorType.OPERATOR_SPECIFIC)
-                || IOperatorSpecificDelegator(delegator).operator() != operator
+            (
+                delegatorType != uint64(DelegatorType.OPERATOR_SPECIFIC)
+                    && delegatorType != uint64(DelegatorType.OPERATOR_NETWORK_SPECIFIC)
+            ) || IOperatorSpecificDelegator(delegator).operator() != operator
         ) {
             revert NotOperatorSpecificVault();
         }