diff --git a/src/renderers/html_cleanup.rs b/src/renderers/html_cleanup.rs index b315782..d07f156 100644 --- a/src/renderers/html_cleanup.rs +++ b/src/renderers/html_cleanup.rs @@ -1,12 +1,16 @@ -use renderers::{sciter_start, get_html, destroy_matching}; +use renderers::{sciter_start, get_html, destroy_matching, destroy_at, find_first}; use config::Config; +use sciter::{Window}; +use std::ops::Deref; +use std::rc::Rc; pub fn html_cleanup(config: Config, input: String) -> Result { let mut root = sciter_start(input); - destroy_matching(&mut root, "meta[content='text/css']"); - destroy_matching(&mut root, "style"); - destroy_matching(&mut root, "title"); - let html = get_html(root); - return Ok(html); + let mut head = find_first(&mut root, "head"); + destroy_at(&mut head, 4); // Sciter doesnt like finding this style tag for some reason + destroy_matching(&mut head, "meta[content='text/css']"); + destroy_matching(&mut head, "style"); + destroy_matching(&mut head, "title"); + return Ok(get_html(root)); }