fix(nodetypes): use default when type not found

xyflow · Jul 16, 2019 · a431148 · a431148
1 parent 53581a8
commit a431148
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/NodeRenderer/index.js b/src/NodeRenderer/index.js
@@ -1,11 +1,16 @@
 import React, { PureComponent } from 'react';
 
 import { Consumer } from '../GraphContext';
+
 class NodeRenderer extends PureComponent {
 
   renderNode(d, onNodeClick) {
     const nodeType = d.data.type || 'default';
-    const NodeComponent = this.props.nodeTypes[nodeType];
+    if (!this.props.nodeTypes[nodeType]) {
+      console.warn(`No node type found for type "${nodeType}". Using fallback type "default".`);
+    }
+
+    const NodeComponent = this.props.nodeTypes[nodeType] || this.props.nodeTypes.default;
 
     return (
       <NodeComponent

diff --git a/src/NodeRenderer/utils.js b/src/NodeRenderer/utils.js
@@ -14,7 +14,7 @@ export function createNodeTypes(nodeTypes) {
     .keys(nodeTypes)
     .filter(k => !['input', 'default', 'output'].includes(k))
     .reduce((res, key) => {
-      res[key] = wrapNode(nodeTypes[key]);
+      res[key] = wrapNode(nodeTypes[key] || DefaultNode);
 
       return res;
     }, {});