fixfix graphviz bug

things-go · Jul 2, 2023 · 3df80c4 · 3df80c4
1 parent d1f3f34
commit 3df80c4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/visualizer._mermaid.go b/visualizer._mermaid.go
@@ -46,7 +46,7 @@ func visualizeMermaidStateDiagram[E constraints.Ordered, S constraints.Ordered](
 		if err != nil {
 			return "", err
 		}
-		buf.WriteString(fmt.Sprintf(`    %v --> %v: %v`, fsm.StateName(ts.State()), fsm.StateName(dst), fsm.EventName(ts.Event())))
+		buf.WriteString(fmt.Sprintf(`    %s --> %s: %s`, fsm.StateName(ts.State()), fsm.StateName(dst), fsm.EventName(ts.Event())))
 		buf.WriteString("\n")
 	}
 	return buf.String(), nil
@@ -104,7 +104,7 @@ func (v *visualizeMermaidFlowChartBuilder[E, S]) writeFlowChartStates() *visuali
 		return v
 	}
 	for _, state := range v.sortedStates {
-		v.buf.WriteString(fmt.Sprintf(`    %s[%v]`, v.statesId[state], v.fsm.StateName(state)))
+		v.buf.WriteString(fmt.Sprintf(`    %s[%s]`, v.statesId[state], v.fsm.StateName(state)))
 		v.buf.WriteString("\n")
 	}
 	v.buf.WriteString("\n")
@@ -120,7 +120,7 @@ func (v *visualizeMermaidFlowChartBuilder[E, S]) writeFlowChartTransitions() *vi
 		if err != nil {
 			return v.setErr(err)
 		}
-		v.buf.WriteString(fmt.Sprintf(`    %s --> |%v| %s`, v.statesId[ts.State()], v.fsm.EventName(ts.Event()), v.statesId[dst]))
+		v.buf.WriteString(fmt.Sprintf(`    %s --> |%s| %s`, v.statesId[ts.State()], v.fsm.EventName(ts.Event()), v.statesId[dst]))
 		v.buf.WriteString("\n")
 	}
 	v.buf.WriteString("\n")

diff --git a/visualizer_graphviz.go b/visualizer_graphviz.go
@@ -61,13 +61,14 @@ func (v *visualizeGraphvizBuilder[E, S]) writeTransitions() *visualizeGraphvizBu
 		if err != nil {
 			return v.setErr(err)
 		}
-		line := fmt.Sprintf(`    "%v" -> "%v" [ label = "%v" ];`, v.fsm.StateName(ts.State()), v.fsm.StateName(dst), v.fsm.EventName(ts.Event()))
+		line := fmt.Sprintf(`    "%s" -> "%s" [ label = "%s" ];`, v.fsm.StateName(ts.State()), v.fsm.StateName(dst), v.fsm.EventName(ts.Event()))
 		if ts.State() == v.fsm.Current() {
 			v.buf.WriteString(line)
+			v.buf.WriteString("\n")
 		} else {
 			b.WriteString(line)
+			b.WriteString("\n")
 		}
-		b.WriteString("\n")
 	}
 	if b.Len() > 0 {
 		v.buf.Write(b.Bytes())
@@ -81,7 +82,7 @@ func (v *visualizeGraphvizBuilder[E, S]) writeStates() *visualizeGraphvizBuilder
 		return v
 	}
 	for _, state := range v.sortedStates {
-		v.buf.WriteString(fmt.Sprintf(`    "%v";`, v.fsm.StateName(state)))
+		v.buf.WriteString(fmt.Sprintf(`    "%s";`, v.fsm.StateName(state)))
 		v.buf.WriteString("\n")
 	}
 	return v