refactor(examples): update custom node example, create rich example

example/src/CustomNode/ColorSelectorNode.js

@@ -0,0 +1,32 @@
+import React from 'react';
+
+import { Handle } from 'react-flow';
+
+export default ({ data, styles }) => {
+  return (
+    <div
+      style={{ background: '#eee', border: '1px solid #ddd', padding: 10, borderRadius: 5, ...styles }}
+    >
+      <Handle
+        type="target"
+        position="left"
+        style={{ background: '#fff' }}
+        onConnect={params => console.log('handle onConnect', params)}
+      />
+      <div>Custom Color Picker Node: <strong>{data.color}</strong></div>
+      <input type="color" onChange={data.onChange} defaultValue={data.color}/>
+      <Handle
+        type="source"
+        position="right"
+        id="a"
+        style={{ top: 10, background: '#fff' }}
+      />
+      <Handle
+        type="source"
+        position="right"
+        id="b"
+        style={{ bottom: 10, top: 'auto', background: '#fff' }}
+      />
+    </div>
+  );
+};

example/src/CustomNode/index.js

@@ -0,0 +1,86 @@
+import React, { useState, useEffect } from 'react';
+
+import Graph, { isEdge, removeElements, addEdge, MiniMap, Controls } from 'react-flow';
+
+import ColorSelectorNode from './ColorSelectorNode';
+
+const onNodeDragStop = node => console.log('drag stop', node);
+const onElementClick = element => console.log('click', element);
+const onLoad = (graph) => {
+  console.log('graph loaded:', graph);
+};
+
+const initBgColor = '#8888e8';
+
+const CustomNodeGraph = () => {
+  const [elements, setElements] = useState([]);
+  const [bgColor, setBgColor] = useState(initBgColor);
+
+  useEffect(() => {
+    const onChange = (evt) => {
+      setElements(els => els.map(e => {
+        if (isEdge(e) || e.id !== '2') {
+          return e;
+        }
+
+        const color = evt.target.value;
+
+        setBgColor(color);
+
+        return {
+          ...e,
+          data: {
+            ...e.data,
+            color
+          }
+        };
+      }));
+    };
+
+    setElements([
+      { id: '1', type: 'input', data: { label: 'An input node' }, position: { x: 0, y: 50 }, sourcePosition: 'right' },
+      { id: '2', type: 'selectorNode', data: { onChange: onChange, color: initBgColor }, position: { x: 250, y: 50 } },
+      { id: '3', type: 'output', data: { label: 'Output A' }, position: { x: 550, y: 25 }, targetPosition: 'left' },
+      { id: '4', type: 'output', data: { label: 'Output B' }, position: { x: 550, y: 100 }, targetPosition: 'left' },
+
+      { id: 'e1-2', source: '1', target: '2', animated: true, style: { stroke: '#fff' } },
+      { id: 'e2a-3', source: '2__a', target: '3', animated: true, style: { stroke: '#fff' } },
+      { id: 'e2b-4', source: '2__b', target: '4', animated: true, style: { stroke: '#fff' } },
+    ])
+  }, []);
+
+  const onElementsRemove = (elementsToRemove) => setElements(els => removeElements(elementsToRemove, els));
+  const onConnect = (params) => setElements(els => addEdge(params, els));
+
+  return (
+    <Graph
+      elements={elements}
+      onElementClick={onElementClick}
+      onElementsRemove={onElementsRemove}
+      onConnect={onConnect}
+      onNodeDragStop={onNodeDragStop}
+      style={{ width: '100%', height: '100%', background: bgColor }}
+      onLoad={onLoad}
+      nodeTypes={{
+        selectorNode: ColorSelectorNode,
+      }}
+      connectionLineStyle={{ stroke: '#ddd', strokeWidth: 2 }}
+      connectionLineType="bezier"
+      backgroundColor="#888"
+      backgroundGap={16}
+      snapToGrid={true}
+      snapGrid={[16, 16]}
+    >
+      <MiniMap
+        nodeColor={n => {
+          if (n.type === 'input') return 'blue';
+          if (n.type === 'selectorNode') return bgColor;
+          if (n.type === 'output') return 'green';
+        }}
+      />
+      <Controls />
+    </Graph>
+  );
+}
+
+export default CustomNodeGraph;

example/src/CustomNodes/InputNode.js

@@ -1,19 +0,0 @@
-import React from 'react';
-
-import { Handle } from 'react-flow';
-
-export default({ data, styles }) => (
-  <div
-    style={{ background: '#FFCC00', padding: 10, borderRadius: 2, ...styles }}
-  >
-    <Handle type="target" position="left" style={{ background: '#999' }} />
-    <div>{data.input}</div>
-    <input onChange={(e) => data.onChange(e.target.value, data)} />
-    <Handle
-      type="source"
-      position="right"
-      style={{ background: '#999' }}
-      isValidConnection={connection => +connection.target % 2 === 0}
-    />
-  </div>
-);

