ability to switch color theme

data.go

@@ -6,12 +6,14 @@ import "time"
 type UIData struct {
 	Services      []*Service
 	Vars          []VarName
+	UITheme       string
 	LastTimestamp time.Time
 }
 
 // NewUIData inits and return new data object.
-func NewUIData(vars []VarName) *UIData {
+func NewUIData(vars []VarName, theme string) *UIData {
 	return &UIData{
-		Vars: vars,
+		Vars:    vars,
+		UITheme: theme,
 	}
 }

main.go

@@ -18,6 +18,7 @@ var (
 	dummy    = flag.Bool("dummy", false, "Use dummy (console) output")
 	self     = flag.Bool("self", false, "Monitor itself")
 	endpoint = flag.String("endpoint", DefaultEndpoint, "URL endpoint for expvars")
+	uiTheme  = flag.String("theme", "dark", "UI color theme. Possible values: dark, default.")
 )
 
 func main() {
@@ -52,7 +53,7 @@ func main() {
 	}
 
 	// Init UIData
-	data := NewUIData(vars)
+	data := NewUIData(vars, ParseUiTheme(*uiTheme))
 	for _, port := range ports {
 		service := NewService(port, vars)
 		data.Services = append(data.Services, service)

ui_multi.go

@@ -24,7 +24,7 @@ func (t *TermUI) Init(data UIData) error {
 		return err
 	}
 
-	termui.UseTheme("helloworld")
+	termui.UseTheme(data.UITheme)
 
 	t.Title = func() *termui.Par {
 		p := termui.NewPar("")

ui_single.go

@@ -25,7 +25,7 @@ func (t *TermUISingle) Init(data UIData) error {
 
 	t.Sparklines = make(map[VarName]*termui.Sparkline)
 
-	termui.UseTheme("helloworld")
+	termui.UseTheme(data.UITheme)
 
 	t.Title = func() *termui.Par {
 		p := termui.NewPar("")

utils.go

@@ -12,6 +12,14 @@ import (
 
 var ErrParsePorts = fmt.Errorf("cannot parse ports argument")
 
+func ParseUiTheme(theme string) string {
+	t := ""
+	if theme == "dark" {
+		t = "helloworld"
+	}
+	return t
+}
+
 // ParseVars returns parsed and validated slice of strings with
 // variables names that will be used for monitoring.
 func ParseVars(vars string) ([]VarName, error) {