e2m/dispatch
4
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add reset-config flag

Browse Source
This commit is contained in:
Ken-Håvard Lieng 2018-04-15 01:25:21 +02:00
parent 39641c315f
commit 0941ed8549
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
commands/dispatch.go
View File

@ -36,7 +36,7 @@ var rootCmd = &cobra.Command{
storage.Initialize(viper.GetString("dir")) storage.Initialize(viper.GetString("dir"))
initConfig(storage.Path.Config()) initConfig(storage.Path.Config(), viper.GetBool("reset_config"))
viper.SetConfigName("config") viper.SetConfigName("config")
viper.AddConfigPath(storage.Path.Root()) viper.AddConfigPath(storage.Path.Root())
@ -77,18 +77,20 @@ func init() {
rootCmd.AddCommand(configCmd) rootCmd.AddCommand(configCmd)
rootCmd.PersistentFlags().String("dir", storage.DefaultDirectory(), "directory to store config and data in") rootCmd.PersistentFlags().String("dir", storage.DefaultDirectory(), "directory to store config and data in")
rootCmd.PersistentFlags().Bool("reset-config", false, "reset to the default configuration, overwriting the current one")
rootCmd.Flags().IntP("port", "p", 80, "port to listen on") rootCmd.Flags().IntP("port", "p", 80, "port to listen on")
rootCmd.Flags().Bool("dev", false, "development mode") rootCmd.Flags().Bool("dev", false, "development mode")
viper.BindPFlag("dir", rootCmd.PersistentFlags().Lookup("dir")) viper.BindPFlag("dir", rootCmd.PersistentFlags().Lookup("dir"))
viper.BindPFlag("reset_config", rootCmd.PersistentFlags().Lookup("reset-config"))
viper.BindPFlag("port", rootCmd.Flags().Lookup("port")) viper.BindPFlag("port", rootCmd.Flags().Lookup("port"))
viper.BindPFlag("dev", rootCmd.Flags().Lookup("dev")) viper.BindPFlag("dev", rootCmd.Flags().Lookup("dev"))
viper.SetDefault("verify_client_certificates", true) viper.SetDefault("verify_client_certificates", true)
} }
func initConfig(configPath string) { func initConfig(configPath string, overwrite bool) {
if _, err := os.Stat(configPath); os.IsNotExist(err) { if _, err := os.Stat(configPath); overwrite || os.IsNotExist(err) {
config, err := assets.Asset("config.default.toml") config, err := assets.Asset("config.default.toml")
if err != nil { if err != nil {
log.Println(err) log.Println(err)