use alloc::vec::Vec;
use core::fmt;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use crate::{
error::ProtoResult,
rr::{RData, RecordData, RecordDataDecodable, RecordType},
serialize::binary::{BinDecoder, BinEncodable, BinEncoder, DecodeError, Restrict},
};
#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
#[derive(Default, Debug, PartialEq, Eq, Hash, Clone)]
#[non_exhaustive]
pub struct NULL {
pub anything: Vec<u8>,
}
impl NULL {
pub const fn new() -> Self {
Self {
anything: Vec::new(),
}
}
pub fn with(anything: Vec<u8>) -> Self {
debug_assert!(!anything.is_empty());
Self { anything }
}
}
impl BinEncodable for NULL {
fn emit(&self, encoder: &mut BinEncoder<'_>) -> ProtoResult<()> {
for b in &self.anything {
encoder.emit(*b)?;
}
Ok(())
}
}
impl<'r> RecordDataDecodable<'r> for NULL {
fn read_data(decoder: &mut BinDecoder<'r>, length: Restrict<u16>) -> Result<Self, DecodeError> {
let rdata_length = length.map(|u| u as usize).unverified();
if rdata_length > 0 {
let anything = decoder.read_vec(rdata_length)?.unverified();
Ok(Self::with(anything))
} else {
Ok(Self::new())
}
}
}
impl RecordData for NULL {
fn try_borrow(data: &RData) -> Option<&Self> {
match data {
RData::NULL(csync) => Some(csync),
_ => None,
}
}
fn record_type(&self) -> RecordType {
RecordType::NULL
}
fn into_rdata(self) -> RData {
RData::NULL(self)
}
}
impl fmt::Display for NULL {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> {
f.write_str(&data_encoding::BASE64.encode(&self.anything))
}
}
#[cfg(test)]
mod tests {
#![allow(clippy::dbg_macro, clippy::print_stdout)]
#[cfg(feature = "std")]
use std::println;
use super::*;
#[test]
fn test() {
let rdata = NULL::with(vec![0, 1, 2, 3, 4, 5, 6, 7]);
let mut bytes = Vec::new();
let mut encoder: BinEncoder<'_> = BinEncoder::new(&mut bytes);
assert!(rdata.emit(&mut encoder).is_ok());
let bytes = encoder.into_bytes();
#[cfg(feature = "std")]
println!("bytes: {bytes:?}");
let mut decoder: BinDecoder<'_> = BinDecoder::new(bytes);
let restrict = Restrict::new(bytes.len() as u16);
let read_rdata = NULL::read_data(&mut decoder, restrict).expect("Decoding error");
assert_eq!(rdata, read_rdata);
}
}