From ba5108694d5ebe71f5bb5ace1f7d96edbd8e3e41 Mon Sep 17 00:00:00 2001
From: Pedro Nauck <pedronauck@gmail.com>
Date: Sun, 2 Sep 2018 19:00:28 -0300
Subject: [PATCH] fix(docz-core): dispose hmr on imports

---
 packages/docz-core/templates/index.tpl.js | 8 +-------
 packages/docz-core/templates/root.tpl.js  | 4 +++-
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/packages/docz-core/templates/index.tpl.js b/packages/docz-core/templates/index.tpl.js
index 6ae7900a0..7c4a71482 100644
--- a/packages/docz-core/templates/index.tpl.js
+++ b/packages/docz-core/templates/index.tpl.js
@@ -1,8 +1,6 @@
 <% if (!isProd) {%>import 'webpack-serve-overlay'<%}%>
 import React from 'react'
 import ReactDOM from 'react-dom'
-
-import { imports } from './imports'
 import Root from './root'
 
 const _onPreRenders = [<% if (onPreRenders) {%><%- onPreRenders %><%}%>]
@@ -14,11 +12,7 @@ const onPostRender = () => _onPostRenders.forEach(f => f && f())
 const root = document.querySelector('#root')
 const render = (Component = Root) => {
   onPreRender()
-  ReactDOM.render(<Component imports={imports} />, root, onPostRender)
-}
-
-if (module.hot) {
-  module.hot.accept('./imports', () => render(Root))
+  ReactDOM.render(<Component />, root, onPostRender)
 }
 
 render(Root)
diff --git a/packages/docz-core/templates/root.tpl.js b/packages/docz-core/templates/root.tpl.js
index b5c402795..8dc77181e 100644
--- a/packages/docz-core/templates/root.tpl.js
+++ b/packages/docz-core/templates/root.tpl.js
@@ -1,11 +1,13 @@
 import React from 'react'
 import { hot } from 'react-hot-loader'
 import Theme from '<%- theme %>'
+
+import { imports } from './imports'
 import db from './db.json'
 
 <% if (wrapper) {%>import Wrapper from '<%- wrapper %>'<%}%>
 
-const Root = ({ imports }) => (
+const Root = () => (
   <Theme
     db={db}
     imports={imports}