choreo-content 0.2.3

//! Content protocol: protobuf `ItemMessage` encoding/decoding, mixin
//! construction, deterministic item-id derivation, and CID <-> sha2-256-digest
//! conversion.
//!
//! On-chain content is a protobuf `ItemMessage`: an ordered list of
//! `MixinPayloadMessage { mixin_id, payload }`, each tagged with a 32-bit mixin
//! ID. The semantic *type* of an item is implied by which mixins are present
//! (a `feed`/`comment` type marker, a `profile` mixin, etc.), so the same
//! encoding serves a decentralized GitHub, a decentralized Wiki, or arbitrary
//! content — only the mixin set differs.
//!
//! The 32-byte `IpfsHash` stored on-chain is the sha2-256 digest of an item's
//! bytes, extracted from an IPFS `CIDv0` multihash (`0x12 0x20 || digest`). This
//! module converts between the CID (Base58, as IPFS returns) and the digest hex
//! (`0x`-prefixed) that is what actually reaches the chain events.

use prost::Message;
use sp_crypto_hashing::blake2_256;

use crate::config::ITEM_ID_NAMESPACE;

// ── Mixin ID constants (pinned to the reference `acuity-dioxus` protocol) ────

/// BCP-47 language tag.
pub const LANGUAGE_MIXIN_ID: u32 = 0x9bc7_a0e6;
/// Human-readable title.
pub const TITLE_MIXIN_ID: u32 = 0x344f_4812;
/// Body text (markdown/plain).
pub const BODY_TEXT_MIXIN_ID: u32 = 0x2d38_2044;
/// Image (full-res + mipmap levels).
pub const IMAGE_MIXIN_ID: u32 = 0x045e_ee8c;
/// Profile (account type + location), used alongside title/body/language.
pub const PROFILE_MIXIN_ID: u32 = 0xbeef_2144;
/// Feed-type marker (empty payload).
pub const FEED_TYPE_MIXIN_ID: u32 = 0xbcec_8faa;
/// Comment-type marker (empty payload).
pub const COMMENT_TYPE_MIXIN_ID: u32 = 0x874a_ba65;

/// Default BCP-47 language tag applied when the caller omits one.
pub const DEFAULT_LANGUAGE_TAG: &str = "en";

// ── Protobuf message types ──────────────────────────────────────────────────

/// The top-level content envelope: an ordered list of mixins.
#[derive(Clone, PartialEq, Message)]
pub struct ItemMessage {
    #[prost(message, repeated, tag = "1")]
    pub mixin_payload: Vec<MixinPayloadMessage>,
}

/// A single tagged mixin: a 32-bit type discriminator plus its payload.
#[derive(Clone, PartialEq, Message)]
pub struct MixinPayloadMessage {
    #[prost(fixed32, tag = "1")]
    pub mixin_id: u32,
    #[prost(bytes = "vec", tag = "2")]
    pub payload: Vec<u8>,
}

/// BCP-47 language tag.
#[derive(Clone, PartialEq, Message)]
pub struct LanguageMixinMessage {
    #[prost(string, tag = "1")]
    pub language_tag: String,
}

/// Human-readable title.
#[derive(Clone, PartialEq, Message)]
pub struct TitleMixinMessage {
    #[prost(string, tag = "1")]
    pub title: String,
}

/// Body text.
#[derive(Clone, PartialEq, Message)]
pub struct BodyTextMixinMessage {
    #[prost(string, tag = "1")]
    pub body_text: String,
}

/// An image reference plus its mipmap pyramid (each level a separate IPFS CID).
#[derive(Clone, PartialEq, Message)]
pub struct ImageMixinMessage {
    #[prost(string, tag = "1")]
    pub filename: String,
    #[prost(uint64, tag = "2")]
    pub filesize: u64,
    #[prost(bytes = "vec", tag = "3")]
    pub ipfs_hash: Vec<u8>,
    #[prost(uint32, tag = "4")]
    pub width: u32,
    #[prost(uint32, tag = "5")]
    pub height: u32,
    #[prost(message, repeated, tag = "6")]
    pub mipmap_level: Vec<MipmapLevelMessage>,
}

/// One mipmap level of an [`ImageMixinMessage`].
#[derive(Clone, PartialEq, Message)]
pub struct MipmapLevelMessage {
    #[prost(uint64, tag = "1")]
    pub filesize: u64,
    #[prost(bytes = "vec", tag = "2")]
    pub ipfs_hash: Vec<u8>,
}

