Fix for empty segments

Author: fasttime

lib/format-code-path.js

@@ -45,6 +45,8 @@ function formatCodePath(codePath, { segmentNames } = { })
         const lines = segment.internal.nodes || [];
         if (segmentNames)
             lines.unshift(`【${id}】`);
+        else if (!lines.length)
+            lines.push(id);
         const label = lines.join('\n');
         text += `state ${JSON.stringify(label)} as ${id}\n`;
     }

test/create-graph-map.js

@@ -6,9 +6,9 @@ const { strict: assert }            = require('assert');
 const { promises: { readFile } }    = require('fs');
 const { join }                      = require('path');
 
-async function test({ code, detail, expectedMappings })
+async function test({ code, detail, expectedMappings, sourceType })
 {
-    const graphMap = createGraphMap(code, { detail });
+    const graphMap = createGraphMap(code, { detail, sourceType });
     const actualKeys = [...graphMap.keys()];
     const expectedKeys = Object.keys(expectedMappings);
     assert.deepEqual(actualKeys, expectedKeys);
@@ -39,7 +39,7 @@ describe
                     {
                         foo();
                     }
-                    bar();                
+                    bar();
                     `,
                     expectedMappings: { 's1': 'graph-1' },
                 },
@@ -87,6 +87,20 @@ describe
             ),
         );
 
+        it
+        (
+            'conditional return statement with segments detail',
+            async () =>
+            await test
+            (
+                {
+                    code:               'if (foo) return;',
+                    sourceType:         'commonjs',
+                    expectedMappings:   { 's1': 'graph-8' },
+                },
+            ),
+        );
+
         it
         (
             'function declaration',

test/expected/graph-8.mermaid

@@ -0,0 +1,16 @@
+stateDiagram-v2
+classDef common fill: white, stroke: black
+class s1_1, s1_2, s1_4, s1_3 common
+classDef unreachable fill: #FF9800, stroke-dasharray: 5 5
+class s1_3 unreachable
+state "Program:enter\nIfStatement:enter\nIdentifier (foo)" as s1_1
+state "ReturnStatement" as s1_2
+state "IfStatement:exit\nProgram:exit" as s1_4
+state "s1_3" as s1_3
+[*] --> s1_1
+s1_1 --> s1_2
+s1_1 --> s1_4
+s1_2 --> s1_3
+s1_3 --> s1_4
+s1_2 --> [*]
+s1_4 --> [*]