diff options
Diffstat (limited to 'init.go')
| -rw-r--r-- | init.go | 59 |
1 files changed, 44 insertions, 15 deletions
diff --git a/init.go b/init.go index 6fd6bc8..3f3440c 100644 --- a/init.go +++ b/init.go @@ -19,8 +19,10 @@ const getwtxt = "0.1" // command line flags var ( - flagVersion *bool = pflag.BoolP("version", "v", false, "Display version information, then exit") - flagHelp *bool = pflag.BoolP("help", "h", false, "") + flagVersion *bool = pflag.BoolP("version", "v", false, "Display version information, then exit.") + flagHelp *bool = pflag.BoolP("help", "h", false, "Display the help screen") + flagConfFile *string = pflag.StringP("config", "c", "getwtxt", "The name of the configuration file without an extension.") + flagConfType *string = pflag.StringP("type", "t", "yml", "The filetype of the configuration file.") ) // config object @@ -80,8 +82,8 @@ func checkFlags() { func initConfig() { - viper.SetConfigName("getwtxt") - viper.SetConfigType("yml") + viper.SetConfigName(*flagConfFile) + viper.SetConfigType(*flagConfType) viper.AddConfigPath(".") viper.AddConfigPath("/usr/local/getwtxt") viper.AddConfigPath("/etc") @@ -89,18 +91,18 @@ func initConfig() { log.Printf("Loading configuration ...\n") if err := viper.ReadInConfig(); err != nil { - log.Printf("Error reading config file: %v\n", err) + log.Printf("%v\n", err) log.Printf("Using defaults ...\n") + } else { + // separate thread to watch for config file changes. + // will log event then run rebindConfig() + viper.WatchConfig() + viper.OnConfigChange(func(e fsnotify.Event) { + log.Printf("Config file change detected. Reloading...\n") + rebindConfig() + }) } - // separate thread to watch for config file changes. - // will log event then run rebindConfig() - viper.WatchConfig() - viper.OnConfigChange(func(e fsnotify.Event) { - log.Printf("Config file change detected. Reloading...\n") - rebindConfig() - }) - viper.SetDefault("ListenPort", 9001) viper.SetDefault("LogFile", "getwtxt.log") viper.SetDefault("DatabasePath", "getwtxt.db") @@ -108,29 +110,43 @@ func initConfig() { viper.SetDefault("ReCacheInterval", "1h") viper.SetDefault("DatabasePushInterval", "5m") + viper.SetDefault("Instance.SiteName", "getwtxt") + viper.SetDefault("Instance.OwnerName", "Anonymous Microblogger") + viper.SetDefault("Instance.Email", "nobody@knows") + viper.SetDefault("Instance.URL", "https://twtxt.example.com") + viper.SetDefault("Instance.Description", "A fast, resilient twtxt registry server written in Go!") + confObj.Mu.Lock() + confObj.Port = viper.GetInt("ListenPort") confObj.LogFile = viper.GetString("LogFile") + confObj.DBPath = viper.GetString("DatabasePath") log.Printf("Using database: %v\n", confObj.DBPath) + confObj.StdoutLogging = viper.GetBool("StdoutLogging") if confObj.StdoutLogging { log.Printf("Logging to stdout\n") } else { log.Printf("Logging to %v\n", confObj.LogFile) } + confObj.CacheInterval = viper.GetDuration("StatusFetchInterval") log.Printf("User status fetch interval: %v\n", confObj.CacheInterval) + confObj.DBInterval = viper.GetDuration("DatabasePushInterval") log.Printf("Database push interval: %v\n", confObj.DBInterval) + confObj.LastCache = time.Now() confObj.LastPush = time.Now() confObj.Version = getwtxt + confObj.Instance.Name = viper.GetString("Instance.SiteName") confObj.Instance.URL = viper.GetString("Instance.URL") confObj.Instance.Owner = viper.GetString("Instance.OwnerName") confObj.Instance.Mail = viper.GetString("Instance.Email") confObj.Instance.Desc = viper.GetString("Instance.Description") + confObj.Mu.Unlock() } @@ -181,16 +197,19 @@ func rebindConfig() { // reassign values to the config object confObj.Mu.Lock() + confObj.LogFile = viper.GetString("LogFile") confObj.DBPath = viper.GetString("DatabasePath") confObj.StdoutLogging = viper.GetBool("StdoutLogging") confObj.CacheInterval = viper.GetDuration("StatusFetchInterval") confObj.DBInterval = viper.GetDuration("DatabasePushInterval") + confObj.Instance.Name = viper.GetString("Instance.SiteName") confObj.Instance.URL = viper.GetString("Instance.URL") confObj.Instance.Owner = viper.GetString("Instance.OwnerName") confObj.Instance.Mail = viper.GetString("Instance.Email") confObj.Instance.Desc = viper.GetString("Instance.Description") + confObj.Mu.Unlock() // reinitialize logging @@ -271,8 +290,18 @@ func titleScreen() { func helpScreen() { fmt.Printf(` - Help File + getwtxt Help + +Command line options: + -h Print this help screen. + -v Print the version information and quit. + -c [--config] Name of an alternate configuration file + to use. Do not include the file extention, + such as ".yml". Must be in the expected + locations. + -t [--type] The file type / extension of the config file. + json, yml, etc. + - Sections: `) } |