fix: allow passing props to NodeViewWrapper and NodeViewContent

ueberdosis · Apr 23, 2021 · cfb3b80 · cfb3b80
1 parent bd9e15d
commit cfb3b80
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/packages/react/src/NodeViewContent.tsx b/packages/react/src/NodeViewContent.tsx
@@ -1,7 +1,7 @@
 import React from 'react'
 
 export interface NodeViewContentProps {
-  className?: string,
+  [key: string]: any,
   as?: React.ElementType,
 }
 
@@ -10,9 +10,12 @@ export const NodeViewContent: React.FC<NodeViewContentProps> = props => {
 
   return (
     <Tag
-      className={props.className}
+      {...props}
       data-node-view-content=""
-      style={{ whiteSpace: 'pre-wrap' }}
+      style={{
+        ...props.style,
+        whiteSpace: 'pre-wrap',
+      }}
     />
   )
 }
diff --git a/packages/react/src/NodeViewWrapper.tsx b/packages/react/src/NodeViewWrapper.tsx
@@ -2,7 +2,7 @@ import React from 'react'
 import { useReactNodeView } from './useReactNodeView'
 
 export interface NodeViewWrapperProps {
-  className?: string,
+  [key: string]: any,
   as?: React.ElementType,
 }
 
@@ -12,12 +12,13 @@ export const NodeViewWrapper: React.FC<NodeViewWrapperProps> = props => {
 
   return (
     <Tag
-      className={props.className}
+      {...props}
       data-node-view-wrapper=""
       onDragStart={onDragStart}
-      style={{ whiteSpace: 'normal' }}
-    >
-     {props.children}
-    </Tag>
+      style={{
+        ...props.style,
+        whiteSpace: 'normal',
+      }}
+    />
   )
 }