From 4af6261771ee3d8252a336f295008e304262083f Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Mon, 11 May 2020 22:05:50 +0200
Subject: [PATCH] feat(examples): add custom edge closes #193

---
 example/src/Edges/CustomEdge.js | 16 ++++++++++++++++
 example/src/Edges/index.js      | 15 +++++++++++----
 src/index.ts                    |  9 ++-------
 3 files changed, 29 insertions(+), 11 deletions(-)
 create mode 100644 example/src/Edges/CustomEdge.js

diff --git a/example/src/Edges/CustomEdge.js b/example/src/Edges/CustomEdge.js
new file mode 100644
index 00000000..9cfd9619
--- /dev/null
+++ b/example/src/Edges/CustomEdge.js
@@ -0,0 +1,16 @@
+import React from 'react';
+
+export default function CustomEdge({
+  id, sourceX, sourceY, targetX, targetY, label, style = {}
+}) {
+  return (
+    <>
+      <path id={id} style={style} className="react-flow__edge-path" d={`M ${sourceX},${sourceY}L ${targetX},${targetY}`} />
+      <text>
+        <textPath href={`#${id}`} style={{ fontSize: '12px' }} startOffset="50%" textAnchor="middle">
+          {label}
+        </textPath>
+      </text>
+    </>
+  );
+};
\ No newline at end of file
diff --git a/example/src/Edges/index.js b/example/src/Edges/index.js
index 8f959066..16333481 100644
--- a/example/src/Edges/index.js
+++ b/example/src/Edges/index.js
@@ -2,6 +2,8 @@ import React, { useState } from 'react';
 
 import ReactFlow, { removeElements, addEdge, MiniMap, Controls } from 'react-flow-renderer';
 
+import CustomEdge from './CustomEdge';
+
 const onNodeDragStop = node => console.log('drag stop', node);
 const onElementClick = element => console.log('click', element);
 const onLoad = (graph) => {
@@ -9,14 +11,15 @@ const onLoad = (graph) => {
 };
 
 const initialElements = [
-  { id: '1', type: 'input', data: { label: 'Input Node 1' }, position: { x: 250, y: 0 } },
+  { id: '1', type: 'input', data: { label: 'Input 1' }, position: { x: 250, y: 0 } },
   { id: '2', data: { label: 'Node 2' }, position: { x: 150, y: 100 } },
   { id: '3', data: { label: 'Node 3' }, position: { x: 250, y: 200 } },
   { id: '4', data: { label: 'Node 4' }, position: { x: 400, y: 300 } },
   { id: '3a', data: { label: 'Node 3a' }, position: { x: 150, y: 300 } },
-  { id: '5', data: { label: 'Node 5' }, position: { x: 450, y: 400 } },
-  { id: '6', type: 'output', data: { label: 'Output Node 5' }, position: { x: 250, y: 550 } },
-  { id: '7', type: 'output', data: { label: 'Output Node 6' }, position: { x: 550, y: 550 } },
+  { id: '5', data: { label: 'Node 5' }, position: { x: 250, y: 400 } },
+  { id: '6', type: 'output', data: { label: 'Output 6' }, position: { x: 50, y: 550 } },
+  { id: '7', type: 'output', data: { label: 'Output 7' }, position: { x: 250, y: 550 } },
+  { id: '8', type: 'output', data: { label: 'Output 8' }, position: { x: 525, y: 600 } },
   { id: 'e1-2', source: '1', target: '2', label: 'bezier edge' },
   { id: 'e2-3', source: '2', target: '3', type: 'step', label: 'step edge' },
   { id: 'e3-4', source: '3', target: '4', type: 'straight', label: 'straight edge' },
@@ -24,6 +27,7 @@ const initialElements = [
   { id: 'e3-5', source: '4', target: '5', animated: true, label: 'animated styled edge', style: { stroke: 'red' } },
   { id: 'e5-6', source: '5', target: '6', label: 'styled label', labelStyle: { fill: 'red', fontWeight: 700 } },
   { id: 'e5-7', source: '5', target: '7', label: 'label with styled bg', labelBgStyle: { fill: '#eee', fillOpacity: 0.7 } },
+  { id: 'e5-8', source: '5', target: '8', type: 'custom', label: 'custom edge' },
 ];
 
 const EdgesFlow = () => {
@@ -42,6 +46,9 @@ const EdgesFlow = () => {
       style={{ width: '100%', height: '100%' }}
       onLoad={onLoad}
       snapToGrid={true}
+      edgeTypes={{
+        custom: CustomEdge
+      }}
     >
       <MiniMap />
       <Controls />
diff --git a/src/index.ts b/src/index.ts
index e856955f..c76d01a6 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -3,12 +3,7 @@ import ReactFlow from './container/ReactFlow';
 export default ReactFlow;
 
 export { default as Handle } from './components/Handle';
+export { default as EdgeText } from './components/Edges/EdgeText';
 export { MiniMap, Controls } from './plugins';
 
-export {
-  isNode,
-  isEdge,
-  removeElements,
-  addEdge,
-  getOutgoers,
-} from './utils/graph';
+export { isNode, isEdge, removeElements, addEdge, getOutgoers } from './utils/graph';