use crate::hash::CryptoHash;
use crate::types::{AccountId, EpochId};
use borsh::{BorshDeserialize, BorshSerialize};
use near_crypto::{KeyType, PublicKey, SecretKey, Signature};
use serde::{Deserialize, Serialize};
use std::fmt;
use std::hash::Hash;
use std::sync::Arc;
#[cfg_attr(feature = "deepsize_feature", derive(deepsize::DeepSizeOf))]
#[derive(
BorshSerialize,
BorshDeserialize,
Clone,
PartialEq,
Eq,
PartialOrd,
Ord,
Hash,
Serialize,
Deserialize,
)]
pub struct PeerId(Arc<PublicKey>);
impl PeerId {
pub fn new(key: PublicKey) -> Self {
Self(Arc::new(key))
}
pub fn public_key(&self) -> &PublicKey {
&self.0
}
}
impl PeerId {
pub fn random() -> Self {
PeerId::new(SecretKey::from_random(KeyType::ED25519).public_key())
}
}
impl fmt::Display for PeerId {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.0)
}
}
impl fmt::Debug for PeerId {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.0)
}
}
#[cfg_attr(feature = "deepsize_feature", derive(deepsize::DeepSizeOf))]
#[derive(BorshSerialize, BorshDeserialize, PartialEq, Eq, Clone, Debug, Hash)]
pub struct AnnounceAccount {
pub account_id: AccountId,
pub peer_id: PeerId,
pub epoch_id: EpochId,
pub signature: Signature,
}
impl AnnounceAccount {
pub fn build_header_hash(
account_id: &AccountId,
peer_id: &PeerId,
epoch_id: &EpochId,
) -> CryptoHash {
CryptoHash::hash_borsh(&(account_id, peer_id, epoch_id))
}
pub fn hash(&self) -> CryptoHash {
AnnounceAccount::build_header_hash(&self.account_id, &self.peer_id, &self.epoch_id)
}
}