aboutsummaryrefslogtreecommitdiff
path: root/lib/mastodon2text/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to 'lib/mastodon2text/src/lib.rs')
-rw-r--r--lib/mastodon2text/src/lib.rs39
1 files changed, 39 insertions, 0 deletions
diff --git a/lib/mastodon2text/src/lib.rs b/lib/mastodon2text/src/lib.rs
new file mode 100644
index 0000000..0359c8a
--- /dev/null
+++ b/lib/mastodon2text/src/lib.rs
@@ -0,0 +1,39 @@
+use lol_html::{element, html_content::ContentType, HtmlRewriter, Settings};
+use std::error::Error;
+
+pub fn convert(input: String) -> Result<String, Box<dyn Error>> {
+ let mut output = Vec::new();
+
+ let mut rewriter = HtmlRewriter::new(
+ Settings {
+ element_content_handlers: vec![
+ element!("span", |el| {
+ el.remove_and_keep_content();
+ Ok(())
+ }),
+ element!("p", |el| {
+ el.append(" ", ContentType::Html);
+ el.remove_and_keep_content();
+ Ok(())
+ }),
+ element!("br", |el| {
+ el.append(" ", ContentType::Html);
+ el.remove_and_keep_content();
+ Ok(())
+ }),
+ element!("a[href]", |el| {
+ el.remove_and_keep_content();
+
+ Ok(())
+ }),
+ ],
+ ..Settings::default()
+ },
+ |c: &[u8]| output.extend_from_slice(c),
+ );
+
+ rewriter.write(input.as_bytes())?;
+ rewriter.end()?;
+
+ Ok(String::from_utf8_lossy(&output).to_string())
+}