use super::secret::SecretKey;
use crate::{JubJubAffine, JubJubExtended};
use dusk_bytes::{Error, HexDebug, Serializable};
use dusk_jubjub::GENERATOR_EXTENDED;
#[cfg(feature = "rkyv-impl")]
use rkyv::{Archive, Deserialize, Serialize};
#[derive(Default, Copy, Clone, HexDebug)]
#[cfg_attr(
feature = "rkyv-impl",
derive(Archive, Serialize, Deserialize),
archive_attr(derive(bytecheck::CheckBytes))
)]
pub struct PublicKey(pub(crate) JubJubExtended);
impl From<&SecretKey> for PublicKey {
fn from(sk: &SecretKey) -> Self {
let public_key = GENERATOR_EXTENDED * sk.0;
PublicKey(public_key)
}
}
impl PartialEq for PublicKey {
fn eq(&self, other: &Self) -> bool {
self.0.get_u() * other.0.get_z() == other.0.get_u() * self.0.get_z()
&& self.0.get_v() * other.0.get_z()
== other.0.get_v() * self.0.get_z()
}
}
impl Eq for PublicKey {}
impl From<JubJubExtended> for PublicKey {
fn from(p: JubJubExtended) -> PublicKey {
PublicKey(p)
}
}
impl From<&JubJubExtended> for PublicKey {
fn from(p: &JubJubExtended) -> PublicKey {
PublicKey(*p)
}
}
impl AsRef<JubJubExtended> for PublicKey {
fn as_ref(&self) -> &JubJubExtended {
&self.0
}
}
impl Serializable<32> for PublicKey {
type Error = Error;
fn to_bytes(&self) -> [u8; 32] {
JubJubAffine::from(self.0).to_bytes()
}
fn from_bytes(bytes: &[u8; 32]) -> Result<Self, Error> {
let public_key: JubJubAffine =
match JubJubAffine::from_bytes(*bytes).into() {
Some(pk) => pk,
None => return Err(Error::InvalidData),
};
Ok(Self(public_key.into()))
}
}
impl PublicKey {
pub const fn from_raw_unchecked(key: JubJubExtended) -> Self {
Self(key)
}
}