From a7b1e7c4663b808737dabddd9d7dbadc3dafaffd Mon Sep 17 00:00:00 2001
From: Jake Howard <git@theorangeone.net>
Date: Wed, 6 Sep 2017 17:52:39 +0100
Subject: [PATCH] Add specific helper functions

---
 src/renderers/html_cleanup.rs | 4 ++--
 src/renderers/mod.rs          | 8 ++++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/renderers/html_cleanup.rs b/src/renderers/html_cleanup.rs
index e4ada8e..c4d6aa6 100644
--- a/src/renderers/html_cleanup.rs
+++ b/src/renderers/html_cleanup.rs
@@ -1,10 +1,10 @@
-use renderers::{sciter_start, get_html, destroy_matching, destroy_at, find_first};
+use renderers::{sciter_start, get_html, destroy_matching, destroy_at, get_head};
 use config::Config;
 
 
 pub fn html_cleanup(config: Config, input: String) -> Result<String, String> {
     let mut root = sciter_start(input);
-    let mut head = find_first(&mut root, "head");
+    let mut head = get_head(&mut root);
     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");
diff --git a/src/renderers/mod.rs b/src/renderers/mod.rs
index 6e33d56..f2b6f0c 100644
--- a/src/renderers/mod.rs
+++ b/src/renderers/mod.rs
@@ -51,3 +51,11 @@ fn destroy_matching(root: &mut Element, selector: &str) {
         ele.destroy().expect("Failed to delete");
     }
 }
+
+fn get_head(root: &mut Element) -> Element {
+    return find_first(root, "head");
+}
+
+fn get_body(root: &mut Element) -> Element {
+    return find_first(root, "body");
+}