add bool support

Author: syntaqx

env.go

@@ -31,6 +31,20 @@ func GetWithFallback(key string, fallback string) string {
 	return fallback
 }
 
+// GetBool returns the value of an environment variable as a boolean.
+func GetBool(key string) bool {
+	return Get(key) == "true"
+}
+
+// GetBoolWithFallback returns the value of an environment variable as a boolean
+// or a fallback value if the environment variable is not set.
+func GetBoolWithFallback(key string, fallback bool) bool {
+	if value, ok := Lookup(key); ok {
+		return value == "true"
+	}
+	return fallback
+}
+
 // Load will read your env file(s) and load them into ENV for this process.
 //
 // Call this function as close as possible to the start of your program

env_test.go

@@ -56,3 +56,50 @@ func TestGetWithFallback(t *testing.T) {
 		})
 	}
 }
+func TestGetBoolWithFallback(t *testing.T) {
+	tests := []struct {
+		setValue  bool
+		key       string
+		fallback  bool
+		envValue  string
+		wantValue bool
+	}{
+		{
+			setValue:  true,
+			key:       "EXISTING_KEY",
+			fallback:  true,
+			envValue:  "true",
+			wantValue: true,
+		},
+		{
+			setValue:  false,
+			key:       "NON_EXISTING_KEY",
+			fallback:  true,
+			envValue:  "",
+			wantValue: true,
+		},
+		{
+			setValue:  true,
+			key:       "EMPTY_VALUE_KEY",
+			fallback:  false,
+			envValue:  "",
+			wantValue: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.key, func(t *testing.T) {
+			// Set the environment variable
+			if tt.setValue {
+				os.Setenv(tt.key, tt.envValue)
+			}
+			// Call the function under test
+			got := GetBoolWithFallback(tt.key, tt.fallback)
+			// Check if the returned value matches the expected value
+			if got != tt.wantValue {
+				t.Errorf("GetBoolWithFallback(%q, %v) = %v, want %v", tt.key, tt.fallback, got, tt.wantValue)
+			}
+			// Unset the environment variable
+			os.Unsetenv(tt.key)
+		})
+	}
+}