-
Notifications
You must be signed in to change notification settings - Fork 89
/
kustomization.go
75 lines (61 loc) · 2.04 KB
/
kustomization.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
package k8sutil
import (
"io/ioutil"
"os"
"sort"
"strings"
"github.com/pkg/errors"
kustomizetypes "sigs.k8s.io/kustomize/api/types"
"sigs.k8s.io/yaml"
)
func ReadKustomizationFromFile(file string) (*kustomizetypes.Kustomization, error) {
b, err := os.ReadFile(file)
if err != nil {
return nil, errors.Wrap(err, "failed to read kustomization file")
}
k := kustomizetypes.Kustomization{}
if err := yaml.Unmarshal(b, &k); err != nil {
return nil, errors.Wrap(err, "failed to unmarshal kustomization")
}
return &k, nil
}
// implementing Len Swap and Less allows sorting the type directly
type kustPatches []kustomizetypes.PatchStrategicMerge
func (s kustPatches) Len() int {
return len(s)
}
func (s kustPatches) Swap(i, j int) {
s[i], s[j] = s[j], s[i]
}
func (s kustPatches) Less(i, j int) bool {
return strings.Compare(string(s[i]), string(s[j])) < 0
}
func WriteKustomizationToFile(kustomization kustomizetypes.Kustomization, file string) error {
cleanedImages := []kustomizetypes.Image{}
// Remove tags and digests and deduplicate image list.
// Tags are removed because we don't want kustomize to change tags, only image names.
// Digests are removed so that more than one digest of the same image can be used (this applies to Tags too).
// When Tags and Digests are not set, kustomize will only rewrite the image name and keep the original tag or digest.
imageDedup := map[string]bool{}
for _, image := range kustomization.Images {
if _, ok := imageDedup[image.Name]; ok {
continue
}
image.NewTag = ""
image.Digest = ""
cleanedImages = append(cleanedImages, image)
imageDedup[image.Name] = true
}
kustomization.Images = cleanedImages
sort.Strings(kustomization.Bases)
sort.Strings(kustomization.Resources)
sort.Sort(kustPatches(kustomization.PatchesStrategicMerge))
b, err := yaml.Marshal(kustomization)
if err != nil {
return errors.Wrap(err, "failed to marshal kustomization")
}
if err := ioutil.WriteFile(file, b, 0644); err != nil {
return errors.Wrap(err, "failed to write kustomization file")
}
return nil
}