/
helper.go
126 lines (106 loc) · 3.29 KB
/
helper.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
package clienthelper
import (
"context"
"fmt"
"io/ioutil"
"os"
"reflect"
authenticationv1 "k8s.io/api/authentication/v1"
authv1 "k8s.io/api/authorization/v1"
apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
kerrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/meta"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/schema"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
apiregistrationv1 "k8s.io/kube-aggregator/pkg/apis/apiregistration/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
)
var (
DefaultScheme = runtime.NewScheme()
)
func init() {
_ = clientgoscheme.AddToScheme(DefaultScheme)
// API extensions are not in the above scheme set,
// and must thus be added separately.
_ = authenticationv1.AddToScheme(DefaultScheme)
_ = apiextensionsv1.AddToScheme(DefaultScheme)
_ = apiregistrationv1.AddToScheme(DefaultScheme)
}
func CurrentNamespace() (string, error) {
namespaceEnv := os.Getenv("NAMESPACE")
if namespaceEnv != "" {
return namespaceEnv, nil
}
namespace, err := ioutil.ReadFile("/var/run/secrets/kubernetes.io/serviceaccount/namespace")
if err != nil {
return "", err
}
return string(namespace), nil
}
// ConvertExtra converts a string array map into the correct kubernetes auth extra value type
func ConvertExtra(orig map[string][]string) map[string]authv1.ExtraValue {
retMap := map[string]authv1.ExtraValue{}
for k, v := range orig {
retMap[k] = v
}
return retMap
}
// ConvertExtraFrom converts a string array map into the correct kubernetes auth extra value type
func ConvertExtraFrom(orig map[string]authenticationv1.ExtraValue) map[string][]string {
retMap := map[string][]string{}
for k, v := range orig {
retMap[k] = v
}
return retMap
}
func GetByIndex(ctx context.Context, c client.Client, obj runtime.Object, index, value string) error {
gvk, err := GVKFrom(obj, c.Scheme())
if err != nil {
return err
}
list, err := c.Scheme().New(gvk.GroupVersion().WithKind(gvk.Kind + "List"))
if err != nil {
// TODO: handle runtime.IsNotRegisteredError(err)
return err
}
err = c.List(ctx, list.(client.ObjectList), client.MatchingFields{index: value})
if err != nil {
return err
}
objs, err := meta.ExtractList(list)
if err != nil {
return err
} else if len(objs) == 0 {
return kerrors.NewNotFound(schema.GroupResource{Group: gvk.Group}, value)
} else if len(objs) > 1 {
return kerrors.NewConflict(schema.GroupResource{Group: gvk.Group}, value, fmt.Errorf("more than 1 object with the value"))
}
val := reflect.ValueOf(obj)
if val.Kind() != reflect.Ptr {
return fmt.Errorf("object not a pointer")
}
val = val.Elem()
newVal := reflect.Indirect(reflect.ValueOf(objs[0]))
if !val.Type().AssignableTo(newVal.Type()) {
return fmt.Errorf("mismatched types")
}
val.Set(newVal)
return nil
}
func GVKFrom(obj runtime.Object, scheme *runtime.Scheme) (schema.GroupVersionKind, error) {
gvks, _, err := scheme.ObjectKinds(obj)
if err != nil {
return schema.GroupVersionKind{}, err
} else if len(gvks) != 1 {
return schema.GroupVersionKind{}, fmt.Errorf("unexpected number of object kinds: %d", len(gvks))
}
return gvks[0], nil
}
func CurrentPodName() (string, error) {
podNameEnv := os.Getenv("POD_NAME")
if podNameEnv != "" {
return podNameEnv, nil
}
return os.Hostname()
}