diff --git a/view/bargraph.go b/view/bargraph.go index 09b20bb4..cc992b2c 100644 --- a/view/bargraph.go +++ b/view/bargraph.go @@ -76,13 +76,18 @@ func BuildStars(data []Bar, maxStars int, starChar string) string { label = fmt.Sprint(bar.Percent) } + labelColor := bar.LabelColor + if labelColor == "" { + labelColor = "default" + } + //write the line buffer.WriteString( fmt.Sprintf( - "%s%s[[%s]%s[white]%s] %s\n", + "%s%s[[%s]%s[default]%s] %s\n", bar.Label, strings.Repeat(" ", longestLabel-len(bar.Label)), - bar.LabelColor, + labelColor, strings.Repeat(starChar, starCount), strings.Repeat(" ", maxStars-starCount), label, diff --git a/view/bargraph_test.go b/view/bargraph_test.go index c43cbc06..b9ccd270 100644 --- a/view/bargraph_test.go +++ b/view/bargraph_test.go @@ -40,7 +40,7 @@ func TestOutput(t *testing.T) { result := BuildStars(makeData(), 20, "*") Equal(t, - "Jun 27, 2018[[]****[white] ] 20\nJul 09, 2018[[red]****************[white] ] 80\nJul 09, 2018[[green]****************[white] ] 80\n", + "Jun 27, 2018[[default]****[default] ] 20\nJul 09, 2018[[red]****************[default] ] 80\nJul 09, 2018[[green]****************[default] ] 80\n", result, ) }