Merge pull request #42 from esumerfd/master

Author: AlexanderGrooff

cmd/parse.go

@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"errors"
+	"fmt"
 	"regexp"
 	"strconv"
 	"strings"
@@ -252,10 +253,10 @@ func mermaidFileToMap(mermaid, styleType string) (*graphProperties, error) {
 	switch lines[0] {
 	case "graph LR", "flowchart LR":
 		graphDirection = "LR"
-	case "graph TD", "flowchart TD":
+	case "graph TD", "flowchart TD", "graph TB", "flowchart TB":
 		graphDirection = "TD"
 	default:
-		return &properties, errors.New("first line should define the graph")
+		return &properties, fmt.Errorf("unsupported graph type '%s'. Supported types: graph TD, graph TB, graph LR, flowchart TD, flowchart TB, flowchart LR", lines[0])
 	}
 	lines = lines[1:]
 

cmd/testdata/ascii/flowchart_tb_simple.txt

@@ -0,0 +1,29 @@
+flowchart TB
+    A --> B
+    B --> C
+---
++---+
+|   |
+| A |
+|   |
++---+
+  |  
+  |  
+  |  
+  |  
+  v  
++---+
+|   |
+| B |
+|   |
++---+
+  |  
+  |  
+  |  
+  |  
+  v  
++---+
+|   |
+| C |
+|   |
++---+

cmd/testdata/ascii/graph_tb_direction.txt

@@ -0,0 +1,26 @@
+graph TB
+subgraph one
+    A --> B
+end
+---
++-------+
+|  one  |
+|       |
+|       |
+| +---+ |
+| |   | |
+| | A | |
+| |   | |
+| +---+ |
+|   |   |
+|   |   |
+|   |   |
+|   |   |
+|   v   |
+| +---+ |
+| |   | |
+| | B | |
+| |   | |
+| +---+ |
+|       |
++-------+