From 12f0f4bee6b47b85cb7b592ea9db006d0c3c0373 Mon Sep 17 00:00:00 2001
From: b1ek <me@blek.codes>
Date: Wed, 22 Feb 2023 16:42:06 +1000
Subject: [PATCH] add 404

---
 index.js | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/index.js b/index.js
index 6e7fd93..cbdddf5 100644
--- a/index.js
+++ b/index.js
@@ -21,15 +21,6 @@ let RedisStore = require("connect-redis")(session)
 
 const { APP_PORT, APP_KEY } = process.env;
 
-app.use((req, res, next) => {
-	req.start = Date.now();
-	res.on('header', (res) => {
-		let time = Date.now() - req.start;
-		console.log(time)
-		res.setHeader('X-Reponse-Time', time);
-	})
-	next();
-});
 app.use(bodyparser.json());
 app.use(bodyparser.urlencoded({ extended: true }));
 app.use(cookie_parse(APP_KEY))
@@ -47,8 +38,17 @@ app.use(require('./routes'));
 app.use(express.static('public'));
 
 // 404
-app.use((req, res, next) => {
-
+app.use(async (req, res, next) => {
+	try {
+		if (res.headersSent) return next();
+		const Helpers = require('./helpers');
+		res.status(404).send(await Helpers.ViewLoader.load('error.pug', {
+			error: '404 Not Found',
+			message: 'The requested page was not found.'
+		}))
+	} catch (err) { 
+		next(err);
+	}
 })