use crate::error::Error;
pub fn apply<T: Bytes>(v: &T, (min, max): (usize, usize)) -> Result<(), Error> {
v.validate_num_bytes(min, max)
}
pub trait Bytes {
fn validate_num_bytes(&self, min: usize, max: usize) -> Result<(), Error>;
}
impl<T: HasBytes> Bytes for T {
fn validate_num_bytes(&self, min: usize, max: usize) -> Result<(), Error> {
super::check_len(self.num_bytes(), min, max)
}
}
impl<T: Bytes> Bytes for Option<T> {
fn validate_num_bytes(&self, min: usize, max: usize) -> Result<(), Error> {
match self {
Some(v) => v.validate_num_bytes(min, max),
None => Ok(()),
}
}
}
pub trait HasBytes {
fn num_bytes(&self) -> usize;
}
macro_rules! impl_via_len {
($(in<$lifetime:lifetime>)? $T:ty) => {
impl<$($lifetime)?> HasBytes for $T {
fn num_bytes(&self) -> usize {
self.len()
}
}
};
}
impl_via_len!(std::string::String);
impl_via_len!(in<'a> &'a std::string::String);
impl_via_len!(in<'a> &'a str);
impl_via_len!(in<'a> std::borrow::Cow<'a, str>);
impl_via_len!(std::rc::Rc<str>);
impl_via_len!(std::sync::Arc<str>);
impl_via_len!(std::boxed::Box<str>);
impl_via_len!(in<'a> &'a [u8]);
impl_via_len!(std::rc::Rc<[u8]>);
impl_via_len!(std::sync::Arc<[u8]>);
impl_via_len!(std::boxed::Box<[u8]>);
impl_via_len!(std::vec::Vec<u8>);
impl<const N: usize> HasBytes for [u8; N] {
fn num_bytes(&self) -> usize {
self.len()
}
}