site: fix npm run site error

ashearer · Oct 31, 2018 · 1f0263c · 1f0263c
1 parent 9e236fc
commit 1f0263c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/site/theme/template/Content/Demo.jsx b/site/theme/template/Content/Demo.jsx
@@ -56,9 +56,12 @@ export default class Demo extends React.Component {
 
   getSourceCode() {
     const { highlightedCode } = this.props;
-    const div = document.createElement('div');
-    div.innerHTML = highlightedCode[1].highlighted;
-    return div.textContent;
+    if (typeof document !== 'undefined') {
+      const div = document.createElement('div');
+      div.innerHTML = highlightedCode[1].highlighted;
+      return div.textContent;
+    }
+    return '';
   }
 
   handleCodeExpand = () => {