feat: optimise PasskeyBinder

zkLinkProtocol · Sep 26, 2024 · a60f60b · a60f60b
1 parent 07fb30d
commit a60f60b
Showing 1 changed file with 10 additions and 31 deletions.
diff --git a/system-contracts/contracts/PasskeyBinder.sol b/system-contracts/contracts/PasskeyBinder.sol
@@ -20,15 +20,13 @@ contract PasskeyBinder is IPasskeyBinder {
         uint256 y;
     }
 
-    // Deprecated
     struct AuthorizedKey {
         address owner;
         P256PublicKey publicKey;
     }
 
     mapping(address account => EnumerableSet.Bytes32Set credentialIdHashSet) private accountToCredentialIdHashSet;
-    mapping(bytes32 credentialIdHash => AuthorizedKey) private authorizedKeys; // Deprecated
-    mapping(bytes32 credentialIdHash => P256PublicKey) private authorizedPublicKeys;
+    mapping(bytes32 credentialIdHash => AuthorizedKey) private authorizedKeys;
 
     /// @dev Event emitted when a P256 key is added
     event AddedP256PublicKey(bytes32 indexed credentialIdHash, address indexed owner, uint256 x, uint256 y);
@@ -43,7 +41,7 @@ contract PasskeyBinder is IPasskeyBinder {
     /// @dev Error emitted when a P256 key is not stored and attempted to be removed
     error KeyDoesNotExist(bytes32 credentialIdHash);
     /// @dev Error emitted when a P256 key is not owned by the caller
-    error DoesNotOwner(bytes32 credentialIdHash);
+    error DoesNotOwnershipOfRights(bytes32 credentialIdHash);
     /// @dev Error emitted when a P256 key is attempted to be add by not EOA
     error DoesNotEOA();
 
@@ -65,9 +63,9 @@ contract PasskeyBinder is IPasskeyBinder {
 
         if (_credentialIdHash == bytes32(0)) revert InvalidCredentialIdHash();
 
-        P256PublicKey memory publicKey = authorizedPublicKeys[_credentialIdHash];
-        if (publicKey.x == 0 && publicKey.y == 0) {
-            authorizedPublicKeys[_credentialIdHash] = P256PublicKey({x: _x, y: _y});
+        address authorizedKeyOwner = authorizedKeys[_credentialIdHash].owner;
+        if (authorizedKeyOwner == address(0)) {
+            authorizedKeys[_credentialIdHash] = AuthorizedKey(sender, P256PublicKey(_x, _y));
         }
 
         accountToCredentialIdHashSet[sender].add(_credentialIdHash);
@@ -83,33 +81,14 @@ contract PasskeyBinder is IPasskeyBinder {
         address sender = msg.sender;
         EnumerableSet.Bytes32Set storage credentialIdHashSet = accountToCredentialIdHashSet[sender];
 
-        bool isOwner = credentialIdHashSet.contains(_credentialIdHash);
-        if (!isOwner) revert DoesNotOwner(_credentialIdHash);
+        bool ownershipOfRight = credentialIdHashSet.contains(_credentialIdHash);
+        if (!ownershipOfRight) revert DoesNotOwnershipOfRights(_credentialIdHash);
 
         accountToCredentialIdHashSet[sender].remove(_credentialIdHash);
 
         emit RemovedP256PublicKey(_credentialIdHash, sender);
     }
 
-    /**
-     * @notice Deprecate owner-specified P256 public key
-     * @param _credentialIdHash The ID Hash of the credential to deprecate
-     */
-    function deprecateP256PublicKey(bytes32 _credentialIdHash) external {
-        address sender = msg.sender;
-        EnumerableSet.Bytes32Set storage credentialIdHashSet = accountToCredentialIdHashSet[sender];
-
-        bool isOwner = credentialIdHashSet.contains(_credentialIdHash);
-        if (!isOwner) revert DoesNotOwner(_credentialIdHash);
-        P256PublicKey memory publicKey = authorizedPublicKeys[_credentialIdHash];
-        if (publicKey.x == 0 && publicKey.y == 0) revert KeyDoesNotExist(_credentialIdHash);
-
-        delete authorizedPublicKeys[_credentialIdHash];
-        accountToCredentialIdHashSet[sender].remove(_credentialIdHash);
-
-        emit DeprecatedP256PublicKey(_credentialIdHash);
-    }
-
     /**
      * @notice Returns authorized key infos by credential id hash
      * @param _account The account to get the authorized key
@@ -122,10 +101,10 @@ contract PasskeyBinder is IPasskeyBinder {
         bytes32 _credentialIdHash
     ) external view returns (uint256 x, uint256 y) {
         EnumerableSet.Bytes32Set storage credentialIdHashSet = accountToCredentialIdHashSet[_account];
-        P256PublicKey memory publicKey = authorizedPublicKeys[_credentialIdHash];
-        bool isOwner = credentialIdHashSet.contains(_credentialIdHash);
+        P256PublicKey memory publicKey = authorizedKeys[_credentialIdHash].publicKey;
+        bool ownershipOfRight = credentialIdHashSet.contains(_credentialIdHash);
 
-        if (isOwner) {
+        if (ownershipOfRight) {
             x = publicKey.x;
             y = publicKey.y;
         } else {