license: Render HTML 404 page from /license

2 files changed, 9 insertions, 0 deletions

diff --git a/license-generator/src/bin/license.rs b/license-generator/src/bin/license.rs
index 4e3c42b..ca29260 100644
--- a/license-generator/src/bin/license.rs
+++ b/license-generator/src/bin/license.rs
@@ -58,6 +58,11 @@ where W: 'a + Write {
     }
 
     fn error_404(&mut self) {
+        let page_404 = include_str!("../../../404.html");
+        response::set_404(self.writer)
+            .and_then(|_|
+                Ok(write!(self.writer, "\n{}", page_404)?)
+            ).unwrap_or(())
     }
 
     fn error_500(&mut self, error: Option<Error>) {
diff --git a/license-generator/src/response.rs b/license-generator/src/response.rs
index 4e838bc..cf91e4e 100644
--- a/license-generator/src/response.rs
+++ b/license-generator/src/response.rs
@@ -6,6 +6,10 @@ pub fn set_403<W: Write>(w: &mut W) -> Result<()> {
     Ok(writeln!(w, "Status: 403")?)
 }
 
+pub fn set_404<W: Write>(w: &mut W) -> Result<()> {
+    Ok(writeln!(w, "Status: 404")?)
+}
+
 pub fn set_405<W: Write>(w: &mut W, allowed_methods: &str) -> Result<()> {
     Ok(
         writeln!(