Update 404 Message

2 files changed, 9 insertions, 3 deletions

diff --git a/app.js b/app.js
index a834f26..68ca3cf 100644
--- a/app.js
+++ b/app.js
@@ -105,7 +105,7 @@ class App {
                                         res.send(this.readfile.GetMain(output.get(this.path.string).meta))
                                     }
                                 } else {
-                                    res.sendStatus(404)
+                                    res.status(404).send(this.readfile.GetMain({title: 'Page not found', description: '', keywords: ''}))
                                 }
                                 break
                         }
@@ -165,7 +165,7 @@ class App {
                             }
                         }
                     } else {
-                        res.send('Not Found')
+                        res.send(JSON.stringify({html: this.readfile.NotFound(), css: ['main.css'], js: ['purify.js', '404.js']}))
                     }
                 })
             })
@@ -184,7 +184,10 @@ class App {
     }
     #FileOpen(x) {
         x.res.set('Content-Type', x.mime)
-        x.data.file.on('open', () => x.data.file.pipe(x.res)).on('error', () => x.res.sendStatus(404))
+        x.data.file.on('open', () => x.data.file.pipe(x.res)).on('error', () => {
+            x.res.set('Content-Type', mime.html)
+            x.res.status(404).send('File not found')
+        })
     }
     #ValidateCookie(req, res, next) {
         const { cookies } = req
diff --git a/source/readfile.js b/source/readfile.js
index def56a1..3cf0a7a 100644
--- a/source/readfile.js
+++ b/source/readfile.js
@@ -90,6 +90,9 @@ class ReadFile {
         
         return streamToPromise(Readable.from(object.links).pipe(object.stream)).then(data => {return data.toString()})
     }
+    NotFound() {
+        return fs.readFileSync('./views/layouts/404.html', 'utf-8')
+    }
     GetFile(x) {
         const path = x.split('/')
         const object = {