lume/cmd/main.go

package lumecmd

import (
	"errors"
	"flag"
	"fmt"

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

func init() {
	RegisterCommand(NewCmdHelp())
	RegisterCommand(NewCmdLs())
	RegisterCommand(NewCmdPoweroff())
	RegisterCommand(NewCmdPoweron())
	RegisterCommand(NewCmdSetColor())
	RegisterCommand(NewCmdSetState())
	RegisterCommand(NewCmdSetWhite())
	RegisterCommand(NewCmdShow())
	RegisterCommand(NewCmdToggle())
	RegisterCommand(NewCmdVersion())
	RegisterCommand(NewCmdBreathe())

	flag.BoolVar(&debugFlag, "debug", false, "debug mode")
	flag.BoolVar(&debugFlag, "d", false, "debug mode")
}

var Version string
var BuildDate string
var GitCommit string
var debugFlag bool

func Main(args []string) (int, error) {
	var config *Config = GetConfig()
	var err error
	var i int

	flag.Parse()
	i = flag.NFlag() + 1

	if len(args) == 1 {
		args = append(args, "help")
	}

	configPath := getConfigPath()
	if configPath == "" {
		err = errors.New("fatal: ~/.lumerc was not found")
		return ExitFailure, err
	}

	if config, err = LoadConfigFile(configPath); err != nil {
		return ExitFailure, err
	}
	config.MergeWithEnv()

	if err = config.Validate(); err != nil {
		return ExitFailure, fmt.Errorf("fatal: %s", err)
	}

	config.Debug = debugFlag

	command := args[i]
	i++

	c := lifx.NewClient(
		config.AccessToken,
		lifx.WithUserAgent(config.userAgent),
		lifx.WithDebug(debugFlag),
	)

	Context := Context{
		Client: c,
		Config: *config,
		Args:   args[i:],
	}

	cmd, ok := GetCommand(command)
	if !ok {
		err = fmt.Errorf("lume: '%s' is not lume command. See 'lume help'", command)
		return ExitFailure, err
	}

	fs := cmd.Flags
	if fs != nil {
		fs.Parse(args[i:])
		Context.Flags = Flags{FlagSet: fs}
	}
	Context.Name = command

	exitCode, err := cmd.Func(Context)
	if err != nil {
		err = fmt.Errorf("fatal: %s", err)
	}

	return exitCode, err
}