/// Account profile mixin: signed account type + free-text location.
#[derive(Clone, PartialEq, Message)]
pub struct ProfileMixinMessage {
    #[prost(int32, tag = "1")]
    pub account_type: i32,
    #[prost(string, tag = "2")]
    pub location: String,
}

/// The account-type taxonomy encoded in a [`ProfileMixinMessage`].
#[derive(Clone, Copy, Debug, PartialEq, Eq, prost::Enumeration)]
#[repr(i32)]
pub enum AccountType {
    Anon = 0,
    Person = 1,
    Project = 2,
    Organization = 3,
    Proxy = 4,
    Parody = 5,
    Bot = 6,
    Shill = 7,
    Test = 8,
}

// ── Content type and structured item I/O ────────────────────────────────────

/// Fixed, protocol-level content-type vocabulary. The type is *encoded* by
/// which mixins an item carries; this enum is the closure over the known
/// marker/profile mixins. `Document` is the default (title + body + language).
#[derive(
    Clone,
    Copy,
    Debug,
    Default,
    PartialEq,
    Eq,
    schemars::JsonSchema,
    serde::Serialize,
    serde::Deserialize,
)]
#[serde(rename_all = "lowercase")]
pub enum ContentType {
    /// Title + body + language — the default item shape.
    #[default]
    Document,
    Feed,
    Comment,
    Profile,
    Image,
}

/// A mipmap level of an image, as uploaded to IPFS (a pre-encoded CID per
/// level).
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct MipmapLevel {
    /// Size of the level's raw bytes.
    pub filesize: u64,
    /// IPFS CID (Base58) of this level.
    pub cid: String,
}

/// Structured image reference carried in an item.
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct ImageSpec {
    pub filename: String,
    pub filesize: u64,
    /// sha2-256 digest hex (`0x`-prefixed) of the full-resolution image.
    pub digest_hex: String,
    pub width: u32,
    pub height: u32,
    pub mipmap_levels: Vec<MipmapLevel>,
}

/// Structured profile fields.
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct ProfileSpec {
    /// Account type (0..=8, see [`AccountType`]).
    pub account_type: i32,
    pub location: String,
}

/// Caller-supplied image reference for a publish.
///
/// The easy path is `path`: the daemon reads the file, builds the JPEG mipmap
/// pyramid, uploads every level to IPFS, and derives the full [`ImageSpec`]
/// (see [`crate::image::build_image_spec`]) — the ported `acuity-dioxus`
/// publishing pipeline. Callers that already have a complete spec (e.g. re-
/// publishing a previously resolved image) can supply `spec` directly.
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct ImageInput {
    /// Local file path of the image to read, process, and upload. Mutually
    /// exclusive with `spec`.
    pub path: Option<String>,
    /// Stored filename override; defaults to the file name in `path`.
    pub filename: Option<String>,
    /// A fully-specified image reference, used verbatim. Mutually exclusive
    /// with `path`.
    pub spec: Option<ImageSpec>,
}

/// A [`ContentInput`] whose image reference has been resolved into a complete
/// [`ImageSpec`] — the input [`encode_item`] actually encodes. Built via
/// [`ContentInput::to_prepared`] (usually through the orchestration layer,
/// which performs the IPFS work for `path`-based inputs).
#[derive(Clone, Debug, Default, PartialEq, Eq)]
pub struct PreparedContent {
    pub content_type: ContentType,
    pub title: Option<String>,
    pub body: Option<String>,
    /// BCP-47 language tag; defaults to [`DEFAULT_LANGUAGE_TAG`].
    pub language: Option<String>,
    pub image: Option<ImageSpec>,
    pub profile: Option<ProfileSpec>,
}

/// Input to a publish: everything a caller wants in a new item revision.
///
/// `image` is an [`ImageInput`] so callers can just point at a local file;
/// it is resolved (reading + uploading as needed) into a [`PreparedContent`]
/// before encoding.
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct ContentInput {
    pub content_type: ContentType,
    pub title: Option<String>,
    pub body: Option<String>,
    /// BCP-47 language tag; defaults to [`DEFAULT_LANGUAGE_TAG`].
    pub language: Option<String>,
    pub image: Option<ImageInput>,
    pub profile: Option<ProfileSpec>,
}

