livekit-datatrack 0.1.12

// Copyright 2026 LiveKit, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use livekit_protocol as proto;
use std::sync::Arc;
use thiserror::Error;

/// Identifier for a data track schema.
///
/// A compound identifier with two components: name and encoding.
///
/// Two IDs are equal only if both components match; the same name with a
/// different encoding refers to a distinct schema. Cloning this type is cheap.
///
/// # Examples
///
/// ```
/// # use livekit_datatrack::api::{DataTrackSchemaId, DataTrackSchemaEncoding};
/// let schema = DataTrackSchemaId::new("my_schema", DataTrackSchemaEncoding::Protobuf);
///
/// assert_eq!(schema.name(), "my_schema");
/// assert_eq!(schema.encoding(), &DataTrackSchemaEncoding::Protobuf);
/// ```
///
#[derive(Clone, Hash, PartialEq, Eq)]
pub struct DataTrackSchemaId {
    inner: Arc<DataTrackSchemaIdInner>,
}

#[derive(Hash, PartialEq, Eq)]
struct DataTrackSchemaIdInner {
    name: String,
    encoding: DataTrackSchemaEncoding,
}

impl std::fmt::Debug for DataTrackSchemaId {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("DataTrackSchemaId")
            .field("name", &self.inner.name)
            .field("encoding", &self.inner.encoding)
            .finish()
    }
}

impl DataTrackSchemaId {
    /// Creates a new schema ID.
    pub fn new(name: impl Into<String>, encoding: DataTrackSchemaEncoding) -> Self {
        let inner = DataTrackSchemaIdInner { name: name.into(), encoding }.into();
        Self { inner }
    }

    /// Returns the name component of the ID.
    pub fn name(&self) -> &str {
        &self.inner.name
    }

    /// Returns the encoding component of the ID.
    pub fn encoding(&self) -> &DataTrackSchemaEncoding {
        &self.inner.encoding
    }
}

/// Encoding used for a schema definition.
///
/// Identifies the interface definition language the schema is written in (e.g. a
/// `.proto` file for [`Protobuf`]). This in turn dictates the wire format of the
/// frames the schema describes, captured by [`DataTrackFrameEncoding`].
///
/// [`Protobuf`]: DataTrackSchemaEncoding::Protobuf
///
#[non_exhaustive]
#[derive(Debug, Clone, Hash, PartialEq, Eq)]
#[cfg_attr(test, derive(fake::Dummy))]
#[cfg_attr(feature = "uniffi", derive(uniffi::Enum))]
pub enum DataTrackSchemaEncoding {
    /// Protocol Buffer IDL, describes [`Protobuf`] encoded frames.
    ///
    /// [`Protobuf`]: DataTrackFrameEncoding::Protobuf
    Protobuf,
    /// FlatBuffer IDL, describes [`Flatbuffer`] encoded frames.
    ///
    /// [`Flatbuffer`]: DataTrackFrameEncoding::Flatbuffer
    Flatbuffer,
    /// ROS 1 Message, describes [`Ros1`] encoded frames.
    ///
    /// [`Ros1`]: DataTrackFrameEncoding::Ros1
    Ros1Msg,
    /// ROS 2 Message, describes [`Cdr`] encoded frames.
    ///
    /// [`Cdr`]: DataTrackFrameEncoding::Cdr
    Ros2Msg,
    /// ROS 2 IDL, describes [`Cdr`] encoded frames.
    ///
    /// [`Cdr`]: DataTrackFrameEncoding::Cdr
    Ros2Idl,
    /// OMG IDL, describes [`Cdr`] encoded frames.
    ///
    /// [`Cdr`]: DataTrackFrameEncoding::Cdr
    OmgIdl,
    /// JSON Schema, describes [`Json`] encoded frames.
    ///
    /// [`Json`]: DataTrackFrameEncoding::Json
    JsonSchema,

    /// Another well-known encoding not known to this client version.
    Other,
    /// An application-specific encoding identified by the contained string.
    ///
    /// Prefer using one of the well-known encodings unless the format is not enumerated.
    /// The identifier must be non-empty and not exceed the server's length limit.
    ///
    Custom(String),
}

