offline_protocol_core/

message.rs

//! Message types and structures.

use crate::types::{AppId, HopCount, LamportClock, Timestamp, UserId, TTL};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fmt;
use uuid::Uuid;

/// Unique message identifier.
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub struct MessageId(Uuid);

impl MessageId {
    /// Generates a new random message ID.
    pub fn new() -> Self {
        Self(Uuid::new_v4())
    }

    /// Creates a message ID from a string representation.
    #[allow(clippy::should_implement_trait)]
    pub fn from_str(id: &str) -> crate::Result<Self> {
        let uuid = Uuid::parse_str(id)
            .map_err(|e| crate::Error::InvalidMessage(format!("Invalid message id: {}", e)))?;
        Ok(Self::from_uuid(uuid))
    }

    /// Creates a message ID from a UUID.
    pub fn from_uuid(uuid: Uuid) -> Self {
        Self(uuid)
    }

    /// Returns the message ID as a string.
    pub fn as_str(&self) -> String {
        self.0.to_string()
    }

    /// Returns the 16-byte representation of the underlying UUID.
    ///
    /// Used by the binary wire codec to carry the id as 16 raw bytes instead of
    /// the 36-character hyphenated string the JSON encoding uses.
    pub fn as_bytes(&self) -> [u8; 16] {
        *self.0.as_bytes()
    }

    /// Reconstructs a message id from its 16-byte representation.
    pub fn from_bytes(bytes: [u8; 16]) -> Self {
        Self(Uuid::from_bytes(bytes))
    }
}

impl Default for MessageId {
    fn default() -> Self {
        Self::new()
    }
}

impl fmt::Display for MessageId {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

/// The type of content carried in a message.
///
/// Adding a variant is an additive wire change: older receivers must keep
/// decoding messages that carry a type they don't know. All three decode
/// paths therefore share the same fallback — an unrecognised value degrades
/// to [`File`](Self::File) instead of failing the containing message:
/// [`parse`](Self::parse) for strings, `content_type_from_u8` for the binary
/// wire codec, and the hand-written [`Deserialize`] below for JSON (the
/// derived impl would reject the whole frame on an unknown variant).
///
/// Both serde impls are hand-written over [`as_wire_str`](Self::as_wire_str)
/// rather than derived, so *every* serde format — not only the
/// self-describing ones — speaks the same strings as the other two decode
/// paths. A derived `Serialize` would emit the variant *index* in a
/// non-self-describing format (postcard, bincode), which the tolerant
/// string-based `Deserialize` would then read back as a zero-length string
/// and degrade to `File` — silently turning `Text` into `File` with no
/// error. Nothing in-tree does that today (the binary wire codec maps to
/// `u8` by hand), but this type is published, so the two directions are
/// kept symmetric in all formats rather than only in JSON.
///
/// Apps should treat `File` as "a file *or* a type this build doesn't
/// know". A degraded value keeps [`is_media`](Self::is_media) `true` while
/// carrying no [`MediaMetadata`] and no transfer, so render that
/// combination as a generic unsupported-content placeholder rather than as
/// a broken attachment.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash)]
pub enum ContentType {
    /// Plain text message.
    #[default]
    Text,
    /// Image attachment (JPEG, PNG, etc.).
    Image,
    /// Video attachment.
    Video,
    /// Audio attachment.
    Audio,
    /// Short voice recording.
    VoiceNote,
    /// Short video recording.
    VideoNote,
    /// Generic file attachment.
    File,
    /// Internal: a chunk belonging to a multi-part file transfer.
    FileChunk,
    /// Structured poll, rendered by the app rather than as plain text.
    Poll,
}

impl ContentType {
    /// Every variant, in declaration order.
    ///
    /// Provably complete — see the `const` block below — which is what lets
    /// `content_type_json_round_trips_every_variant` stand in for the
    /// compiler over [`parse`](Self::parse).
    pub const ALL: [Self; 9] = [
        Self::Text,
        Self::Image,
        Self::Video,
        Self::Audio,
        Self::VoiceNote,
        Self::VideoNote,
        Self::File,
        Self::FileChunk,
        Self::Poll,
    ];

    /// Returns `true` for types that carry binary media data.
    pub fn is_media(&self) -> bool {
        !matches!(self, Self::Text | Self::Poll)
    }

    /// The canonical wire string for this type.
    ///
    /// The single source of truth for the variant → string direction,
    /// shared by [`Display`](fmt::Display), [`Serialize`] and the JSON
    /// [`Deserialize`]. Exhaustive, so rustc — not a test — guarantees
    /// every variant has exactly one wire string.
    pub const fn as_wire_str(self) -> &'static str {
        match self {
            Self::Text => "text",
            Self::Image => "image",
            Self::Video => "video",
            Self::Audio => "audio",
            Self::VoiceNote => "voice_note",
            Self::VideoNote => "video_note",
            Self::File => "file",
            Self::FileChunk => "file_chunk",
            Self::Poll => "poll",
        }
    }

    /// Parses a content type from its string representation.
    ///
    /// Falls back to `File` for unrecognised strings. This is the inverse of
    /// [`as_wire_str`](Self::as_wire_str), and the one mapping rustc cannot
    /// check: the `_` arm swallows a variant whose arm was forgotten, which
    /// would make that variant decode as `File` even between same-version
    /// peers. `content_type_json_round_trips_every_variant` is what pins it,
    /// iterating [`ALL`](Self::ALL) so it cannot silently miss a variant.
    pub fn parse(s: &str) -> Self {
        match s {
            "text" => Self::Text,
            "image" => Self::Image,
            "video" => Self::Video,
            "audio" => Self::Audio,
            "voice_note" => Self::VoiceNote,
            "video_note" => Self::VideoNote,
            "file" => Self::File,
            "file_chunk" => Self::FileChunk,
            "poll" => Self::Poll,
            _ => Self::File,
        }
    }
}

