diff --git a/src/components/Nodes/wrapNode.tsx b/src/components/Nodes/wrapNode.tsx index 52c34f36..439d8537 100644 --- a/src/components/Nodes/wrapNode.tsx +++ b/src/components/Nodes/wrapNode.tsx @@ -41,6 +41,7 @@ export default (NodeComponent: ComponentType) => { isParent, noPanClassName, noDragClassName, + initialized, }: WrapNodeProps) => { const store = useStoreApi(); const updateNodeDimensions = useStore(selector); @@ -134,6 +135,7 @@ export default (NodeComponent: ComponentType) => { zIndex, transform: `translate(${xPos}px,${yPos}px)`, pointerEvents: hasPointerEvents ? 'all' : 'none', + visibility: initialized ? 'visible' : 'hidden', ...style, }} onMouseEnter={onMouseEnterHandler} diff --git a/src/container/NodeRenderer/index.tsx b/src/container/NodeRenderer/index.tsx index ba3ea773..ec90bc5c 100644 --- a/src/container/NodeRenderer/index.tsx +++ b/src/container/NodeRenderer/index.tsx @@ -122,6 +122,7 @@ const NodeRenderer = (props: NodeRendererProps) => { isParent={!!node[internalsSymbol]?.isParent} noDragClassName={props.noDragClassName} noPanClassName={props.noPanClassName} + initialized={!!node.width && !!node.height} /> ); })} diff --git a/src/types/nodes.ts b/src/types/nodes.ts index ae0bb6f7..0045ae34 100644 --- a/src/types/nodes.ts +++ b/src/types/nodes.ts @@ -64,8 +64,7 @@ export interface WrapNodeProps { isConnectable: boolean; xPos: number; yPos: number; - width?: number | null; - height?: number | null; + initialized: boolean; isSelectable: boolean; isDraggable: boolean; selectNodesOnDrag: boolean;