example/src/CustomNodes/SpecialNode.js

@@ -1,36 +0,0 @@
-import React from 'react';
-
-import { Handle } from 'react-flow';
-
-export default ({ data, styles }) => {
-  return (
-    <div
-      style={{ background: '#FFCC00', padding: 10, borderRadius: 2, ...styles }}
-    >
-      <Handle
-        type="target"
-        position="top"
-        id="a"
-        style={{ left: 10, background: '#999' }}
-        onConnect={params => console.log('handle onConnect', params)}
-      />
-      <Handle
-        type="target"
-        position="top"
-        id="b"
-        style={{ left: 30, background: '#999' }}
-      />
-      <div>I am <strong>special</strong>!<br />{data.label}</div>
-      <select onChange={(e) => data.onChange(e.target.value, data)}>
-        <option value="1">1</option>
-        <option value="2">2</option>
-        <option value="3">3</option>
-      </select>
-      <Handle
-        type="source"
-        position="bottom"
-        style={{ left: 10, background: '#999' }}
-      />
-    </div>
-  );
-};

example/src/CustomNodes/index.js

@@ -1,119 +0,0 @@
-import React, { useState, useEffect } from 'react';
-
-import Graph, { isEdge, removeElements, addEdge, MiniMap, Controls } from 'react-flow';
-
-import SpecialNode from './SpecialNode';
-import InputNode from './InputNode';
-
-const onNodeDragStop = node => console.log('drag stop', node);
-const onElementClick = element => console.log('click', element);
-const onLoad = (graph) => console.log('graph loaded:', graph);
-
-const CustomNodesGraph = () => {
-  const [elements, setElements] = useState([]);
-
-  useEffect(() => {
-    const onChange = (option, d) => {
-      setElements(els => els.map(e => {
-        if (isEdge(e) || e.id !== '6') {
-          return e;
-        }
-
-        return {
-          ...e,
-          data: {
-            ...e.data,
-            label: `Option ${option} selected.`
-          }
-        };
-      }));
-    };
-
-    const onChangeInput = (input, d) => {
-      setElements(els => els.map(e => {
-        if (isEdge(e) || e.id !== '8') {
-          return e;
-        }
-
-        return {
-          ...e,
-          data: {
-            ...e.data,
-            input: input || 'write something'
-          }
-        };
-      }));
-    };
-
-    setElements([
-      { id: '1', type: 'input', data: { label: '1 Tests' }, position: { x: 250, y: 5 } },
-      { id: '2', data: { label: '2 This is a node This is a node This is a node This is a node' }, position: { x: 100, y: 100 } },
-      { id: '3', data: { label: '3 I bring my own style' }, position: { x: 100, y: 200 }, style: { background: '#eee', color: '#222', border: '1px solid #bbb' } },
-      { id: '4', type: 'output', data: { label: '4 nody nodes' }, position: { x: 50, y: 300 } },
-      { id: '5', type: 'default', data: { label: '5 Another node'}, position: { x: 400, y: 300 } },
-      { id: '6', type: 'special', data: { onChange: onChange, label: '6 no option selected' }, position: { x: 425, y: 375 } },
-      { id: '7', type: 'output', data: { label: '7 output' }, position: { x: 250, y: 500 } },
-      { id: '8', type: 'text', data: { onChange: onChangeInput, input: 'write something' }, position: { x: 350, y: 100 } },
-      { id: 'e1-2', source: '1', target: '2', animated: true },
-      { id: 'e1-8', source: '1', target: '8', animated: true },
-      { id: 'e2-3', source: '2', target: '3', animated: true },
-      { id: 'e3-4', source: '3', target: '4', animated: true },
-      { id: 'e3-5', source: '3', target: '5', animated: true },
-      { id: 'e5-6b', source: '5', target: '6__b', animated: true  },
-      { id: 'e5-6a', source: '5', target: '6__a', animated: true },
-      { id: 'e6-7', source: '6', target: '7', animated: true },
-    ])
-  }, []);
-
-  const addRandomNode = () => {
-    setElements(els => els.concat({
-      id: (els.length + 1).toString(),
-      data: { label: 'Added node' },
-      position: { x: Math.random() * window.innerWidth, y: Math.random() * window.innerHeight }
-    }));
-  };
-  const onElementsRemove = (elementsToRemove) => setElements(els => removeElements(elementsToRemove, els));
-  const onConnect = (params) => setElements(els => addEdge(params, els));
-
-  return (
-    <Graph
-      elements={elements}
-      onElementClick={onElementClick}
-      onElementsRemove={onElementsRemove}
-      onConnect={onConnect}
-      onNodeDragStop={onNodeDragStop}
-      style={{ width: '100%', height: '100%' }}
-      onLoad={onLoad}
-      nodeTypes={{
-        special: SpecialNode,
-        text: InputNode
-      }}
-      connectionLineStyle={{ stroke: '#ddd', strokeWidth: 2 }}
-      connectionLineType="bezier"
-      backgroundColor="#888"
-      backgroundGap={16}
-      snapToGrid={true}
-      snapGrid={[16, 16]}
-    >
-      <MiniMap
-        nodeColor={n => {
-          if (n.type === 'input') return 'blue';
-          if (n.type === 'output') return 'green';
-          if (n.type === 'default') return 'red';
-
-          return '#FFCC00';
-        }}
-      />
-      <Controls />
-      <button
-        type="button"
-        onClick={addRandomNode}
-        style={{ position: 'absolute', right: 10, top: 10, zIndex: 4 }}
-      >
-        add node
-      </button>
-    </Graph>
-  );
-}
-
-export default CustomNodesGraph;