/// Pins [`ContentType::ALL`] complete and correctly ordered at compile time.
///
/// Adding a variant fails to compile twice over: the match below is
/// exhaustive, so the new variant must be named here, and the arm naming it
/// then indexes past the end of `ALL` (a deny-by-default `unconditional_panic`
/// error) until the array grows too. The loop then const-evaluates each entry,
/// pinning position as well as membership.
///
/// This matters because `ALL` is what the round-trip test iterates to cover
/// [`ContentType::parse`]'s unavoidable `_` arm. A hand-maintained list would
/// leave "forgot the `parse` arm" and "forgot the test entry" as the same
/// omission, so the test could never catch it.
const _: () = {
    const fn listed_at_own_index(ct: ContentType) -> bool {
        match ct {
            ContentType::Text => matches!(ContentType::ALL[0], ContentType::Text),
            ContentType::Image => matches!(ContentType::ALL[1], ContentType::Image),
            ContentType::Video => matches!(ContentType::ALL[2], ContentType::Video),
            ContentType::Audio => matches!(ContentType::ALL[3], ContentType::Audio),
            ContentType::VoiceNote => matches!(ContentType::ALL[4], ContentType::VoiceNote),
            ContentType::VideoNote => matches!(ContentType::ALL[5], ContentType::VideoNote),
            ContentType::File => matches!(ContentType::ALL[6], ContentType::File),
            ContentType::FileChunk => matches!(ContentType::ALL[7], ContentType::FileChunk),
            ContentType::Poll => matches!(ContentType::ALL[8], ContentType::Poll),
        }
    }

    let mut i = 0;
    while i < ContentType::ALL.len() {
        assert!(listed_at_own_index(ContentType::ALL[i]));
        i += 1;
    }
};

/// Serializes as the [`wire string`](ContentType::as_wire_str), in every
/// format rather than only the self-describing ones. Byte-identical to the
/// derived `snake_case` impl on JSON; see the type docs for why the derived
/// impl is not used.
impl Serialize for ContentType {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.serialize_str(self.as_wire_str())
    }
}

/// Tolerant decoding: any string is accepted and routed through
/// [`ContentType::parse`], so a variant added by a newer sender degrades to
/// `File` on this receiver rather than failing the whole `Message` (or
/// sealed rich body) it rides in. Non-string values still error — that is
/// malformed data, not additive evolution.
impl<'de> Deserialize<'de> for ContentType {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        struct ContentTypeVisitor;

        impl serde::de::Visitor<'_> for ContentTypeVisitor {
            type Value = ContentType;

            fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
                f.write_str("a content type string")
            }

            fn visit_str<E: serde::de::Error>(self, s: &str) -> Result<ContentType, E> {
                Ok(ContentType::parse(s))
            }
        }

        deserializer.deserialize_str(ContentTypeVisitor)
    }
}

impl fmt::Display for ContentType {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(self.as_wire_str())
    }
}

/// Metadata describing a media attachment.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MediaMetadata {
    /// MIME type (e.g. "image/jpeg", "video/mp4").
    pub mime_type: String,

    /// Original file name.
    pub file_name: String,

    /// File size in bytes.
    pub file_size: u64,

    /// Duration in milliseconds (audio/video/voice-note/video-note).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub duration_ms: Option<u64>,

    /// Width in pixels (images and video).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub width: Option<u32>,

    /// Height in pixels (images and video).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub height: Option<u32>,

    /// Small base64-encoded thumbnail (< 2 KB) for preview.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub thumbnail_base64: Option<String>,

    /// Stable media identifier assigned by the application.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub media_id: Option<String>,

    /// URL to fetch the full media from (cloud-stored media).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub download_url: Option<String>,

    /// URL to fetch a thumbnail from (cloud-stored media).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub thumbnail_url: Option<String>,

    /// Content-encryption key for cloud-stored media (base64).
    ///
    /// Secret material: must only travel inside an end-to-end-encrypted
    /// payload, never in cleartext message fields visible to relays.
    /// Enforced at the transport wire chokepoint, which strips it (via
    /// [`MediaMetadata::without_secrets`]) from every outbound cleartext
    /// frame, and by the telemetry scrubber, which redacts it
    /// unconditionally before events reach a sink.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub encryption_key: Option<String>,

    /// Initialization vector for the cloud-media content encryption (base64).
    /// Secret material — same handling as [`Self::encryption_key`].
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub iv: Option<String>,

    /// Integrity hash of the encrypted cloud-media blob (base64).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub ciphertext_hash: Option<String>,

    /// Sticker pack provider (sticker messages).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub sticker_provider: Option<String>,

    /// Provider-scoped sticker identifier (sticker messages).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub sticker_remote_id: Option<String>,

    /// Sticker rendering kind (e.g. "static", "animated", "lottie").
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub sticker_kind: Option<String>,
}

impl MediaMetadata {
    /// True when any confidential field ([`Self::encryption_key`],
    /// [`Self::iv`]) is set.
    pub fn has_secrets(&self) -> bool {
        self.encryption_key.is_some() || self.iv.is_some()
    }

