mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type Root struct {
|
|
Config string
|
|
|
|
LogLevel zerolog.Level
|
|
LogTime string
|
|
LogJson bool
|
|
LogNocolor bool
|
|
LogDir string
|
|
}
|
|
|
|
func (Root) Init(cmd *cobra.Command) error {
|
|
cmd.PersistentFlags().StringP("config", "c", "", "configuration file path")
|
|
if err := viper.BindPFlag("config", cmd.PersistentFlags().Lookup("config")); err != nil {
|
|
return err
|
|
}
|
|
|
|
// just a shortcut
|
|
cmd.PersistentFlags().BoolP("debug", "d", false, "enable debug mode")
|
|
if err := viper.BindPFlag("debug", cmd.PersistentFlags().Lookup("debug")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("log.level", "info", "set log level (trace, debug, info, warn, error, fatal, panic, disabled)")
|
|
if err := viper.BindPFlag("log.level", cmd.PersistentFlags().Lookup("log.level")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("log.time", "unix", "time format used in logs (unix, unixms, unixmicro)")
|
|
if err := viper.BindPFlag("log.time", cmd.PersistentFlags().Lookup("log.time")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("log.json", false, "logs in JSON format")
|
|
if err := viper.BindPFlag("log.json", cmd.PersistentFlags().Lookup("log.json")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("log.nocolor", false, "no ANSI colors in non-JSON output")
|
|
if err := viper.BindPFlag("log.nocolor", cmd.PersistentFlags().Lookup("log.nocolor")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("log.dir", "", "logging directory to store logs")
|
|
if err := viper.BindPFlag("log.dir", cmd.PersistentFlags().Lookup("log.dir")); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Root) Set() {
|
|
s.Config = viper.GetString("config")
|
|
|
|
logLevel := viper.GetString("log.level")
|
|
level, err := zerolog.ParseLevel(logLevel)
|
|
if err != nil {
|
|
log.Warn().Msgf("unknown log level %s", logLevel)
|
|
} else {
|
|
s.LogLevel = level
|
|
}
|
|
|
|
logTime := viper.GetString("log.time")
|
|
switch logTime {
|
|
case "unix":
|
|
s.LogTime = zerolog.TimeFormatUnix
|
|
case "unixms":
|
|
s.LogTime = zerolog.TimeFormatUnixMs
|
|
case "unixmicro":
|
|
s.LogTime = zerolog.TimeFormatUnixMicro
|
|
default:
|
|
log.Warn().Msgf("unknown log time %s", logTime)
|
|
}
|
|
|
|
s.LogJson = viper.GetBool("log.json")
|
|
s.LogNocolor = viper.GetBool("log.nocolor")
|
|
s.LogDir = viper.GetString("log.dir")
|
|
|
|
if viper.GetBool("debug") && s.LogLevel != zerolog.TraceLevel {
|
|
s.LogLevel = zerolog.DebugLevel
|
|
}
|
|
|
|
// support for NO_COLOR env variable: https://no-color.org/
|
|
if os.Getenv("NO_COLOR") != "" {
|
|
s.LogNocolor = true
|
|
}
|
|
}
|