From dcdb1c43d5d97ccc4da11bf01082e55cce5626e5 Mon Sep 17 00:00:00 2001
From: Henning Dieterichs <henning.dieterichs@live.de>
Date: Thu, 20 Feb 2020 13:51:00 +0100
Subject: [PATCH] Fixes graphviz graph visualizer (undefined labels).

---
 .../GraphVisualizer/GraphvizGraphVisualizer.tsx     | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/extension/ui/src/Visualizers/GraphVisualizer/GraphvizGraphVisualizer.tsx b/extension/ui/src/Visualizers/GraphVisualizer/GraphvizGraphVisualizer.tsx
index db83e1e..8834e1e 100644
--- a/extension/ui/src/Visualizers/GraphVisualizer/GraphvizGraphVisualizer.tsx
+++ b/extension/ui/src/Visualizers/GraphVisualizer/GraphvizGraphVisualizer.tsx
@@ -36,8 +36,8 @@ export class GraphvizGraphVisualizer extends VisualizationProvider {
 
 @observer
 export class GraphvizGraphViewer extends React.Component<{
-	nodes: { id: string; label: string }[];
-	edges: { from: string; to: string; label: string }[];
+	nodes: { id: string; label?: string }[];
+	edges: { from: string; to: string; label?: string }[];
 }> {
 	render() {
 		const { nodes, edges } = this.props;
@@ -45,7 +45,10 @@ export class GraphvizGraphViewer extends React.Component<{
             digraph MyGraph {
                 ${nodes
 					.map(
-						n => `"${n.id}" [ label = ${JSON.stringify(n.label)} ];`
+						n =>
+							`"${n.id}" [ label = ${JSON.stringify(
+								n.label !== undefined ? n.label : n.id
+							)} ];`
 					)
 					.join("\n ")}
                 ${edges
@@ -53,7 +56,9 @@ export class GraphvizGraphViewer extends React.Component<{
 						e =>
 							`"${e.from}" -> "${
 								e.to
-							}" [ label = ${JSON.stringify(e.label)} ];`
+							}" [ label = ${JSON.stringify(
+								e.label !== undefined ? e.label : ""
+							)} ];`
 					)
 					.join("\n")}
             }