mirror of
https://github.com/netbirdio/netbird.git
synced 2026-10-03 03:59:07 +02:00
[management,misc] Fix configuration migration checks
The initial migration let an empty legacy Datadir override Management defaults and pulled Viper into WASM builds through the common util package. Normalize the Management data directory, isolate template expansion from the loader dependency, and split source binding to satisfy the quality gate.
This commit is contained in:
+99
-54
@@ -28,8 +28,7 @@
|
||||
// })
|
||||
//
|
||||
// Set [Options.AllowMissing] when the service must start without a configuration
|
||||
// file. [Options.Transform] can preprocess file contents before decoding, such
|
||||
// as with [ExpandEnvTemplate].
|
||||
// file. [Options.Transform] can preprocess file contents before decoding.
|
||||
package config
|
||||
|
||||
import (
|
||||
@@ -215,66 +214,112 @@ func bindConfigSources(
|
||||
defer delete(visiting, configType)
|
||||
|
||||
for i := range configType.NumField() {
|
||||
field := configType.Field(i)
|
||||
if !field.IsExported() {
|
||||
continue
|
||||
if err := bindConfigField(
|
||||
v,
|
||||
configType.Field(i),
|
||||
prefix,
|
||||
tagName,
|
||||
flagSet,
|
||||
bindEnvironment,
|
||||
bindFlags,
|
||||
visiting,
|
||||
); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
key, inline, skip := configFieldKey(field, tagName)
|
||||
if skip {
|
||||
continue
|
||||
}
|
||||
if inline {
|
||||
key = prefix
|
||||
} else if prefix != "" {
|
||||
key = prefix + "." + key
|
||||
}
|
||||
func bindConfigField(
|
||||
v *viper.Viper,
|
||||
field reflect.StructField,
|
||||
prefix string,
|
||||
tagName string,
|
||||
flagSet *pflag.FlagSet,
|
||||
bindEnvironment bool,
|
||||
bindFlags bool,
|
||||
visiting map[reflect.Type]bool,
|
||||
) error {
|
||||
if !field.IsExported() {
|
||||
return nil
|
||||
}
|
||||
|
||||
fieldEnvironment := field.Tag.Get("env")
|
||||
fieldFlags := field.Tag.Get("flag")
|
||||
bindFieldEnvironment := bindEnvironment && fieldEnvironment != "-"
|
||||
bindFieldFlags := bindFlags && fieldFlags != "-"
|
||||
key, inline, skip := configFieldKey(field, tagName)
|
||||
if skip {
|
||||
return nil
|
||||
}
|
||||
if inline {
|
||||
key = prefix
|
||||
} else if prefix != "" {
|
||||
key = prefix + "." + key
|
||||
}
|
||||
|
||||
fieldType := field.Type
|
||||
for fieldType.Kind() == reflect.Pointer {
|
||||
fieldType = fieldType.Elem()
|
||||
}
|
||||
if fieldType.Kind() == reflect.Struct && !isScalarUnmarshaler(fieldType) {
|
||||
if !visiting[fieldType] {
|
||||
if err := bindConfigSources(
|
||||
v,
|
||||
fieldType,
|
||||
key,
|
||||
tagName,
|
||||
flagSet,
|
||||
bindFieldEnvironment,
|
||||
bindFieldFlags,
|
||||
visiting,
|
||||
); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
fieldEnvironment := field.Tag.Get("env")
|
||||
fieldFlags := field.Tag.Get("flag")
|
||||
bindEnvironment = bindEnvironment && fieldEnvironment != "-"
|
||||
bindFlags = bindFlags && fieldFlags != "-"
|
||||
|
||||
if key == "" {
|
||||
return fmt.Errorf("empty config key for field %s", field.Name)
|
||||
fieldType := field.Type
|
||||
for fieldType.Kind() == reflect.Pointer {
|
||||
fieldType = fieldType.Elem()
|
||||
}
|
||||
if fieldType.Kind() == reflect.Struct && !isScalarUnmarshaler(fieldType) {
|
||||
if visiting[fieldType] {
|
||||
return nil
|
||||
}
|
||||
if bindFieldEnvironment {
|
||||
if err := bindEnvironmentVariable(v, key, fieldEnvironment); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if bindFieldFlags && flagSet != nil && fieldFlags != "" {
|
||||
flagName, flag, err := selectFlag(flagSet, fieldFlags)
|
||||
if err != nil {
|
||||
return fmt.Errorf("config field %s: %w", field.Name, err)
|
||||
}
|
||||
if err := v.BindPFlag(key, flag); err != nil {
|
||||
return fmt.Errorf("bind flag %s: %w", flagName, err)
|
||||
}
|
||||
return bindConfigSources(
|
||||
v,
|
||||
fieldType,
|
||||
key,
|
||||
tagName,
|
||||
flagSet,
|
||||
bindEnvironment,
|
||||
bindFlags,
|
||||
visiting,
|
||||
)
|
||||
}
|
||||
|
||||
return bindScalarSources(
|
||||
v,
|
||||
field,
|
||||
key,
|
||||
fieldEnvironment,
|
||||
fieldFlags,
|
||||
flagSet,
|
||||
bindEnvironment,
|
||||
bindFlags,
|
||||
)
|
||||
}
|
||||
|
||||
func bindScalarSources(
|
||||
v *viper.Viper,
|
||||
field reflect.StructField,
|
||||
key string,
|
||||
environmentName string,
|
||||
flagNames string,
|
||||
flagSet *pflag.FlagSet,
|
||||
bindEnvironment bool,
|
||||
bindFlags bool,
|
||||
) error {
|
||||
if key == "" {
|
||||
return fmt.Errorf("empty config key for field %s", field.Name)
|
||||
}
|
||||
if bindEnvironment {
|
||||
if err := bindEnvironmentVariable(v, key, environmentName); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if !bindFlags || flagSet == nil || flagNames == "" {
|
||||
return nil
|
||||
}
|
||||
|
||||
flagName, flag, err := selectFlag(flagSet, flagNames)
|
||||
if err != nil {
|
||||
return fmt.Errorf("config field %s: %w", field.Name, err)
|
||||
}
|
||||
if err := v.BindPFlag(key, flag); err != nil {
|
||||
return fmt.Errorf("bind flag %s: %w", flagName, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -11,6 +11,8 @@ import (
|
||||
"github.com/spf13/pflag"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/netbirdio/netbird/util/envtemplate"
|
||||
)
|
||||
|
||||
type testConfig struct {
|
||||
@@ -138,7 +140,7 @@ server:
|
||||
|
||||
cfg, err := Load(configPath, defaultTestConfig(), Options{
|
||||
TagName: "yaml",
|
||||
Transform: ExpandEnvTemplate,
|
||||
Transform: envtemplate.Expand,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, ":8443", cfg.Server.Address, "The transform should run before decoding")
|
||||
|
||||
@@ -1,34 +0,0 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
)
|
||||
|
||||
// ExpandEnvTemplate substitutes Go-template references with environment values.
|
||||
func ExpandEnvTemplate(data []byte) ([]byte, error) {
|
||||
tmpl, err := template.New("config").Parse(string(data))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("parse environment template: %w", err)
|
||||
}
|
||||
|
||||
var output bytes.Buffer
|
||||
if err := tmpl.Execute(&output, environmentMap()); err != nil {
|
||||
return nil, fmt.Errorf("execute environment template: %w", err)
|
||||
}
|
||||
return output.Bytes(), nil
|
||||
}
|
||||
|
||||
func environmentMap() map[string]string {
|
||||
environment := make(map[string]string)
|
||||
for _, entry := range os.Environ() {
|
||||
key, value, ok := strings.Cut(entry, "=")
|
||||
if ok {
|
||||
environment[key] = value
|
||||
}
|
||||
}
|
||||
return environment
|
||||
}
|
||||
Reference in New Issue
Block a user