use super::Prefix;
use crate::derivation::{self_addressing::SelfAddressing, DerivationCode};
use crate::error::Error;
use base64::decode_config;
use core::str::FromStr;
use serde::{Deserialize, Deserializer, Serialize, Serializer};
#[derive(Debug, PartialEq, Clone)]
pub struct SelfAddressingPrefix {
pub derivation: SelfAddressing,
pub digest: Vec<u8>,
}
impl SelfAddressingPrefix {
pub fn new(code: SelfAddressing, digest: Vec<u8>) -> Self {
Self {
derivation: code,
digest,
}
}
pub fn verify_binding(&self, sed: &[u8]) -> bool {
self.derivation.digest(sed) == self.digest
}
}
impl FromStr for SelfAddressingPrefix {
type Err = Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let code = SelfAddressing::from_str(s)?;
if s.len() == code.prefix_b64_len() {
Ok(Self::new(
code,
decode_config(&s[code.code_len()..code.prefix_b64_len()], base64::URL_SAFE)?,
))
} else {
Err(Error::SemanticError(format!(
"Incorrect Prefix Length: {}",
s
)))
}
}
}
impl Prefix for SelfAddressingPrefix {
fn derivative(&self) -> &[u8] {
&self.digest
}
fn derivation_code(&self) -> String {
self.derivation.to_str()
}
}
impl Serialize for SelfAddressingPrefix {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_str(&self.to_str())
}
}
impl<'de> Deserialize<'de> for SelfAddressingPrefix {
fn deserialize<D>(deserializer: D) -> Result<SelfAddressingPrefix, D::Error>
where
D: Deserializer<'de>,
{
let s = String::deserialize(deserializer)?;
SelfAddressingPrefix::from_str(&s).map_err(serde::de::Error::custom)
}
}
impl Default for SelfAddressingPrefix {
fn default() -> Self {
Self {
derivation: SelfAddressing::Blake3_256,
digest: vec![],
}
}
}