36 lines
1.0 KiB
Rust
36 lines
1.0 KiB
Rust
use crate::traits::Signer;
|
|
use std::sync::RwLock;
|
|
use types::{Keypair, Signature};
|
|
|
|
/// A test-only struct used to simulate a Beacon Node.
|
|
pub struct LocalSigner {
|
|
keypair: Keypair,
|
|
should_sign: RwLock<bool>,
|
|
}
|
|
|
|
impl LocalSigner {
|
|
/// Produce a new LocalSigner with signing enabled by default.
|
|
pub fn new(keypair: Keypair) -> Self {
|
|
Self {
|
|
keypair,
|
|
should_sign: RwLock::new(true),
|
|
}
|
|
}
|
|
|
|
/// If set to `false`, the service will refuse to sign all messages. Otherwise, all messages
|
|
/// will be signed.
|
|
pub fn enable_signing(&self, enabled: bool) {
|
|
*self.should_sign.write().unwrap() = enabled;
|
|
}
|
|
}
|
|
|
|
impl Signer for LocalSigner {
|
|
fn sign_block_proposal(&self, message: &[u8], domain: u64) -> Option<Signature> {
|
|
Some(Signature::new(message, domain, &self.keypair.sk))
|
|
}
|
|
|
|
fn sign_randao_reveal(&self, message: &[u8], domain: u64) -> Option<Signature> {
|
|
Some(Signature::new(message, domain, &self.keypair.sk))
|
|
}
|
|
}
|