Simplify editOn* helpers and improve consistency

Author: acusti

src/component/handlers/edit/editOnInput.js

@@ -20,7 +20,6 @@ var DraftModifier = require('DraftModifier');
 var DraftOffsetKey = require('DraftOffsetKey');
 var EditorState = require('EditorState');
 var UserAgent = require('UserAgent');
-var ReactDOM = require('ReactDOM');
 
 var findAncestorOffsetKey = require('findAncestorOffsetKey');
 var nullthrows = require('nullthrows');
@@ -47,9 +46,8 @@ function editOnInput(editor: DraftEditor): void {
     editor._pendingStateFromBeforeInput = undefined;
   }
 
-  var domSelection = ReactDOM.findDOMNode(
-    editor.refs.editor,
-  ).ownerDocument.defaultView.getSelection();
+  const editorNode = nullthrows(editor.editor);
+  const domSelection = editorNode.ownerDocument.defaultView.getSelection();
 
   var {anchorNode, isCollapsed} = domSelection;
   const isNotTextNode = anchorNode.nodeType !== Node.TEXT_NODE;

src/component/handlers/edit/editOnSelect.js

@@ -16,10 +16,9 @@
 import type DraftEditor from 'DraftEditor.react';
 
 var EditorState = require('EditorState');
-var ReactDOM = require('ReactDOM');
 
 var getDraftEditorSelection = require('getDraftEditorSelection');
-const invariant = require('invariant');
+const nullthrows = require('nullthrows');
 
 function editOnSelect(editor: DraftEditor): void {
   if (
@@ -30,16 +29,8 @@ function editOnSelect(editor: DraftEditor): void {
   }
 
   var editorState = editor.props.editorState;
-  const editorNode = ReactDOM.findDOMNode(editor.editorContainer);
-  invariant(editorNode, 'Missing editorNode');
-  invariant(
-    editorNode.firstChild.nodeType === 1,
-    'editorNode.firstChild is not an Element',
-  );
-  var documentSelection = getDraftEditorSelection(
-    editorState,
-    editorNode.firstChild,
-  );
+  const editorNode = nullthrows(editor.editor);
+  var documentSelection = getDraftEditorSelection(editorState, editorNode);
   var updatedSelectionState = documentSelection.selectionState;
 
   if (updatedSelectionState !== editorState.getSelection()) {