brian/user-config
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

More Helper Functions

Browse Source 
Maybe some cleanup?
This commit is contained in:
Brian Buller 2017-03-01 06:58:34 -06:00
parent acc14262cb
commit bc9246ba22
2 changed files with 37 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
config.go
View File

@ -50,6 +50,11 @@ func (c *Config) SetDateTime(k string, v time.Time) error {
return c.generalConfig.SetDateTime(k, v) return c.generalConfig.SetDateTime(k, v)
} }
// SetArray saves a string slice in the <c.name>.conf file
func (c *Config) SetArray(k string, v []string) error {
return c.generalConfig.SetArray(k, v)
}
// Get at the config level retrieves a value from the <c.name>.conf file // Get at the config level retrieves a value from the <c.name>.conf file
func (c *Config) Get(k string) string { func (c *Config) Get(k string) string {
return c.generalConfig.Get(k) return c.generalConfig.Get(k)
@ -72,9 +77,13 @@ func (c *Config) GetDateTime(k string) (time.Time, error) {
return c.generalConfig.GetDateTime(k) return c.generalConfig.GetDateTime(k)
} }
func (c *Config) GetArray(k string) ([]string, error) {
return c.generalConfig.GetArray(k)
}
// DeleteKey at the config level removes a key from the <c.name>.conf file // DeleteKey at the config level removes a key from the <c.name>.conf file
func (c *Config) DeleteKey(k string) { func (c *Config) DeleteKey(k string) error {
c.generalConfig.DeleteKey(k) return c.generalConfig.DeleteKey(k)
} }
// GetConfigPath just returns the config path // GetConfigPath just returns the config path

30
config_file.go
View File

@ -3,6 +3,7 @@ package userConfig
import ( import (
"bytes" "bytes"
"encoding/json"
"errors" "errors"
"io/ioutil" "io/ioutil"
"strconv" "strconv"
@ -99,6 +100,15 @@ func (gf *GeneralConfig) SetDateTime(k string, v time.Time) error {
return gf.Set(k, v.Format(time.RFC3339)) return gf.Set(k, v.Format(time.RFC3339))
} }
// SetArray sets a string slice value (as a string) in the config file
func (gf *GeneralConfig) SetArray(k string, v []string) error {
b, e := json.Marshal(v)
if e != nil {
return e
}
return gf.SetBytes(k, b)
}
// Get gets a key/value pair from gf // Get gets a key/value pair from gf
func (gf *GeneralConfig) Get(k string) string { func (gf *GeneralConfig) Get(k string) string {
return gf.Values[k] return gf.Values[k]
@ -113,7 +123,7 @@ func (gf *GeneralConfig) GetInt(k string) (int, error) {
// GetDateTime gets a key/value pair from gf and returns it as a time.Time // GetDateTime gets a key/value pair from gf and returns it as a time.Time
// An error if it can't be converted // An error if it can't be converted
func (gf *GeneralConfig) GetDateTime(k string) (time.Time, error) { func (gf *GeneralConfig) GetDateTime(k string) (time.Time, error) {
return time.Parse(time.RFC3339, k) return time.Parse(time.RFC3339, gf.Get(k))
} }
// GetBytes gets a key/value pair from gf and returns it as a byte slice // GetBytes gets a key/value pair from gf and returns it as a byte slice
@ -122,7 +132,19 @@ func (gf *GeneralConfig) GetBytes(k string) []byte {
return []byte(gf.Get(k)) return []byte(gf.Get(k))
} }
// DeleteKey removes a key from the file func (gf *GeneralConfig) GetArray(k string) ([]string, error) {
func (gf *GeneralConfig) DeleteKey(k string) { var ret []string
delete(gf.Values, k) err := json.Unmarshal(gf.GetBytes(k), &ret)
return ret, err
}
// DeleteKey removes a key from the file
func (gf *GeneralConfig) DeleteKey(k string) error {
oldVal := gf.Get(k)
delete(gf.Values, k)
if err := gf.Save(); err != nil {
gf.Values[k] = oldVal
return err
}
return nil
} }