Add CLI output sort functions

cmd/ls.go

@@ -4,10 +4,6 @@ import (
 	"flag"
 )
 
-var (
-	idWidth, locationWidth, groupWidth, labelWidth, lastSeenWidth, powerWidth int
-)
-
 func init() {
 	var cmdName string = "ls"
 	fs := flag.NewFlagSet(cmdName, flag.ExitOnError)

cmd/util.go

@@ -2,11 +2,12 @@ package lumecmd
 
 import (
 	"fmt"
+	"sort"
 	"strconv"
 	"strings"
 	"time"
 
-	"git.kill0.net/chill9/lume"
+	lifx "git.kill0.net/chill9/lume"
 )
 
 func powerColor(s string) string {
@@ -28,78 +29,88 @@ func statusColor(s lifx.Status) string {
 }
 
 func PrintResults(res []lifx.Result) {
-	var length, idWidth, labelWidth, statusWidth int
+	var length int
+	var widths map[string]int
+
+	widths = make(map[string]int)
 
 	for _, r := range res {
 		length = len(r.Id)
-		if idWidth < length {
+		if widths["id"] < length {
-			idWidth = length
+			widths["id"] = length
 		}
 
 		length = len(r.Label)
-		if labelWidth < length {
+		if widths["label"] < length {
-			labelWidth = length
+			widths["label"] = length
 		}
 
 		length = len(r.Status)
-		if statusWidth < length {
+		if widths["status"] < length {
-			statusWidth = length
+			widths["status"] = length
 		}
 	}
 
+	sortResults(res)
+
 	for _, r := range res {
 		fmt.Printf("%*s %*s %*s\n",
-			idWidth, r.Id,
+			widths["id"], r.Id,
-			labelWidth, r.Label,
+			widths["label"], r.Label,
-			statusWidth, statusColor(r.Status))
+			widths["status"], statusColor(r.Status))
 	}
 }
 
 func PrintLights(lights []lifx.Light) {
 	var length int
+	var widths map[string]int
+
+	widths = make(map[string]int)
 
 	for _, l := range lights {
 		length = len(l.Id)
-		if idWidth < length {
+		if widths["id"] < length {
-			idWidth = length
+			widths["id"] = length
 		}
 
 		length = len(l.Location.Name)
-		if locationWidth < length {
+		if widths["location"] < length {
-			locationWidth = length
+			widths["location"] = length
 		}
 
 		length = len(l.Group.Name)
-		if groupWidth < length {
+		if widths["group"] < length {
-			groupWidth = length
+			widths["group"] = length
 		}
 
 		length = len(l.Label)
-		if labelWidth < length {
+		if widths["label"] < length {
-			labelWidth = length
+			widths["label"] = length
 		}
 
 		length = len(l.LastSeen.Local().Format(time.RFC3339))
-		if lastSeenWidth < length {
+		if widths["last_seen"] < length {
-			lastSeenWidth = length
+			widths["last_seen"] = length
 		}
 
 		length = len(l.Power)
-		if powerWidth < length {
+		if widths["power"] < length {
-			powerWidth = length
+			widths["power"] = length
 		}
 	}
 
+	sortLights(lights)
+
 	fmt.Printf("total %d\n", len(lights))
 	for _, l := range lights {
 		fmt.Printf(
 			"%*s %*s %*s %*s %*s %-*s\n",
-			idWidth, l.Id,
+			widths["id"], l.Id,
-			locationWidth, l.Location.Name,
+			widths["loction"], l.Location.Name,
-			groupWidth, l.Group.Name,
+			widths["group"], l.Group.Name,
-			labelWidth, l.Label,
+			widths["label"], l.Label,
-			lastSeenWidth, l.LastSeen.Local().Format(time.RFC3339),
+			widths["last_seen"], l.LastSeen.Local().Format(time.RFC3339),
-			powerWidth, powerColor(l.Power),
+			widths["power"], powerColor(l.Power),
 		)
 	}
 }
@@ -121,3 +132,22 @@ func parseRGB(s string) (lifx.RGBColor, error) {
 	}
 	return lifx.NewRGBColor(uint8(r), uint8(g), uint8(b))
 }
+
+func sortLights(lights []lifx.Light) {
+	sort.Slice(lights, func(i, j int) bool {
+		if lights[i].Group.Name < lights[j].Group.Name {
+			return true
+		}
+		if lights[i].Group.Name > lights[j].Group.Name {
+			return false
+		}
+		return lights[i].Label < lights[j].Label
+
+	})
+}
+
+func sortResults(res []lifx.Result) {
+	sort.Slice(res, func(i, j int) bool {
+		return res[i].Label < res[j].Label
+	})
+}