128 lines
2.3 KiB
Go
128 lines
2.3 KiB
Go
package lumecmd
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"git.kill0.net/chill9/lifx-go"
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
var userAgent string
|
|
|
|
func init() {
|
|
userAgent = initUserAgent()
|
|
|
|
RegisterCommand(NewCmdHelp())
|
|
RegisterCommand(NewCmdLs())
|
|
RegisterCommand(NewCmdPoweroff())
|
|
RegisterCommand(NewCmdPoweron())
|
|
RegisterCommand(NewCmdSetColor())
|
|
RegisterCommand(NewCmdSetState())
|
|
RegisterCommand(NewCmdSetWhite())
|
|
RegisterCommand(NewCmdShow())
|
|
RegisterCommand(NewCmdToggle())
|
|
RegisterCommand(NewCmdVersion())
|
|
}
|
|
|
|
var Version string = "0.1.0-pre"
|
|
|
|
const lumercFile string = ".lumerc"
|
|
|
|
func Main(args []string) (int, error) {
|
|
var config Config
|
|
var err error
|
|
|
|
if len(args) == 1 {
|
|
args = append(args, "help")
|
|
}
|
|
|
|
configPath := getConfigPath()
|
|
if configPath == "" {
|
|
err = errors.New("fatal: ~/.lumerc was not found")
|
|
return ExitFailure, err
|
|
}
|
|
|
|
if _, err := toml.DecodeFile(configPath, &config); err != nil {
|
|
err = fmt.Errorf("fatal: failed to parse %s; %w", configPath, err)
|
|
return ExitFailure, err
|
|
}
|
|
|
|
envAccessToken := os.Getenv("LIFX_ACCESS_TOKEN")
|
|
if envAccessToken != "" {
|
|
config.AccessToken = envAccessToken
|
|
}
|
|
|
|
if err = config.Validate(); err != nil {
|
|
return ExitFailure, fmt.Errorf("fatal: %s", err)
|
|
}
|
|
|
|
flag.Parse()
|
|
|
|
command := args[1]
|
|
|
|
c := lifx.NewClient(
|
|
config.AccessToken,
|
|
lifx.WithUserAgent(userAgent),
|
|
)
|
|
|
|
cmdArgs := CmdArgs{
|
|
Client: c,
|
|
Config: config,
|
|
}
|
|
|
|
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
|
|
fs.Parse(args[2:])
|
|
|
|
cmdArgs.Flags = Flags{FlagSet: fs}
|
|
cmdArgs.Name = command
|
|
exitCode, err := cmd.Func(cmdArgs)
|
|
if err != nil {
|
|
err = fmt.Errorf("fatal: %s", err)
|
|
}
|
|
|
|
return exitCode, err
|
|
}
|
|
|
|
func getConfigPath() string {
|
|
var tryPath, configPath string
|
|
|
|
// ~/.lumerc
|
|
homeDir, err := os.UserHomeDir()
|
|
if err == nil {
|
|
tryPath = path.Join(homeDir, lumercFile)
|
|
if _, err := os.Stat(tryPath); !os.IsNotExist(err) {
|
|
configPath = tryPath
|
|
}
|
|
}
|
|
|
|
// ./.lumerc
|
|
cwd, err := os.Getwd()
|
|
if err == nil {
|
|
tryPath = path.Join(cwd, lumercFile)
|
|
if _, err := os.Stat(tryPath); !os.IsNotExist(err) {
|
|
configPath = tryPath
|
|
}
|
|
}
|
|
|
|
return configPath
|
|
}
|
|
|
|
func initUserAgent() string {
|
|
var b strings.Builder
|
|
|
|
b.WriteString("lume")
|
|
b.WriteRune('/')
|
|
b.WriteString(Version)
|
|
return b.String()
|
|
}
|