package lumecmd import ( "flag" "fmt" "strconv" "git.kill0.net/chill9/lume" ) type Config struct { AccessToken string Colors map[string]string `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 } 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 = "" ) 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") } commandRegistry[name] = cmd return nil } func GetCommand(name string) (Command, bool) { cmd, ok := commandRegistry[name] return cmd, ok }