From d40c74e54ce22ed936980e59b3ecc6c721956128 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Tue, 21 Jul 2020 16:33:38 +0200
Subject: [PATCH] chore(examples): add edge types example

---
 example/src/EdgeTypes/index.js |  56 ++++++++++++++++++
 example/src/EdgeTypes/utils.js | 104 +++++++++++++++++++++++++++++++++
 example/src/index.js           |   5 ++
 3 files changed, 165 insertions(+)
 create mode 100644 example/src/EdgeTypes/index.js
 create mode 100644 example/src/EdgeTypes/utils.js

diff --git a/example/src/EdgeTypes/index.js b/example/src/EdgeTypes/index.js
new file mode 100644
index 00000000..7fef539d
--- /dev/null
+++ b/example/src/EdgeTypes/index.js
@@ -0,0 +1,56 @@
+import React, { useState } from 'react';
+
+import ReactFlow, { removeElements, addEdge, MiniMap, isNode, Controls, Background } from 'react-flow-renderer';
+import { getElements } from './utils';
+
+const onLoad = (reactFlowInstance) => {
+  reactFlowInstance.fitView();
+
+  console.log(reactFlowInstance.getElements());
+};
+
+const initialElements = getElements(10, 10);
+
+const EdgeTypesFlow = () => {
+  const [elements, setElements] = useState(initialElements);
+  const onElementsRemove = (elementsToRemove) => setElements((els) => removeElements(elementsToRemove, els));
+  const onConnect = (params) => setElements((els) => addEdge(params, els));
+
+  const updatePos = () => {
+    setElements((elms) => {
+      return elms.map((el) => {
+        if (isNode(el)) {
+          return {
+            ...el,
+            position: {
+              x: Math.random() * window.innerWidth,
+              y: Math.random() * window.innerHeight,
+            },
+          };
+        }
+
+        return el;
+      });
+    });
+  };
+
+  return (
+    <ReactFlow
+      elements={elements}
+      onLoad={onLoad}
+      onElementsRemove={onElementsRemove}
+      onConnect={onConnect}
+      minZoom={0.2}
+    >
+      <MiniMap />
+      <Controls />
+      <Background />
+
+      <button onClick={updatePos} style={{ position: 'absolute', right: 10, top: 30, zIndex: 4 }}>
+        change pos
+      </button>
+    </ReactFlow>
+  );
+};
+
+export default EdgeTypesFlow;
diff --git a/example/src/EdgeTypes/utils.js b/example/src/EdgeTypes/utils.js
new file mode 100644
index 00000000..baf832a3
--- /dev/null
+++ b/example/src/EdgeTypes/utils.js
@@ -0,0 +1,104 @@
+const nodeWidth = 80;
+const nodeGapWidth = nodeWidth * 2;
+const nodeStyle = { width: nodeWidth, fontSize: 11, color: 'white' };
+
+const sourceTargetPositions = [
+  { source: 'bottom', target: 'top' },
+  { source: 'right', target: 'left' },
+];
+const nodeColors = [
+  ['#1e9e99', '#4cb3ac', '#6ec9c0', '#8ddfd4'],
+  ['#0f4c75', '#1b5d8b', '#276fa1', '#3282b8'],
+];
+const edgeTypes = ['default', 'step', 'smoothstep', 'straight'];
+const offsets = [
+  {
+    x: 0,
+    y: -nodeGapWidth,
+  },
+  {
+    x: nodeGapWidth,
+    y: -nodeGapWidth,
+  },
+  {
+    x: nodeGapWidth,
+    y: 0,
+  },
+  {
+    x: nodeGapWidth,
+    y: nodeGapWidth,
+  },
+  {
+    x: 0,
+    y: nodeGapWidth,
+  },
+  {
+    x: -nodeGapWidth,
+    y: nodeGapWidth,
+  },
+  {
+    x: -nodeGapWidth,
+    y: 0,
+  },
+  {
+    x: -nodeGapWidth,
+    y: -nodeGapWidth,
+  },
+];
+
+let id = 0;
+const getNodeId = () => (id++).toString();
+
+export function getElements() {
+  const initialElements = [];
+
+  for (let sourceTargetIndex = 0; sourceTargetIndex < sourceTargetPositions.length; sourceTargetIndex++) {
+    const currSourceTargetPos = sourceTargetPositions[sourceTargetIndex];
+
+    for (let edgeTypeIndex = 0; edgeTypeIndex < edgeTypes.length; edgeTypeIndex++) {
+      const currEdgeType = edgeTypes[edgeTypeIndex];
+
+      for (let offsetIndex = 0; offsetIndex < offsets.length; offsetIndex++) {
+        const currOffset = offsets[offsetIndex];
+
+        const style = { ...nodeStyle, background: nodeColors[sourceTargetIndex][edgeTypeIndex] };
+        const sourcePosition = {
+          x: offsetIndex * nodeWidth * 4,
+          y: edgeTypeIndex * 300 + sourceTargetIndex * edgeTypes.length * 300,
+        };
+        const sourceId = getNodeId();
+        const sourceData = { label: `Source ${sourceId}` };
+        const sourceNode = {
+          id: sourceId,
+          style,
+          data: sourceData,
+          position: sourcePosition,
+          sourcePosition: currSourceTargetPos.source,
+          targetPosition: currSourceTargetPos.target,
+        };
+
+        const targetId = getNodeId();
+        const targetData = { label: `Target ${targetId}` };
+        const targetPosition = {
+          x: sourcePosition.x + currOffset.x,
+          y: sourcePosition.y + currOffset.y,
+        };
+        const targetNode = {
+          id: targetId,
+          style,
+          data: targetData,
+          position: targetPosition,
+          sourcePosition: currSourceTargetPos.source,
+          targetPosition: currSourceTargetPos.target,
+        };
+
+        initialElements.push(sourceNode);
+        initialElements.push(targetNode);
+
+        initialElements.push({ id: `${sourceId}-${targetId}`, source: sourceId, target: targetId, type: currEdgeType });
+      }
+    }
+  }
+
+  return initialElements;
+}
diff --git a/example/src/index.js b/example/src/index.js
index 5eb71cfa..51b9c00f 100644
--- a/example/src/index.js
+++ b/example/src/index.js
@@ -13,6 +13,7 @@ import Validation from './Validation';
 import Horizontal from './Horizontal';
 import Provider from './Provider';
 import Hidden from './Hidden';
+import EdgeTypes from './EdgeTypes';
 
 import './index.css';
 
@@ -69,6 +70,10 @@ const routes = [
     path: '/hidden',
     component: Hidden,
   },
+  {
+    path: '/edge-types',
+    component: EdgeTypes,
+  },
 ];
 
 const navLinks = routes.filter((route) => route.label);