87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package configuration
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/authelia/authelia/v4/internal/configuration/schema"
|
|
"github.com/authelia/authelia/v4/internal/utils"
|
|
)
|
|
|
|
// koanfEnvironmentCallback returns a koanf callback to map the environment vars to Configuration keys.
|
|
func koanfEnvironmentCallback(keyMap map[string]string, ignoredKeys []string, prefix, delimiter string) func(key, value string) (finalKey string, finalValue any) {
|
|
return func(key, value string) (finalKey string, finalValue any) {
|
|
if k, ok := keyMap[key]; ok {
|
|
return k, value
|
|
}
|
|
|
|
if utils.IsStringInSlice(key, ignoredKeys) {
|
|
return "", nil
|
|
}
|
|
|
|
formattedKey := strings.TrimPrefix(key, prefix)
|
|
formattedKey = strings.ReplaceAll(strings.ToLower(formattedKey), delimiter, constDelimiter)
|
|
|
|
if utils.IsStringInSlice(formattedKey, schema.Keys) {
|
|
return formattedKey, value
|
|
}
|
|
|
|
return key, value
|
|
}
|
|
}
|
|
|
|
// koanfEnvironmentSecretsCallback returns a koanf callback to map the environment vars to Configuration keys.
|
|
func koanfEnvironmentSecretsCallback(keyMap map[string]string, validator *schema.StructValidator) func(key, value string) (finalKey string, finalValue any) {
|
|
return func(key, value string) (finalKey string, finalValue any) {
|
|
k, ok := keyMap[key]
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
|
|
switch v, err := loadSecret(value); err {
|
|
case nil:
|
|
return k, v
|
|
default:
|
|
switch {
|
|
case os.IsNotExist(err):
|
|
validator.Push(fmt.Errorf(errFmtSecretOSNotExist, value, k, err))
|
|
|
|
return "", nil
|
|
case os.IsPermission(err):
|
|
validator.Push(fmt.Errorf(errFmtSecretOSPermission, value, k, err))
|
|
|
|
return "", nil
|
|
default:
|
|
validator.Push(fmt.Errorf(errFmtSecretOSError, value, k, err))
|
|
|
|
return "", nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func koanfCommandLineWithMappingCallback(mapping map[string]string, includeValidKeys, includeUnchangedKeys bool) func(flag *pflag.Flag) (string, any) {
|
|
return func(flag *pflag.Flag) (string, any) {
|
|
if !includeUnchangedKeys && !flag.Changed {
|
|
return "", nil
|
|
}
|
|
|
|
if actualKey, ok := mapping[flag.Name]; ok {
|
|
return actualKey, flag.Value.String()
|
|
}
|
|
|
|
if includeValidKeys {
|
|
formattedKey := strings.ReplaceAll(flag.Name, "-", "_")
|
|
|
|
if utils.IsStringInSlice(formattedKey, schema.Keys) {
|
|
return formattedKey, flag.Value.String()
|
|
}
|
|
}
|
|
|
|
return "", nil
|
|
}
|
|
}
|