Add selector getter function to module

zkemail · Jun 18, 2024 · c9aaa0d · c9aaa0d
1 parent 5fbedbf
commit c9aaa0d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/src/interfaces/IRecoveryModule.sol b/src/interfaces/IRecoveryModule.sol
@@ -5,4 +5,5 @@ interface IRecoveryModule {
     function recover(address account, bytes memory recoveryCalldata) external;
     function getTrustedRecoveryManager() external returns (address);
     function getAllowedValidators(address account) external view returns (address[] memory);
+    function getAllowedSelectors(address account) external view returns (bytes4[] memory);
 }
diff --git a/src/modules/EmailRecoveryModule.sol b/src/modules/EmailRecoveryModule.sol
@@ -212,14 +212,27 @@ contract EmailRecoveryModule is ERC7579ExecutorBase, IRecoveryModule {
         return emailRecoveryManager;
     }
 
-    function getAllowedValidators(address account) external view returns (address[] memory) {
+    function getAllowedValidators(address account) public view returns (address[] memory) {
         ValidatorList storage validatorList = validators[account];
 
         // getEntriesPaginated pageCount cannot be zero
         uint256 pageCount = validatorList.count == 0 ? 1 : validatorList.count;
-        (address[] memory array,) =
+        (address[] memory allowedValidators,) =
             validatorList.validators.getEntriesPaginated(SENTINEL, pageCount);
-        return array;
+
+        return allowedValidators;
+    }
+
+    function getAllowedSelectors(address account) external view returns (bytes4[] memory) {
+        address[] memory allowedValidators = getAllowedValidators(account);
+        uint256 allowedValidatorsLength = allowedValidators.length;
+
+        bytes4[] memory selectors = new bytes4[](allowedValidatorsLength);
+        for (uint256 i; i < allowedValidatorsLength; i++) {
+            selectors[i] = allowedSelectors[allowedValidators[i]][account];
+        }
+
+        return selectors;
     }
 
     /*//////////////////////////////////////////////////////////////////////////