impl ContentInput {
    /// Merge this input with an already-resolved image spec (the result of
    /// resolving [`ImageInput::path`], or `spec` copied verbatim) into the
    /// fully-specified form [`encode_item`] consumes.
    #[must_use]
    pub fn to_prepared(&self, image: Option<ImageSpec>) -> PreparedContent {
        PreparedContent {
            content_type: self.content_type,
            title: self.title.clone(),
            body: self.body.clone(),
            language: self.language.clone(),
            image,
            profile: self.profile.clone(),
        }
    }
}

/// The decoded, field-extracted view of an item (what a read tool returns).
#[derive(
    Clone, Debug, Default, PartialEq, Eq, schemars::JsonSchema, serde::Serialize, serde::Deserialize,
)]
pub struct DecodedItem {
    /// The type inferred from the present marker/profile mixins.
    pub content_type: ContentType,
    pub title: Option<String>,
    pub body: Option<String>,
    pub language: Option<String>,
    pub image: Option<ImageSpec>,
    pub profile: Option<ProfileSpec>,
}

/// Encode a [`ContentInput`] into the protobuf `ItemMessage` bytes, tagging the
/// right mixin set for the chosen [`ContentType`].
///
/// Returns an error when an embedded image's digest hex or mipmap CID is
/// malformed (see [`encode_image_mixin`]); a well-formed item always succeeds.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Cid`] when the embedded image's
/// `digest_hex` or any mipmap CID is malformed (propagated from
/// [`encode_image_mixin`]); otherwise never fails.
pub fn encode_item(input: &PreparedContent) -> Result<Vec<u8>, crate::ContentError> {
    let language = input
        .language
        .clone()
        .unwrap_or_else(|| DEFAULT_LANGUAGE_TAG.to_string());

    let mut mixins: Vec<MixinPayloadMessage> = Vec::new();

    // Type markers come first (empty payloads for feed/comment).
    match input.content_type {
        ContentType::Feed => mixins.push(marker(FEED_TYPE_MIXIN_ID)),
        ContentType::Comment => mixins.push(marker(COMMENT_TYPE_MIXIN_ID)),
        _ => {}
    }

    // Ordinal content mixins.
    mixins.push(MixinPayloadMessage {
        mixin_id: LANGUAGE_MIXIN_ID,
        payload: LanguageMixinMessage {
            language_tag: language,
        }
        .encode_to_vec(),
    });

    if let Some(title) = &input.title {
        mixins.push(MixinPayloadMessage {
            mixin_id: TITLE_MIXIN_ID,
            payload: TitleMixinMessage {
                title: title.clone(),
            }
            .encode_to_vec(),
        });
    }
    if let Some(body) = &input.body {
        mixins.push(MixinPayloadMessage {
            mixin_id: BODY_TEXT_MIXIN_ID,
            payload: BodyTextMixinMessage {
                body_text: body.clone(),
            }
            .encode_to_vec(),
        });
    }
    if let Some(image) = &input.image {
        mixins.push(MixinPayloadMessage {
            mixin_id: IMAGE_MIXIN_ID,
            payload: encode_image_mixin(image)?,
        });
    }
    if let Some(profile) = &input.profile {
        mixins.push(MixinPayloadMessage {
            mixin_id: PROFILE_MIXIN_ID,
            payload: ProfileMixinMessage {
                account_type: profile.account_type,
                location: profile.location.clone(),
            }
            .encode_to_vec(),
        });
    }

    Ok(ItemMessage {
        mixin_payload: mixins,
    }
    .encode_to_vec())
}

/// Decode `ItemMessage` bytes into an extracted [`DecodedItem`]. Unknown mixins
/// are ignored (forward compatibility); a missing/ill-formed mixin degrades
/// that field to `None` rather than failing the whole decode.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Content`] only when the top-level
/// `ItemMessage` protobuf cannot be decoded from `bytes`; ill-formed
/// individual mixins degrade their field to `None` instead.
pub fn decode_item(bytes: &[u8]) -> Result<DecodedItem, crate::ContentError> {
    let item = ItemMessage::decode(bytes)
        .map_err(|e| crate::ContentError::Content(format!("failed to decode item payload: {e}")))?;

    let mut out = DecodedItem {
        content_type: infer_content_type(&item),
        ..Default::default()
    };

    for mixin in &item.mixin_payload {
        match mixin.mixin_id {
            LANGUAGE_MIXIN_ID => {
                out.language = LanguageMixinMessage::decode(mixin.payload.as_slice())
                    .ok()
                    .map(|m| m.language_tag);
            }
            TITLE_MIXIN_ID => {
                out.title = TitleMixinMessage::decode(mixin.payload.as_slice())
                    .ok()
                    .map(|m| m.title);
            }
            BODY_TEXT_MIXIN_ID => {
                out.body = BodyTextMixinMessage::decode(mixin.payload.as_slice())
                    .ok()
                    .map(|m| m.body_text);
            }
            IMAGE_MIXIN_ID => {
                out.image = ImageMixinMessage::decode(mixin.payload.as_slice())
                    .ok()
                    .map(decode_image_mixin);
            }
            PROFILE_MIXIN_ID => {
                out.profile = ProfileMixinMessage::decode(mixin.payload.as_slice())
                    .ok()
                    .map(|m| ProfileSpec {
                        account_type: m.account_type,
                        location: m.location,
                    });
            }
            _ => {}
        }
    }
    Ok(out)
}

