diff --git a/components/site/src/lib.rs b/components/site/src/lib.rs
index 53ebc9d..b217795 100644
--- a/components/site/src/lib.rs
+++ b/components/site/src/lib.rs
@@ -526,6 +526,7 @@ impl Site {
if self.config.generate_rss {
self.render_rss_feed()?;
}
+ self.render_404()?;
self.render_robots()?;
// `render_categories` and `render_tags` will check whether the config allows
// them to render or not
@@ -660,6 +661,15 @@ impl Site {
Ok(())
}
+ /// Renders 404.html
+ pub fn render_404(&self) -> Result<()> {
+ ensure_directory_exists(&self.output_path)?;
+ create_file(
+ &self.output_path.join("404.html"),
+ &render_template("404.html", &self.tera, &Context::new(), &self.config.theme)?
+ )
+ }
+
/// Renders robots.txt
pub fn render_robots(&self) -> Result<()> {
ensure_directory_exists(&self.output_path)?;
diff --git a/components/templates/src/builtins/404.html b/components/templates/src/builtins/404.html
new file mode 100644
index 0000000..f7d50b1
--- /dev/null
+++ b/components/templates/src/builtins/404.html
@@ -0,0 +1,10 @@
+
+
+
+ File Not Found: 404.
+
+
+ Oops!
+ File Not Found: 404.
+
+
diff --git a/components/templates/src/lib.rs b/components/templates/src/lib.rs
index 1c49403..ce0fe91 100644
--- a/components/templates/src/lib.rs
+++ b/components/templates/src/lib.rs
@@ -24,6 +24,7 @@ lazy_static! {
pub static ref GUTENBERG_TERA: Tera = {
let mut tera = Tera::default();
tera.add_raw_templates(vec![
+ ("404.html", include_str!("builtins/404.html")),
("rss.xml", include_str!("builtins/rss.xml")),
("sitemap.xml", include_str!("builtins/sitemap.xml")),
("robots.txt", include_str!("builtins/robots.txt")),
diff --git a/src/cmd/serve.rs b/src/cmd/serve.rs
index af22cc7..a5158e5 100644
--- a/src/cmd/serve.rs
+++ b/src/cmd/serve.rs
@@ -59,9 +59,11 @@ enum ChangeKind {
// errors
const LIVE_RELOAD: &'static str = include_str!("livereload.js");
-struct ErrCatcher;
+struct NotFoundHandler {
+ rendered_template: PathBuf,
+}
-impl Middleware for ErrCatcher {
+impl Middleware for NotFoundHandler {
fn start(&self, _req: &mut HttpRequest) -> actix_web::Result {
Ok(Started::Done)
}
@@ -72,17 +74,14 @@ impl Middleware for ErrCatcher {
mut resp: HttpResponse,
) -> actix_web::Result {
if http::StatusCode::NOT_FOUND == resp.status() {
- let not_found_page = "static/error/404.html";
- if let Ok(mut fh) = File::open(¬_found_page) {
- println!("Using {} to handle missing file.", ¬_found_page);
- let mut buf: Vec = vec![];
- let _ = fh.read_to_end(&mut buf)?;
- resp.replace_body(buf);
- resp.headers_mut().insert(
- http::header::CONTENT_TYPE,
- http::header::HeaderValue::from_static("text/html"),
- );
- }
+ let mut fh = File::open(&self.rendered_template)?;
+ let mut buf: Vec = vec![];
+ let _ = fh.read_to_end(&mut buf)?;
+ resp.replace_body(buf);
+ resp.headers_mut().insert(
+ http::header::CONTENT_TYPE,
+ http::header::HeaderValue::from_static("text/html"),
+ );
}
Ok(Response::Done(resp))
}
@@ -184,7 +183,7 @@ pub fn serve(interface: &str, port: &str, output_dir: &str, base_url: &str, conf
let sys = actix::System::new("http-server");
server::new(move || {
App::new()
- .middleware(ErrCatcher)
+ .middleware(NotFoundHandler { rendered_template: static_root.join("404.html") })
.resource(r"/livereload.js", |r| r.f(livereload_handler))
// Start a webserver that serves the `output_dir` directory
.handler(r"/", fs::StaticFiles::new(&static_root)