Install wildcard handler if path includes '/404'

gatsbyjs · Mar 7, 2016 · f7067f6 · f7067f6
1 parent 1e44024
commit f7067f6
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib/isomorphic/create-routes.js b/lib/isomorphic/create-routes.js
@@ -20,6 +20,7 @@ module.exports = (files, pagesReq) => {
     pages,
     templates,
   }
+  let notFound = null
   const templatesHash = {}
   templatesHash.root = routes
   templatesHash['/'] = routes
@@ -159,7 +160,22 @@ module.exports = (files, pagesReq) => {
         parentTemplateFile,
       })
     }
+
+    if (includes(page.path, '/404')) {
+      notFound = {
+        path: '*',
+        component: handler,
+        page,
+        pages,
+        templates,
+        parentTemplateFile,
+      }
+    }
   })
 
+  if (notFound) {
+    routes.childRoutes.push(notFound)
+  }
+
   return routes
 }