Emmanuel BENOîT
610cbf28f8
* An example configuration file has been added.
* The data structure that corresponds to the configuration has been
defined, and functions to load it have been added.
* Logging using logrus and command line arguments that configure
logging and set the configuration file's path have been added.
* Opening the UNIX socket has been implemented.
73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/karrick/golf"
|
|
)
|
|
|
|
type (
|
|
// This structure contains all values that may be set from the command line.
|
|
cliFlags struct {
|
|
// The path to the configuration file.
|
|
cfgFile string
|
|
// Quiet mode. Will disable logging to stderr.
|
|
quiet bool
|
|
// The log level.
|
|
logLevel string
|
|
// A file to write logs into.
|
|
logFile string
|
|
// Graylog server to send logs to (using GELF/UDP). Format is <hostname>:<port>.
|
|
logGraylog string
|
|
// Send logs to syslog.
|
|
logSyslog bool
|
|
}
|
|
)
|
|
|
|
// Parse command line options.
|
|
func parseCommandLine() cliFlags {
|
|
var help bool
|
|
flags := cliFlags{}
|
|
|
|
golf.StringVarP(&flags.cfgFile, 'c', "config", "/etc/fetch-certificates.yml", "Path to the configuration file.")
|
|
golf.StringVarP(&flags.logFile, 'f', "log-file", "", "Path to the log file.")
|
|
golf.StringVarP(&flags.logGraylog, 'g', "log-graylog", "", "Log to Graylog server (format: <host>:<port>).")
|
|
golf.BoolVarP(&help, 'h', "help", false, "Display command line help and exit.")
|
|
golf.StringVarP(&flags.logLevel, 'L', "log-level", "info", "Log level to use.")
|
|
golf.BoolVarP(&flags.quiet, 'q', "quiet", false, "Quiet mode; prevents logging to stderr.")
|
|
golf.BoolVarP(&flags.logSyslog, 's', "syslog", false, "Log to local syslog.")
|
|
|
|
golf.Parse()
|
|
if help {
|
|
golf.Usage()
|
|
os.Exit(0)
|
|
}
|
|
return flags
|
|
}
|
|
|
|
func main() {
|
|
// This utility will load its configuration then start listening on
|
|
// a UNIX socket. It will be handle messages that can :
|
|
// - stop the program,
|
|
// - update the configuration,
|
|
// - check a single entry for replacement,
|
|
// - check all entries for replacement.
|
|
// Both check commands include a flag that will force replacement.
|
|
|
|
flags := parseCommandLine()
|
|
err := configureLogging(flags)
|
|
if err != nil {
|
|
log.WithField("error", err).Fatal("Failed to configure logging.")
|
|
}
|
|
|
|
cfg, err := loadConfiguration(flags.cfgFile)
|
|
if err != nil {
|
|
log.WithField("error", err).Fatal("Failed to load initial configuration.")
|
|
}
|
|
|
|
listener, err := initSocket(cfg.Socket)
|
|
if err != nil {
|
|
log.WithField("error", err).Fatal("Failed to initialize socket.")
|
|
}
|
|
listener.Close()
|
|
}
|