mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
2.9 KiB
118 lines
2.9 KiB
/*
|
|
Copyright The Helm Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package values
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
"sigs.k8s.io/yaml"
|
|
|
|
"helm.sh/helm/pkg/cli"
|
|
"helm.sh/helm/pkg/getter"
|
|
"helm.sh/helm/pkg/strvals"
|
|
)
|
|
|
|
type Options struct {
|
|
ValueFiles []string
|
|
StringValues []string
|
|
Values []string
|
|
}
|
|
|
|
// MergeValues merges values from files specified via -f/--values and
|
|
// directly via --set or --set-string, marshaling them to YAML
|
|
func (opts *Options) MergeValues(settings cli.EnvSettings) (map[string]interface{}, error) {
|
|
base := map[string]interface{}{}
|
|
|
|
// User specified a values files via -f/--values
|
|
for _, filePath := range opts.ValueFiles {
|
|
currentMap := map[string]interface{}{}
|
|
|
|
bytes, err := readFile(filePath, settings)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := yaml.Unmarshal(bytes, ¤tMap); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to parse %s", filePath)
|
|
}
|
|
// Merge with the previous map
|
|
base = mergeMaps(base, currentMap)
|
|
}
|
|
|
|
// User specified a value via --set
|
|
for _, value := range opts.Values {
|
|
if err := strvals.ParseInto(value, base); err != nil {
|
|
return nil, errors.Wrap(err, "failed parsing --set data")
|
|
}
|
|
}
|
|
|
|
// User specified a value via --set-string
|
|
for _, value := range opts.StringValues {
|
|
if err := strvals.ParseIntoString(value, base); err != nil {
|
|
return nil, errors.Wrap(err, "failed parsing --set-string data")
|
|
}
|
|
}
|
|
|
|
return base, nil
|
|
}
|
|
|
|
func mergeMaps(a, b map[string]interface{}) map[string]interface{} {
|
|
out := make(map[string]interface{}, len(a))
|
|
for k, v := range a {
|
|
out[k] = v
|
|
}
|
|
for k, v := range b {
|
|
if v, ok := v.(map[string]interface{}); ok {
|
|
if bv, ok := out[k]; ok {
|
|
if bv, ok := bv.(map[string]interface{}); ok {
|
|
out[k] = mergeMaps(bv, v)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
out[k] = v
|
|
}
|
|
return out
|
|
}
|
|
|
|
// readFile load a file from stdin, the local directory, or a remote file with a url.
|
|
func readFile(filePath string, settings cli.EnvSettings) ([]byte, error) {
|
|
if strings.TrimSpace(filePath) == "-" {
|
|
return ioutil.ReadAll(os.Stdin)
|
|
}
|
|
u, _ := url.Parse(filePath)
|
|
p := getter.All(settings)
|
|
|
|
// FIXME: maybe someone handle other protocols like ftp.
|
|
getterConstructor, err := p.ByScheme(u.Scheme)
|
|
|
|
if err != nil {
|
|
return ioutil.ReadFile(filePath)
|
|
}
|
|
|
|
getter, err := getterConstructor(getter.WithURL(filePath))
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
data, err := getter.Get(filePath)
|
|
return data.Bytes(), err
|
|
}
|