    /// Returns a copy with the confidential fields cleared.
    ///
    /// The cloud-media content key (and its IV) grant access to the media
    /// itself, so they must only ever travel inside an end-to-end-encrypted
    /// payload. Every path that writes metadata into a cleartext surface —
    /// the transport wire encoders, the telemetry event scrubber — routes
    /// through this before emitting. The remaining fields (URLs, hashes,
    /// sticker identity, dimensions) are display hints, not secrets, and are
    /// preserved.
    pub fn without_secrets(&self) -> Self {
        Self {
            encryption_key: None,
            iv: None,
            ..self.clone()
        }
    }
}

/// Information about a forwarded message.
///
/// **Trust model:** `ForwardInfo` is an unverified attribution hint, not a
/// cryptographic proof. A malicious client can forge the `original_sender` or
/// reset `forward_count`. UI layers should treat this as a display-level hint
/// and must not rely on it for access-control or security decisions.
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct ForwardInfo {
    /// The original sender of the message.
    pub original_sender: UserId,
    /// The original message ID.
    pub original_message_id: MessageId,
    /// The original timestamp (wall-clock, for display).
    pub original_timestamp: Timestamp,
    /// Number of times this message has been forwarded.
    pub forward_count: u32,
}

impl ForwardInfo {
    /// Creates `ForwardInfo` from a message being forwarded.
    ///
    /// If the message was already forwarded, the original attribution is preserved
    /// and `forward_count` is incremented. Otherwise, the message's sender becomes
    /// the original sender with `forward_count = 1`.
    pub fn from_message(message: &Message) -> Self {
        match &message.forwarded_from {
            Some(existing) => ForwardInfo {
                original_sender: existing.original_sender.clone(),
                original_message_id: existing.original_message_id.clone(),
                original_timestamp: existing.original_timestamp,
                forward_count: existing.forward_count.saturating_add(1),
            },
            None => ForwardInfo {
                original_sender: message.sender.clone(),
                original_message_id: message.id.clone(),
                original_timestamp: message.timestamp,
                forward_count: 1,
            },
        }
    }
}

/// Quoted-reply context for rendering a reply preview without a local copy
/// of the original message.
///
/// **Trust model:** like [`ForwardInfo`], this is an unverified display-level
/// hint copied by the sending client. A malicious client can forge any field.
/// UI layers should treat this as a display-level hint and must not rely on
/// it for access-control or security decisions.
///
/// On an MLS-encrypted message this outer field additionally sits outside
/// the AEAD boundary (any relay hop can inject or rewrite it in transit),
/// so the protocol layer drops it on decryption: it never surfaces on an
/// `encrypted: true` event. The end-to-end-sealed envelope is the only
/// trusted carrier for reply context on encrypted messages.
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct ReplyContext {
    /// Sender of the message being replied to.
    pub sender: UserId,
    /// Text (or excerpt) of the message being replied to.
    pub text: String,
    /// Timestamp of the quoted message (wall-clock, for display).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub timestamp: Option<Timestamp>,
    /// Short human-readable label for quoted media (e.g. a file name).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub reply_media_label: Option<String>,
    /// Content type of the quoted message, as a display string (e.g. "image").
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub reply_content_type: Option<String>,
}

/// Message priority levels.
///
/// Serializes lowercase; the capitalized aliases keep Debug-cased values
/// parseable — the UniFFI `receive_message` JSON has always emitted
/// `"Medium"` etc., and that JSON must round-trip back through
/// `forward_message`.
#[derive(
    Debug, Clone, Copy, Default, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize,
)]
#[serde(rename_all = "lowercase")]
pub enum MessagePriority {
    /// Low priority - can be delayed or dropped under congestion.
    #[serde(alias = "Low")]
    Low,
    /// Medium priority - default for most messages.
    #[default]
    #[serde(alias = "Medium")]
    Medium,
    /// High priority - important messages that should be delivered quickly.
    #[serde(alias = "High")]
    High,
    /// Critical priority - emergency messages, highest delivery guarantee.
    #[serde(alias = "Critical")]
    Critical,
}

impl MessagePriority {
    /// Returns a numeric score for the priority (higher = more important).
    pub fn score(&self) -> u8 {
        match self {
            Self::Low => 1,
            Self::Medium => 2,
            Self::High => 3,
            Self::Critical => 4,
        }
    }

    /// Returns a stable string representation suitable for event payloads.
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::Low => "Low",
            Self::Medium => "Medium",
            Self::High => "High",
            Self::Critical => "Critical",
        }
    }
}

/// Wire codec selected for encoding a message on its next send.
///
/// Carried in-process only (never serialized) and defaults to
/// [`WireCodec::Json`], the universally-understood encoding. The transport
/// manager stamps [`WireCodec::BinaryV1`] just before send when the recipient is
/// known to support the compact binary format; mesh transports honor the stamp
/// at drain time while the internet transport always emits JSON.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub enum WireCodec {
    /// Legacy JSON encoding — the permanent interoperability floor.
    #[default]
    Json,
    /// Compact binary wire codec, version 1 (see [`crate::wire`]).
    BinaryV1,
}

