import {
$getBlockElementNodesInSelection,
$getNodeFromSelection,
- $insertNewBlockNodeAtSelection, $selectionContainsNodeType,
+ $insertNewBlockNodeAtSelection, $selectionContainsNodeType, $selectSingleNode,
$toggleSelectionBlockNodeType,
getLastSelection
} from "./selection";
editor.update(() => {
const codeBlock = $createCodeBlockNode();
codeBlock.setCode(selection?.getTextContent() || '');
- $insertNewBlockNodeAtSelection(codeBlock, true);
+
+ const selectionNodes = $getBlockElementNodesInSelection(selection);
+ const firstSelectionNode = selectionNodes[0];
+ const extraNodes = selectionNodes.slice(1);
+ if (firstSelectionNode) {
+ firstSelectionNode.replace(codeBlock);
+ extraNodes.forEach(n => n.remove());
+ } else {
+ $insertNewBlockNodeAtSelection(codeBlock, true);
+ }
+
$openCodeEditorForNode(editor, codeBlock);
- codeBlock.selectStart();
+ $selectSingleNode(codeBlock);
});
} else {
$openCodeEditorForNode(editor, codeBlock);