diff --git a/src/processors/head_cleanup.rs b/src/processors/head_cleanup.rs index 4de4f70..b44b793 100644 --- a/src/processors/head_cleanup.rs +++ b/src/processors/head_cleanup.rs @@ -1,4 +1,4 @@ -use html::{sciter_start, get_html, destroy_matching, destroy_at, get_head}; +use html::{sciter_start, get_html, destroy_matching, destroy_at, get_head, get_body}; use sciter::Element; use config::Config; @@ -9,7 +9,12 @@ pub fn head_cleanup(config: Config, input: String) -> Result { destroy_matching(&mut head, "meta[content='text/css']"); destroy_matching(&mut head, "style"); destroy_matching(&mut head, "title"); + let mut meta_charset = Element::create_at("meta", &mut head).unwrap(); meta_charset.set_attribute("charset", "UTF-8"); + + let mut body = get_body(&mut root); + body.set_attribute("class", "content").expect("Failed to set body class"); + return Ok(get_html(root)); }