/// Infer the [`ContentType`] of a decoded item from its marker/profile mixins.
#[must_use]
pub fn infer_content_type(item: &ItemMessage) -> ContentType {
    for mixin in &item.mixin_payload {
        match mixin.mixin_id {
            FEED_TYPE_MIXIN_ID => return ContentType::Feed,
            COMMENT_TYPE_MIXIN_ID => return ContentType::Comment,
            PROFILE_MIXIN_ID => return ContentType::Profile,
            _ => {}
        }
    }
    if item
        .mixin_payload
        .iter()
        .any(|m| m.mixin_id == IMAGE_MIXIN_ID)
    {
        return ContentType::Image;
    }
    ContentType::Document
}

/// Build an empty-payload marker mixin.
fn marker(mixin_id: u32) -> MixinPayloadMessage {
    MixinPayloadMessage {
        mixin_id,
        payload: Vec::new(),
    }
}

/// Encode an [`ImageSpec`] into the `ImageMixinMessage` payload bytes.
///
/// Returns an error if the image digest hex or any mipmap CID is malformed,
/// rather than silently encoding an empty `ipfs_hash`. A bad image reference
/// must fail the whole publish so it can never land on-chain.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Cid`] when the image's `digest_hex` is
/// not 32 bytes of `0x` hex (via [`multihash_bytes`]) or any mipmap level's
/// `cid` is not a sha2-256 `CIDv0` multihash (via [`cid_to_multihash_bytes`]).
pub fn encode_image_mixin(image: &ImageSpec) -> Result<Vec<u8>, crate::ContentError> {
    let mipmap_levels = image
        .mipmap_levels
        .iter()
        .map(|l| {
            Ok(MipmapLevelMessage {
                filesize: l.filesize,
                // The reference client (`acuity-dioxus`) stores the FULL
                // 34-byte multihash (`0x12 0x20 ‖ digest`) in these fields
                // and turns them into CIDs with a bare `bs58::encode` — so
                // the header must be present here or its `api/v0/cat` calls
                // get a base58-of-raw-digest string that Kubo rejects (500).
                ipfs_hash: cid_to_multihash_bytes(&l.cid)?,
            })
        })
        .collect::<Result<Vec<_>, crate::ContentError>>()?;

    let message = ImageMixinMessage {
        filename: image.filename.clone(),
        filesize: image.filesize,
        ipfs_hash: multihash_bytes(&image.digest_hex)?,
        width: image.width,
        height: image.height,
        mipmap_level: mipmap_levels,
    };
    Ok(message.encode_to_vec())
}

/// Decode an `ImageMixinMessage` payload into an [`ImageSpec`], converting each
/// level's digest back to a CID for display.
#[must_use]
pub fn decode_image_mixin(msg: ImageMixinMessage) -> ImageSpec {
    // Accept both the reference form (34-byte multihash with the `0x12 0x20`
    // header) and the legacy form this crate briefly wrote (bare 32-byte
    // digest) so older items keep decoding.
    ImageSpec {
        filename: msg.filename,
        filesize: msg.filesize,
        digest_hex: bytes_to_hex(&digest_from_bytes(&msg.ipfs_hash)),
        width: msg.width,
        height: msg.height,
        mipmap_levels: msg
            .mipmap_level
            .into_iter()
            .map(|l| MipmapLevel {
                filesize: l.filesize,
                cid: bytes_to_cid(&digest_from_bytes(&l.ipfs_hash)),
            })
            .collect(),
    }
}

