diff options
Diffstat (limited to 'init.go')
| -rw-r--r-- | init.go | 80 |
1 files changed, 54 insertions, 26 deletions
diff --git a/init.go b/init.go index f37968c..fdadca8 100644 --- a/init.go +++ b/init.go @@ -1,6 +1,7 @@ package main import ( + "fmt" "log" "os" "time" @@ -12,9 +13,10 @@ import ( // command line flags var ( - port *int = pflag.IntP("port", "p", 9001, "getwtxt will serve from this port") - logfile *string = pflag.StringP("logfile", "l", "getwtxt.log", "File for logging output") - twtxtfile *string = pflag.StringP("twtxtfile", "f", "/var/twtxt/twtxt.txt", "Registry file for getwtxt") + port *int = pflag.IntP("port", "p", 9001, "getwtxt will serve from this port") + logfile *string = pflag.StringP("logfile", "l", "getwtxt.log", "File for logging output") + twtxtfile *string = pflag.StringP("twtxtfile", "f", "/var/twtxt/twtxt.txt", "Registry file for getwtxt") + stdoutLogging *bool = pflag.BoolP("stdout", "o", true, "Log to stdout rather than to a file") ) // config object @@ -24,24 +26,23 @@ var confObj = &configuration{} var closelog = make(chan bool, 1) func init() { - + titleScreen() initConfig() - - pflag.Parse() - viper.BindPFlags(pflag.CommandLine) - initLogging() } func initConfig() { - viper.SetConfigName("getwtxt") + pflag.Parse() + viper.BindPFlags(pflag.CommandLine) + viper.SetConfigName("getwtxt") viper.AddConfigPath(".") viper.AddConfigPath("/usr/local/getwtxt") viper.AddConfigPath("/etc") viper.AddConfigPath("/usr/local/etc") + log.Printf("Loading configuration ...\n") if err := viper.ReadInConfig(); err != nil { log.Printf("Error reading config file: %v\n", err) log.Printf("Using defaults ...\n") @@ -62,40 +63,67 @@ func initConfig() { confObj.port = viper.GetInt("port") confObj.logfile = viper.GetString("logfile") confObj.twtxtfile = viper.GetString("twtxtfile") + confObj.stdoutLogging = viper.GetBool("stdoutLogging") } func initLogging() { - logfile, err := os.OpenFile(confObj.logfile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600) - if err != nil { - log.Printf("Could not open log file: %v\n", err) - } + // only open a log file if it's necessary + if !confObj.stdoutLogging { - // Listen for the signal to close the log file - // in a separate thread - go func() { - <-closelog - log.Printf("Closing log file ...\n") - err = logfile.Close() + logfile, err := os.OpenFile(confObj.logfile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600) if err != nil { - log.Printf("Couldn't close log file: %v\n", err) + log.Printf("Could not open log file: %v\n", err) } - }() - log.SetOutput(logfile) + // Listen for the signal to close the log file + // in a separate thread + go func() { + <-closelog + log.Printf("Closing log file ...\n") + err = logfile.Close() + if err != nil { + log.Printf("Couldn't close log file: %v\n", err) + } + }() + + log.SetOutput(logfile) + + } else { + log.SetOutput(os.Stdout) + } } func rebindConfig() { + // signal to close the log file then wait + if !confObj.stdoutLogging { + closelog <- true + time.Sleep(50 * time.Millisecond) + } + // reassign values to the config object confObj.port = viper.GetInt("port") confObj.logfile = viper.GetString("logfile") confObj.twtxtfile = viper.GetString("twtxtfile") - - // signal to close the log file then wait - closelog <- true - time.Sleep(50 * time.Millisecond) + confObj.stdoutLogging = viper.GetBool("stdoutLogging") // reinitialize logging initLogging() } + +func titleScreen() { + fmt.Printf(` + + _ _ _ + __ _ ___| |___ _| |___ _| |_ + / _ |/ _ \ __\ \ /\ / / __\ \/ / __| +| (_| | __/ |_ \ V V /| |_ > <| |_ + \__, |\___|\__| \_/\_/ \__/_/\_\\__| + |___/ + version ` + getwtxt + ` + github.com/gbmor/getwtxt + GPL v3 + +`) +} |