fix: remove data server from build

doczjs · Mar 29, 2019 · 8715129 · 8715129
1 parent 84b23ea
commit 8715129
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/docz-core/templates/root.tpl.js b/core/docz-core/templates/root.tpl.js
@@ -1,5 +1,5 @@
 import React from 'react'
-import { Link, Router, Routes, useDataServer } from 'docz'
+import { Link, Router, Routes<% if (isProd) {%>, useDataServer<%}%> } from 'docz'
 <% if (!isProd) {%>import { hot } from 'react-hot-loader'<%}%>
 import Theme from '<%- theme %>'
 
@@ -8,7 +8,7 @@ import database from './db.json'
 <% if (wrapper) {%>import Wrapper from '<%- wrapper %>'<%}%>
 
 const Root = () => {
-  <% if (websocketUrl) {%>useDataServer('<%- websocketUrl %>')<%}%>
+  <% if (websocketUrl || isProd) {%>useDataServer('<%- websocketUrl %>')<%}%>
   return (
     <Theme
       <% if (wrapper) {%>wrapper={Wrapper}<%}%>