/// Normalize a mixin hash field to the bare 32-byte digest: strip the sha2-256
/// multihash header when present, pass raw 32-byte digests through unchanged.
fn digest_from_bytes(bytes: &[u8]) -> Vec<u8> {
    // The len check above guarantees indices 0..2 are in bounds.
    if bytes.len() == 34 && bytes.first() == Some(&0x12) && bytes.get(1) == Some(&0x20) {
        bytes.get(2..).unwrap_or(bytes).to_vec()
    } else {
        bytes.to_vec()
    }
}

/// Extract a single mixin's payload by ID.
#[must_use]
pub fn decode_single_mixin<M>(item: &ItemMessage, mixin_id: u32) -> Option<M>
where
    M: Message + Default,
{
    item.mixin_payload
        .iter()
        .find(|m| m.mixin_id == mixin_id)
        .and_then(|m| M::decode(m.payload.as_slice()).ok())
}

// ── Item-id derivation ──────────────────────────────────────────────────────

/// Derive a deterministic item ID from the publishing account, a caller-
/// supplied nonce, and the fixed namespace:
///
/// `blake2_256(SCALE(account_id) ++ SCALE(nonce) ++ SCALE(namespace=1000))`
///
/// The result matches what `pallet-content` computes, so a client can predict
/// an item's ID before it exists and subscribe to it.
#[must_use]
pub fn derive_item_id(account_id: [u8; 32], nonce: [u8; 32]) -> [u8; 32] {
    let payload = [
        parity_scale_codec::Encode::encode(&account_id),
        parity_scale_codec::Encode::encode(&nonce),
        parity_scale_codec::Encode::encode(&ITEM_ID_NAMESPACE),
    ]
    .concat();
    blake2_256(&payload)
}

// ── Hex / CID helpers ──────────────────────────────────────────────────────

/// `[u8; 32]` -> `0x`-prefixed hex string.
#[must_use]
pub fn bytes_to_hex(bytes: &[u8]) -> String {
    format!("0x{}", hex::encode(bytes))
}

/// `0x`-prefixed hex (32 bytes) -> `[u8; 32]`.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Cid`] when `hex_value` is not valid hex
/// or does not decode to exactly 32 bytes.
pub fn hex_to_bytes(hex_value: &str) -> Result<[u8; 32], crate::ContentError> {
    let raw = hex::decode(hex_value.trim_start_matches("0x"))
        .map_err(|_| crate::ContentError::Cid(format!("invalid hex value {hex_value}")))?;
    raw.try_into()
        .map_err(|_| crate::ContentError::Cid(format!("expected 32 bytes for {hex_value}")))
}

/// sha2-256 digest hex (`0x`) -> IPFS `CIDv0` (Base58), prefixing the
/// `0x12 0x20` multihash header.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Cid`] (via [`hex_to_bytes`]) when
/// `hex_value` is not exactly 32 bytes of `0x`-prefixed hex.
pub fn digest_hex_to_cid(hex_value: &str) -> Result<String, crate::ContentError> {
    let digest = hex_to_bytes(hex_value)?;
    let mut multihash = Vec::with_capacity(34);
    multihash.push(0x12);
    multihash.push(0x20);
    multihash.extend_from_slice(&digest);
    Ok(bs58::encode(multihash).into_string())
}

/// IPFS `CIDv0` (Base58 sha2-256) -> `0x`-prefixed digest hex.
///
/// # Errors
///
/// Fails with [`crate::ContentError::Cid`] when `cid` is not valid Base58,
/// does not decode to a 34-byte multihash with the `0x12 0x20` sha2-256
/// header, or the digest cannot round-trip through [`hex_to_bytes`].
pub fn cid_to_digest_hex(cid: &str) -> Result<String, crate::ContentError> {
    let multihash = bs58::decode(cid)
        .into_vec()
        .map_err(|_| crate::ContentError::Cid(format!("failed to decode CID {cid}")))?;
    // The != 34 check above guarantees the header indices are in bounds.
    if multihash.len() != 34 || multihash.first() != Some(&0x12) || multihash.get(1) != Some(&0x20)
    {
        return Err(crate::ContentError::Cid(format!(
            "CID {cid} is not a sha2-256 CIDv0 multihash"
        )));
    }
    Ok(format!(
        "0x{}",
        hex::encode(multihash.get(2..).unwrap_or(&multihash))
    ))
}

