Merge "Update Rsync"

1 files changed, 93 insertions, 0 deletions

diff --git a/src/rsync/pkg/client/apply.go b/src/rsync/pkg/client/apply.go
new file mode 100644
index 00000000..96233a31
--- /dev/null
+++ b/src/rsync/pkg/client/apply.go
@@ -0,0 +1,93 @@
+/*
+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.
+*/
+// Based on Code: https://github.com/johandry/klient
+package client
+
+import (
+	"k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/cli-runtime/pkg/resource"
+	"k8s.io/kubectl/pkg/util"
+)
+
+// Apply creates a resource with the given content
+func (c *Client) Apply(content []byte) error {
+	r := c.ResultForContent(content, nil)
+	return c.ApplyResource(r)
+}
+
+// ApplyFiles create the resource(s) from the given filenames (file, directory or STDIN) or HTTP URLs
+func (c *Client) ApplyFiles(filenames ...string) error {
+	r := c.ResultForFilenameParam(filenames, nil)
+	return c.ApplyResource(r)
+}
+
+// ApplyResource applies the given resource. Create the resources with `ResultForFilenameParam` or `ResultForContent`
+func (c *Client) ApplyResource(r *resource.Result) error {
+	if err := r.Err(); err != nil {
+		return err
+	}
+
+	// Is ServerSideApply requested
+	if c.ServerSideApply {
+		return r.Visit(serverSideApply)
+	}
+
+	return r.Visit(apply)
+}
+
+func apply(info *resource.Info, err error) error {
+	if err != nil {
+		return failedTo("apply", info, err)
+	}
+
+	// If it does not exists, just create it
+	current, err := resource.NewHelper(info.Client, info.Mapping).Get(info.Namespace, info.Name, info.Export)
+	if err != nil {
+		if !errors.IsNotFound(err) {
+			return failedTo("retrieve current configuration", info, err)
+		}
+		if err := util.CreateApplyAnnotation(info.Object, unstructured.UnstructuredJSONScheme); err != nil {
+			return failedTo("set annotation", info, err)
+		}
+		return create(info, nil)
+	}
+
+	// If exists, patch it
+	return patch(info, current)
+}
+
+func serverSideApply(info *resource.Info, err error) error {
+	if err != nil {
+		return failedTo("serverside apply", info, err)
+	}
+
+	data, err := runtime.Encode(unstructured.UnstructuredJSONScheme, info.Object)
+	if err != nil {
+		return failedTo("encode for the serverside apply", info, err)
+	}
+
+	options := metav1.PatchOptions{
+		// TODO: Find out how to get the force conflict flag
+		// Force:        &forceConflicts,
+		// FieldManager: FieldManager,
+	}
+	obj, err := resource.NewHelper(info.Client, info.Mapping).Patch(info.Namespace, info.Name, types.ApplyPatchType, data, &options)
+	if err != nil {
+		return failedTo("serverside patch", info, err)
+	}
+	info.Refresh(obj, true)
+	return nil
+}