feat(connectionline): custom component option #516

example/src/CustomConnectionLine/index.js

@@ -0,0 +1,25 @@
+import React, { useState } from 'react';
+import ReactFlow, { removeElements, addEdge, Background } from 'react-flow-renderer';
+
+import ConnectionLine from './ConnectionLine';
+
+const initialElements = [{ id: '1', type: 'input', data: { label: 'Node 1' }, position: { x: 250, y: 5 } }];
+
+const ConnectionLineFlow = () => {
+  const [elements, setElements] = useState(initialElements);
+  const onElementsRemove = (elementsToRemove) => setElements((els) => removeElements(elementsToRemove, els));
+  const onConnect = (params) => setElements((els) => addEdge(params, els));
+
+  return (
+    <ReactFlow
+      elements={elements}
+      connectionLineComponent={ConnectionLine}
+      onElementsRemove={onElementsRemove}
+      onConnect={onConnect}
+    >
+      <Background variant="lines" />
+    </ReactFlow>
+  );
+};
+
+export default ConnectionLineFlow;