example/src/Rich/index.js

@@ -0,0 +1,77 @@
+import React, { useState, useEffect } from 'react';
+
+import Graph, { removeElements, addEdge, MiniMap, Controls } from 'react-flow';
+
+const onNodeDragStop = node => console.log('drag stop', node);
+const onElementClick = element => console.log('click', element);
+const onLoad = (graph) => console.log('graph loaded:', graph);
+
+const initialElements = [
+  { id: '1', type: 'input', data: { label: 'Input Node 1' }, position: { x: 250, y: 5 } },
+  { id: '2', data: { label: 'Node 2' }, position: { x: 150, y: 75 } },
+  {
+    id: '3', data: { label: 'Node 3 with custom style' }, position: { x: 250, y: 150 },
+    style: { background: '#eee', color: '#222', border: '1px solid #bbb' },
+    sourcePosition: 'right'
+  },
+  { id: '4', data: { label: 'Node 4' }, position: { x: 500, y: 200 }, targetPosition: 'left' },
+  { id: '5', type: 'output', data: { label: 'Output Node 5'}, position: { x: 300, y: 300 } },
+  { id: '6', type: 'output', data: { label: 'Output Node 6' }, position: { x: 600, y: 400 } },
+  { id: 'e1-2', source: '1', target: '2', animated: true },
+  { id: 'e2-3', source: '2', target: '3', animated: true },
+  { id: 'e3-4', source: '3', target: '4', animated: true },
+  { id: 'e3-5', source: '4', target: '5', animated: true, type: 'step' },
+  { id: 'e5-6b', source: '4', target: '6', type: 'step'  },
+]
+
+const RichGraph = () => {
+  const [elements, setElements] = useState(initialElements);
+
+  const addRandomNode = () => {
+    setElements(els => els.concat({
+      id: (els.length + 1).toString(),
+      data: { label: 'Added node' },
+      position: { x: Math.random() * window.innerWidth, y: Math.random() * window.innerHeight }
+    }));
+  };
+  const onElementsRemove = (elementsToRemove) => setElements(els => removeElements(elementsToRemove, els));
+  const onConnect = (params) => setElements(els => addEdge(params, els));
+
+  return (
+    <Graph
+      elements={elements}
+      onElementClick={onElementClick}
+      onElementsRemove={onElementsRemove}
+      onConnect={onConnect}
+      onNodeDragStop={onNodeDragStop}
+      style={{ width: '100%', height: '100%' }}
+      onLoad={onLoad}
+      connectionLineStyle={{ stroke: '#ddd', strokeWidth: 2 }}
+      connectionLineType="bezier"
+      backgroundColor="#888"
+      backgroundGap={16}
+      snapToGrid={true}
+      snapGrid={[16, 16]}
+    >
+      <MiniMap
+        nodeColor={n => {
+          if (n.type === 'input') return 'blue';
+          if (n.type === 'output') return 'green';
+          if (n.type === 'default') return 'red';
+
+          return '#FFCC00';
+        }}
+      />
+      <Controls />
+      <button
+        type="button"
+        onClick={addRandomNode}
+        style={{ position: 'absolute', right: 10, top: 10, zIndex: 4 }}
+      >
+        add node
+      </button>
+    </Graph>
+  );
+}
+
+export default RichGraph;

example/src/index.js

@@ -2,7 +2,8 @@ import React from 'react';
 import ReactDOM from 'react-dom';
 import { BrowserRouter as Router, Route, Switch } from 'react-router-dom';
 
-import CustomNodes from './CustomNodes';
+import CustomNode from './CustomNode';
+import Rich from './Rich';
 import Basic from './Basic';
 import Empty from './Empty';
 import Inactive from './Inactive';
@@ -21,8 +22,11 @@ ReactDOM.render((
       <Route path="/inactive">
         <Inactive />
       </Route>
+      <Route path="/custom-node">
+        <CustomNode />
+      </Route>
       <Route path="/">
-        <CustomNodes />
+        <Rich />
       </Route>
     </Switch>
   </Router>