/// A message in the Offline Protocol.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Message {
    /// Unique message identifier.
    pub id: MessageId,

    /// Sender's user ID.
    pub sender: UserId,

    /// Recipient's user ID.
    pub recipient: UserId,

    /// Application ID that this message belongs to.
    pub app_id: AppId,

    /// Message priority.
    pub priority: MessagePriority,

    /// Time-to-live: maximum hops remaining.
    pub ttl: TTL,

    /// Number of hops this message has traversed.
    pub hop_count: HopCount,

    /// Timestamp when the message was created (wall-clock, for display only).
    pub timestamp: Timestamp,

    /// Lamport logical clock for causal ordering across devices.
    #[serde(default)]
    pub lamport_clock: LamportClock,

    /// The type of content this message carries.
    #[serde(default)]
    pub content_type: ContentType,

    /// Message content (text, JSON-serialized file chunk metadata, etc.).
    pub content: String,

    /// Raw payload for file chunk data carried by the message envelope.
    /// This avoids wrapping chunk bytes in `FileChunk` JSON/base64 content.
    /// Note: current transport serialization still uses JSON for `Message`.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub binary_content: Option<Vec<u8>>,

    /// Media metadata (present for non-text content types).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub media_metadata: Option<MediaMetadata>,

    /// Optional metadata for application-specific use.
    #[serde(default)]
    pub metadata: HashMap<String, String>,

    /// Whether this message requires an ACK.
    #[serde(default = "default_requires_ack")]
    pub requires_ack: bool,

    /// ID of the message this is replying to (optional).
    #[serde(default)]
    pub reply_to_msg: Option<MessageId>,

    /// Forwarding attribution (present when this message was forwarded from another user).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub forwarded_from: Option<ForwardInfo>,

    /// Quoted-reply context (present when this message quotes another).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub reply_context: Option<ReplyContext>,

    /// Transport-verified peer identity.
    ///
    /// Set by the transport layer when a message is received, binding the message
    /// to the physical peer that delivered it. This field is **never serialized**
    /// over the wire — it exists only in-process to prevent sender spoofing.
    ///
    /// When `Some`, the protocol layer validates that `sender` matches this value
    /// before processing security-sensitive control messages.
    ///
    /// # Security
    ///
    /// This field is private to prevent application code from bypassing sender
    /// verification. Use [`set_transport_peer_id`](Self::set_transport_peer_id)
    /// (transport layers only) and [`transport_peer_id`](Self::transport_peer_id)
    /// to interact with it.
    #[serde(skip)]
    transport_peer_id: Option<String>,

    /// Wire codec to use when this message is next serialized for the wire.
    ///
    /// In-process only (`#[serde(skip)]`); stamped by the transport manager from
    /// per-peer capability just before send. Defaults to [`WireCodec::Json`].
    #[serde(skip)]
    wire_codec: WireCodec,
}

fn default_requires_ack() -> bool {
    true
}

impl Message {
    /// Creates a new message.
    ///
    /// # Arguments
    ///
    /// * `sender` - Sender's user ID
    /// * `recipient` - Recipient's user ID
    /// * `app_id` - Application ID
    /// * `content` - Message content
    pub fn new(
        sender: UserId,
        recipient: UserId,
        app_id: AppId,
        content: impl Into<String>,
    ) -> Self {
        Self {
            id: MessageId::new(),
            sender,
            recipient,
            app_id,
            priority: MessagePriority::default(),
            ttl: TTL::default(),
            hop_count: HopCount::new(),
            timestamp: Timestamp::now(),
            lamport_clock: LamportClock::default(),
            content_type: ContentType::default(),
            content: content.into(),
            binary_content: None,
            media_metadata: None,
            metadata: HashMap::new(),
            requires_ack: true,
            reply_to_msg: None,
            forwarded_from: None,
            reply_context: None,
            transport_peer_id: None,
            wire_codec: WireCodec::Json,
        }
    }

    /// Creates a message builder for more control over message creation.
    pub fn builder(sender: UserId, recipient: UserId, app_id: AppId) -> MessageBuilder {
        MessageBuilder::new(sender, recipient, app_id)
    }

    /// Increments the hop count.
    ///
    /// # Returns
    ///
    /// Returns `Ok(())` if successful, `Err` if hop count overflows.
    pub fn increment_hop(&mut self) -> crate::Result<()> {
        self.hop_count = self
            .hop_count
            .increment()
            .ok_or(crate::Error::InvalidHopCount(u8::MAX))?;
        Ok(())
    }

    /// Decrements the TTL.
    ///
    /// # Returns
    ///
    /// Returns `Ok(())` if successful, `Err` if TTL is exhausted.
    pub fn decrement_ttl(&mut self) -> crate::Result<()> {
        self.ttl = self.ttl.decrement().ok_or(crate::Error::InvalidTTL(0))?;
        Ok(())
    }

    /// Checks if the message's TTL is exhausted.
    pub fn is_ttl_exhausted(&self) -> bool {
        self.ttl.is_exhausted()
    }

    /// Serializes the message to JSON.
    pub fn to_json(&self) -> crate::Result<String> {
        serde_json::to_string(self).map_err(|e| crate::Error::SerializationError(e.to_string()))
    }

    /// Deserializes a message from JSON.
    pub fn from_json(json: &str) -> crate::Result<Self> {
        serde_json::from_str(json).map_err(|e| crate::Error::DeserializationError(e.to_string()))
    }

    /// Sets the transport-verified peer identity.
    ///
    /// # Security
    ///
    /// This method must only be called by transport layer implementations
    /// (e.g., `BleTransport::on_message_received_from`) to bind a message to
    /// the physical peer that delivered it. The protocol layer uses this value
    /// to reject control messages where the claimed `sender` does not match the
    /// transport-authenticated peer.
    ///
    /// This method is `pub` because transport implementations live in a
    /// separate crate (`offline-protocol-transport`). It is **not** exposed
    /// through UniFFI bindings, and the underlying field is private +
    /// `#[serde(skip)]`, so application code cannot set it via deserialization
    /// or direct field access.
    ///
    /// **Do not call from application code or expose via FFI.**
    ///
    /// # Errors
    ///
    /// Returns an error if `peer_id` is empty, since an empty transport peer
    /// identity cannot meaningfully authenticate a sender.
    #[doc(hidden)]
    pub fn set_transport_peer_id(&mut self, peer_id: String) -> crate::Result<()> {
        if peer_id.is_empty() {
            return Err(crate::Error::InvalidMessage(
                "transport_peer_id must not be empty".to_string(),
            ));
        }
        self.transport_peer_id = Some(peer_id);
        Ok(())
    }

