From 1c8baefce74640c26214752105dc635988be08e7 Mon Sep 17 00:00:00 2001
From: Luke Childs <lukechilds123@gmail.com>
Date: Fri, 13 Jan 2017 11:54:55 +0700
Subject: [PATCH] Use our error page for 404s

Otherwise non existent routes fall back to the default express 404
---
 controllers/error404.js | 5 +++++
 controllers/index.js    | 3 ++-
 index.js                | 1 +
 3 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 controllers/error404.js

diff --git a/controllers/error404.js b/controllers/error404.js
new file mode 100644
index 0000000..8835b36
--- /dev/null
+++ b/controllers/error404.js
@@ -0,0 +1,5 @@
+module.exports = (req, res) => {
+  const statusCode = 404;
+  const error = '404 Not Found';
+  res.status(statusCode).render('error.html', { error });
+};
diff --git a/controllers/index.js b/controllers/index.js
index cd15edd..a1ebb81 100644
--- a/controllers/index.js
+++ b/controllers/index.js
@@ -3,5 +3,6 @@ module.exports = {
   node:         require('./node'),
   about:        require('./about'),
   noConnection: require('./no-connection'),
-  error:        require('./error')
+  error:        require('./error'),
+  error404:     require('./error404')
 };
diff --git a/index.js b/index.js
index 8f3a37f..90dfff5 100644
--- a/index.js
+++ b/index.js
@@ -39,6 +39,7 @@ app.use('/sw.js', express.static(`${__dirname}/public/sw.js`, { maxAge: '1 hour'
 app.use(express.static(`${__dirname}/public`, { maxAge: '1 year' }));
 
 // Errors
+app.use(controllers.error404);
 app.use(controllers.error);
 
 // Start app