Skip to content

Commit f103c8d

Browse files
committed
refactor(signer/client): re-export bls types and rename methods for consistency
1 parent 0bcd2d7 commit f103c8d

File tree

5 files changed

+15
-17
lines changed

5 files changed

+15
-17
lines changed

bin/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ pub mod prelude {
1010
load_pbs_custom_config, StartCommitModuleConfig,
1111
},
1212
pbs::{BuilderEvent, BuilderEventClient, OnBuilderApiEvent},
13-
signer::{EcdsaPublicKey, EcdsaSignature},
13+
signer::{BlsPublicKey, BlsSignature, EcdsaPublicKey, EcdsaSignature},
1414
utils::{
1515
initialize_pbs_tracing_log, initialize_tracing_log, utcnow_ms, utcnow_ns, utcnow_sec,
1616
utcnow_us,

crates/common/src/commit/client.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,14 +93,14 @@ impl SignerClient {
9393
self.request_signature(&request.into()).await
9494
}
9595

96-
pub async fn request_proxy_ecdsa_signature(
96+
pub async fn request_proxy_signature_ecdsa(
9797
&self,
9898
request: SignProxyRequest<EcdsaPublicKey>,
9999
) -> Result<EcdsaSignature, SignerClientError> {
100100
self.request_signature(&request.into()).await
101101
}
102102

103-
pub async fn request_proxy_bls_signature(
103+
pub async fn request_proxy_signature_bls(
104104
&self,
105105
request: SignProxyRequest<BlsPublicKey>,
106106
) -> Result<BlsSignature, SignerClientError> {
@@ -132,7 +132,7 @@ impl SignerClient {
132132
Ok(signed_proxy_delegation)
133133
}
134134

135-
pub async fn generate_bls_proxy_key(
135+
pub async fn generate_proxy_key_bls(
136136
&self,
137137
consensus_pubkey: BlsPublicKey,
138138
) -> Result<SignedProxyDelegation<BlsPublicKey>, SignerClientError> {
@@ -143,7 +143,7 @@ impl SignerClient {
143143
Ok(bls_signed_proxy_delegation)
144144
}
145145

146-
pub async fn generate_ecdsa_proxy_key(
146+
pub async fn generate_proxy_key_ecdsa(
147147
&self,
148148
consensus_pubkey: BlsPublicKey,
149149
) -> Result<SignedProxyDelegation<EcdsaPublicKey>, SignerClientError> {

crates/common/src/signer/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
pub mod schemes;
22

33
pub use schemes::{
4-
bls::{BlsSecretKey, BlsSigner},
4+
bls::{BlsPublicKey, BlsSecretKey, BlsSignature, BlsSigner},
55
ecdsa::{EcdsaPublicKey, EcdsaSecretKey, EcdsaSignature, EcdsaSigner},
66
};
77

crates/common/src/signer/schemes/bls.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
1-
use alloy::rpc::types::beacon::{
2-
constants::BLS_DST_SIG, BlsPublicKey as BlsPublicKeyInner, BlsSignature,
3-
};
1+
pub use alloy::rpc::types::beacon::BlsSignature;
2+
use alloy::rpc::types::beacon::{constants::BLS_DST_SIG, BlsPublicKey as BlsPublicKeyInner};
43
use blst::BLST_ERROR;
54
use derive_more::derive::{Deref, Display, From, Into, LowerHex};
65
use serde::{Deserialize, Serialize};

examples/da_commit/src/main.rs

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
use std::time::Duration;
22

3-
use alloy::rpc::types::beacon::BlsPublicKey;
43
use commit_boost::prelude::*;
54
use eyre::{OptionExt, Result};
65
use lazy_static::lazy_static;
@@ -51,19 +50,19 @@ impl DaCommitService {
5150
let pubkey = *pubkeys.consensus.first().ok_or_eyre("no key available")?;
5251
info!("Registered validator {pubkey}");
5352

54-
let proxy_delegation_bls = self.config.signer_client.generate_bls_proxy_key(pubkey).await?;
53+
let proxy_delegation_bls = self.config.signer_client.generate_proxy_key_bls(pubkey).await?;
5554
info!("Obtained a BLS proxy delegation:\n{proxy_delegation_bls}");
5655
let proxy_bls = proxy_delegation_bls.message.proxy;
5756

5857
let proxy_delegation_ecdsa =
59-
self.config.signer_client.generate_ecdsa_proxy_key(pubkey).await?;
58+
self.config.signer_client.generate_proxy_key_ecdsa(pubkey).await?;
6059
info!("Obtained an ECDSA proxy delegation:\n{proxy_delegation_ecdsa}");
6160
let proxy_ecdsa = proxy_delegation_ecdsa.message.proxy;
6261

6362
let mut data = 0;
6463

6564
loop {
66-
self.send_request(data, pubkey.into(), proxy_bls.into(), proxy_ecdsa).await?;
65+
self.send_request(data, pubkey, proxy_bls, proxy_ecdsa).await?;
6766
sleep(Duration::from_secs(self.config.extra.sleep_secs)).await;
6867
data += 1;
6968
}
@@ -78,16 +77,16 @@ impl DaCommitService {
7877
) -> Result<()> {
7978
let datagram = Datagram { data };
8079

81-
let request = SignConsensusRequest::builder(pubkey.into()).with_msg(&datagram);
80+
let request = SignConsensusRequest::builder(pubkey).with_msg(&datagram);
8281
let signature = self.config.signer_client.request_consensus_signature(request);
8382

84-
let proxy_request_bls = SignProxyRequest::builder(proxy_bls.into()).with_msg(&datagram);
83+
let proxy_request_bls = SignProxyRequest::builder(proxy_bls).with_msg(&datagram);
8584
let proxy_signature_bls =
86-
self.config.signer_client.request_proxy_bls_signature(proxy_request_bls);
85+
self.config.signer_client.request_proxy_signature_bls(proxy_request_bls);
8786

8887
let proxy_request_ecdsa = SignProxyRequest::builder(proxy_ecdsa).with_msg(&datagram);
8988
let proxy_signature_ecdsa =
90-
self.config.signer_client.request_proxy_ecdsa_signature(proxy_request_ecdsa);
89+
self.config.signer_client.request_proxy_signature_ecdsa(proxy_request_ecdsa);
9190

9291
let (signature, proxy_signature_bls, proxy_signature_ecdsa) = {
9392
let res = tokio::join!(signature, proxy_signature_bls, proxy_signature_ecdsa);

0 commit comments

Comments
 (0)