/// Encoding used for frames pushed on a data track.
///
/// The serialization format of the frame bytes (e.g. [`Protobuf`]); the structure
/// of those bytes is described by a schema, see [`DataTrackSchemaEncoding`].
///
/// [`Protobuf`]: DataTrackFrameEncoding::Protobuf
///
#[non_exhaustive]
#[derive(Debug, Clone, Hash, PartialEq, Eq)]
#[cfg_attr(test, derive(fake::Dummy))]
#[cfg_attr(feature = "uniffi", derive(uniffi::Enum))]
pub enum DataTrackFrameEncoding {
    /// ROS 1, must be described by a [`Ros1Msg`] schema.
    ///
    /// [`Ros1Msg`]: DataTrackSchemaEncoding::Ros1Msg
    Ros1,
    /// CDR, must be described by a [`Ros2Msg`], [`Ros2Idl`], or [`OmgIdl`] schema.
    ///
    /// [`Ros2Msg`]: DataTrackSchemaEncoding::Ros2Msg
    /// [`Ros2Idl`]: DataTrackSchemaEncoding::Ros2Idl
    /// [`OmgIdl`]: DataTrackSchemaEncoding::OmgIdl
    Cdr,
    /// Protocol Buffer, must be described by a [`Protobuf`] schema.
    ///
    /// [`Protobuf`]: DataTrackSchemaEncoding::Protobuf
    Protobuf,
    /// FlatBuffer, must be described by a [`Flatbuffer`] schema.
    ///
    /// [`Flatbuffer`]: DataTrackSchemaEncoding::Flatbuffer
    Flatbuffer,
    /// CBOR, self-describing.
    Cbor,
    /// MessagePack, self-describing.
    Msgpack,
    /// JSON, self-describing or described by a [`JsonSchema`] schema.
    ///
    /// [`JsonSchema`]: DataTrackSchemaEncoding::JsonSchema
    Json,

    /// Another well-known encoding not known to this client version.
    Other,
    /// An application-specific encoding identified by the contained string.
    ///
    /// Prefer using one of the well-known encodings unless the format is not enumerated.
    /// The identifier must be non-empty and not exceed the server's length limit.
    ///
    Custom(String),
}

/// An error that can occur when validating data track schema metadata.
#[derive(Debug, Error, PartialEq)]
pub enum DataTrackSchemaError {
    /// Frame encoding is required when providing schema ID.
    #[error("Frame encoding is required when providing schema ID")]
    MissingFrameEncoding,

    /// Schema ID is required for frame encoding that is not self-describing.
    #[error("Schema ID is required for frame encoding that is not self-describing")]
    MissingSchemaId,

    /// Specified schema and frame encodings are incompatible.
    #[error("Specified schema and frame encodings are incompatible")]
    Incompatible,
}

/// Validates that the given frame and schema encodings are compatible.
pub(crate) fn validate_schema(
    frame_encoding: Option<&DataTrackFrameEncoding>,
    schema_encoding: Option<&DataTrackSchemaEncoding>,
) -> Result<(), DataTrackSchemaError> {
    match (frame_encoding, schema_encoding) {
        (None, Some(_)) => Err(DataTrackSchemaError::MissingFrameEncoding),
        (Some(frame_encoding), None) => match frame_encoding.is_self_describing() {
            Some(false) => Err(DataTrackSchemaError::MissingSchemaId),
            _ => Ok(()),
        },
        (Some(frame_encoding), Some(schema_encoding)) => {
            match frame_encoding.is_described_by(schema_encoding) {
                Some(false) => Err(DataTrackSchemaError::Incompatible),
                _ => Ok(()),
            }
        }
        (None, None) => Ok(()), // Not using schema metadata
    }
}

impl DataTrackFrameEncoding {
    /// Returns whether the frame encoding is self-describing (i.e. requires no schema).
    fn is_self_describing(&self) -> Option<bool> {
        match self {
            Self::Cbor | Self::Msgpack | Self::Json => Some(true),
            Self::Other | Self::Custom(_) => None, // Cannot be determined
            _ => Some(false),
        }
    }

