fix: check if node.prop is defined

mermaid-js · Nov 28, 2021 · 28bb07d · 28bb07d
1 parent c2db1d4
commit 28bb07d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/dagre-wrapper/nodes.js b/src/dagre-wrapper/nodes.js
@@ -325,14 +325,16 @@ const rect = (parent, node) => {
     .attr('width', totalWidth)
     .attr('height', totalHeight);
 
-  const propKeys = new Set(Object.keys(node.props));
-  if (node.props?.borders) {
-    applyNodePropertyBorders(rect, node.props.borders, totalWidth, totalHeight);
-    propKeys.delete('borders');
+  if (node.props) {
+    const propKeys = new Set(Object.keys(node.props));
+    if (node.props.borders) {
+      applyNodePropertyBorders(rect, node.props.borders, totalWidth, totalHeight);
+      propKeys.delete('borders');
+    }
+    propKeys.forEach((propKey) => {
+      log.warn(`Unknown node property ${propKey}`);
+    });
   }
-  propKeys.forEach((propKey) => {
-    log.warn(`Unknown node property ${propKey}`);
-  });
 
   updateNodeBounds(node, rect);