archive/md-pdf-rs
Archived
1
Fork 0
Code Releases Packages Activity

Set body class

Browse source
This commit is contained in:
Jake Howard 2017-09-24 17:28:15 +01:00
parent 1132c78234
commit a6d6185f9f
Signed by: jake
GPG key ID: 57AFB45680EDD477
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/processors/head_cleanup.rs
View file

@ -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 sciter::Element;
use config::Config; use config::Config;
@ -9,7 +9,12 @@ pub fn head_cleanup(config: Config, input: String) -> Result<String, String> {
destroy_matching(&mut head, "meta[content='text/css']"); destroy_matching(&mut head, "meta[content='text/css']");
destroy_matching(&mut head, "style"); destroy_matching(&mut head, "style");
destroy_matching(&mut head, "title"); destroy_matching(&mut head, "title");
let mut meta_charset = Element::create_at("meta", &mut head).unwrap(); let mut meta_charset = Element::create_at("meta", &mut head).unwrap();
meta_charset.set_attribute("charset", "UTF-8"); 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)); return Ok(get_html(root));
} }