    /// Returns the transport-verified peer identity, if set.
    pub fn transport_peer_id(&self) -> Option<&str> {
        self.transport_peer_id.as_deref()
    }

    /// Selects the wire codec used the next time this message is serialized.
    ///
    /// Stamped by the transport manager from negotiated per-peer capability.
    /// Application code should not call this: an incorrect choice can make a
    /// message undecodable to a peer that does not support the codec.
    #[doc(hidden)]
    pub fn set_wire_codec(&mut self, codec: WireCodec) {
        self.wire_codec = codec;
    }

    /// Returns the wire codec selected for this message's next serialization.
    pub fn wire_codec(&self) -> WireCodec {
        self.wire_codec
    }

    /// Serializes the message to binary (MessagePack-like JSON bytes).
    pub fn to_bytes(&self) -> crate::Result<Vec<u8>> {
        serde_json::to_vec(self).map_err(|e| crate::Error::SerializationError(e.to_string()))
    }

    /// Deserializes a message from binary.
    pub fn from_bytes(bytes: &[u8]) -> crate::Result<Self> {
        serde_json::from_slice(bytes).map_err(|e| crate::Error::DeserializationError(e.to_string()))
    }

    /// Serializes the message using the compact binary wire codec (v1).
    ///
    /// See [`crate::wire`] for the format. This is an additive alternative to
    /// [`to_bytes`](Self::to_bytes) (JSON); the two are distinguishable by their
    /// first byte, so a receiver can accept either without negotiation.
    pub fn to_wire_v1_bytes(&self) -> crate::Result<Vec<u8>> {
        let media_metadata_json =
            match &self.media_metadata {
                Some(m) => Some(serde_json::to_vec(m).map_err(|e| {
                    crate::Error::SerializationError(format!("media_metadata: {e}"))
                })?),
                None => None,
            };
        let forwarded_from_json =
            match &self.forwarded_from {
                Some(f) => Some(serde_json::to_vec(f).map_err(|e| {
                    crate::Error::SerializationError(format!("forwarded_from: {e}"))
                })?),
                None => None,
            };
        let mut metadata: Vec<(String, String)> = self
            .metadata
            .iter()
            .map(|(k, v)| (k.clone(), v.clone()))
            .collect();
        // Deterministic order (a HashMap iterates nondeterministically).
        metadata.sort_by(|a, b| a.0.cmp(&b.0));

        // A long canonical-base64 content tail (an MLS envelope, typically)
        // rides raw in the ext TLV instead of paying the 4/3 base64 inflation;
        // the split is verified byte-exact at encode time, so decoding
        // reconstructs the identical string. See [`crate::wire::EXT_TAG_B64_TAIL`].
        let (content, mut ext) = match crate::wire::split_b64_tail(&self.content) {
            Some((head, raw)) => (head.to_string(), vec![(crate::wire::EXT_TAG_B64_TAIL, raw)]),
            None => (self.content.clone(), Vec::new()),
        };

        // The quoted-reply context rides the ext TLV as a JSON blob (like
        // `media_metadata_json`) because the v1 DTO is frozen. Old decoders
        // skip the tag, losing only the reply preview — the same degradation
        // a legacy JSON receiver applies by ignoring the unknown field.
        if let Some(rc) = &self.reply_context {
            ext.push((
                crate::wire::EXT_TAG_REPLY_CONTEXT,
                serde_json::to_vec(rc)
                    .map_err(|e| crate::Error::SerializationError(format!("reply_context: {e}")))?,
            ));
        }

        let wire = crate::wire::WireMessageV1 {
            id: self.id.as_bytes(),
            sender: self.sender.as_str().to_string(),
            recipient: self.recipient.as_str().to_string(),
            app_id: self.app_id.as_str().to_string(),
            priority: crate::wire::priority_to_u8(self.priority),
            ttl: self.ttl.value(),
            hop_count: self.hop_count.value(),
            timestamp: self.timestamp.as_millis(),
            lamport_clock: self.lamport_clock.value(),
            content_type: crate::wire::content_type_to_u8(self.content_type),
            content,
            binary_content: self.binary_content.clone(),
            media_metadata_json,
            metadata,
            requires_ack: self.requires_ack,
            reply_to_msg: self.reply_to_msg.as_ref().map(|m| m.as_bytes()),
            forwarded_from_json,
            ext,
        };
        crate::wire::encode(&wire)
    }