/// `0x`-prefixed digest hex -> raw digest bytes, for the protobuf mixin.
fn multihash_bytes(hex_value: &str) -> Result<Vec<u8>, crate::ContentError> {
    let digest = hex_to_bytes(hex_value)?;
    let mut multihash = Vec::with_capacity(34);
    multihash.push(0x12);
    multihash.push(0x20);
    multihash.extend_from_slice(&digest);
    Ok(multihash)
}

/// IPFS `CIDv0` -> full multihash bytes (with the `0x12 0x20` header), for the
/// protobuf mixin — the reference wire form. Validates the multihash shape
/// and the 32-byte digest length like [`cid_to_digest_bytes`].
fn cid_to_multihash_bytes(cid: &str) -> Result<Vec<u8>, crate::ContentError> {
    let digest = cid_to_digest_bytes(cid)?;
    let mut multihash = Vec::with_capacity(34);
    multihash.push(0x12);
    multihash.push(0x20);
    multihash.extend_from_slice(&digest);
    Ok(multihash)
}

/// Raw 32-byte digest -> IPFS `CIDv0` (Base58).
#[must_use]
pub fn bytes_to_cid(digest: &[u8]) -> String {
    let mut multihash = Vec::with_capacity(34);
    multihash.push(0x12);
    multihash.push(0x20);
    multihash.extend_from_slice(digest);
    bs58::encode(multihash).into_string()
}

/// IPFS `CIDv0` -> raw 32-byte digest, for the protobuf mixin. Performs both the
/// multihash-form validation (in [`cid_to_digest_hex`]) and the 32-byte length
/// check (in [`hex_to_bytes`]), returning an error instead of an empty payload
/// on any malformed input.
fn cid_to_digest_bytes(cid: &str) -> Result<Vec<u8>, crate::ContentError> {
    Ok(hex_to_bytes(&cid_to_digest_hex(cid)?)?.to_vec())
}

