contracts: register message key when register candidate

Author: txhsl

contracts/solidity/Governance.sol

@@ -126,7 +126,10 @@ contract Governance is IGovernance, ReentrancyGuard, GovProxyUpgradeable {
         return candidateList.values();
     }
 
-    function registerCandidate(uint shareRate) external payable {
+    function registerCandidate(
+        uint shareRate,
+        bytes calldata pubkey
+    ) external payable {
         if (tx.origin != msg.sender) revert Errors.OnlyEOA();
         if (msg.value != registerFee) revert Errors.InsufficientValue();
         if (shareRate > 1000) revert Errors.InvalidShareRate();
@@ -140,6 +143,9 @@ contract Governance is IGovernance, ReentrancyGuard, GovProxyUpgradeable {
         // record share rate and balance
         shareRateOf[msg.sender] = shareRate;
         candidateBalanceOf[msg.sender] = msg.value;
+
+        // register message key
+        IKeyManagement(KEY_MANAGEMENT).registerMessageKey(msg.sender, pubkey);
     }
 
     function exitCandidate() external {

contracts/solidity/KeyManagement.sol

@@ -69,7 +69,7 @@ contract KeyManagement is GovProxyUpgradeable, IKeyManagement {
             revert Errors.SideCallNotAllowed();
         if (tx.origin != candidate) revert Errors.OnlyEOA();
         if (pubkey.length != 65) revert Errors.InvalidMessageKey();
-        messagePubkeys[msg.sender] = pubkey;
+        messagePubkeys[candidate] = pubkey;
     }
 
     function reshare(bytes calldata pvss, bytes[] calldata messages) external {

contracts/solidity/interfaces/IGovernance.sol

@@ -11,7 +11,10 @@ interface IGovernance {
     event Persist(address[] validators);
 
     // register to be a candidate with gas
-    function registerCandidate(uint shareRate) external payable;
+    function registerCandidate(
+        uint shareRate,
+        bytes calldata pubkey
+    ) external payable;
 
     // exit candidates and wait for withdraw
     function exitCandidate() external;

contracts/solidity/test/MockContract.sol

@@ -6,8 +6,9 @@ import {IGovernance} from "../interfaces/IGovernance.sol";
 contract MockContract {
     function call_registerCandidate(
         IGovernance governanceAddr,
-        uint shareRate
+        uint shareRate,
+        bytes calldata pubkey
     ) public payable {
-        governanceAddr.registerCandidate{value: msg.value}(shareRate);
+        governanceAddr.registerCandidate{value: msg.value}(shareRate, pubkey);
     }
 }