nurse/config/env.go
2022-05-13 16:46:31 +02:00

61 lines
1.1 KiB
Go

package config
import (
"os"
"path"
. "strings"
)
const (
ServerKeyPrefix = "NURSE_SERVER"
EndpointKeyPrefix = "NURSE_ENDPOINT"
)
func ServersFromEnv() (map[string]Server, error) {
servers := make(map[string]Server)
for _, kv := range os.Environ() {
key, value, valid := Cut(kv, "=")
if !valid {
continue
}
if !HasPrefix(key, ServerKeyPrefix) {
continue
}
serverName := ToLower(Trim(Replace(key, ServerKeyPrefix, "", -1), "_"))
srv := Server{}
if err := srv.UnmarshalURL(value); err != nil {
return nil, err
}
servers[serverName] = srv
}
return servers, nil
}
func EndpointsFromEnv() (map[Route]EndpointSpec, error) {
endpoints := make(map[Route]EndpointSpec)
for _, kv := range os.Environ() {
key, value, valid := Cut(kv, "=")
if !valid {
continue
}
if !HasPrefix(key, EndpointKeyPrefix) {
continue
}
endpointRoute := path.Join(Split(ToLower(Trim(Replace(key, EndpointKeyPrefix, "", -1), "_")), "_")...)
spec := EndpointSpec{}
if err := spec.Parse(value); err != nil {
return nil, err
}
endpoints[Route(endpointRoute)] = spec
}
return endpoints, nil
}