mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-14 02:53:50 +00:00
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
|
|
rotatelogs "github.com/lestrrat-go/file-rotatelogs"
|
|
"github.com/rifflock/lfshook"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
prefixed "github.com/x-cray/logrus-prefixed-formatter"
|
|
)
|
|
|
|
var RootCmd = &cobra.Command{
|
|
Use: "bbgo",
|
|
Short: "bbgo trade bot",
|
|
Long: "bitcoin trader",
|
|
|
|
// SilenceUsage is an option to silence usage when an error occurs.
|
|
SilenceUsage: true,
|
|
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
RootCmd.PersistentFlags().Bool("debug", false, "debug flag")
|
|
|
|
// A flag can be 'persistent' meaning that this flag will be available to
|
|
// the command it's assigned to as well as every command under that command.
|
|
// For global flags, assign a flag as a persistent flag on the root.
|
|
RootCmd.PersistentFlags().String("slack-token", "", "slack token")
|
|
RootCmd.PersistentFlags().String("slack-trading-channel", "dev-bbgo", "slack trading channel")
|
|
RootCmd.PersistentFlags().String("slack-error-channel", "bbgo-error", "slack error channel")
|
|
}
|
|
|
|
func Run() {
|
|
viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
|
|
|
|
// Enable environment variable binding, the env vars are not overloaded yet.
|
|
viper.AutomaticEnv()
|
|
|
|
// setup the config paths for looking up the config file
|
|
viper.AddConfigPath("config")
|
|
viper.AddConfigPath("$HOME/.bbgo")
|
|
viper.AddConfigPath("/etc/bbgo")
|
|
|
|
// set the config file name and format for loading the config file.
|
|
viper.SetConfigName("bbgo")
|
|
viper.SetConfigType("yaml")
|
|
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("failed to load config file")
|
|
}
|
|
|
|
// Once the flags are defined, we can bind config keys with flags.
|
|
if err := viper.BindPFlags(RootCmd.PersistentFlags()); err != nil {
|
|
log.WithError(err).Errorf("failed to bind persistent flags. please check the flag settings.")
|
|
}
|
|
|
|
if err := viper.BindPFlags(RootCmd.Flags()); err != nil {
|
|
log.WithError(err).Errorf("failed to bind local flags. please check the flag settings.")
|
|
}
|
|
|
|
log.SetFormatter(&prefixed.TextFormatter{})
|
|
|
|
logger := log.StandardLogger()
|
|
if viper.GetBool("debug") {
|
|
logger.SetLevel(log.DebugLevel)
|
|
}
|
|
|
|
environment := os.Getenv("BBGO_ENV")
|
|
switch environment {
|
|
case "production", "prod":
|
|
|
|
writer, err := rotatelogs.New(
|
|
path.Join("log", "access_log.%Y%m%d"),
|
|
rotatelogs.WithLinkName("access_log"),
|
|
// rotatelogs.WithMaxAge(24 * time.Hour),
|
|
rotatelogs.WithRotationTime(time.Duration(24)*time.Hour),
|
|
)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
logger.AddHook(
|
|
lfshook.NewHook(
|
|
lfshook.WriterMap{
|
|
log.DebugLevel: writer,
|
|
log.InfoLevel: writer,
|
|
log.WarnLevel: writer,
|
|
log.ErrorLevel: writer,
|
|
log.FatalLevel: writer,
|
|
},
|
|
&log.JSONFormatter{},
|
|
),
|
|
)
|
|
}
|
|
|
|
if err := RootCmd.Execute(); err != nil {
|
|
log.WithError(err).Fatalf("cannot execute command")
|
|
}
|
|
}
|