From 53a1f43ce2301a83c4109564790c694751f4dab3 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Tue, 13 Nov 2018 17:14:52 +0100
Subject: license: Render HTML 404 page from /license

---
 license-generator/src/bin/license.rs | 5 +++++
 license-generator/src/response.rs    | 4 ++++
 2 files changed, 9 insertions(+)

(limited to 'license-generator')

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!(
-- 
cgit v1.2.3