cd93ad8081
Module path changed from forge.lthn.ai/core/go-config to forge.lthn.ai/core/config. Package name remains "config" — only the module path changed. Co-Authored-By: Virgil <virgil@lethean.io>
52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
"iter"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// Env returns an iterator over environment variables with the given prefix,
|
|
// providing them as dot-notation keys and values.
|
|
//
|
|
// For example, with prefix "CORE_CONFIG_":
|
|
//
|
|
// CORE_CONFIG_FOO_BAR=baz -> yields ("foo.bar", "baz")
|
|
func Env(prefix string) iter.Seq2[string, any] {
|
|
return func(yield func(string, any) bool) {
|
|
for _, env := range os.Environ() {
|
|
if !strings.HasPrefix(env, prefix) {
|
|
continue
|
|
}
|
|
|
|
parts := strings.SplitN(env, "=", 2)
|
|
if len(parts) != 2 {
|
|
continue
|
|
}
|
|
|
|
name := parts[0]
|
|
value := parts[1]
|
|
|
|
// Strip prefix and convert to dot notation
|
|
key := strings.TrimPrefix(name, prefix)
|
|
key = strings.ToLower(key)
|
|
key = strings.ReplaceAll(key, "_", ".")
|
|
|
|
if !yield(key, value) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// LoadEnv parses environment variables with the given prefix and returns
|
|
// them as a flat map with dot-notation keys.
|
|
//
|
|
// Deprecated: Use Env for iterative access or collect into a map manually.
|
|
func LoadEnv(prefix string) map[string]any {
|
|
result := make(map[string]any)
|
|
for k, v := range Env(prefix) {
|
|
result[k] = v
|
|
}
|
|
return result
|
|
}
|