    /// Deserializes a message from a compact binary wire frame (v1).
    ///
    /// Runs the same identifier and Lamport-clock validation as the JSON path by
    /// routing every field through its validating constructor, so a hostile
    /// binary frame is rejected identically to a hostile JSON one. The
    /// transport-verified peer identity is never on the wire and is left unset.
    pub fn from_wire_v1_bytes(data: &[u8]) -> crate::Result<Self> {
        let wire = crate::wire::decode(data)?;
        // Reassemble a base64 content tail carried raw in the ext TLV. Senders
        // emit at most one tag-1 entry; only the first is honored so a hostile
        // frame cannot splice several. Unknown tags are ignored (forward
        // compatibility — see the tag registry on `EXT_TAG_B64_TAIL`).
        let mut content = wire.content;
        if let Some((_, raw)) = wire
            .ext
            .iter()
            .find(|(tag, _)| *tag == crate::wire::EXT_TAG_B64_TAIL)
        {
            content.push_str(&crate::wire::encode_b64_tail(raw));
        }
        let media_metadata =
            match wire.media_metadata_json {
                Some(bytes) => Some(serde_json::from_slice(&bytes).map_err(|e| {
                    crate::Error::DeserializationError(format!("media_metadata: {e}"))
                })?),
                None => None,
            };
        let forwarded_from =
            match wire.forwarded_from_json {
                Some(bytes) => Some(serde_json::from_slice(&bytes).map_err(|e| {
                    crate::Error::DeserializationError(format!("forwarded_from: {e}"))
                })?),
                None => None,
            };
        // Like tag 1, only the first tag-2 entry is honored; a malformed
        // payload rejects the frame, matching the JSON path where a malformed
        // `reply_context` field rejects the whole message.
        let reply_context =
            match wire
                .ext
                .iter()
                .find(|(tag, _)| *tag == crate::wire::EXT_TAG_REPLY_CONTEXT)
            {
                Some((_, bytes)) => Some(serde_json::from_slice(bytes).map_err(|e| {
                    crate::Error::DeserializationError(format!("reply_context: {e}"))
                })?),
                None => None,
            };
        Ok(Self {
            id: MessageId::from_bytes(wire.id),
            sender: UserId::new(wire.sender)?,
            recipient: UserId::new(wire.recipient)?,
            app_id: AppId::new(wire.app_id)?,
            priority: crate::wire::priority_from_u8(wire.priority),
            ttl: TTL::from_value(wire.ttl),
            hop_count: HopCount::from_value(wire.hop_count),
            timestamp: Timestamp::from_millis(wire.timestamp),
            lamport_clock: LamportClock::from_value(wire.lamport_clock),
            content_type: crate::wire::content_type_from_u8(wire.content_type),
            content,
            binary_content: wire.binary_content,
            media_metadata,
            metadata: wire.metadata.into_iter().collect(),
            requires_ack: wire.requires_ack,
            reply_to_msg: wire.reply_to_msg.map(MessageId::from_bytes),
            forwarded_from,
            reply_context,
            transport_peer_id: None,
            wire_codec: WireCodec::Json,
        })
    }
}

/// Builder for creating messages with custom settings.
pub struct MessageBuilder {
    sender: UserId,
    recipient: UserId,
    app_id: AppId,
    content_type: ContentType,
    content: String,
    media_metadata: Option<MediaMetadata>,
    priority: MessagePriority,
    ttl: TTL,
    lamport_clock: LamportClock,
    metadata: HashMap<String, String>,
    requires_ack: bool,
    reply_to_msg: Option<MessageId>,
    forwarded_from: Option<ForwardInfo>,
    reply_context: Option<ReplyContext>,
}

impl MessageBuilder {
    /// Creates a new message builder.
    pub fn new(sender: UserId, recipient: UserId, app_id: AppId) -> Self {
        Self {
            sender,
            recipient,
            app_id,
            content_type: ContentType::default(),
            content: String::new(),
            media_metadata: None,
            priority: MessagePriority::default(),
            ttl: TTL::default(),
            lamport_clock: LamportClock::default(),
            metadata: HashMap::new(),
            requires_ack: true,
            reply_to_msg: None,
            forwarded_from: None,
            reply_context: None,
        }
    }

    /// Sets the content type.
    pub fn content_type(mut self, content_type: ContentType) -> Self {
        self.content_type = content_type;
        self
    }

    /// Sets the message content.
    pub fn content(mut self, content: impl Into<String>) -> Self {
        self.content = content.into();
        self
    }

    /// Sets the media metadata.
    pub fn media_metadata(mut self, meta: MediaMetadata) -> Self {
        self.media_metadata = Some(meta);
        self
    }

    /// Sets the message priority.
    pub fn priority(mut self, priority: MessagePriority) -> Self {
        self.priority = priority;
        self
    }

    /// Sets the TTL.
    pub fn ttl(mut self, ttl: TTL) -> Self {
        self.ttl = ttl;
        self
    }

    /// Adds metadata to the message.
    pub fn metadata(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        self.metadata.insert(key.into(), value.into());
        self
    }

    /// Sets whether the message requires an ACK.
    pub fn requires_ack(mut self, requires_ack: bool) -> Self {
        self.requires_ack = requires_ack;
        self
    }

    /// Sets the message this is replying to.
    pub fn reply_to_msg(mut self, reply_to: MessageId) -> Self {
        self.reply_to_msg = Some(reply_to);
        self
    }

    /// Sets forwarding attribution on the message.
    pub fn forwarded_from(mut self, info: ForwardInfo) -> Self {
        self.forwarded_from = Some(info);
        self
    }

    /// Sets the quoted-reply context on the message.
    pub fn reply_context(mut self, ctx: ReplyContext) -> Self {
        self.reply_context = Some(ctx);
        self
    }

    /// Sets the Lamport clock value for this message.
    pub fn lamport_clock(mut self, clock: LamportClock) -> Self {
        self.lamport_clock = clock;
        self
    }

