pub mod error;
pub mod messages;
pub mod storage;
use self::storage::{ChunkAddress, DbcAddress, RegisterAddress};
use libp2p::{
kad::{kbucket::Distance, KBucketKey as Key},
PeerId,
};
use serde::{Deserialize, Serialize};
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum NetworkAddress {
PeerId(Vec<u8>),
ChunkAddress(ChunkAddress),
DbcAddress(DbcAddress),
RegisterAddress(RegisterAddress),
}
impl NetworkAddress {
pub fn from_chunk_address(chunk_address: ChunkAddress) -> Self {
NetworkAddress::ChunkAddress(chunk_address)
}
pub fn from_dbc_address(dbc_address: DbcAddress) -> Self {
NetworkAddress::DbcAddress(dbc_address)
}
pub fn from_register_address(register_address: RegisterAddress) -> Self {
NetworkAddress::RegisterAddress(register_address)
}
pub fn from_peer(peer_id: PeerId) -> Self {
NetworkAddress::PeerId(peer_id.to_bytes())
}
pub fn as_bytes(&self) -> Vec<u8> {
match self {
NetworkAddress::PeerId(bytes) => bytes.to_vec(),
NetworkAddress::ChunkAddress(chunk_address) => chunk_address.name().0.to_vec(),
NetworkAddress::DbcAddress(dbc_address) => dbc_address.name().0.to_vec(),
NetworkAddress::RegisterAddress(register_address) => register_address.id().0.to_vec(),
}
}
pub fn as_kbucket_key(&self) -> Key<Vec<u8>> {
Key::new(self.as_bytes())
}
pub fn distance(&self, other: &NetworkAddress) -> Distance {
self.as_kbucket_key().distance(&other.as_kbucket_key())
}
}
impl std::fmt::Debug for NetworkAddress {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let name_str = match self {
NetworkAddress::PeerId(_) => "NetworkAddress::PeerId(".to_string(),
NetworkAddress::ChunkAddress(chunk_address) => {
format!("NetworkAddress::ChunkAddress({:?} - ", chunk_address.name())
}
NetworkAddress::DbcAddress(dbc_address) => {
format!("NetworkAddress::DbcAddress({:?} - ", dbc_address.name())
}
NetworkAddress::RegisterAddress(register_address) => format!(
"NetworkAddress::RegisterAddress({:?} - ",
register_address.id()
),
};
write!(f, "{name_str}{:?})", self.as_bytes())
}
}