feat(renderer): pass nodeTypes

2019-07-16 12:33:05 +02:00
parent e40ae6d841
commit f330b0c4ee
6 changed files with 36 additions and 8 deletions
@@ -9,7 +9,22 @@ class NodeRenderer extends PureComponent {

  renderNode(d, onNodeClick) {
    const nodeType = d.data.type || 'default';
-    const NodeComponent = this.props.nodeTypes[nodeType] || DefaultNode;
+    let NodeComponent = null;
+
+    switch (nodeType) {
+      case 'input': {
+        NodeComponent = this.props.nodeTypes.input || InputNode; break;
+      }
+      case 'default': {
+        NodeComponent = this.props.nodeTypes.default || DefaultNode; break;
+      }
+      case 'output': {
+        NodeComponent = this.props.nodeTypes.output || OutputNode; break;
+      }
+      default: {
+        NodeComponent = this.props.nodeTypes[nodeType] || DefaultNode;
+      }
+    }

    return (
      <NodeComponent