fix(gatsby-theme-docz): hot reload

doczjs · Feb 19, 2019 · bf4e440 · bf4e440
1 parent bc1303d
commit bf4e440
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/core/docz/src/utils/createState.tsx b/core/docz/src/utils/createState.tsx
@@ -41,9 +41,6 @@ export function create<T = any>(initial: T): State<T> {
       public componentWillUnmount(): void {
         listeners.clear()
       }
-      public shouldComponentUpdate(nextProps: any, nextState: any): boolean {
-        return !equal(this.state, nextState)
-      }
       public render(): React.ReactNode {
         return (
           <ctx.Provider value={this.state}>{this.props.children}</ctx.Provider>

diff --git a/core/gatsby-theme-docz/package.json b/core/gatsby-theme-docz/package.json
@@ -28,6 +28,7 @@
     "gatsby-plugin-compile-es6-packages": "^1.0.6",
     "gatsby-plugin-manifest": "^2.0.17",
     "gatsby-plugin-react-helmet": "^3.0.6",
+    "lodash": "^4.17.11",
     "prop-types": "^15.7.2",
     "react": "^16.8.2",
     "react-dom": "^16.8.2",

diff --git a/core/gatsby-theme-docz/src/node/createPagesStatefully.js b/core/gatsby-theme-docz/src/node/createPagesStatefully.js
@@ -2,10 +2,11 @@ const path = require('path')
 const chokidar = require('chokidar')
 const { Entries } = require('docz-core')
 const { parseConfig } = require('../utils/parseConfig')
+const { get } = require('lodash/fp')
 
 const getEntry = async (filepath, entries) => {
   const map = await entries.get()
-  return map[filepath]
+  return get(filepath, map)
 }
 
 const mountRoute = (base = '/', route) => {
@@ -25,8 +26,8 @@ module.exports = async ({ store, actions }, opts) => {
   })
 
   const handleCreatePage = async filepath => {
-    const component = path.join(paths.root, filepath)
     const entry = await getEntry(filepath, entries)
+    const component = path.join(paths.root, filepath)
 
     if (entry) {
       createPage({
@@ -50,10 +51,16 @@ module.exports = async ({ store, actions }, opts) => {
     }
   }
 
+  const allEntries = await entries.get()
+  await Promise.all(
+    Object.values(allEntries).map(async value => {
+      handleCreatePage(value.filepath)
+    })
+  )
+
   return new Promise((resolve, reject) => {
     watcher
       .on('add', handleCreatePage)
-      .on('change', handleCreatePage)
       .on('unlink', handleDeletePage)
       .on('ready', () => resolve())
       .on('error', err => reject(err))