-
Notifications
You must be signed in to change notification settings - Fork 90
/
yaml.go
87 lines (70 loc) · 2.29 KB
/
yaml.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package kotsutil
import (
"bytes"
"github.com/pkg/errors"
"github.com/replicatedhq/kots/pkg/util"
yaml "github.com/replicatedhq/yaml/v3"
k8syaml "sigs.k8s.io/yaml"
)
// FixUpYAML is a general purpose function that will ensure that YAML is compatible with KOTS
// This ensures that lines aren't wrapped at 80 chars which breaks template functions
func FixUpYAML(inputContent []byte) ([]byte, error) {
docs := util.ConvertToSingleDocs(inputContent)
fixedUpDocs := make([][]byte, 0)
for _, doc := range docs {
yamlObj := map[string]interface{}{}
err := yaml.Unmarshal(doc, &yamlObj)
if err != nil {
return nil, errors.Wrap(err, "failed to unmarshal yaml")
}
fixedUpDoc, err := util.MarshalIndent(2, yamlObj)
if err != nil {
return nil, errors.Wrap(err, "failed to marshal yaml")
}
fixedUpDocs = append(fixedUpDocs, fixedUpDoc)
}
// MarshalIndent add a line break at the end of each file
return bytes.Join(fixedUpDocs, []byte("---\n")), nil
}
// RemoveNilFieldsFromYAML removes nil fields from a yaml document.
// This is necessary because kustomize will fail to apply a kustomization if these fields contain nil values: https://github.com/kubernetes-sigs/kustomize/issues/5050
func RemoveNilFieldsFromYAML(input []byte) ([]byte, error) {
var data map[string]interface{}
err := k8syaml.Unmarshal([]byte(input), &data)
if err != nil {
return nil, errors.Wrap(err, "failed to unmarshal yaml")
}
removedItems := removeNilFieldsFromMap(data)
if !removedItems {
// no changes were made, return the original input
return input, nil
}
output, err := k8syaml.Marshal(data)
if err != nil {
return nil, errors.Wrap(err, "failed to marshal yaml")
}
return output, nil
}
func removeNilFieldsFromMap(input map[string]interface{}) bool {
removedItems := false
for key, value := range input {
if value == nil {
delete(input, key)
removedItems = true
continue
}
if valueMap, ok := value.(map[string]interface{}); ok {
removedItems = removeNilFieldsFromMap(valueMap) || removedItems
continue
}
if valueSlice, ok := value.([]interface{}); ok {
for idx := range valueSlice {
if itemMap, ok := valueSlice[idx].(map[string]interface{}); ok {
removedItems = removeNilFieldsFromMap(itemMap) || removedItems
}
}
continue
}
}
return removedItems
}