refactor(parser): wrap some objects and elements with struct

src/objects/inline_src.rs

@@ -1,52 +1,71 @@
 use memchr::{memchr, memchr2};
 
-/// returns (language, option, body, offset)
-#[inline]
-pub fn parse(text: &str) -> Option<(&str, Option<&str>, &str, usize)> {
-    debug_assert!(text.starts_with("src_"));
-
-    let (lang, off) = memchr2(b'[', b'{', text.as_bytes())
-        .map(|i| (&text[4..i], i))
-        .filter(|(lang, off)| {
-            *off != 4 && lang.as_bytes().iter().all(|c| !c.is_ascii_whitespace())
-        })?;
-
-    let (option, off) = if text.as_bytes()[off] == b'[' {
-        memchr(b']', text[off..].as_bytes())
-            .filter(|&i| text[off..off + i].as_bytes().iter().all(|c| *c != b'\n'))
-            .map(|i| (Some(&text[off + 1..off + i]), off + i + 1))?
-    } else {
-        (None, off)
-    };
-
-    let (body, off) = memchr(b'}', text[off..].as_bytes())
-        .map(|i| (&text[off + 1..off + i], off + i + 1))
-        .filter(|(body, _)| body.as_bytes().iter().all(|c| *c != b'\n'))?;
-
-    Some((lang, option, body, off))
+#[cfg_attr(test, derive(PartialEq))]
+#[derive(Debug)]
+pub struct InlineSrc<'a> {
+    pub lang: &'a str,
+    pub option: Option<&'a str>,
+    pub body: &'a str,
 }
 
-#[cfg(test)]
-mod tests {
-    #[test]
-    fn parse() {
-        use super::parse;
+impl<'a> InlineSrc<'a> {
+    #[inline]
+    pub(crate) fn parse(text: &str) -> Option<(InlineSrc<'_>, usize)> {
+        debug_assert!(text.starts_with("src_"));
 
-        assert_eq!(
-            parse("src_C{int a = 0;}"),
-            Some(("C", None, "int a = 0;", "src_C{int a = 0;}".len()))
-        );
-        assert_eq!(
-            parse("src_xml[:exports code]{<tag>text</tag>}"),
-            Some((
-                "xml",
-                Some(":exports code"),
-                "<tag>text</tag>",
-                "src_xml[:exports code]{<tag>text</tag>}".len()
-            ))
-        );
-        assert_eq!(parse("src_xml[:exports code]{<tag>text</tag>"), None);
-        assert_eq!(parse("src_[:exports code]{<tag>text</tag>}"), None);
-        // assert_eq!(parse("src_xml[:exports code]"), None);
+        let (lang, off) = memchr2(b'[', b'{', text.as_bytes())
+            .map(|i| (&text["src_".len()..i], i))
+            .filter(|(lang, off)| {
+                *off != 4 && lang.as_bytes().iter().all(|c| !c.is_ascii_whitespace())
+            })?;
+
+        let (option, off) = if text.as_bytes()[off] == b'[' {
+            memchr(b']', text[off..].as_bytes())
+                .filter(|&i| text[off..off + i].as_bytes().iter().all(|c| *c != b'\n'))
+                .map(|i| (Some(&text[off + 1..off + i]), off + i + 1))?
+        } else {
+            (None, off)
+        };
+
+        let (body, off) = memchr(b'}', text[off..].as_bytes())
+            .map(|i| (&text[off + 1..off + i], off + i + 1))
+            .filter(|(body, _)| body.as_bytes().iter().all(|c| *c != b'\n'))?;
+
+        Some((InlineSrc { lang, option, body }, off))
     }
 }
+
+#[test]
+fn parse() {
+    assert_eq!(
+        InlineSrc::parse("src_C{int a = 0;}"),
+        Some((
+            InlineSrc {
+                lang: "C",
+                option: None,
+                body: "int a = 0;"
+            },
+            "src_C{int a = 0;}".len()
+        ))
+    );
+    assert_eq!(
+        InlineSrc::parse("src_xml[:exports code]{<tag>text</tag>}"),
+        Some((
+            InlineSrc {
+                lang: "xml",
+                option: Some(":exports code"),
+                body: "<tag>text</tag>",
+            },
+            "src_xml[:exports code]{<tag>text</tag>}".len()
+        ))
+    );
+    assert_eq!(
+        InlineSrc::parse("src_xml[:exports code]{<tag>text</tag>"),
+        None
+    );
+    assert_eq!(
+        InlineSrc::parse("src_[:exports code]{<tag>text</tag>}"),
+        None
+    );
+    // assert_eq!(parse("src_xml[:exports code]"), None);
+}