Skip to content

Commit ca13fec

Browse files
committed
refactor(config): cleaned up code a bit more
1 parent 2e93e2a commit ca13fec

File tree

2 files changed

+13
-26
lines changed

2 files changed

+13
-26
lines changed

cmd/root.go

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,12 @@ func Execute() {
3737

3838
func init() {
3939
cobra.OnInitialize(initConfig)
40-
confDir, err := config.GetConfigDir()
41-
cobra.CheckErr(err)
42-
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", fmt.Sprintf("config file (default is %v)", confDir))
40+
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", fmt.Sprintf("config file (default is %v)", config.GetConfigDir()))
4341
// Instead of profiles for now, I recommend just passing in different config files.
4442

4543
}
4644

4745
// initConfig reads in config file and ENV variables if set.
4846
func initConfig() {
49-
err := config.Configure(cfgFile)
50-
cobra.CheckErr(err)
47+
config.Configure(cfgFile)
5148
}

lib/config/config.go

Lines changed: 11 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -16,29 +16,24 @@ import (
1616
)
1717

1818
// Configures Toolsium based on the cfgFile path passed in.
19-
func Configure(cfgFile string) error { return t.Configure(cfgFile) }
20-
func (t *Toolsium) Configure(cfgFile string) (err error) {
21-
var installDir string
19+
func Configure(cfgFile string) { t.Configure(cfgFile) }
20+
func (t *Toolsium) Configure(cfgFile string) {
2221
if cfgFile != "" {
2322
// Use config file from the flag.
2423
t.Config.SetConfigFile(cfgFile)
2524
} else {
26-
// Find installDir directory.
27-
installDir, err = t.GetConfigDir()
28-
2925
// Search config in home directory with name ".toolsium" (without extension).
30-
t.ConfigureViper(installDir)
26+
t.ConfigureViper(t.GetConfigDir())
3127
}
3228
// If a config file is found, read it in.
3329
if err := t.Config.ReadInConfig(); err == nil {
3430
log.Infof("Using config file: %v", t.Config.ConfigFileUsed())
3531
}
36-
return
3732
}
3833

3934
// Returns the current Config Directory
40-
func GetConfigDir() (string, error) { return t.GetConfigDir() }
41-
func (t *Toolsium) GetConfigDir() (path string, err error) {
35+
func GetConfigDir() string { return t.GetConfigDir() }
36+
func (t *Toolsium) GetConfigDir() (path string) {
4237
if t.ConfigDir != "" {
4338
path = t.ConfigDir
4439
} else {
@@ -63,11 +58,9 @@ func (t *Toolsium) SetConfigDir(confDir string) (err error) {
6358
// Returns the current Config Directory
6459
//
6560
// TODO(JR): Change usage of cfgFileName to allow changes to name
66-
func GetConfigFile() (string, error) { return t.GetConfigFile() }
67-
func (t *Toolsium) GetConfigFile() (path string, err error) {
68-
directory, err := t.GetConfigDir()
69-
path = filepath.Join(directory, cfgFileName)
70-
return
61+
func GetConfigFile() string { return t.GetConfigFile() }
62+
func (t *Toolsium) GetConfigFile() string {
63+
return filepath.Join(t.GetConfigDir(), cfgFileName)
7164
}
7265

7366
// TODO(JR): Add SetConfigFile
@@ -84,11 +77,11 @@ func (t *Toolsium) ConfigureViper(installDir string) {
8477
}
8578

8679
func (t *Toolsium) createConfigDirectory() {
87-
path, _ := t.GetConfigDir()
80+
path := t.GetConfigDir()
8881
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
8982
err := os.Mkdir(path, os.ModePerm)
9083
if err != nil {
91-
log.Println(err)
84+
log.Errorln(err)
9285
}
9386
}
9487
}
@@ -97,10 +90,7 @@ func (t *Toolsium) createConfigDirectory() {
9790
func CreateConfig() error { return t.CreateConfig() }
9891
func (t *Toolsium) CreateConfig() (err error) {
9992
t.createConfigDirectory()
100-
cfgFile, err := t.GetConfigFile()
101-
if err != nil {
102-
return
103-
}
93+
cfgFile := t.GetConfigFile()
10494
log.Infof("Wrote Config to %v", cfgFile)
10595
err = t.Config.WriteConfigAs(cfgFile)
10696
return

0 commit comments

Comments
 (0)