How to conditionally render node view in React

[piszczu4]

I have a figure node which consists of two nodes media (required) and caption (optional). Here is the view:

	return (
		<NodeViewWrapper
			as="div"
			className={
				"media-with-caption-node-view " +
				(isFloat
					? "f-" + node.attrs.dataFloat
					: isAlign
					? "justify-" + node.attrs.dataAlign
					: "")
			}
		>
			<figure>
				<NodeViewContent></NodeViewContent>
			</figure>
		</NodeViewWrapper>
	);
}

I want to add attribute caption to this node and use to to either render or not the capton view. Currently <NodeViewContent></NodeViewContent> renders the views of both figure and caption - can I somehow split it? Or what is the recommended way to implement this kind of things? Add a command which deletes and adds a caption node?