-
Notifications
You must be signed in to change notification settings - Fork 474
/
utils.go
329 lines (293 loc) · 11.4 KB
/
utils.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package webhook
import (
"reflect"
"regexp"
"slices"
"strings"
"github.com/coreos/go-systemd/v22/unit"
"github.com/go-logr/logr"
corev1 "k8s.io/api/core/v1"
vpaautoscalingv1 "k8s.io/autoscaler/vertical-pod-autoscaler/pkg/apis/autoscaling.k8s.io/v1"
extensionsv1alpha1 "github.com/gardener/gardener/pkg/apis/extensions/v1alpha1"
)
// LogMutation provides a log message.
func LogMutation(logger logr.Logger, kind, namespace, name string) {
logger.Info("Mutating resource", "kind", kind, "namespace", namespace, "name", name)
}
// AppendUniqueUnit appens a unit only if it does not exist.
func AppendUniqueUnit(units *[]extensionsv1alpha1.Unit, unit extensionsv1alpha1.Unit) {
for _, un := range *units {
if un.Name == unit.Name {
return
}
}
*units = append(*units, unit)
}
// splitCommandLineRegex is used to split command line arguments by white space or "\".
var splitCommandLineRegex = regexp.MustCompile(`[\\\s]+`)
// DeserializeCommandLine de-serializes the given string to a slice of command line elements by splitting it
// on white space and the "\" character.
func DeserializeCommandLine(s string) []string {
return splitCommandLineRegex.Split(s, -1)
}
// SerializeCommandLine serializes the given command line elements slice to a string by joining the first
// n+1 elements with a space " ", and all subsequent elements with the given separator.
func SerializeCommandLine(command []string, n int, sep string) string {
if len(command) <= n {
return strings.Join(command, " ")
}
if n == 0 {
return strings.Join(command, sep)
}
return strings.Join(command[0:n], " ") + " " + strings.Join(command[n:], sep)
}
// ContainerWithName returns the first container with the specified name from the slice, or nil if not found.
func ContainerWithName(containers []corev1.Container, name string) *corev1.Container {
for i, container := range containers {
if container.Name == name {
return &containers[i]
}
}
return nil
}
// PVCWithName returns the first PersistentVolumeClaim with the specified name from the slice, or nil if not found.
func PVCWithName(pvcs []corev1.PersistentVolumeClaim, name string) *corev1.PersistentVolumeClaim {
for i, pvc := range pvcs {
if pvc.Name == name {
return &pvcs[i]
}
}
return nil
}
// UnitWithName returns the first unit with the specified name from the slice, or nil if not found.
func UnitWithName(units []extensionsv1alpha1.Unit, name string) *extensionsv1alpha1.Unit {
for i, unit := range units {
if unit.Name == name {
return &units[i]
}
}
return nil
}
// FileWithPath returns the first file with the specified path from the slice, or nil if not found.
func FileWithPath(files []extensionsv1alpha1.File, path string) *extensionsv1alpha1.File {
for i, file := range files {
if file.Path == path {
return &files[i]
}
}
return nil
}
// UnitOptionWithSectionAndName returns the first unit option with the specified section and name from the slice, or nil if not found.
func UnitOptionWithSectionAndName(opts []*unit.UnitOption, section, name string) *unit.UnitOption {
for i, opt := range opts {
if opt.Section == section && opt.Name == name {
return opts[i]
}
}
return nil
}
// EnsureStringWithPrefix ensures that a string having the given prefix exists in the given slice
// and all matches are with a value equal to prefix + value.
func EnsureStringWithPrefix(items []string, prefix, value string) []string {
if StringWithPrefixIndex(items, prefix) < 0 {
return append(items, prefix+value)
}
for i, item := range items {
if strings.HasPrefix(item, prefix) {
items[i] = prefix + value
}
}
return items
}
// EnsureNoStringWithPrefix ensures that a string having the given prefix does not exist in the given slice.
func EnsureNoStringWithPrefix(items []string, prefix string) []string {
return slices.DeleteFunc(items, func(s string) bool {
return strings.HasPrefix(s, prefix)
})
}
// EnsureStringWithPrefixContains ensures that a string having the given prefix exists in the given slice
// and all matches contain the given value in a list separated by sep.
func EnsureStringWithPrefixContains(items []string, prefix, value, sep string) []string {
if StringWithPrefixIndex(items, prefix) < 0 {
return append(items, prefix+value)
}
for i, item := range items {
if !strings.HasPrefix(item, prefix) {
continue
}
values := strings.Split(strings.TrimPrefix(items[i], prefix), sep)
if slices.Index(values, value) < 0 {
values = append(values, value)
items[i] = prefix + strings.Join(values, sep)
}
}
return items
}
// EnsureNoStringWithPrefixContains ensures that either a string having the given prefix does not exist in the given slice,
// or it doesn't contain the given value in a list separated by sep.
func EnsureNoStringWithPrefixContains(items []string, prefix, value, sep string) []string {
for i, item := range items {
if !strings.HasPrefix(item, prefix) {
continue
}
values := strings.Split(strings.TrimPrefix(items[i], prefix), sep)
if j := slices.Index(values, value); j >= 0 {
values = append(values[:j], values[j+1:]...)
items[i] = prefix + strings.Join(values, sep)
}
}
return items
}
// EnsureEnvVarWithName ensures that a EnvVar with a name equal to the name of the given EnvVar exists
// in the given slice and the first item in the list would be equal to the given EnvVar.
func EnsureEnvVarWithName(items []corev1.EnvVar, item corev1.EnvVar) []corev1.EnvVar {
i := slices.IndexFunc(items, func(ev corev1.EnvVar) bool {
return ev.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureNoEnvVarWithName ensures that a EnvVar with the given name does not exist in the given slice.
func EnsureNoEnvVarWithName(items []corev1.EnvVar, name string) []corev1.EnvVar {
return slices.DeleteFunc(items, func(ev corev1.EnvVar) bool {
return ev.Name == name
})
}
// EnsureVolumeMountWithName ensures that a VolumeMount with a name equal to the name of the given VolumeMount exists
// in the given slice and the first item in the list would be equal to the given VolumeMount.
func EnsureVolumeMountWithName(items []corev1.VolumeMount, item corev1.VolumeMount) []corev1.VolumeMount {
i := slices.IndexFunc(items, func(vm corev1.VolumeMount) bool {
return vm.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureNoVolumeMountWithName ensures that a VolumeMount with the given name does not exist in the given slice.
func EnsureNoVolumeMountWithName(items []corev1.VolumeMount, name string) []corev1.VolumeMount {
return slices.DeleteFunc(items, func(vm corev1.VolumeMount) bool {
return vm.Name == name
})
}
// EnsureVolumeWithName ensures that a Volume with a name equal to the name of the given Volume exists
// in the given slice and the first item in the list would be equal to the given Volume.
func EnsureVolumeWithName(items []corev1.Volume, item corev1.Volume) []corev1.Volume {
i := slices.IndexFunc(items, func(v corev1.Volume) bool {
return v.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureNoVolumeWithName ensures that a Volume with the given name does not exist in the given slice.
func EnsureNoVolumeWithName(items []corev1.Volume, name string) []corev1.Volume {
return slices.DeleteFunc(items, func(v corev1.Volume) bool {
return v.Name == name
})
}
// EnsureContainerWithName ensures that a Container with a name equal to the name of the given Container exists
// in the given slice and the first item in the list would be equal to the given Container.
func EnsureContainerWithName(items []corev1.Container, item corev1.Container) []corev1.Container {
i := slices.IndexFunc(items, func(c corev1.Container) bool {
return c.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureNoContainerWithName ensures that a Container with the given name does not exist in the given slice.
func EnsureNoContainerWithName(items []corev1.Container, name string) []corev1.Container {
return slices.DeleteFunc(items, func(c corev1.Container) bool {
return c.Name == name
})
}
// EnsureVPAContainerResourcePolicyWithName ensures that a container policy with a name equal to the name of the given
// container policy exists in the given slice and the first item in the list would be equal to the given container policy.
func EnsureVPAContainerResourcePolicyWithName(items []vpaautoscalingv1.ContainerResourcePolicy, item vpaautoscalingv1.ContainerResourcePolicy) []vpaautoscalingv1.ContainerResourcePolicy {
i := slices.IndexFunc(items, func(crp vpaautoscalingv1.ContainerResourcePolicy) bool {
return crp.ContainerName == item.ContainerName
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsurePVCWithName ensures that a PVC with a name equal to the name of the given PVC exists
// in the given slice and the first item in the list would be equal to the given PVC.
func EnsurePVCWithName(items []corev1.PersistentVolumeClaim, item corev1.PersistentVolumeClaim) []corev1.PersistentVolumeClaim {
i := slices.IndexFunc(items, func(pvc corev1.PersistentVolumeClaim) bool {
return pvc.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureNoPVCWithName ensures that a PVC with the given name does not exist in the given slice.
func EnsureNoPVCWithName(items []corev1.PersistentVolumeClaim, name string) []corev1.PersistentVolumeClaim {
return slices.DeleteFunc(items, func(pvc corev1.PersistentVolumeClaim) bool {
return pvc.Name == name
})
}
// EnsureUnitOption ensures the given unit option exist in the given slice.
func EnsureUnitOption(items []*unit.UnitOption, item *unit.UnitOption) []*unit.UnitOption {
i := slices.IndexFunc(items, func(uo *unit.UnitOption) bool {
return reflect.DeepEqual(uo, item)
})
if i < 0 {
return append(items, item)
}
return items
}
// EnsureFileWithPath ensures that a file with a path equal to the path of the given file exists in the given slice
// and is equal to the given file.
func EnsureFileWithPath(items []extensionsv1alpha1.File, item extensionsv1alpha1.File) []extensionsv1alpha1.File {
i := slices.IndexFunc(items, func(f extensionsv1alpha1.File) bool {
return f.Path == item.Path
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureUnitWithName ensures that an unit with a name equal to the name of the given unit exists in the given slice
// and is equal to the given unit.
func EnsureUnitWithName(items []extensionsv1alpha1.Unit, item extensionsv1alpha1.Unit) []extensionsv1alpha1.Unit {
i := slices.IndexFunc(items, func(u extensionsv1alpha1.Unit) bool {
return u.Name == item.Name
})
if i < 0 {
return append(items, item)
}
items[i] = item
return items
}
// EnsureAnnotationOrLabel ensures the given key/value exists in the annotationOrLabelMap map.
func EnsureAnnotationOrLabel(annotationOrLabelMap map[string]string, key, value string) map[string]string {
if annotationOrLabelMap == nil {
annotationOrLabelMap = make(map[string]string, 1)
}
annotationOrLabelMap[key] = value
return annotationOrLabelMap
}
// StringWithPrefixIndex returns the index of the first occurrence of a string having the given prefix in the given slice, or -1 if not found.
func StringWithPrefixIndex(items []string, prefix string) int {
return slices.IndexFunc(items, func(s string) bool {
return strings.HasPrefix(s, prefix)
})
}