Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

allow all config options to also be envVars using subpaths #70

Merged
merged 2 commits into from
Jan 8, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 36 additions & 2 deletions stim/config.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package stim

import (
"fmt"
"io/ioutil"
"os"
"path"
Expand All @@ -13,7 +14,15 @@ import (
)

func (stim *Stim) ConfigGetRaw(configKey string) interface{} {
var envCV interface{} = nil
configValue := stim.config.Get(configKey)
if strings.Contains(configKey, ".") {
envCK := strings.ReplaceAll(configKey, ".", "-")
envCV = stim.config.Get(envCK)
}
if envCV != nil {
return envCV
}
if configValue != nil {
return configValue
}
Expand All @@ -22,21 +31,45 @@ func (stim *Stim) ConfigGetRaw(configKey string) interface{} {
}

func (stim *Stim) ConfigGetString(configKey string) string {
var envCV string = ""
configValue := stim.config.GetString(configKey)
if strings.Contains(configKey, ".") {
envCK := strings.ReplaceAll(configKey, ".", "-")
envCV = stim.config.GetString(envCK)
}
if envCV != "" {
return envCV
}
return configValue
}

// GetConfigBool takes a config key and returns the boolean result
func (stim *Stim) ConfigGetBool(configKey string) bool {
configValue := stim.config.Get(configKey)
configValue := stim.ConfigGetRaw(configKey)
if configValue != nil {
return configValue.(bool)
switch v := configValue.(type) {
case bool:
return v
case string:
lc := strings.ToLower(v)
return lc == "true" || lc == "t"
default:
stim.log.Warn("Unknown type for bool, type:{}, value of:{}", fmt.Sprintf("%T", v), v)
}

}
return false
}

func (stim *Stim) ConfigHasValue(configKey string) bool {
configValue := stim.config.Get(configKey)
if strings.Contains(configKey, ".") {
envCK := strings.ReplaceAll(configKey, ".", "-")
envCV := stim.config.Get(envCK)
if envCV != nil {
return true
}
}
if configValue != nil {
return true
}
Expand Down Expand Up @@ -141,6 +174,7 @@ func (stim *Stim) writeConfigData(config map[string]interface{}) error {
stim.log.Debug("Problem writing configfile:{}", err)
return err
}
stim.getConfigData()
return nil
}

Expand Down
87 changes: 87 additions & 0 deletions stim/config_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,87 @@
package stim

import (
"testing"

"github.com/PremiereGlobal/stim/pkg/stimlog"
"github.com/spf13/viper"
"gotest.tools/assert"
)

func check(e error) {
if e != nil {
panic(e)
}
}

func TestSimpleGetString(t *testing.T) {
stim := &Stim{
config: viper.New(),
}
stim.config.SetDefault("TEST-VALUE-ONE", "ONE")

simple := stim.ConfigGetString("test-value-one")
simple2 := stim.ConfigGetString("test.value-one")
none := stim.ConfigGetString("test.value-two")

assert.Equal(t, "ONE", simple, "Values not Equal")
assert.Equal(t, "ONE", simple2, "Values not Equal")
assert.Equal(t, "", none, "Values not Equal")
}

func TestSimpleGetRaw(t *testing.T) {
stim := &Stim{
config: viper.New(),
}
stim.config.SetDefault("TEST-VALUE-ONE", "ONE")

simple := stim.ConfigGetRaw("test-value-one")
simple2 := stim.ConfigGetRaw("test.value-one")
none := stim.ConfigGetRaw("test.value-two")

assert.Equal(t, "ONE", simple, "Values not Equal")
assert.Equal(t, "ONE", simple2, "Values not Equal")
assert.Equal(t, nil, none, "Values not Equal")
}

func TestSimpleGetBool(t *testing.T) {
stim := &Stim{
config: viper.New(),
log: stimlog.GetLogger(),
}
stim.config.SetDefault("TEST-VALUE-TRUE", true)
stim.config.SetDefault("TEST-VALUE-TRUE-STRING", "true")
stim.config.SetDefault("TEST-VALUE-TRUE-STRING2", "bad")
stim.config.SetDefault("TEST-VALUE-TRUE-INT", 100)

simple := stim.ConfigGetBool("test-value-true")
simple2 := stim.ConfigGetBool("test.value-true")
simple3 := stim.ConfigGetBool("test.value-true.string")
simple4 := stim.ConfigGetBool("test.value-true.string2")
simple5 := stim.ConfigGetBool("test.value-true.int")
none := stim.ConfigGetBool("test.value-false")

assert.Equal(t, true, simple, "Values not Equal")
assert.Equal(t, true, simple2, "Values not Equal")
assert.Equal(t, true, simple3, "Values not Equal")
assert.Equal(t, false, simple4, "Values not Equal")
assert.Equal(t, false, simple5, "Values not Equal")
assert.Equal(t, false, none, "Values not Equal")
}

func TestHasValue(t *testing.T) {
stim := &Stim{
config: viper.New(),
}
stim.config.SetDefault("TEST-VALUE-TRUE", true)

yes := stim.ConfigHasValue("test-value-true")
yes2 := stim.ConfigHasValue("test.value-true")
yes3 := stim.ConfigHasValue("test.value.true")
no := stim.ConfigHasValue("test-value-false")

assert.Equal(t, true, yes, "Values not Equal")
assert.Equal(t, true, yes2, "Values not Equal")
assert.Equal(t, true, yes3, "Values not Equal")
assert.Equal(t, false, no, "Values not Equal")
}