/// Abbreviate a long hex string to `first10...last8` for display.
#[must_use]
pub fn short_hex(value: &str) -> String {
    if value.len() <= 18 {
        value.to_string()
    } else {
        // Hex strings are ASCII, so these are always char boundaries; .get
        // keeps the truncation total against any future non-hex input.
        let head = value.get(..10).unwrap_or("");
        let tail = value.get(value.len().saturating_sub(8)..).unwrap_or("");
        format!("{head}...{tail}")
    }
}

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

    #[test]
    fn derive_item_id_is_deterministic_and_nonce_sensitive() {
        let account = [7u8; 32];
        let nonce_a = [1u8; 32];
        let nonce_b = [2u8; 32];

        assert_eq!(
            derive_item_id(account, nonce_a),
            derive_item_id(account, nonce_a)
        );
        assert_ne!(
            derive_item_id(account, nonce_a),
            derive_item_id(account, nonce_b)
        );
        // Changing the account changes the id.
        assert_ne!(
            derive_item_id(account, nonce_a),
            derive_item_id([8u8; 32], nonce_a)
        );
        // Non-zero.
        assert_ne!(derive_item_id(account, nonce_a), [0u8; 32]);
    }

    #[test]
    fn hex_helpers_round_trip_and_validate() {
        let bytes = [0xabu8; 32];
        let encoded = bytes_to_hex(&bytes);
        assert_eq!(encoded, format!("0x{}", "ab".repeat(32)));
        assert_eq!(hex_to_bytes(&encoded).unwrap(), bytes);
        // string_slice has no test-allowance config; the 0x prefix is ASCII
        // so the .get fallback is unreachable.
        assert_eq!(hex_to_bytes(encoded.get(2..).unwrap_or("")).unwrap(), bytes);

        assert!(hex_to_bytes("0xzz").is_err());
        assert!(hex_to_bytes("0x1234").is_err());
    }

    #[test]
    fn cid_helpers_round_trip() {
        let digest = format!("0x{}", "11".repeat(32));
        let cid = digest_hex_to_cid(&digest).unwrap();
        assert_eq!(cid_to_digest_hex(&cid).unwrap(), digest);
        // short_hex keeps the 0x prefix in its leading 10 chars ("0x" + 8 nibbles).
        assert_eq!(
            short_hex(&digest),
            format!("0x{}...{}", "11111111", "11111111")
        );
    }

    #[test]
    fn cid_rejects_non_sha256_multihash() {
        let not_sha256 = bs58::encode(
            [0x13u8, 0x20]
                .into_iter()
                .chain([0u8; 32])
                .collect::<Vec<_>>(),
        )
        .into_string();
        assert!(cid_to_digest_hex(&not_sha256).is_err());
    }

    #[test]
    fn image_mixin_stores_reference_multihash_form() {
        // acuity-dioxus turns `mipmap_level.ipfs_hash` into a CID with a bare
        // `bs58::encode`, so the bytes must be a full 34-byte sha2-256
        // multihash — not the bare digest (which yields an unparseable CID
        // and a Kubo 500). Regression for the interop break.
        let digest = format!("0x{}", "22".repeat(32));
        let cid = digest_hex_to_cid(&digest).unwrap();
        let input = spec_input("photo.jpg", &digest, &cid);
        let payload = encode_image_mixin(prepare(&input).image.as_ref().unwrap()).unwrap();
        let msg = ImageMixinMessage::decode(payload.as_slice()).unwrap();

        let expected_multihash = [[0x12u8, 0x20].as_slice(), &[0x22u8; 32]].concat();
        assert_eq!(msg.ipfs_hash, expected_multihash);
        assert_eq!(msg.mipmap_level[0].ipfs_hash, expected_multihash);

        // Round-trips through decode (both spec fields come back as before).
        let decoded = decode_image_mixin(msg);
        assert_eq!(decoded.digest_hex, digest);
        assert_eq!(decoded.mipmap_levels[0].cid, cid);
    }

    #[test]
    fn decode_image_mixin_accepts_legacy_bare_digests() {
        // Items published before the multihash fix carried the raw 32-byte
        // digest; they must still decode to the same spec fields.
        let msg = ImageMixinMessage {
            filename: "legacy.jpg".into(),
            filesize: 1,
            ipfs_hash: vec![0x22; 32],
            width: 10,
            height: 10,
            mipmap_level: vec![MipmapLevelMessage {
                filesize: 1,
                ipfs_hash: vec![0x22; 32],
            }],
        };
        let decoded = decode_image_mixin(msg);
        assert_eq!(decoded.digest_hex, format!("0x{}", "22".repeat(32)));
        assert_eq!(
            decoded.mipmap_levels[0].cid,
            digest_hex_to_cid(&format!("0x{}", "22".repeat(32))).unwrap()
        );
    }

    #[test]
    fn encode_decode_document_round_trip() {
        let input = ContentInput {
            content_type: ContentType::Document,
            title: Some("Hello".into()),
            body: Some("World".into()),
            language: Some("en".into()),
            image: None,
            profile: None,
        };
        let bytes = encode_item(&input.to_prepared(None)).unwrap();
        let decoded = decode_item(&bytes).unwrap();

        assert_eq!(decoded.content_type, ContentType::Document);
        assert_eq!(decoded.title.as_deref(), Some("Hello"));
        assert_eq!(decoded.body.as_deref(), Some("World"));
        assert_eq!(decoded.language.as_deref(), Some("en"));
        assert!(decoded.image.is_none());
        assert!(decoded.profile.is_none());
    }

    #[test]
    fn encode_decode_feed_includes_marker() {
        let input = ContentInput {
            content_type: ContentType::Feed,
            title: Some("Feed".into()),
            body: None,
            language: None,
            image: None,
            profile: None,
        };
        let item =
            ItemMessage::decode(encode_item(&input.to_prepared(None)).unwrap().as_slice()).unwrap();
        assert_eq!(infer_content_type(&item), ContentType::Feed);
        assert!(
            item.mixin_payload
                .iter()
                .any(|m| m.mixin_id == FEED_TYPE_MIXIN_ID && m.payload.is_empty())
        );
        // Default language applied.
        assert_eq!(
            decode_single_mixin::<LanguageMixinMessage>(&item, LANGUAGE_MIXIN_ID)
                .unwrap()
                .language_tag,
            DEFAULT_LANGUAGE_TAG
        );
    }

    #[test]
    fn encode_decode_comment_and_profile_types() {
        let comment = ContentInput {
            content_type: ContentType::Comment,
            title: None,
            body: Some("a comment".into()),
            language: None,
            image: None,
            profile: None,
        };
        let c_item =
            ItemMessage::decode(encode_item(&comment.to_prepared(None)).unwrap().as_slice())
                .unwrap();
        assert_eq!(infer_content_type(&c_item), ContentType::Comment);

        let profile = ContentInput {
            content_type: ContentType::Profile,
            title: Some("Alice".into()),
            body: Some("bio".into()),
            language: None,
            image: None,
            profile: Some(ProfileSpec {
                account_type: AccountType::Project as i32,
                location: "Earth".into(),
            }),
        };
        let p_item =
            ItemMessage::decode(encode_item(&profile.to_prepared(None)).unwrap().as_slice())
                .unwrap();
        assert_eq!(infer_content_type(&p_item), ContentType::Profile);
        let decoded = decode_item(&encode_item(&profile.to_prepared(None)).unwrap()).unwrap();
        assert_eq!(decoded.content_type, ContentType::Profile);
        assert_eq!(decoded.title.as_deref(), Some("Alice"));
        assert_eq!(
            decoded.profile.as_ref().map(|p| p.location.as_str()),
            Some("Earth")
        );
        assert_eq!(decoded.profile.as_ref().map(|p| p.account_type), Some(2));
    }

    /// A fully-specified image input built from the parts the tests vary.
    fn spec_input(filename: &str, digest_hex: &str, cid: &str) -> ContentInput {
        ContentInput {
            content_type: ContentType::Image,
            title: None,
            body: None,
            language: None,
            image: Some(ImageInput {
                path: None,
                filename: None,
                spec: Some(ImageSpec {
                    filename: filename.into(),
                    filesize: 12345,
                    digest_hex: digest_hex.into(),
                    width: 800,
                    height: 600,
                    mipmap_levels: vec![MipmapLevel {
                        filesize: 100,
                        cid: cid.into(),
                    }],
                }),
            }),
            profile: None,
        }
    }

    /// Prepare a [`ContentInput`] for encoding the way the orchestration
    /// layer does: copy a `spec`-based image through, drop `path`-based ones
    /// (those need IPFS and are covered in `image.rs`'s tests instead).
    fn prepare(input: &ContentInput) -> PreparedContent {
        input.to_prepared(input.image.as_ref().and_then(|i| i.spec.clone()))
    }

    #[test]
    fn encode_decode_image_round_trip() {
        let digest = format!("0x{}", "22".repeat(32));
        let input = spec_input("photo.jpg", &digest, &digest_hex_to_cid(&digest).unwrap());
        let prepared = prepare(&input);
        let item = ItemMessage::decode(encode_item(&prepared).unwrap().as_slice()).unwrap();
        assert_eq!(infer_content_type(&item), ContentType::Image);
        let decoded = decode_item(&encode_item(&prepared).unwrap()).unwrap();
        let image = decoded.image.expect("image should decode");
        assert_eq!(image.filename, "photo.jpg");
        assert_eq!(image.filesize, 12345);
        assert_eq!(image.width, 800);
        assert_eq!(image.height, 600);
        // Round-trips the full-res digest and the mipmap CID.
        assert_eq!(image.digest_hex, digest);
        assert_eq!(image.mipmap_levels.len(), 1);
        assert_eq!(image.mipmap_levels[0].filesize, 100);
        assert_eq!(
            image.mipmap_levels[0].cid,
            digest_hex_to_cid(&digest).unwrap()
        );
    }

    #[test]
    fn decode_unknown_mixins_are_ignored() {
        let mut item = ItemMessage {
            mixin_payload: vec![MixinPayloadMessage {
                mixin_id: 0xdead_beef,
                payload: vec![1, 2, 3],
            }],
        };
        // A valid title mixin alongside the unknown one.
        item.mixin_payload.push(MixinPayloadMessage {
            mixin_id: TITLE_MIXIN_ID,
            payload: TitleMixinMessage {
                title: "kept".into(),
            }
            .encode_to_vec(),
        });
        let decoded = decode_item(&item.encode_to_vec()).unwrap();
        assert_eq!(decoded.title.as_deref(), Some("kept"));
        assert_eq!(decoded.content_type, ContentType::Document);
    }

    #[test]
    fn encode_item_rejects_malformed_image_digest() {
        // A malformed image digest must fail the whole encode rather than
        // silently producing a payload with an empty `ipfs_hash`.
        let input = spec_input("bad.jpg", "0xnothex", "unused");
        assert!(encode_item(&prepare(&input)).is_err());
    }

    #[test]
    fn encode_item_rejects_malformed_mipmap_cid() {
        // A malformed mipmap CID must fail the whole encode too.
        let input = spec_input("b.jpg", &format!("0x{}", "22".repeat(32)), "not-a-cid");
        assert!(encode_item(&prepare(&input)).is_err());
    }
}