fix(docz-theme-default): refresh editor after it's mounted (#406) (#506)

doczjs · Dec 13, 2018 · 06b0721 · 06b0721
1 parent 4ba0e52
commit 06b0721
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/docz-theme-default/src/components/ui/Editor/index.tsx b/packages/docz-theme-default/src/components/ui/Editor/index.tsx
@@ -103,7 +103,7 @@ export class Editor extends Component<EditorProps, EditorState> {
     const editorProps = (config: any) => ({
       value: this.state.code,
       className: editorClassName,
-      editorDidMount: this.removeLastLine,
+      editorDidMount: this.onEditorDidMount,
       onBeforeChange: this.handleChange,
       options: {
         ...options,
@@ -124,6 +124,13 @@ export class Editor extends Component<EditorProps, EditorState> {
     )
   }
 
+  private onEditorDidMount = (editor: any) => {
+    if (editor) {
+      this.removeLastLine(editor)
+      editor.refresh()
+    }
+  }
+
   private removeLastLine = (editor: any) => {
     if (editor && !this.props.withLastLine && this.props.readOnly) {
       const lastLine = editor.lastLine()