3 files changed, 67 insertions, 0 deletions
diff --git a/src/k8splugin/plugins/generic/plugin.go b/src/k8splugin/plugins/generic/plugin.go
index 0711466f..aa503097 100644
--- a/src/k8splugin/plugins/generic/plugin.go
+++ b/src/k8splugin/plugins/generic/plugin.go
@@ -91,6 +91,62 @@ func (g genericPlugin) Create(yamlFilePath string, namespace string, client plug
 	return createdObj.GetName(), nil
 }
 
+// Update deployment object in a specific Kubernetes cluster
+func (g genericPlugin) Update(yamlFilePath string, namespace string, client plugin.KubernetesConnector) (string, error) {
+        if namespace == "" {
+                namespace = "default"
+        }
+
+        //Decode the yaml file to create a runtime.Object
+        unstruct := &unstructured.Unstructured{}
+        //Ignore the returned obj as we expect the data in unstruct
+        _, err := utils.DecodeYAML(yamlFilePath, unstruct)
+        if err != nil {
+                return "", pkgerrors.Wrap(err, "Decode deployment object error")
+        }
+
+        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")
+        }
+
+        //Add the tracking label to all resources created here
+        labels := unstruct.GetLabels()
+        //Check if labels exist for this object
+        if labels == nil {
+                labels = map[string]string{}
+        }
+        labels[config.GetConfiguration().KubernetesLabelName] = client.GetInstanceID()
+        unstruct.SetLabels(labels)
+
+        // This checks if the resource we are creating has a podSpec in it
+        // Eg: Deployment, StatefulSet, Job etc..
+        // If a PodSpec is found, the label will be added to it too.
+        plugin.TagPodsIfPresent(unstruct, client.GetInstanceID())
+
+        gvr := mapping.Resource
+        var updatedObj *unstructured.Unstructured
+
+        switch mapping.Scope.Name() {
+        case meta.RESTScopeNameNamespace:
+                updatedObj, err = dynClient.Resource(gvr).Namespace(namespace).Update(unstruct, metav1.UpdateOptions{})
+        case meta.RESTScopeNameRoot:
+                updatedObj, err = dynClient.Resource(gvr).Update(unstruct, metav1.UpdateOptions{})
+        default:
+                return "", pkgerrors.New("Got an unknown RESTSCopeName for mapping: " + gvk.String())
+        }
+
+        if err != nil {
+                return "", pkgerrors.Wrap(err, "Update object error")
+        }
+
+        return updatedObj.GetName(), nil
+}
+
 // Get an existing resource hosted in a specific Kubernetes cluster
 func (g genericPlugin) Get(resource helm.KubernetesResource,
 	namespace string, client plugin.KubernetesConnector) (string, error) {
diff --git a/src/k8splugin/plugins/namespace/plugin.go b/src/k8splugin/plugins/namespace/plugin.go
index feb2aa61..cfc395bb 100644
--- a/src/k8splugin/plugins/namespace/plugin.go
+++ b/src/k8splugin/plugins/namespace/plugin.go
@@ -107,3 +107,8 @@ func (p namespacePlugin) List(gvk schema.GroupVersionKind, namespace string, cli
 
 	return result, nil
 }
+
+func (p namespacePlugin) Update(yamlFilePath string, namespace string, client plugin.KubernetesConnector) (string, error) {
+
+   return "", nil
+}
diff --git a/src/k8splugin/plugins/service/plugin.go b/src/k8splugin/plugins/service/plugin.go
index 4c1f37be..28191767 100644
--- a/src/k8splugin/plugins/service/plugin.go
+++ b/src/k8splugin/plugins/service/plugin.go
@@ -137,3 +137,9 @@ func (p servicePlugin) Get(resource helm.KubernetesResource, namespace string, c
 
 	return service.Name, nil
 }
+
+func (p servicePlugin) Update(yamlFilePath string, namespace string, client plugin.KubernetesConnector) (string, error) {
+
+        return "", nil
+
+}