custom.rs

use binary_codec::{FromBytes, ToBytes};
use serde::{Deserialize, Serialize};
use serde_with::base64::{Base64, UrlSafe};
use serde_with::formats::Unpadded;
use serde_with::serde_as;

/// Custom request/response body for sub-protocols, containing a protocol ID and raw data
#[serde_as]
#[derive(Debug, FromBytes, ToBytes, Serialize, Deserialize, PartialEq, Clone)]
pub struct CustomBody {
    /// Protocol ID to identify the sub-protocol this body belongs to
    pub protocol: u16,

    /// Raw data for the sub-protocol, which can be parsed according to the protocol ID
    /// Represented as base64url (no padding) in TOML/JSON.
    #[serde_as(as = "Base64<UrlSafe, Unpadded>")]
    pub data: Vec<u8>,
}

#[cfg(test)]
mod tests {
    use binary_codec::{BinaryDeserializer, BinarySerializer};

    use crate::packets::{request::PlabbleRequestPacket, response::PlabbleResponsePacket};

    #[test]
    fn can_serialize_and_deserialize_custom_request() {
        let packet: PlabbleRequestPacket = toml::from_str(
            r#"
            version = 1
            
            [header]
            packet_type = "Custom"
            flag1 = true
            flag2 = false
            flag3 = true
            flag4 = false

            [body]
            protocol = 42
            data = "AQIDBAU"
        "#,
        )
        .unwrap();

        let serialized = packet.to_bytes(None).unwrap();
        assert_eq!("01011110", format!("{:08b}", serialized[1]));
        assert_eq!("015e002a0102030405", hex::encode(&serialized));

        let deserialized = PlabbleRequestPacket::from_bytes(&serialized, None).unwrap();
        assert_eq!(packet, deserialized);
    }

    #[test]
    fn can_serialize_and_deserialize_custom_response() {
        let packet: PlabbleResponsePacket = toml::from_str(
            r#"
            version = 1
            
            [header]
            packet_type = "Custom"
            request_counter = 7
            flag1 = true
            flag2 = false
            flag3 = true
            flag4 = false

            [body]
            protocol = 42
            data = "AQIDBAU"
        "#,
        )
        .unwrap();

        let serialized = packet.to_bytes(None).unwrap();
        assert_eq!("01011110", format!("{:08b}", serialized[1]));
        assert_eq!("015e0007002a0102030405", hex::encode(&serialized));

        let deserialized = PlabbleResponsePacket::from_bytes(&serialized, None).unwrap();
        assert_eq!(packet, deserialized);
    }
}