colorization

Author: MrZoidberg

cmd/power.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/MrZoidberg/megarac/api"
 	"github.com/MrZoidberg/megarac/lgr"
+	"github.com/fatih/color"
 	"github.com/urfave/cli/v2"
 )
 
@@ -42,7 +43,7 @@ func PowerOn(c *cli.Context) error {
 	}
 
 	if format == OutputFormatText {
-		lgr.Logger.Logf("[INFO] Server %s is powering on", profile.Host)
+		lgr.Logger.Logf("[INFO] Server %s is powering %s", profile.Host, lgr.ColorFormat(color.FgGreen, "on"))
 	} else {
 		result := map[string]interface{}{
 			"host": profile.Host,
@@ -94,7 +95,7 @@ func PowerOff(c *cli.Context) error {
 	}
 
 	if format == OutputFormatText {
-		lgr.Logger.Logf("[INFO] Server %s is powering off", profile.Host)
+		lgr.Logger.Logf("[INFO] Server %s is powering %s", profile.Host, lgr.ColorFormat(color.FgRed, "off"))
 	} else {
 		result := map[string]interface{}{
 			"host": profile.Host,
@@ -152,7 +153,13 @@ func PowerStatus(c *cli.Context) error {
 	}
 
 	if format == OutputFormatText {
-		lgr.Logger.Logf("[INFO] Power status for %s: %v", profile.Host, powerStatus)
+		powerStatusStr := ""
+		if powerStatus == "on" {
+			powerStatusStr = lgr.ColorFormat(color.FgGreen, powerStatus)
+		} else {
+			powerStatusStr = lgr.ColorFormat(color.FgRed, powerStatus)
+		}
+		lgr.Logger.Logf("[INFO] Power status for %s: %v", profile.Host, powerStatusStr)
 	} else {
 		result := map[string]interface{}{
 			"host": profile.Host,

cmd/sensor.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/MrZoidberg/megarac/api"
 	"github.com/MrZoidberg/megarac/lgr"
+	"github.com/fatih/color"
 	"github.com/urfave/cli/v2"
 )
 
@@ -61,12 +62,14 @@ func SensorList(c *cli.Context) error {
 			if !showAll && sensor.State == "inactive" || sensor.Accessible == "inaccessible" {
 				continue
 			}
+
 			if find && !strings.Contains(sensor.Name, find_str) {
 				continue
 			}
 
 			fmt.Fprintf(w, "%d\t%s\t%s\t%s %s\t%s\t%s\n", sensor.ID, sensor.Name, sensor.Type,
-				sensor.Reading, sensor.Unit, sensor.Alert, sensor.State)
+				colorizeReading(sensor.Reading, sensor.Alert, sensor.State), sensor.Unit,
+				colorizeAlert(sensor.Alert), colorizeState(sensor.State))
 		}
 		w.Flush()
 
@@ -94,3 +97,35 @@ func SensorList(c *cli.Context) error {
 	}
 	return nil
 }
+
+func colorizeAlert(alert string) string {
+	if len(alert) > 0 {
+		return lgr.ColorFormat(color.FgRed, alert)
+	}
+	return "\t"
+}
+
+func colorizeReading(reading string, alert string, state string) string {
+	if len(alert) > 0 {
+		return lgr.ColorFormat(color.FgRed, reading)
+	}
+	switch state {
+	case "inactive":
+		return lgr.ColorFormat(color.FgWhite, reading)
+	case "active":
+		return lgr.ColorFormat(color.FgGreen, reading)
+	default:
+		return reading
+	}
+}
+
+func colorizeState(state string) string {
+	switch state {
+	case "inactive":
+		return lgr.ColorFormat(color.FgRed, state)
+	case "active":
+		return lgr.ColorFormat(color.FgGreen, state)
+	default:
+		return state
+	}
+}

lgr/logger.go

@@ -9,6 +9,7 @@ import (
 
 var Logger lgr.L
 
+// SetupLog sets up logger
 func SetupLog() {
 	logOpts := []lgr.Option{lgr.Format("{{.Message}}")} // default to discard
 	dbg := os.Getenv("DEBUG") == "true"
@@ -29,3 +30,8 @@ func SetupLog() {
 
 	Logger = lgr.Std
 }
+
+// ColorFormat returns formatted string with color
+func ColorFormat(clr color.Attribute, format string, args ...interface{}) string {
+	return color.New(clr).Sprintf(format, args...)
+}