Add empty unstyled block after code block

Ref #56

Author: mxstbr

src/modifiers/handleNewCodeBlock.js

@@ -1,5 +1,6 @@
 import changeCurrentBlockType from "./changeCurrentBlockType";
 import insertEmptyBlock from "./insertEmptyBlock";
+import splitBlockAndChange from './splitBlockAndChange';
 import { CODE_BLOCK_REGEX } from "../constants";
 
 const handleNewCodeBlock = editorState => {
@@ -19,7 +20,8 @@ const handleNewCodeBlock = editorState => {
     if (language) {
       data.language = language;
     }
-    return changeCurrentBlockType(editorState, "code-block", "", data);
+    const editorStateWithCodeBlock = changeCurrentBlockType(editorState, "code-block", "", data);
+    return splitBlockAndChange(editorStateWithCodeBlock, undefined, undefined, false)
   }
   const type = currentBlock.getType();
   if (type === "code-block" && isLast) {

src/modifiers/splitBlockAndChange.js

@@ -3,20 +3,21 @@ import { EditorState, Modifier } from "draft-js";
 const splitBlockAndChange = (
   editorState,
   type = "unstyled",
-  blockMetadata = {}
+  blockMetadata = {},
+  selectNewBlock = true
 ) => {
   let currentContent = editorState.getCurrentContent();
-  let selection = editorState.getSelection();
-  currentContent = Modifier.splitBlock(currentContent, selection);
-  selection = currentContent.getSelectionAfter();
+  const currentSelection = editorState.getSelection();
+  currentContent = Modifier.splitBlock(currentContent, currentSelection);
+  const selection = currentContent.getSelectionAfter();
   const key = selection.getStartKey();
   const blockMap = currentContent.getBlockMap();
   const block = blockMap.get(key);
   const data = block.getData().merge(blockMetadata);
   const newBlock = block.merge({ type, data });
   const newContentState = currentContent.merge({
     blockMap: blockMap.set(key, newBlock),
-    selectionAfter: selection,
+    selectionAfter: selectNewBlock ? selection : currentSelection,
   });
 
   return EditorState.push(editorState, newContentState, "split-block");