-
Notifications
You must be signed in to change notification settings - Fork 479
/
config_references.go
75 lines (67 loc) · 1.86 KB
/
config_references.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 config
import (
"github.com/grafana/agent/pkg/util/structwalk"
prom "github.com/prometheus-operator/prometheus-operator/pkg/apis/monitoring/v1"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
// AssetReference is a namespaced Secret or ConfigMap selector.
type AssetReference struct {
Namespace string
Reference prom.SecretOrConfigMap
}
// AssetReferences returns all secret or configmap selectors used throughout v.
func AssetReferences(v interface{}) []AssetReference {
var refs []AssetReference
w := assetReferencesWalker{
addReference: func(ar AssetReference) {
refs = append(refs, ar)
},
}
structwalk.Walk(&w, v)
return refs
}
type assetReferencesWalker struct {
namespace string
addReference func(ar AssetReference)
}
func (arw *assetReferencesWalker) Visit(v interface{}) (w structwalk.Visitor) {
if v == nil {
return nil
}
// If we've come across a namespaced object, create a new visitor for that
// namespace.
if o, ok := v.(metav1.Object); ok {
return &assetReferencesWalker{
namespace: o.GetNamespace(),
addReference: arw.addReference,
}
}
switch sel := v.(type) {
case corev1.SecretKeySelector:
if sel.Key != "" && sel.Name != "" {
arw.addReference(AssetReference{
Namespace: arw.namespace,
Reference: prom.SecretOrConfigMap{Secret: &sel},
})
}
case *corev1.SecretKeySelector:
arw.addReference(AssetReference{
Namespace: arw.namespace,
Reference: prom.SecretOrConfigMap{Secret: sel},
})
case corev1.ConfigMapKeySelector:
if sel.Key != "" && sel.Name != "" {
arw.addReference(AssetReference{
Namespace: arw.namespace,
Reference: prom.SecretOrConfigMap{ConfigMap: &sel},
})
}
case *corev1.ConfigMapKeySelector:
arw.addReference(AssetReference{
Namespace: arw.namespace,
Reference: prom.SecretOrConfigMap{ConfigMap: sel},
})
}
return arw
}