131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package lumecmd
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
lifx "git.kill0.net/chill9/lume"
|
|
)
|
|
|
|
const (
|
|
ExitSuccess = iota
|
|
ExitFailure
|
|
)
|
|
|
|
type Config struct {
|
|
AccessToken string `toml:"access_token"`
|
|
Colors map[string][]float32 `toml:"colors"`
|
|
}
|
|
|
|
type CmdArgs struct {
|
|
Flags Flags
|
|
Client *lifx.Client
|
|
Config Config
|
|
}
|
|
|
|
type Flags struct {
|
|
*flag.FlagSet
|
|
}
|
|
|
|
type Command struct {
|
|
Name string
|
|
Func func(CmdArgs) (int, error)
|
|
Flags *flag.FlagSet
|
|
Use string
|
|
Short string
|
|
Long string
|
|
}
|
|
|
|
var commandRegistry = make(map[string]Command)
|
|
|
|
var (
|
|
defaultSelector string = "all"
|
|
defaultDuration float64 = 1.0
|
|
defaultPower string = ""
|
|
defaultColor string = ""
|
|
defaultBrightness string = ""
|
|
defaultInfrared string = ""
|
|
defaultFast bool = false
|
|
defaultWhiteKelvin string = ""
|
|
defaultWhiteName string = ""
|
|
defaultHue string = ""
|
|
defaultSaturation string = ""
|
|
defaultRGB string = ""
|
|
defaultName string = ""
|
|
)
|
|
|
|
func (f Flags) String(name string) string {
|
|
return f.FlagSet.Lookup(name).Value.String()
|
|
}
|
|
|
|
func (f Flags) Float32(name string) float32 {
|
|
val, _ := strconv.ParseFloat(f.String(name), 32)
|
|
return float32(val)
|
|
}
|
|
|
|
func (f Flags) Float64(name string) float64 {
|
|
val, _ := strconv.ParseFloat(f.String(name), 64)
|
|
return val
|
|
}
|
|
|
|
func (f Flags) Int16(name string) int16 {
|
|
val, _ := strconv.ParseInt(f.String(name), 10, 16)
|
|
return int16(val)
|
|
}
|
|
|
|
func (f Flags) Bool(name string) bool {
|
|
val, _ := strconv.ParseBool(f.String(name))
|
|
return val
|
|
}
|
|
|
|
func RegisterCommand(name string, cmd Command) error {
|
|
if _, ok := commandRegistry[name]; ok {
|
|
return fmt.Errorf("%s command is already registered")
|
|
}
|
|
cmd.Name = name
|
|
commandRegistry[name] = cmd
|
|
return nil
|
|
}
|
|
|
|
func GetCommand(name string) (Command, bool) {
|
|
cmd, ok := commandRegistry[name]
|
|
return cmd, ok
|
|
}
|
|
|
|
// Validate configuration struct
|
|
func (c *Config) Validate() error {
|
|
var err error
|
|
if c.AccessToken == "" {
|
|
err = errors.New("access_token is not set")
|
|
}
|
|
|
|
if err = c.validateColors(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (c *Config) validateColors() (err error) {
|
|
if len(c.Colors) > 0 {
|
|
for name, hsb := range c.Colors {
|
|
if len(hsb) != 3 {
|
|
return fmt.Errorf("color '%s' needs three values", name)
|
|
}
|
|
h, s, b := hsb[0], hsb[1], hsb[2]
|
|
if h < 0 || h > 360 {
|
|
return fmt.Errorf("color '%s' hue value must be between 0.0-360.0", name)
|
|
}
|
|
if s < 0 || b > 1 {
|
|
return fmt.Errorf("color '%s' saturation value must be between 0.0-1.0", name)
|
|
}
|
|
if b < 0 || b > 1 {
|
|
return fmt.Errorf("color '%s' brightness value must be between 0.0-1.0", name)
|
|
}
|
|
}
|
|
}
|
|
return err
|
|
}
|