162 lines
2.9 KiB
Go
162 lines
2.9 KiB
Go
package lumecmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
lifx "git.kill0.net/chill9/lume"
|
|
)
|
|
|
|
func powerColor(s string) string {
|
|
fs := "\033[1;31m%s\033[0m"
|
|
if s == "on" {
|
|
fs = "\033[1;32m%s\033[0m"
|
|
}
|
|
|
|
return fmt.Sprintf(fs, s)
|
|
}
|
|
|
|
func statusColor(s lifx.Status) string {
|
|
fs := "\033[1;31m%s\033[0m"
|
|
if s == "ok" {
|
|
fs = "\033[1;32m%s\033[0m"
|
|
}
|
|
|
|
return fmt.Sprintf(fs, s)
|
|
}
|
|
|
|
func PrintResults(res []lifx.Result) {
|
|
var length int
|
|
var widths map[string]int
|
|
|
|
widths = make(map[string]int)
|
|
|
|
for _, r := range res {
|
|
length = len(r.Id)
|
|
if widths["id"] < length {
|
|
widths["id"] = length
|
|
}
|
|
|
|
length = len(r.Label)
|
|
if widths["label"] < length {
|
|
widths["label"] = length
|
|
}
|
|
|
|
length = len(r.Status)
|
|
if widths["status"] < length {
|
|
widths["status"] = length
|
|
}
|
|
}
|
|
|
|
sortResults(res)
|
|
|
|
for _, r := range res {
|
|
fmt.Printf("%*s %*s %*s\n",
|
|
widths["id"], r.Id,
|
|
widths["label"], r.Label,
|
|
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 widths["id"] < length {
|
|
widths["id"] = length
|
|
}
|
|
|
|
length = len(l.Location.Name)
|
|
if widths["location"] < length {
|
|
widths["location"] = length
|
|
}
|
|
|
|
length = len(l.Group.Name)
|
|
if widths["group"] < length {
|
|
widths["group"] = length
|
|
}
|
|
|
|
length = len(l.Label)
|
|
if widths["label"] < length {
|
|
widths["label"] = length
|
|
}
|
|
|
|
length = len(l.LastSeen.Local().Format(time.RFC3339))
|
|
if widths["last_seen"] < length {
|
|
widths["last_seen"] = length
|
|
}
|
|
|
|
length = len(l.Power)
|
|
if widths["power"] < 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",
|
|
widths["id"], l.Id,
|
|
widths["loction"], l.Location.Name,
|
|
widths["group"], l.Group.Name,
|
|
widths["label"], l.Label,
|
|
widths["last_seen"], l.LastSeen.Local().Format(time.RFC3339),
|
|
widths["power"], powerColor(l.Power),
|
|
)
|
|
}
|
|
}
|
|
|
|
func parseRGB(s string) (lifx.RGBColor, error) {
|
|
var c lifx.RGBColor
|
|
rgb := strings.SplitN(s, ",", 3)
|
|
r, err := strconv.ParseUint(rgb[0], 10, 8)
|
|
if err != nil {
|
|
return c, err
|
|
}
|
|
g, err := strconv.ParseUint(rgb[1], 10, 8)
|
|
if err != nil {
|
|
return c, err
|
|
}
|
|
b, err := strconv.ParseUint(rgb[2], 10, 8)
|
|
if err != nil {
|
|
return c, err
|
|
}
|
|
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
|
|
})
|
|
}
|
|
|
|
func ExitWithCode(code int, err error) {
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err)
|
|
}
|
|
os.Exit(code)
|
|
}
|