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
|
/*
Copyright 2018 Intel Corporation.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package main
import (
"log"
pkgerrors "github.com/pkg/errors"
"k8s.io/apimachinery/pkg/api/meta"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/runtime/schema"
"k8s.io/client-go/kubernetes/scheme"
utils "k8splugin/internal"
"k8splugin/internal/app"
"k8splugin/internal/helm"
)
type genericPlugin struct {
}
// Create deployment object in a specific Kubernetes cluster
func (g genericPlugin) Create(yamlFilePath string, namespace string, client *app.KubernetesClient) (string, error) {
if namespace == "" {
namespace = "default"
}
//Decode the yaml file to create a runtime.Object
obj, err := utils.DecodeYAML(yamlFilePath, nil)
if err != nil {
return "", pkgerrors.Wrap(err, "Decode deployment object error")
}
//Convert the runtime.Object to an unstructured Object
unstruct := &unstructured.Unstructured{}
err = scheme.Scheme.Convert(obj, unstruct, nil)
if err != nil {
return "", pkgerrors.Wrap(err, "Converting to unstructured object")
}
dynClient := client.GetDynamicClient()
mapper := client.GetMapper()
gvk := unstruct.GroupVersionKind()
mapping, err := mapper.RESTMapping(schema.GroupKind{Group: gvk.Group, Kind: gvk.Kind}, gvk.Version)
if err != nil {
return "", pkgerrors.Wrap(err, "Mapping kind to resource error")
}
gvr := mapping.Resource
var createdObj *unstructured.Unstructured
switch mapping.Scope.Name() {
case meta.RESTScopeNameNamespace:
createdObj, err = dynClient.Resource(gvr).Namespace(namespace).Create(unstruct, metav1.CreateOptions{})
case meta.RESTScopeNameRoot:
createdObj, err = dynClient.Resource(gvr).Create(unstruct, metav1.CreateOptions{})
default:
return "", pkgerrors.New("Got an unknown RESTSCopeName for mapping: " + gvk.String())
}
if err != nil {
return "", pkgerrors.Wrap(err, "Create object error")
}
return createdObj.GetName(), nil
}
// Delete an existing deployment hosted in a specific Kubernetes cluster
func (g genericPlugin) Delete(resource helm.KubernetesResource, namespace string, client *app.KubernetesClient) error {
if namespace == "" {
namespace = "default"
}
deletePolicy := metav1.DeletePropagationForeground
opts := &metav1.DeleteOptions{
PropagationPolicy: &deletePolicy,
}
dynClient := client.GetDynamicClient()
mapper := client.GetMapper()
mapping, err := mapper.RESTMapping(schema.GroupKind{
Group: resource.GVK.Group,
Kind: resource.GVK.Kind,
}, resource.GVK.Version)
if err != nil {
return pkgerrors.Wrap(err, "Mapping kind to resource error")
}
gvr := mapping.Resource
log.Printf("Using gvr: %s, %s, %s", gvr.Group, gvr.Version, gvr.Resource)
switch mapping.Scope.Name() {
case meta.RESTScopeNameNamespace:
err = dynClient.Resource(gvr).Namespace(namespace).Delete(resource.Name, opts)
case meta.RESTScopeNameRoot:
err = dynClient.Resource(gvr).Delete(resource.Name, opts)
default:
return pkgerrors.New("Got an unknown RESTSCopeName for mapping: " + resource.GVK.String())
}
if err != nil {
return pkgerrors.Wrap(err, "Delete object error")
}
return nil
}
// ExportedVariable is what we will look for when calling the generic plugin
var ExportedVariable genericPlugin
|
