diff --git a/env/boolean.go b/env/boolean.go new file mode 100644 index 0000000..1ceb5ef --- /dev/null +++ b/env/boolean.go @@ -0,0 +1,35 @@ +package env + +import ( + "os" + "strings" +) + +// Get an environment variable as a boolean, specifying a default value if its +// not set or can't be parsed properly into a bool +func Bool(key string, defaultValue bool) bool { + val, exists := os.LookupEnv(key) + if !exists { + return defaultValue + } + truthy := map[string]bool{ + "true": true, "t": true, "yes": true, "y": true, "on": true, "1": true, + "enable": true, "enabled": true, "active": true, "affirmative": true, + } + + falsy := map[string]bool{ + "false": false, "f": false, "no": false, "n": false, "off": false, "0": false, + "disable": false, "disabled": false, "inactive": false, "negative": false, + } + + normalized := strings.TrimSpace(strings.ToLower(val)) + + if val, ok := truthy[normalized]; ok { + return val + } + if val, ok := falsy[normalized]; ok { + return val + } + + return defaultValue +} diff --git a/env/duration.go b/env/duration.go new file mode 100644 index 0000000..92b5bfe --- /dev/null +++ b/env/duration.go @@ -0,0 +1,23 @@ +package env + +import ( + "os" + "strconv" + "time" +) + +// Get an environment variable as a time.Duration, specifying a default value if its +// not set or can't be parsed properly +func Duration(key string, defaultValue time.Duration) time.Duration { + val, exists := os.LookupEnv(key) + if !exists { + return time.Duration(defaultValue) + } + + intVal, err := strconv.Atoi(val) + if err != nil { + return time.Duration(defaultValue) + } + return time.Duration(intVal) + +} diff --git a/env/go.mod b/env/go.mod new file mode 100644 index 0000000..ea2e0e7 --- /dev/null +++ b/env/go.mod @@ -0,0 +1,3 @@ +module git.haelnorr.com/h/golib/env + +go 1.25.5 diff --git a/env/int.go b/env/int.go new file mode 100644 index 0000000..18a3a3b --- /dev/null +++ b/env/int.go @@ -0,0 +1,37 @@ +package env + +import ( + "os" + "strconv" +) + +// Get an environment variable as an int, specifying a default value if its +// not set or can't be parsed properly into an int +func Int(key string, defaultValue int) int { + val, exists := os.LookupEnv(key) + if !exists { + return defaultValue + } + + intVal, err := strconv.Atoi(val) + if err != nil { + return defaultValue + } + return intVal +} + +// Get an environment variable as an int64, specifying a default value if its +// not set or can't be parsed properly into an int64 +func Int64(key string, defaultValue int64) int64 { + val, exists := os.LookupEnv(key) + if !exists { + return defaultValue + } + + intVal, err := strconv.ParseInt(val, 10, 64) + if err != nil { + return defaultValue + } + return intVal + +} diff --git a/env/string.go b/env/string.go new file mode 100644 index 0000000..ef04e11 --- /dev/null +++ b/env/string.go @@ -0,0 +1,14 @@ +package env + +import ( + "os" +) + +// Get an environment variable, specifying a default value if its not set +func String(key string, defaultValue string) string { + val, exists := os.LookupEnv(key) + if !exists { + return defaultValue + } + return val +}