fix: check for NodeSelection within posToDOMRect

ueberdosis · May 24, 2021 · a4ec4ff · a4ec4ff
1 parent 9e83413
commit a4ec4ff
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/core/src/helpers/posToDOMRect.ts b/packages/core/src/helpers/posToDOMRect.ts
@@ -1,7 +1,16 @@
+import isNodeSelection from './isNodeSelection'
 import { EditorView } from 'prosemirror-view'
 import coordsAtPos from './coordsAtPos'
 
 export default function posToDOMRect(view: EditorView, from: number, to: number): DOMRect {
+  if (isNodeSelection(view.state.selection)) {
+    const node = view.nodeDOM(from) as HTMLElement
+
+    if (node && node.getBoundingClientRect) {
+      return node.getBoundingClientRect()
+    }
+  }
+
   const start = coordsAtPos(view, from)
   const end = coordsAtPos(view, to, true)
   const top = Math.min(start.top, end.top)