NodeViewWrapper forwardRef

2021-06-11 08:04:53 +02:00
parent 2502932fa0
commit 83525ec4f1
1 changed files with 17 additions and 14 deletions
--- a/packages/react/src/NodeViewWrapper.tsx
+++ b/packages/react/src/NodeViewWrapper.tsx
@@ -6,13 +6,15 @@ export interface NodeViewWrapperProps {
  as?: React.ElementType,
 }

-export const NodeViewWrapper: React.FC<NodeViewWrapperProps> = props => {
+export const NodeViewWrapper: React.FC<NodeViewWrapperProps> =
+  React.forwardRef((props, ref) => {
    const { onDragStart } = useReactNodeView()
    const Tag = props.as || 'div'

    return (
      <Tag
        {...props}
+        ref={ref}
        data-node-view-wrapper=""
        onDragStart={onDragStart}
        style={{
@@ -22,3 +24,4 @@ export const NodeViewWrapper: React.FC<NodeViewWrapperProps> = props => {
      />
    )
  }
+}