xmtp_content_types/
markdown.rs

1use std::collections::HashMap;
2
3use xmtp_proto::xmtp::mls::message_contents::{ContentTypeId, EncodedContent};
4
5use super::{CodecError, ContentCodec};
6
7pub struct MarkdownCodec {}
8
9impl MarkdownCodec {
10    const AUTHORITY_ID: &'static str = "xmtp.org";
11    pub const TYPE_ID: &'static str = "markdown";
12    const ENCODING_KEY: &'static str = "encoding";
13    const ENCODING_UTF8: &'static str = "UTF-8";
14    pub const MAJOR_VERSION: u32 = 1;
15    pub const MINOR_VERSION: u32 = 0;
16}
17
18impl ContentCodec<String> for MarkdownCodec {
19    fn content_type() -> ContentTypeId {
20        ContentTypeId {
21            authority_id: MarkdownCodec::AUTHORITY_ID.to_string(),
22            type_id: MarkdownCodec::TYPE_ID.to_string(),
23            version_major: MarkdownCodec::MAJOR_VERSION,
24            version_minor: MarkdownCodec::MINOR_VERSION,
25        }
26    }
27
28    fn encode(markdown: String) -> Result<EncodedContent, CodecError> {
29        Ok(EncodedContent {
30            r#type: Some(MarkdownCodec::content_type()),
31            parameters: HashMap::from([(
32                MarkdownCodec::ENCODING_KEY.to_string(),
33                MarkdownCodec::ENCODING_UTF8.to_string(),
34            )]),
35            fallback: None,
36            compression: None,
37            content: markdown.into_bytes(),
38        })
39    }
40
41    fn decode(content: EncodedContent) -> Result<String, CodecError> {
42        let encoding = content
43            .parameters
44            .get(MarkdownCodec::ENCODING_KEY)
45            .map_or(MarkdownCodec::ENCODING_UTF8, String::as_str);
46        if encoding != MarkdownCodec::ENCODING_UTF8 {
47            return Err(CodecError::Decode(format!(
48                "Unsupported text encoding {encoding}"
49            )));
50        }
51        let markdown = std::str::from_utf8(&content.content)
52            .map_err(|utf8_err| CodecError::Decode(utf8_err.to_string()))?;
53        Ok(markdown.to_string())
54    }
55
56    fn should_push() -> bool {
57        true
58    }
59}
60
61#[cfg(test)]
62pub(crate) mod tests {
63    use crate::{ContentCodec, markdown::MarkdownCodec};
64
65    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
66    #[cfg_attr(not(target_arch = "wasm32"), test)]
67    fn can_encode_and_decode_markdown() {
68        let markdown = "# Hello, world!";
69        let encoded_content =
70            MarkdownCodec::encode(markdown.to_string()).expect("Should encode successfully");
71        let decoded_content =
72            MarkdownCodec::decode(encoded_content).expect("Should decode successfully");
73        assert!(decoded_content == markdown);
74    }
75}
xmtp_content_types/markdown.rs

xmtp_content_types/
markdown.rs
