From 9cd4f124d03b009ac4bacaea0bf1f8a130fbd689 Mon Sep 17 00:00:00 2001
From: Jake Howard <git@theorangeone.net>
Date: Wed, 6 Sep 2017 20:07:06 +0100
Subject: [PATCH] Change generator message

---
 src/processors/mod.rs     |  9 +++++----
 src/processors/rebrand.rs | 11 +++++++++++
 2 files changed, 16 insertions(+), 4 deletions(-)
 create mode 100644 src/processors/rebrand.rs

diff --git a/src/processors/mod.rs b/src/processors/mod.rs
index 50c9a75..9b70f2b 100644
--- a/src/processors/mod.rs
+++ b/src/processors/mod.rs
@@ -1,8 +1,9 @@
 use config::Config;
 
-pub mod head_cleanup;
-pub mod strip_blank;
+mod head_cleanup;
+mod strip_blank;
+mod rebrand;
 
 
-pub const PROCESSORS: [fn(Config, String) -> Result<String, String>; 2] =
-    [head_cleanup::head_cleanup, strip_blank::strip_blank];
+pub const PROCESSORS: [fn(Config, String) -> Result<String, String>; 3] =
+    [head_cleanup::head_cleanup, rebrand::rebrand, strip_blank::strip_blank];
diff --git a/src/processors/rebrand.rs b/src/processors/rebrand.rs
new file mode 100644
index 0000000..2f03db8
--- /dev/null
+++ b/src/processors/rebrand.rs
@@ -0,0 +1,11 @@
+use html::{sciter_start, get_html, get_head, find_first};
+use config::Config;
+
+
+pub fn rebrand(config: Config, input: String) -> Result<String, String> {
+    let mut root = sciter_start(input);
+    let mut head = get_head(&mut root);
+    let mut ele = find_first(&mut root, "meta[name='generator']");
+    ele.set_attribute("content", crate_name!());
+    return Ok(get_html(root));
+}