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::mqtt_string::MqttString;
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;
#[derive(PartialEq, Eq, Builder, Clone, Getters, CopyGetters)]
#[builder(no_std, derive(Debug), pattern = "owned", setter(into), build_fn(skip))]
pub struct GenericUnsubscribe<PacketIdType>
where
PacketIdType: IsPacketId,
{
#[builder(private)]
fixed_header: [u8; 1],
#[builder(private)]
remaining_length: VariableByteInteger,
#[builder(private)]
packet_id_buf: PacketIdType::Buffer,
#[builder(default = "Vec::new()", setter(custom))]
entry_bufs: Vec<MqttString>,
}
pub type Unsubscribe = GenericUnsubscribe<u16>;
impl<PacketIdType> GenericUnsubscribe<PacketIdType>
where
PacketIdType: IsPacketId,
{
pub fn builder() -> GenericUnsubscribeBuilder<PacketIdType> {
GenericUnsubscribeBuilder::<PacketIdType>::default()
}
pub fn packet_type() -> PacketType {
PacketType::Unsubscribe
}
pub fn packet_id(&self) -> PacketIdType {
PacketIdType::from_buffer(self.packet_id_buf.as_ref())
}
pub fn entries(&self) -> &Vec<MqttString> {
&self.entry_bufs
}
pub fn parse(data: &[u8]) -> Result<(Self, usize), MqttError> {
let mut cursor = 0;
let buffer_size = mem::size_of::<<PacketIdType as IsPacketId>::Buffer>();
if data.len() < buffer_size {
return Err(MqttError::MalformedPacket);
}
let packet_id = PacketIdType::from_buffer(&data[0..buffer_size]);
let packet_id_buf = packet_id.to_buffer();
cursor += buffer_size;
let mut entries = Vec::new();
while cursor < data.len() {
let (mqtt_string, consumed) = MqttString::decode(&data[cursor..])?;
entries.push(mqtt_string);
cursor += consumed;
}
if entries.is_empty() {
return Err(MqttError::ProtocolError);
}
let remaining_size = buffer_size + entries.iter().map(|e| e.size()).sum::<usize>();
let remaining_length = VariableByteInteger::from_u32(remaining_size as u32).unwrap();
let unsubscribe = GenericUnsubscribe {
fixed_header: [FixedHeader::Unsubscribe as u8],
remaining_length,
packet_id_buf,
entry_bufs: entries,
};
Ok((unsubscribe, cursor))
}
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()));
for entry in &self.entry_bufs {
bufs.extend(entry.to_buffers());
}
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());
for entry in &self.entry_bufs {
buf.append(&mut entry.to_continuous_buffer());
}
buf
}
}
impl<PacketIdType> GenericUnsubscribeBuilder<PacketIdType>
where
PacketIdType: IsPacketId,
{
pub fn packet_id(mut self, id: PacketIdType) -> Self {
self.packet_id_buf = Some(id.to_buffer());
self
}
pub fn entries<I, T>(mut self, entries: I) -> Result<Self, MqttError>
where
I: IntoIterator<Item = T>,
T: TryInto<MqttString>,
T::Error: Into<MqttError>,
{
let mqtt_strings: Result<Vec<_>, _> = entries
.into_iter()
.map(|item| item.try_into().map_err(Into::into))
.collect();
self.entry_bufs = Some(mqtt_strings?);
Ok(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);
}
if self.entry_bufs.as_ref().map_or(true, |e| e.is_empty()) {
return Err(MqttError::ProtocolError);
}
Ok(())
}
pub fn build(self) -> Result<GenericUnsubscribe<PacketIdType>, MqttError> {
self.validate()?;
let packet_id_buf = self.packet_id_buf.unwrap();
let entries = self.entry_bufs.unwrap_or_default();
let packet_id_size = mem::size_of::<<PacketIdType as IsPacketId>::Buffer>();
let entries_size = entries.iter().map(|e| e.size()).sum::<usize>();
let remaining = packet_id_size + entries_size;
let remaining_length = VariableByteInteger::from_u32(remaining as u32).unwrap();
Ok(GenericUnsubscribe {
fixed_header: [FixedHeader::Unsubscribe as u8],
remaining_length,
packet_id_buf,
entry_bufs: entries,
})
}
}
impl<PacketIdType> Serialize for GenericUnsubscribe<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.entry_bufs.is_empty() {
field_count += 1;
}
let mut state = serializer.serialize_struct("Unsubscribe", field_count)?;
state.serialize_field("type", "unsubscribe")?;
state.serialize_field("packet_id", &self.packet_id())?;
if !self.entry_bufs.is_empty() {
state.serialize_field("entries", &self.entry_bufs)?;
}
state.end()
}
}
impl<PacketIdType> fmt::Display for GenericUnsubscribe<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 GenericUnsubscribe<PacketIdType>
where
PacketIdType: IsPacketId + Serialize,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}
impl<PacketIdType> GenericPacketTrait for GenericUnsubscribe<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 GenericUnsubscribe<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)
}
}