109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package environment
|
|
|
|
import (
|
|
"dario.cat/mergo"
|
|
|
|
"github.com/helmfile/helmfile/pkg/maputil"
|
|
"github.com/helmfile/helmfile/pkg/yaml"
|
|
)
|
|
|
|
type Environment struct {
|
|
Name string
|
|
KubeContext string
|
|
Values map[string]any
|
|
Defaults map[string]any
|
|
// IsCLIOverride indicates if this environment contains CLI overrides from --state-values-set
|
|
// When true, arrays are merged element-by-element. When false, arrays are replaced.
|
|
IsCLIOverride bool
|
|
}
|
|
|
|
var EmptyEnvironment Environment
|
|
|
|
// New return Environment with default name and values
|
|
func New(name string) *Environment {
|
|
return &Environment{
|
|
Name: name,
|
|
KubeContext: "",
|
|
Values: map[string]any{},
|
|
Defaults: map[string]any{},
|
|
}
|
|
}
|
|
|
|
func (e Environment) DeepCopy() Environment {
|
|
valuesBytes, err := yaml.Marshal(e.Values)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var values map[string]any
|
|
if err := yaml.Unmarshal(valuesBytes, &values); err != nil {
|
|
panic(err)
|
|
}
|
|
values, err = maputil.CastKeysToStrings(values)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
defaultsBytes, err := yaml.Marshal(e.Defaults)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var defaults map[string]any
|
|
if err := yaml.Unmarshal(defaultsBytes, &defaults); err != nil {
|
|
panic(err)
|
|
}
|
|
defaults, err = maputil.CastKeysToStrings(defaults)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return Environment{
|
|
Name: e.Name,
|
|
KubeContext: e.KubeContext,
|
|
Values: values,
|
|
Defaults: defaults,
|
|
IsCLIOverride: e.IsCLIOverride,
|
|
}
|
|
}
|
|
|
|
func (e *Environment) Merge(other *Environment) (*Environment, error) {
|
|
if e == nil {
|
|
if other != nil {
|
|
copy := other.DeepCopy()
|
|
return ©, nil
|
|
}
|
|
return nil, nil
|
|
}
|
|
copy := e.DeepCopy()
|
|
if other != nil {
|
|
if err := mergo.Merge(©, other, mergo.WithOverride); err != nil {
|
|
return nil, err
|
|
}
|
|
// Preserve the IsCLIOverride flag from other
|
|
if other.IsCLIOverride {
|
|
copy.IsCLIOverride = true
|
|
}
|
|
}
|
|
return ©, nil
|
|
}
|
|
|
|
func (e *Environment) GetMergedValues() (map[string]any, error) {
|
|
vals := map[string]any{}
|
|
|
|
// For CLI overrides (--state-values-set), use MergeMapsWithArrayMerge to handle array indices
|
|
// For helmfile composition, use regular MergeMaps which replaces arrays (documented behavior)
|
|
if e.IsCLIOverride {
|
|
vals = maputil.MergeMapsWithArrayMerge(vals, e.Defaults)
|
|
vals = maputil.MergeMapsWithArrayMerge(vals, e.Values)
|
|
} else {
|
|
vals = maputil.MergeMaps(vals, e.Defaults)
|
|
vals = maputil.MergeMaps(vals, e.Values)
|
|
}
|
|
|
|
vals, err := maputil.CastKeysToStrings(vals)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return vals, nil
|
|
}
|