    /// Returns whether the frame encoding can be described by the given schema encoding.
    fn is_described_by(&self, schema_encoding: &DataTrackSchemaEncoding) -> Option<bool> {
        use DataTrackSchemaEncoding as SchemaEncoding;
        match (self, schema_encoding) {
            (Self::Ros1, SchemaEncoding::Ros1Msg)
            | (Self::Cdr, SchemaEncoding::Ros2Msg)
            | (Self::Cdr, SchemaEncoding::Ros2Idl)
            | (Self::Cdr, SchemaEncoding::OmgIdl)
            | (Self::Protobuf, SchemaEncoding::Protobuf)
            | (Self::Flatbuffer, SchemaEncoding::Flatbuffer)
            | (Self::Json, SchemaEncoding::JsonSchema) => Some(true),
            (Self::Other, _) | (Self::Custom(_), _) => None, // Cannot be determined
            _ => Some(false),
        }
    }
}

impl From<proto::DataTrackSchemaId> for DataTrackSchemaId {
    fn from(msg: proto::DataTrackSchemaId) -> Self {
        let encoding = msg.encoding.map(Into::into).unwrap_or(DataTrackSchemaEncoding::Other);
        DataTrackSchemaId::new(msg.name, encoding)
    }
}

impl From<DataTrackSchemaId> for proto::DataTrackSchemaId {
    fn from(value: DataTrackSchemaId) -> Self {
        Self { name: value.name().to_string(), encoding: Some(value.encoding().clone().into()) }
    }
}

impl From<proto::DataTrackSchemaEncoding> for DataTrackSchemaEncoding {
    fn from(msg: proto::DataTrackSchemaEncoding) -> Self {
        use proto::data_track_schema_encoding::{Value, WellKnownSchemaEncoding as WellKnown};
        match msg.value {
            Some(Value::WellKnown(value)) => match WellKnown::try_from(value) {
                Ok(WellKnown::Protobuf) => Self::Protobuf,
                Ok(WellKnown::Flatbuffer) => Self::Flatbuffer,
                Ok(WellKnown::Ros1Msg) => Self::Ros1Msg,
                Ok(WellKnown::Ros2Msg) => Self::Ros2Msg,
                Ok(WellKnown::Ros2Idl) => Self::Ros2Idl,
                Ok(WellKnown::OmgIdl) => Self::OmgIdl,
                Ok(WellKnown::JsonSchema) => Self::JsonSchema,
                // Unspecified or a value introduced after this client version.
                Ok(WellKnown::Unspecified) | Err(_) => Self::Other,
            },
            Some(Value::Custom(name)) => Self::Custom(name),
            None => Self::Other,
        }
    }
}

impl From<DataTrackSchemaEncoding> for proto::DataTrackSchemaEncoding {
    fn from(value: DataTrackSchemaEncoding) -> Self {
        use proto::data_track_schema_encoding::{Value, WellKnownSchemaEncoding as WellKnown};
        let well_known = match value {
            DataTrackSchemaEncoding::Protobuf => WellKnown::Protobuf,
            DataTrackSchemaEncoding::Flatbuffer => WellKnown::Flatbuffer,
            DataTrackSchemaEncoding::Ros1Msg => WellKnown::Ros1Msg,
            DataTrackSchemaEncoding::Ros2Msg => WellKnown::Ros2Msg,
            DataTrackSchemaEncoding::Ros2Idl => WellKnown::Ros2Idl,
            DataTrackSchemaEncoding::OmgIdl => WellKnown::OmgIdl,
            DataTrackSchemaEncoding::JsonSchema => WellKnown::JsonSchema,
            DataTrackSchemaEncoding::Other => WellKnown::Unspecified,
            DataTrackSchemaEncoding::Custom(name) => {
                return Self { value: Some(Value::Custom(name)) }
            }
        };
        Self { value: Some(Value::WellKnown(well_known as i32)) }
    }
}

