use ant_protocol::storage::AddressParseError;
use std::hash::Hash;
use xor_name::XorName;
pub mod private;
pub mod public;
mod helpers;
mod stream;
pub use stream::DataStream;
#[derive(Clone, Copy, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub struct DataAddress(XorName);
impl DataAddress {
pub fn new(xor_name: XorName) -> Self {
Self(xor_name)
}
pub fn xorname(&self) -> &XorName {
&self.0
}
pub fn to_hex(&self) -> String {
hex::encode(self.0)
}
pub fn from_hex(hex: &str) -> Result<Self, AddressParseError> {
let bytes = hex::decode(hex)?;
let xor = XorName(
bytes
.try_into()
.map_err(|_| AddressParseError::InvalidLength)?,
);
Ok(Self(xor))
}
}
impl std::fmt::Display for DataAddress {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", &self.to_hex())
}
}
impl serde::Serialize for DataAddress {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
self.0.serialize(serializer)
}
}
impl<'de> serde::Deserialize<'de> for DataAddress {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let xor_name = XorName::deserialize(deserializer)?;
Ok(Self(xor_name))
}
}
impl std::fmt::Debug for DataAddress {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", &self.to_hex())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_serialize_deserialize() {
let name = XorName::random(&mut rand::thread_rng());
let data_address = DataAddress::new(name);
let name_serialized = rmp_serde::to_vec_named(&name).unwrap();
let serialized = rmp_serde::to_vec_named(&data_address).unwrap();
assert_eq!(name_serialized, serialized);
let deserialized: DataAddress = rmp_serde::from_slice(&serialized).unwrap();
assert_eq!(data_address, deserialized);
}
}