handle db page not found improved

src/routes/pages.js

@@ -2,15 +2,15 @@ const {renderPage} = require('../render/page')
 const {get} = require('koa-route')
 const getPathWithNoTrailingSlash = require('../lib/url').getPathWithNoTrailingSlash
 
-module.exports = function () {
+module.exports = function (handleNotFound) {
 	return get('*', async ctx => {
 		const path = getPathWithNoTrailingSlash(ctx.path)
 		const body = await renderPage(path)
 		if (body) {
 			ctx.type = 'text/html'
 			ctx.body = body
-		} else {
-			ctx.status = 404
-		}
+        } else {
+            await handleNotFound(ctx)
+        }
 	})
 }

src/server.js

@@ -33,6 +33,12 @@ const dirPublic = path.join(__dirname, '../public')
 
 handlebars.registerHelper('raw-helper', options => options.fn())
 
+function notFound() {
+    return async ctx => {
+        ctx.status = 404
+    }
+}
+
 export default function (config) {
 	const $ = convert
 
@@ -77,7 +83,7 @@ export default function (config) {
 	server.use(documents.getDocuments())
 	server.use(document.getDocument())
 	server.use(schedule.getSchedule())
-	server.use(pages())
+    server.use(pages(notFound()))
 
     return createServer(server.callback())
 }