impl From<proto::DataTrackFrameEncoding> for DataTrackFrameEncoding {
    fn from(msg: proto::DataTrackFrameEncoding) -> Self {
        use proto::data_track_frame_encoding::{Value, WellKnownFrameEncoding as WellKnown};
        match msg.value {
            Some(Value::WellKnown(value)) => match WellKnown::try_from(value) {
                Ok(WellKnown::Ros1) => Self::Ros1,
                Ok(WellKnown::Cdr) => Self::Cdr,
                Ok(WellKnown::Protobuf) => Self::Protobuf,
                Ok(WellKnown::Flatbuffer) => Self::Flatbuffer,
                Ok(WellKnown::Cbor) => Self::Cbor,
                Ok(WellKnown::Msgpack) => Self::Msgpack,
                Ok(WellKnown::Json) => Self::Json,
                // Unspecified or a value introduced after this client version.
                Ok(WellKnown::Unspecified) | Err(_) => Self::Other,
            },
            Some(Value::Custom(name)) => Self::Custom(name),
            None => Self::Other,
        }
    }
}

impl From<DataTrackFrameEncoding> for proto::DataTrackFrameEncoding {
    fn from(value: DataTrackFrameEncoding) -> Self {
        use proto::data_track_frame_encoding::{Value, WellKnownFrameEncoding as WellKnown};
        let well_known = match value {
            DataTrackFrameEncoding::Ros1 => WellKnown::Ros1,
            DataTrackFrameEncoding::Cdr => WellKnown::Cdr,
            DataTrackFrameEncoding::Protobuf => WellKnown::Protobuf,
            DataTrackFrameEncoding::Flatbuffer => WellKnown::Flatbuffer,
            DataTrackFrameEncoding::Cbor => WellKnown::Cbor,
            DataTrackFrameEncoding::Msgpack => WellKnown::Msgpack,
            DataTrackFrameEncoding::Json => WellKnown::Json,
            DataTrackFrameEncoding::Other => WellKnown::Unspecified,
            DataTrackFrameEncoding::Custom(name) => {
                return Self { value: Some(Value::Custom(name)) }
            }
        };
        Self { value: Some(Value::WellKnown(well_known as i32)) }
    }
}

impl From<DataTrackSchemaId> for proto::DataBlobKey {
    fn from(id: DataTrackSchemaId) -> Self {
        Self { key: Some(proto::data_blob_key::Key::SchemaId(id.into())) }
    }
}

#[cfg(test)]
impl fake::Dummy<fake::Faker> for DataTrackSchemaId {
    fn dummy_with_rng<R: rand::Rng + ?Sized>(_: &fake::Faker, rng: &mut R) -> Self {
        use fake::{Fake, Faker};
        let name: String = Faker.fake_with_rng(rng);
        let encoding: DataTrackSchemaEncoding = Faker.fake_with_rng(rng);
        Self::new(name, encoding)
    }
}

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

    #[test]
    fn test_validate_schema_not_specified() {
        assert_eq!(validate_schema(None, None), Ok(()));
    }

    #[test]
    fn test_validate_schema_self_describing() {
        assert_eq!(validate_schema(Some(&DataTrackFrameEncoding::Json), None), Ok(()));
    }

    #[test]
    fn test_validate_schema_compatible_encodings() {
        assert_eq!(
            validate_schema(
                Some(&DataTrackFrameEncoding::Cdr),
                Some(&DataTrackSchemaEncoding::Ros2Idl)
            ),
            Ok(())
        );
    }

    #[test]
    fn test_validate_schema_custom() {
        assert_eq!(
            validate_schema(
                Some(&DataTrackFrameEncoding::Custom("my-frame-encoding".to_string())),
                Some(&DataTrackSchemaEncoding::Custom("my-schema-encoding".to_string()))
            ),
            Ok(())
        );
    }

    #[test]
    fn test_validate_schema_missing_frame_encoding() {
        assert_eq!(
            validate_schema(None, Some(&DataTrackSchemaEncoding::Protobuf)),
            Err(DataTrackSchemaError::MissingFrameEncoding)
        );
    }

    #[test]
    fn test_validate_schema_missing_schema_id() {
        assert_eq!(
            validate_schema(Some(&DataTrackFrameEncoding::Protobuf), None),
            Err(DataTrackSchemaError::MissingSchemaId)
        );
    }

    #[test]
    fn test_validate_schema_incompatible() {
        assert_eq!(
            validate_schema(
                Some(&DataTrackFrameEncoding::Json),
                Some(&DataTrackSchemaEncoding::Protobuf)
            ),
            Err(DataTrackSchemaError::Incompatible)
        );
    }
}