scte104 0.3.1

ANSI/SCTE 104 2023 — Automation System to Compression System Communications API (single/multiple operation messages + DPI operations).
Documentation
//! insert_avail_descriptor_request_data() — ANSI/SCTE 104 2023 §9.8.4, Table 9-26 (opID 0x010A).
//!
//! Supplemental usage. Adds an avail_descriptor to the resulting SCTE 35 section.

use alloc::vec::Vec;

use crate::error::{Error, Result};
use crate::traits::OperationDef;
use broadcast_common::{Parse, Serialize};

/// `opID` for insert_avail_descriptor_request (§8.3, Table 8-4).
pub const OP_ID: u16 = 0x010A;

/// insert_avail_descriptor_request_data() — §9.8.4, Table 9-26.
#[derive(Debug, Clone, PartialEq, Eq, Default)]
#[cfg_attr(feature = "serde", derive(serde::Serialize))]
pub struct InsertAvailDescriptor {
    /// `num_provider_avails` — 1 byte.
    pub num_provider_avails: u8,
    /// Loop of `provider_avail_id` (4 bytes each).
    pub provider_avail_ids: Vec<u32>,
}

impl<'a> Parse<'a> for InsertAvailDescriptor {
    type Error = Error;
    fn parse(bytes: &'a [u8]) -> Result<Self> {
        if bytes.is_empty() {
            return Err(Error::BufferTooShort {
                need: 1,
                have: 0,
                what: "insert_avail_descriptor num_provider_avails",
            });
        }
        let count = bytes[0] as usize;
        let need = 1 + count * 4;
        if bytes.len() < need {
            return Err(Error::BufferTooShort {
                need,
                have: bytes.len(),
                what: "insert_avail_descriptor provider_avail_ids",
            });
        }
        let mut ids = Vec::with_capacity(count);
        for i in 0..count {
            let off = 1 + i * 4;
            ids.push(u32::from_be_bytes([
                bytes[off],
                bytes[off + 1],
                bytes[off + 2],
                bytes[off + 3],
            ]));
        }
        Ok(Self {
            num_provider_avails: count as u8,
            provider_avail_ids: ids,
        })
    }
}

impl Serialize for InsertAvailDescriptor {
    type Error = Error;
    fn serialized_len(&self) -> usize {
        1 + self.provider_avail_ids.len() * 4
    }
    fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
        let need = self.serialized_len();
        if buf.len() < need {
            return Err(Error::OutputBufferTooSmall {
                need,
                have: buf.len(),
            });
        }
        buf[0] = self.num_provider_avails;
        for (i, &id) in self.provider_avail_ids.iter().enumerate() {
            let off = 1 + i * 4;
            buf[off..off + 4].copy_from_slice(&id.to_be_bytes());
        }
        Ok(need)
    }
}

impl OperationDef<'_> for InsertAvailDescriptor {
    const OP_ID: u16 = OP_ID;
    const NAME: &'static str = "INSERT_AVAIL_DESCRIPTOR";
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn round_trip() {
        let op = InsertAvailDescriptor {
            num_provider_avails: 2,
            provider_avail_ids: alloc::vec![0xAAAA_BBBB, 0xCCCC_DDDD],
        };
        let bytes = op.to_bytes();
        assert_eq!(bytes.len(), 9);
        let back = InsertAvailDescriptor::parse(&bytes).unwrap();
        assert_eq!(op, back);
    }

    #[test]
    fn zero_avails_round_trip() {
        let op = InsertAvailDescriptor {
            num_provider_avails: 0,
            provider_avail_ids: alloc::vec![],
        };
        let bytes = op.to_bytes();
        assert_eq!(bytes.len(), 1);
        let back = InsertAvailDescriptor::parse(&bytes).unwrap();
        assert_eq!(op, back);
    }

    #[test]
    fn mutate_field_changes_output() {
        let op = InsertAvailDescriptor {
            num_provider_avails: 1,
            provider_avail_ids: alloc::vec![0xAAAA_BBBB],
        };
        let bytes = op.to_bytes();
        let mut op2 = op.clone();
        op2.provider_avail_ids[0] = 0x1111_2222;
        assert_ne!(op2.to_bytes(), bytes);
    }
}