use serde::{Deserialize, Deserializer, Serialize, Serializer};
use std::{
cmp::Ordering,
fmt::{self, Debug, Formatter},
hash::{Hash, Hasher},
};
use threshold_crypto::{PublicKeySet, SecretKeyShare};
#[derive(Clone)]
pub struct DkgResult {
pub public_key_set: PublicKeySet,
pub secret_key_share: Option<SecretKeyShare>,
}
impl DkgResult {
pub fn new(public_key_set: PublicKeySet, secret_key_share: Option<SecretKeyShare>) -> Self {
Self {
public_key_set,
secret_key_share,
}
}
}
impl Debug for DkgResult {
fn fmt(&self, formatter: &mut Formatter) -> fmt::Result {
write!(
formatter,
"DkgResult({:?}, {})",
self.public_key_set,
self.secret_key_share.is_some()
)
}
}
#[derive(Clone)]
pub struct DkgResultWrapper(pub DkgResult);
impl Debug for DkgResultWrapper {
fn fmt(&self, formatter: &mut Formatter) -> fmt::Result {
self.0.fmt(formatter)
}
}
impl DkgResultWrapper {
fn comparison_value(&self) -> &PublicKeySet {
&self.0.public_key_set
}
}
impl PartialEq for DkgResultWrapper {
fn eq(&self, rhs: &Self) -> bool {
self.comparison_value().eq(&rhs.comparison_value())
}
}
impl Eq for DkgResultWrapper {}
impl Hash for DkgResultWrapper {
fn hash<H: Hasher>(&self, state: &mut H) {
self.comparison_value().hash(state)
}
}
impl PartialOrd for DkgResultWrapper {
fn partial_cmp(&self, rhs: &Self) -> Option<Ordering> {
self.comparison_value().partial_cmp(&rhs.comparison_value())
}
}
impl Ord for DkgResultWrapper {
fn cmp(&self, rhs: &Self) -> Ordering {
self.comparison_value().cmp(&rhs.comparison_value())
}
}
impl Serialize for DkgResultWrapper {
fn serialize<S: Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
self.0.public_key_set.serialize(s)
}
}
impl<'a> Deserialize<'a> for DkgResultWrapper {
fn deserialize<D: Deserializer<'a>>(deserializer: D) -> Result<Self, D::Error> {
PublicKeySet::deserialize(deserializer).map(|public_key_set| {
DkgResultWrapper(DkgResult {
public_key_set,
secret_key_share: None,
})
})
}
}