request_header.rs

use std::cell::RefCell;

use binary_codec::{FromBytes, ToBytes};
use serde::{Deserialize, Serialize};

use crate::{core::BucketId, packets::header::type_and_flags::RequestPacketType};

/// Plabble Packet request header
#[derive(FromBytes, ToBytes, Serialize, Deserialize, PartialEq, Debug, Clone)]
pub struct PlabbleRequestHeader {
    /// Internal packet type field for binary serialization/deserialization
    #[serde(skip_serializing, skip_deserializing)]
    #[bits = 4]
    #[variant_for("packet_type")]
    _type: RefCell<u8>,

    /// Packet type (derived from `_type`)
    #[variant_by = "packet_type"]
    #[serde(flatten)]
    pub packet_type: RequestPacketType,

    /// ID of the bucket, if needed by the type
    /// The following types require the bucket ID in the header:
    /// Get, Stream, Patch, Put, Delete, Subscribe
    #[toggled_by_variant = "packet_type=2|3|5|6|7|8"]
    pub id: Option<BucketId>,
}

impl PlabbleRequestHeader {
    /// Create new packet header for specific request packet type
    pub fn new(packet_type: RequestPacketType, id: Option<BucketId>) -> Self {
        Self {
            _type: RefCell::new(packet_type.get_discriminator()),
            packet_type,
            id,
        }
    }

    /// Indicates if SESSION packet
    pub fn is_session_packet(&self) -> bool {
        matches!(self.packet_type, RequestPacketType::Session { .. })
    }

    pub fn preprocess(&self) {
        self._type.replace(self.packet_type.get_discriminator());
    }
}