    /// Builds the message.
    pub fn build(self) -> Message {
        Message {
            id: MessageId::new(),
            sender: self.sender,
            recipient: self.recipient,
            app_id: self.app_id,
            priority: self.priority,
            ttl: self.ttl,
            hop_count: HopCount::new(),
            timestamp: Timestamp::now(),
            lamport_clock: self.lamport_clock,
            content_type: self.content_type,
            content: self.content,
            binary_content: None,
            media_metadata: self.media_metadata,
            metadata: self.metadata,
            requires_ack: self.requires_ack,
            reply_to_msg: self.reply_to_msg,
            forwarded_from: self.forwarded_from,
            reply_context: self.reply_context,
            transport_peer_id: None,
            wire_codec: WireCodec::Json,
        }
    }
}

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

    fn create_test_message() -> Message {
        Message::new(
            UserId::new("alice").unwrap(),
            UserId::new("bob").unwrap(),
            AppId::new("test-app").unwrap(),
            "Hello, Bob!",
        )
    }

    #[test]
    fn test_message_creation() {
        let msg = create_test_message();
        assert_eq!(msg.sender.as_str(), "alice");
        assert_eq!(msg.recipient.as_str(), "bob");
        assert_eq!(msg.content, "Hello, Bob!");
        assert_eq!(msg.priority, MessagePriority::Medium);
        assert_eq!(msg.hop_count.value(), 0);
    }

    #[test]
    fn test_message_builder() {
        let msg = Message::builder(
            UserId::new("alice").unwrap(),
            UserId::new("bob").unwrap(),
            AppId::new("test-app").unwrap(),
        )
        .content("Test message")
        .priority(MessagePriority::High)
        .metadata("key1", "value1")
        .requires_ack(false)
        .build();

        assert_eq!(msg.content, "Test message");
        assert_eq!(msg.priority, MessagePriority::High);
        assert_eq!(msg.metadata.get("key1").unwrap(), "value1");
        assert!(!msg.requires_ack);
    }

    #[test]
    fn reply_context_json_round_trip_and_default() {
        let mut msg = create_test_message();
        msg.reply_context = Some(ReplyContext {
            sender: UserId::new("carol").unwrap(),
            text: "quoted".into(),
            timestamp: Some(Timestamp::from_millis(1_700_000_000_000)),
            reply_media_label: None,
            reply_content_type: Some("text".into()),
        });
        let decoded = Message::from_bytes(&msg.to_bytes().unwrap()).unwrap();
        assert_eq!(decoded.reply_context, msg.reply_context);

        // Absent on the wire → defaults to None (old sender, new receiver).
        let plain = create_test_message();
        let json = serde_json::to_string(&plain).unwrap();
        assert!(!json.contains("reply_context"));
        let decoded = Message::from_bytes(json.as_bytes()).unwrap();
        assert!(decoded.reply_context.is_none());
    }

    #[test]
    fn media_metadata_without_secrets_clears_only_key_material() {
        let meta = MediaMetadata {
            mime_type: "image/jpeg".into(),
            file_name: "x.jpg".into(),
            file_size: 10,
            duration_ms: None,
            width: Some(1),
            height: Some(1),
            thumbnail_base64: None,
            media_id: Some("m1".into()),
            download_url: Some("https://cdn.example/m1".into()),
            thumbnail_url: Some("https://cdn.example/m1-t".into()),
            encryption_key: Some("a2V5".into()),
            iv: Some("aXY=".into()),
            ciphertext_hash: Some("aGFzaA==".into()),
            sticker_provider: Some("prov".into()),
            sticker_remote_id: Some("rid".into()),
            sticker_kind: Some("static".into()),
        };
        assert!(meta.has_secrets());

        let clean = meta.without_secrets();
        assert!(!clean.has_secrets());
        assert!(clean.encryption_key.is_none());
        assert!(clean.iv.is_none());
        // Everything that isn't key material survives.
        assert_eq!(clean.media_id, meta.media_id);
        assert_eq!(clean.download_url, meta.download_url);
        assert_eq!(clean.thumbnail_url, meta.thumbnail_url);
        assert_eq!(clean.ciphertext_hash, meta.ciphertext_hash);
        assert_eq!(clean.sticker_provider, meta.sticker_provider);
        assert_eq!(clean.sticker_remote_id, meta.sticker_remote_id);
        assert_eq!(clean.sticker_kind, meta.sticker_kind);
        assert_eq!(clean.file_name, meta.file_name);
    }

    #[test]
    fn message_json_ignores_unknown_fields() {
        // A legacy receiver's tolerance for fields it doesn't know is the
        // compat story for `reply_context` (new sender, old receiver); prove
        // Message keeps that tolerance.
        let mut json = serde_json::to_value(create_test_message()).unwrap();
        json["some_future_field"] = serde_json::Value::String("x".into());
        let bytes = serde_json::to_vec(&json).unwrap();
        assert!(Message::from_bytes(&bytes).is_ok());
    }

    #[test]
    fn test_message_hop_operations() {
        let mut msg = create_test_message();
        assert_eq!(msg.hop_count.value(), 0);

        msg.increment_hop().unwrap();
        assert_eq!(msg.hop_count.value(), 1);
    }

    #[test]
    fn test_message_ttl_operations() {
        let mut msg = create_test_message();
        let initial_ttl = msg.ttl.value();

        msg.decrement_ttl().unwrap();
        assert_eq!(msg.ttl.value(), initial_ttl - 1);
    }

    #[test]
    fn test_message_serialization() {
        let msg = create_test_message();

        // JSON serialization
        let json = msg.to_json().unwrap();
        let deserialized = Message::from_json(&json).unwrap();
        assert_eq!(msg.id, deserialized.id);
        assert_eq!(msg.sender, deserialized.sender);
        assert_eq!(msg.content, deserialized.content);

        // Binary serialization
        let bytes = msg.to_bytes().unwrap();
        let deserialized = Message::from_bytes(&bytes).unwrap();
        assert_eq!(msg.id, deserialized.id);
    }

    #[test]
    fn test_message_priority_score() {
        assert_eq!(MessagePriority::Low.score(), 1);
        assert_eq!(MessagePriority::Medium.score(), 2);
        assert_eq!(MessagePriority::High.score(), 3);
        assert_eq!(MessagePriority::Critical.score(), 4);
    }

    #[test]
    fn test_forward_info_serialization() {
        use crate::types::Timestamp;

        let info = ForwardInfo {
            original_sender: UserId::new("alice").unwrap(),
            original_message_id: MessageId::new(),
            original_timestamp: Timestamp::now(),
            forward_count: 1,
        };

        let msg = Message::builder(
            UserId::new("bob").unwrap(),
            UserId::new("charlie").unwrap(),
            AppId::new("test-app").unwrap(),
        )
        .content("Forwarded message")
        .forwarded_from(info.clone())
        .build();

        assert!(msg.forwarded_from.is_some());
        let fwd = msg.forwarded_from.as_ref().unwrap();
        assert_eq!(fwd.original_sender.as_str(), "alice");
        assert_eq!(fwd.forward_count, 1);

        // Round-trip through JSON
        let json = msg.to_json().unwrap();
        let deserialized = Message::from_json(&json).unwrap();
        let fwd2 = deserialized.forwarded_from.as_ref().unwrap();
        assert_eq!(fwd2.original_sender.as_str(), "alice");
        assert_eq!(fwd2.forward_count, 1);
    }

    #[test]
    fn test_backward_compat_no_forwarded_from() {
        // Old messages without forwarded_from should deserialize with None
        let msg = create_test_message();
        let json = msg.to_json().unwrap();

        // Verify forwarded_from is not in the JSON (skip_serializing_if = None)
        assert!(!json.contains("forwarded_from"));

        let deserialized = Message::from_json(&json).unwrap();
        assert!(deserialized.forwarded_from.is_none());
    }

    #[test]
    fn test_forward_chain_increment() {
        // Simulate: Alice sends original → Bob forwards (count=1) → Charlie forwards (count=2)
        let original_msg = Message::builder(
            UserId::new("alice").unwrap(),
            UserId::new("bob").unwrap(),
            AppId::new("test-app").unwrap(),
        )
        .content("Hello from Alice")
        .build();

        // Bob forwards to Charlie — ForwardInfo::from_message creates attribution
        let first_forward = ForwardInfo::from_message(&original_msg);
        assert_eq!(first_forward.original_sender.as_str(), "alice");
        assert_eq!(first_forward.original_message_id, original_msg.id);
        assert_eq!(first_forward.forward_count, 1);

        // Charlie forwards to Dave — should carry original_sender and increment count
        let forwarded_msg = Message::builder(
            UserId::new("bob").unwrap(),
            UserId::new("charlie").unwrap(),
            AppId::new("test-app").unwrap(),
        )
        .content("Hello from Alice")
        .forwarded_from(first_forward)
        .build();

        let second_forward = ForwardInfo::from_message(&forwarded_msg);
        assert_eq!(second_forward.original_sender.as_str(), "alice");
        assert_eq!(second_forward.forward_count, 2);
        assert_eq!(second_forward.original_message_id, original_msg.id);
    }

    #[test]
    fn test_forward_count_saturates_at_max() {
        use crate::types::Timestamp;

        let saturated = ForwardInfo {
            original_sender: UserId::new("alice").unwrap(),
            original_message_id: MessageId::new(),
            original_timestamp: Timestamp::now(),
            forward_count: u32::MAX,
        };

        let msg = Message::builder(
            UserId::new("bob").unwrap(),
            UserId::new("charlie").unwrap(),
            AppId::new("test-app").unwrap(),
        )
        .content("Hello")
        .forwarded_from(saturated)
        .build();

        let next = ForwardInfo::from_message(&msg);
        assert_eq!(next.forward_count, u32::MAX);
    }

    #[test]
    fn content_type_json_round_trips_every_variant() {
        // Iterating `ALL` (compile-time-pinned complete) rather than a
        // hand-written list is what makes this test able to catch a variant
        // added without a `ContentType::parse` arm — the one string mapping
        // rustc cannot check for itself.
        for ct in ContentType::ALL {
            let json = serde_json::to_string(&ct).unwrap();
            assert_eq!(json, format!("\"{ct}\""));
            assert_eq!(serde_json::from_str::<ContentType>(&json).unwrap(), ct);
            assert_eq!(ContentType::parse(ct.as_wire_str()), ct);
        }
    }

    #[test]
    fn content_type_round_trips_in_non_self_describing_formats() {
        // Both serde directions speak strings, so a format that carries no
        // field/variant names round-trips too. A derived `Serialize` would
        // write the variant index here, which the tolerant string-based
        // `Deserialize` would read back as "" and degrade to `File` —
        // silently turning `Text` into `File`.
        for ct in ContentType::ALL {
            let bytes = postcard::to_allocvec(&ct).unwrap();
            assert_eq!(postcard::from_bytes::<ContentType>(&bytes).unwrap(), ct);
        }
    }

    #[test]
    fn unknown_content_type_degrades_to_file_on_json_decode() {
        // A variant added by a newer sender must not fail this receiver's
        // decode: all three paths (JSON, `parse`, binary u8) share the
        // `File` fallback.
        assert_eq!(
            serde_json::from_str::<ContentType>("\"some_future_type\"").unwrap(),
            ContentType::File
        );
        // A non-string is malformed data, not additive evolution — still an
        // error.
        assert!(serde_json::from_str::<ContentType>("7").is_err());
    }

    #[test]
    fn message_with_unknown_content_type_still_decodes() {
        let msg = create_test_message();
        let mut value = serde_json::to_value(&msg).unwrap();
        value["content_type"] = serde_json::Value::String("some_future_type".to_string());
        let decoded: Message = serde_json::from_value(value).unwrap();
        assert_eq!(decoded.content_type, ContentType::File);
        assert_eq!(decoded.content, msg.content);
        assert_eq!(decoded.id, msg.id);
    }
}