Emmanuel BENOîT
d3466b9609
With logrus' default formatter it boils down to a choice between the colored version OR the human-unreadable list of fields, so the latter is preferable.
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
lrh_wr "github.com/sirupsen/logrus/hooks/writer"
|
|
)
|
|
|
|
type (
|
|
// This structure contains all values that may be set from the command line.
|
|
cliFlags struct {
|
|
// The path to the configuration file.
|
|
cfgFile string
|
|
// The name of the instance, to be used in logs.
|
|
instance string
|
|
// Quiet mode. Will disable logging to stderr.
|
|
quiet bool
|
|
// The log level.
|
|
logLevel string
|
|
// A file to write logs into.
|
|
logFile string
|
|
}
|
|
)
|
|
|
|
var (
|
|
// The logging context.
|
|
log *logrus.Entry
|
|
)
|
|
|
|
// Parse command line options.
|
|
func parseCommandLine() cliFlags {
|
|
flags := cliFlags{}
|
|
flag.StringVar(&flags.cfgFile, "c", "graylog-groups.yml", "Configuration file.")
|
|
flag.StringVar(&flags.instance, "i", "", "Instance identifier.")
|
|
flag.BoolVar(&flags.quiet, "q", false, "Quiet mode.")
|
|
flag.StringVar(&flags.logLevel, "L", "", "Log level.")
|
|
flag.StringVar(&flags.logFile, "log-file", "", "Log file.")
|
|
flag.Parse()
|
|
return flags
|
|
}
|
|
|
|
// Initialize the logging context.
|
|
func getLoggingContext(instance string) *logrus.Entry {
|
|
logFields := logrus.Fields{
|
|
"application": "graylog",
|
|
"component": "graylog-groups",
|
|
}
|
|
if instance != "" {
|
|
logFields["instance"] = instance
|
|
}
|
|
return logrus.WithFields(logFields)
|
|
}
|
|
|
|
// Configure the log level
|
|
func toLogLevel(cliLevel string) logrus.Level {
|
|
if cliLevel == "" {
|
|
return logrus.InfoLevel
|
|
}
|
|
lvl, err := logrus.ParseLevel(cliLevel)
|
|
if err == nil {
|
|
return lvl
|
|
}
|
|
log.WithField("level", cliLevel).Warning("Invalid log level on command line")
|
|
return logrus.InfoLevel
|
|
}
|
|
|
|
// Add a file writer hook to the logging library.
|
|
func configureLogFile(path string) {
|
|
file, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
|
|
if err == nil {
|
|
log.Logger.AddHook(&lrh_wr.Hook{
|
|
Writer: file,
|
|
LogLevels: logrus.AllLevels,
|
|
})
|
|
} else {
|
|
log.WithFields(logrus.Fields{
|
|
"error": err,
|
|
"file": path,
|
|
}).Error("Could not open log file")
|
|
}
|
|
}
|
|
|
|
// Configure the logging library based on the various command line flags.
|
|
func configureLogging(flags cliFlags) {
|
|
log = getLoggingContext(flags.instance)
|
|
log.Logger.SetFormatter(&logrus.TextFormatter{
|
|
DisableColors: true,
|
|
FullTimestamp: true,
|
|
})
|
|
log.Logger.SetLevel(toLogLevel(flags.logLevel))
|
|
if flags.logFile != "" {
|
|
configureLogFile(flags.logFile)
|
|
}
|
|
if flags.quiet {
|
|
log.Logger.SetOutput(ioutil.Discard)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flags := parseCommandLine()
|
|
configureLogging(flags)
|
|
configuration := loadConfiguration(flags)
|
|
glUsers := getGraylogUsers(configuration.Graylog)
|
|
groups := readLdapGroups(configuration)
|
|
applyMapping(configuration, glUsers, groups)
|
|
}
|