Added support for bool

Author: Attumm

README.md

@@ -16,7 +16,7 @@ Getting vars out has the same level of complexity as setting the value.
 ## Features
 - Simple to use
 - supports command line and environment variables 
-- Support for types: str, int, ~~bool~~, ~~map~~
+- Support for types: str, int, bool, ~~map~~
 - Singleton, makes it easy to use in program anywhere in the code-base
 - Supports help text with --help on the binary
 - Ease of use in any environment examples: linux, docker, k8

settingo/main.go

@@ -6,10 +6,20 @@ import (
 	"strconv"
 )
 
+func truthiness(s string) bool {
+	thruthy := map[string]bool{
+		"y":    true,
+		"true": true,
+		"yes":  true,
+	}
+	return thruthy[s]
+}
+
 type Settings struct {
 	msg        map[string]string
 	VarString  map[string]string
 	VarInt     map[string]int
+	VarBool    map[string]bool
 	Parsers    map[string]func(string) string
 	ParsersInt map[string]func(int) int
 }
@@ -28,6 +38,11 @@ func (s *Settings) SetInt(flagName string, defaultVar int, message string) {
 	s.VarInt[flagName] = defaultVar
 }
 
+func (s *Settings) SetBool(flagName string, defaultVar bool, message string) {
+	s.msg[flagName] = message
+	s.VarBool[flagName] = defaultVar
+}
+
 func (s *Settings) SetParsed(flagName, defaultVar, message string, parserFunc func(string) string) {
 	s.msg[flagName] = message
 	s.VarString[flagName] = defaultVar
@@ -48,6 +63,10 @@ func (s Settings) GetInt(flagName string) int {
 	return s.VarInt[flagName]
 }
 
+func (s Settings) GetBool(flagName string) bool {
+	return s.VarBool[flagName]
+}
+
 func (s *Settings) HandleCMDLineInput() {
 	parsedString := make(map[string]*string)
 	for key, val := range s.VarString {
@@ -59,6 +78,11 @@ func (s *Settings) HandleCMDLineInput() {
 		var newV = flag.Int(key, val, s.msg[key])
 		parsedInt[key] = newV
 	}
+	parsedBool := make(map[string]*string)
+	for key, val := range s.VarBool {
+		var newV = flag.String(key, strconv.FormatBool(val), s.msg[key])
+		parsedBool[key] = newV
+	}
 	flag.Parse()
 
 	for key, val := range parsedString {
@@ -75,6 +99,9 @@ func (s *Settings) HandleCMDLineInput() {
 			s.VarInt[key] = *val
 		}
 	}
+	for key, val := range parsedBool {
+		s.VarBool[key] = truthiness(*val)
+	}
 }
 
 func (s *Settings) HandleOSInput() {
@@ -92,6 +119,12 @@ func (s *Settings) HandleOSInput() {
 			}
 		}
 	}
+	for key, _ := range s.VarBool {
+		varEnv, found := os.LookupEnv(key)
+		if found {
+			s.VarBool[key] = truthiness(varEnv)
+		}
+	}
 }
 
 func (s *Settings) Parse() {
@@ -105,4 +138,5 @@ var SETTINGS = Settings{
 	VarInt:     make(map[string]int),
 	Parsers:    make(map[string]func(string) string),
 	ParsersInt: make(map[string]func(int) int),
+	VarBool:    make(map[string]bool),
 }

settingo/main_test.go

@@ -0,0 +1,69 @@
+package settingo
+
+import (
+	"os"
+	"testing"
+)
+
+func Test_types_default(t *testing.T) {
+	expected := "default_value_for_foobar"
+	expectedInt := 42
+	expectedBool := true
+
+	SETTINGS.Set("FOOBAR", expected, "help text")
+	SETTINGS.SetInt("FOOBAR_INT", expectedInt, "help text")
+	SETTINGS.SetBool("FOOBAR_BOOL", expectedBool, "help text")
+
+	//SETTINGS.Parse()
+
+	foobar := SETTINGS.Get("FOOBAR")
+	if foobar != expected {
+		t.Error(foobar, " != ", expected)
+	}
+
+	foobarInt := SETTINGS.GetInt("FOOBAR_INT")
+	if foobarInt != expectedInt {
+		t.Error(foobarInt, " != ", expectedInt)
+	}
+
+	foobarBool := SETTINGS.GetBool("FOOBAR_BOOL")
+	if foobarBool != expectedBool {
+		t.Error(foobarBool, " != ", expectedBool)
+	}
+}
+
+func Test_types_os_env(t *testing.T) {
+
+	expected := "other value"
+	os.Setenv("FOOBAR", expected)
+	defaultStr := "default value"
+
+	expectedInt := 44
+	os.Setenv("FOOBAR_INT", "44")
+	defaultInt := 42
+
+	os.Setenv("FOOBAR_BOOL", "y")
+	expectedBool := true
+	defaultBool := false
+
+	SETTINGS.Set("FOOBAR", defaultStr, "help text")
+	SETTINGS.SetInt("FOOBAR_INT", defaultInt, "help text")
+	SETTINGS.SetBool("FOOBAR_BOOL", defaultBool, "help text")
+
+	SETTINGS.Parse()
+
+	foobar := SETTINGS.Get("FOOBAR")
+	if foobar != expected {
+		t.Error(foobar, " != ", expected)
+	}
+
+	foobarInt := SETTINGS.GetInt("FOOBAR_INT")
+	if foobarInt != expectedInt {
+		t.Error(foobarInt, " != ", expectedInt)
+	}
+
+	foobarBool := SETTINGS.GetBool("FOOBAR_BOOL")
+	if foobarBool != expectedBool {
+		t.Error(foobarBool, " != ", expectedBool)
+	}
+}