lume/cmd/command.go

package lumecmd

import (
	"errors"
	"flag"
	"fmt"
	"strconv"

	"git.kill0.net/chill9/lifx-go"
)

const (
	ExitSuccess = iota
	ExitFailure
)

type Config struct {
	AccessToken  string               `toml:"access_token"`
	OutputFormat string               `toml:"output_format"`
	Colors       map[string][]float32 `toml:"colors"`
}

type CmdArgs struct {
	Flags  Flags
	Args   []string
	Client *lifx.Client
	Config Config
	Name   string
}

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  = ""
	defaultOutputFormat 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(cmd Command) error {
	if _, ok := commandRegistry[cmd.Name]; ok {
		return fmt.Errorf("%s command is already registered")
	}
	commandRegistry[cmd.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
}