refactor(example): cleanup

example/scripts/ExampleGraph.js

@@ -0,0 +1,193 @@
+import React, { PureComponent } from 'react';
+
+import Graph, { isEdge, removeElements, getOutgoers, MiniMap } from '../../src';
+
+import SpecialNode from './SpecialNode';
+import InputNode from './InputNode';
+
+const onNodeDragStop = node => console.log('drag stop', node);
+
+class App extends PureComponent {
+  constructor() {
+    super();
+
+    const onChange = (option, d) => {
+      this.setState(prevState => (
+        {elements: prevState.elements.map(e => {
+          if (isEdge(e) || e.id !== '6') {
+            return e;
+          }
+
+          return {
+            ...e,
+            data: {
+              ...e.data,
+              label: `Option ${option} selected.`
+            }
+          };
+        })}
+      ));
+    }
+
+    const onChangeInput = (input, d) => {
+      this.setState(prevState => (
+        {elements: prevState.elements.map(e => {
+          if (isEdge(e) || e.id !== '8') {
+            return e;
+          }
+
+          if (e.id === '8') {
+            return {
+              ...e,
+              data: {
+                ...e.data,
+                input: input || 'write something'
+              }
+            };
+          }
+        })}
+      ));
+    }
+
+    this.state = {
+      graphLoaded: false,
+      elements: [
+        { 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, 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: '9', type: 'text', data: { label: 'right' }, position: { x: 600, y: 100 } },
+        { source: '1', target: '2', animated: true },
+        { source: '1', target: '8', animated: true },
+        { source: '8', target: '9', animated: true },
+        { source: '2', target: '3' },
+        { source: '3', target: '4', type: 'step' },
+        { source: '3', target: '5' },
+        { source: '5', target: '6__b' },
+        { source: '5', target: '6__a', type: 'step', animated: true, style: { stroke: '#FFCC00' } },
+        { source: '6', target: '7', style: { stroke: '#FFCC00' }},
+      ]
+    };
+
+    this.onElementClick = this.onElementClick.bind(this);
+    this.onConnect = this.onConnect.bind(this);
+  }
+
+  onLoad(graphInstance) {
+    console.log('graph loaded:', graphInstance);
+    window.rg = graphInstance;
+
+    this.graphInstance = graphInstance;
+    this.graphInstance.fitView({ padding: 0.1 });
+    this.setState({
+      graphLoaded: true
+    });
+  }
+
+  onFitView() {
+    this.graphInstance.fitView();
+  }
+
+  onAdd() {
+    this.setState(prevState => ({
+      ...prevState,
+      elements: prevState.elements.concat({
+        id: (prevState.elements.length + 1).toString(),
+        data: { label: 'Added node' },
+        position: { x: Math.random() * window.innerWidth, y: Math.random() * window.innerHeight }
+      })
+    }));
+  }
+
+  onElementClick(element) {
+    console.log('click', element);
+    console.log('outgoers', getOutgoers(element, this.state.elements));
+  }
+
+  onZoomIn() {
+    this.graphInstance.zoomIn();
+  }
+
+  onZoomOut() {
+    this.graphInstance.zoomOut();
+  }
+
+  onElementsRemove(elementsToRemove) {
+    this.setState(prevState => ({
+      elements: removeElements(prevState.elements, elementsToRemove)
+    }));
+  }
+
+  onConnect(params) {
+    console.log('connect', params);
+    this.setState(prevState => ({
+      elements: prevState.elements.concat(params)
+    }));
+  }
+
+  render() {
+    return (
+      <Graph
+        elements={this.state.elements}
+        onElementClick={this.onElementClick}
+        onElementsRemove={elements => this.onElementsRemove(elements)}
+        onConnect={this.onConnect}
+        onNodeDragStop={onNodeDragStop}
+        style={{ width: '100%', height: '100%' }}
+        onLoad={graphInstance => this.onLoad(graphInstance)}
+        nodeTypes={{
+          special: SpecialNode,
+          text: InputNode
+        }}
+        connectionLineStyle={{ stroke: '#ddd', strokeWidth: 2 }}
+        connectionLineType="bezier"
+      >
+        <MiniMap
+          style={{ position: 'absolute', right: 10, bottom: 10 }}
+          nodeColor={n => {
+            if (n.type === 'input') return 'blue';
+            if (n.type === 'output') return 'green';
+            if (n.type === 'default') return 'red';
+
+            return '#FFCC00';
+          }}
+        />
+        <button
+          type="button"
+          onClick={() => this.onAdd()}
+          style={{ position: 'absolute', right: 10, top: 10, zIndex: 4 }}
+        >
+          add
+        </button>
+        {this.state.graphLoaded && (
+          <div className="controls">
+            <button
+              type="button"
+              onClick={() => this.onFitView()}
+            >
+              fit
+            </button>
+            <button
+              type="button"
+              onClick={() => this.onZoomIn()}
+            >
+              zoom in
+            </button>
+            <button
+              type="button"
+              onClick={() => this.onZoomOut()}
+            >
+              zoom out
+            </button>
+          </div>
+        )}
+      </Graph>
+    );
+  }
+}
+
+export default App;