Dawid Pogorzelski
4a0574554a
* Fix loglevel parsing * Bump dependencies * Change --filter to --target * Update documentation * Import log handler as "loghandler" * Change command color to purple * Change flag name to target * Update formatting
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/apex/log"
|
|
homedir "github.com/mitchellh/go-homedir"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var cfgFile string
|
|
var version string
|
|
var commit string
|
|
var date string
|
|
|
|
//Execute runs the root command
|
|
func Execute() {
|
|
// cobra.OnInitialize(initConfig)
|
|
var rootCmd = &cobra.Command{
|
|
Use: "speedrun",
|
|
Short: "Cloud first command execution",
|
|
Version: fmt.Sprintf("%s, commit: %s, date: %s", version, commit, date),
|
|
SilenceUsage: true,
|
|
SilenceErrors: true,
|
|
}
|
|
|
|
rootCmd.AddCommand(initCmd)
|
|
rootCmd.AddCommand(keyCmd)
|
|
rootCmd.AddCommand(runCmd)
|
|
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
dir := filepath.Join(home, ".speedrun")
|
|
path := filepath.Join(dir, "config.toml")
|
|
|
|
rootCmd.PersistentFlags().StringVarP(&cfgFile, "config", "c", path, "config file")
|
|
rootCmd.PersistentFlags().StringP("loglevel", "l", "info", "Log level")
|
|
viper.BindPFlag("loglevel", rootCmd.PersistentFlags().Lookup("loglevel"))
|
|
|
|
if err := rootCmd.Execute(); err != nil {
|
|
log.Fatal(err.Error())
|
|
}
|
|
}
|
|
|
|
func initConfig() {
|
|
dir, file := filepath.Split(cfgFile)
|
|
viper.SetConfigName(file)
|
|
viper.SetConfigType("toml")
|
|
viper.AddConfigPath(dir)
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
log.Fatal("Run `speedrun init` first")
|
|
}
|
|
} else {
|
|
log.Fatal(err.Error())
|
|
}
|
|
}
|
|
lvl, err := log.ParseLevel(viper.GetString("loglevel"))
|
|
if err != nil {
|
|
log.Fatalf("Couldn't parse log level: %s", err)
|
|
return
|
|
}
|
|
log.SetLevel(lvl)
|
|
|
|
}
|