use alloc::vec::Vec;
use core::fmt;
use core::mem;
use derive_builder::Builder;
#[cfg(feature = "std")]
use std::io::IoSlice;
use serde::ser::{SerializeStruct, Serializer};
use serde::Serialize;
use getset::{CopyGetters, Getters};
use crate::mqtt::packet::packet_type::{FixedHeader, PacketType};
use crate::mqtt::packet::variable_byte_integer::VariableByteInteger;
use crate::mqtt::packet::GenericPacketDisplay;
use crate::mqtt::packet::GenericPacketTrait;
use crate::mqtt::packet::IsPacketId;
use crate::mqtt::result_code::MqttError;
use crate::mqtt::result_code::PubackReasonCode;
#[derive(PartialEq, Eq, Builder, Clone, Getters, CopyGetters)]
#[builder(no_std, derive(Debug), pattern = "owned", setter(into), build_fn(skip))]
pub struct GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId,
{
#[builder(private)]
fixed_header: [u8; 1],
#[builder(private)]
remaining_length: VariableByteInteger,
#[builder(private)]
packet_id_buf: PacketIdType::Buffer,
#[builder(private)]
reason_code_buf: Option<[u8; 1]>,
}
pub type Puback = GenericPuback<u16>;
impl<PacketIdType> GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId,
{
pub fn builder() -> GenericPubackBuilder<PacketIdType> {
GenericPubackBuilder::<PacketIdType>::default()
}
pub const fn packet_type() -> PacketType {
PacketType::Puback
}
pub fn packet_id(&self) -> PacketIdType {
PacketIdType::from_buffer(self.packet_id_buf.as_ref())
}
pub fn reason_code(&self) -> Option<PubackReasonCode> {
self.reason_code_buf
.as_ref()
.and_then(|buf| PubackReasonCode::try_from(buf[0]).ok())
}
pub fn size(&self) -> usize {
1 + self.remaining_length.size() + self.remaining_length.to_u32() as usize
}
#[cfg(feature = "std")]
pub fn to_buffers(&self) -> Vec<IoSlice<'_>> {
let mut bufs = Vec::new();
bufs.push(IoSlice::new(&self.fixed_header));
bufs.push(IoSlice::new(self.remaining_length.as_bytes()));
bufs.push(IoSlice::new(self.packet_id_buf.as_ref()));
if let Some(buf) = &self.reason_code_buf {
bufs.push(IoSlice::new(buf));
}
bufs
}
pub fn to_continuous_buffer(&self) -> Vec<u8> {
let mut buf = Vec::new();
buf.extend_from_slice(&self.fixed_header);
buf.extend_from_slice(self.remaining_length.as_bytes());
buf.extend_from_slice(self.packet_id_buf.as_ref());
if let Some(rc_buf) = &self.reason_code_buf {
buf.extend_from_slice(rc_buf);
}
buf
}
pub fn parse(data: &[u8]) -> Result<(Self, usize), MqttError> {
let mut cursor = 0;
let buffer_size = core::mem::size_of::<<PacketIdType as IsPacketId>::Buffer>();
if data.len() < buffer_size {
return Err(MqttError::MalformedPacket);
}
let all_zeros = &data[0..buffer_size].iter().all(|&b| b == 0);
if *all_zeros {
return Err(MqttError::MalformedPacket);
}
let mut packet_id_buf = PacketIdType::Buffer::default();
packet_id_buf
.as_mut()
.copy_from_slice(&data[0..buffer_size]);
cursor += buffer_size;
let reason_code_buf = if cursor < data.len() {
let rc = data[cursor];
let _ = PubackReasonCode::try_from(rc).map_err(|_| MqttError::MalformedPacket)?;
cursor += 1;
Some([rc])
} else {
None
};
let remaining_size = buffer_size + reason_code_buf.as_ref().map_or(0, |_| 1);
let puback = GenericPuback {
fixed_header: [FixedHeader::Puback.as_u8()],
remaining_length: VariableByteInteger::from_u32(remaining_size as u32).unwrap(),
packet_id_buf,
reason_code_buf,
};
Ok((puback, cursor))
}
}
impl<PacketIdType> GenericPubackBuilder<PacketIdType>
where
PacketIdType: IsPacketId,
{
pub fn packet_id(mut self, id: PacketIdType) -> Self {
self.packet_id_buf = Some(id.to_buffer());
self
}
pub fn reason_code(mut self, rc: PubackReasonCode) -> Self {
self.reason_code_buf = Some(Some([rc as u8]));
self
}
fn validate(&self) -> Result<(), MqttError> {
if self.packet_id_buf.is_none() {
return Err(MqttError::MalformedPacket);
}
let packet_id_bytes = self.packet_id_buf.as_ref().unwrap().as_ref();
let all_zeros = packet_id_bytes.iter().all(|&b| b == 0);
if all_zeros {
return Err(MqttError::MalformedPacket);
}
Ok(())
}
pub fn build(self) -> Result<GenericPuback<PacketIdType>, MqttError> {
self.validate()?;
let packet_id_buf = self.packet_id_buf.unwrap();
let reason_code_buf = self.reason_code_buf.flatten();
let mut remaining = mem::size_of::<PacketIdType>();
if reason_code_buf.is_some() {
remaining += 1;
}
let remaining_length = VariableByteInteger::from_u32(remaining as u32).unwrap();
Ok(GenericPuback {
fixed_header: [FixedHeader::Puback.as_u8()],
remaining_length,
packet_id_buf,
reason_code_buf,
})
}
}
impl<PacketIdType> Serialize for GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId + Serialize,
{
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let mut field_count = 2;
if self.reason_code_buf.is_some() {
field_count += 1; }
let mut state = serializer.serialize_struct("puback", field_count)?;
state.serialize_field("type", PacketType::Puback.as_str())?;
state.serialize_field("packet_id", &self.packet_id())?;
if self.reason_code_buf.is_some() {
state.serialize_field("reason_code", &self.reason_code())?;
}
state.end()
}
}
impl<PacketIdType> fmt::Display for GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId + Serialize,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match serde_json::to_string(self) {
Ok(json) => write!(f, "{json}"),
Err(e) => write!(f, "{{\"error\": \"{e}\"}}"),
}
}
}
impl<PacketIdType> fmt::Debug for GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId + Serialize,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}
impl<PacketIdType> GenericPacketTrait for GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId,
{
fn size(&self) -> usize {
self.size()
}
#[cfg(feature = "std")]
fn to_buffers(&self) -> Vec<IoSlice<'_>> {
self.to_buffers()
}
fn to_continuous_buffer(&self) -> Vec<u8> {
self.to_continuous_buffer()
}
}
impl<PacketIdType> GenericPacketDisplay for GenericPuback<PacketIdType>
where
PacketIdType: IsPacketId + Serialize,
{
fn fmt_debug(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
core::fmt::Debug::fmt(self, f)
}
fn fmt_display(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
core::fmt::Display::fmt(self, f)
}
}