taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

WTF-553 Fix config.yml creation issue on first run

Browse Source
This commit is contained in:
Chris Cummer 2019-08-29 21:24:02 -07:00
parent d08a5baf60
commit a8bde9ecbe
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
flags/flags.go
View File

@ -17,6 +17,8 @@ type Flags struct {
Module string `short:"m" long:"module" optional:"yes" description:"Display info about a specific module, i.e.: 'wtf -m=todo'"`
Profile bool `short:"p" long:"profile" optional:"yes" description:"Profile application memory usage"`
Version bool `short:"v" long:"version" description:"Show version info"`
hasCustom bool
}
// NewFlags creates an instance of Flags
@ -48,7 +50,7 @@ func (flags *Flags) RenderIf(version string, config *config.Config) {
// HasCustomConfig returns TRUE if a config path was passed in, FALSE if one was not
func (flags *Flags) HasCustomConfig() bool {
return len(flags.Config) > 0
return flags.hasCustom
}
// HasModule returns TRUE if a module name was passed in, FALSE if one was not
@ -70,15 +72,18 @@ func (flags *Flags) Parse() {
}
}
// If no config file is explicitly passed in as a param,
// set the flag to the default config file
if !flags.HasCustomConfig() {
homeDir, err := utils.Home()
if err != nil {
fmt.Printf("Error: %v\n", err)
os.Exit(1)
}
flags.Config = filepath.Join(homeDir, ".config", "wtf", "config.yml")
// If we have a custom config, then we're done parsing parameters, we don't need to
// generate the default value
flags.hasCustom = (len(flags.Config) > 0)
if flags.hasCustom == true {
return
}
// If no config file is explicitly passed in as a param then set the flag to the default config file
homeDir, err := utils.Home()
if err != nil {
fmt.Printf("Error: %v\n", err)
os.Exit(1)
}
flags.Config = filepath.Join(homeDir, ".config", "wtf", "config.yml")
}