diff options
-rw-r--r-- | license-generator/src/bin/license.rs | 5 | ||||
-rw-r--r-- | license-generator/src/response.rs | 4 |
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!( |