62 lines
1.1 KiB
Go
62 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.UnmarshalText([]byte(value)); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
endpoints[Route(endpointRoute)] = spec
|
||
|
|
}
|
||
|
|
|
||
|
|
return endpoints, nil
|
||
|
|
}
|