create work command (#134)

Signed-off-by: Jian Qiu <[email protected]>

Author: qiujian16

pkg/cmd/create/cmd.go

@@ -5,6 +5,7 @@ import (
 	"github.com/spf13/cobra"
 	"k8s.io/cli-runtime/pkg/genericclioptions"
 	"open-cluster-management.io/clusteradm/pkg/cmd/create/clusterset"
+	"open-cluster-management.io/clusteradm/pkg/cmd/create/work"
 	genericclioptionsclusteradm "open-cluster-management.io/clusteradm/pkg/genericclioptions"
 )
 
@@ -16,6 +17,7 @@ func NewCmd(clusteradmFlags *genericclioptionsclusteradm.ClusteradmFlags, stream
 	}
 
 	cmd.AddCommand(clusterset.NewCmd(clusteradmFlags, streams))
+	cmd.AddCommand(work.NewCmd(clusteradmFlags, streams))
 
 	return cmd
 }

pkg/cmd/create/work/cmd.go

@@ -0,0 +1,53 @@
+// Copyright Contributors to the Open Cluster Management project
+package work
+
+import (
+	"fmt"
+
+	genericclioptionsclusteradm "open-cluster-management.io/clusteradm/pkg/genericclioptions"
+	clusteradmhelpers "open-cluster-management.io/clusteradm/pkg/helpers"
+
+	"github.com/spf13/cobra"
+	"k8s.io/cli-runtime/pkg/genericclioptions"
+)
+
+var example = `
+# create a work.
+%[1]s create work work-example -f xxx.yaml --cluster cluster1
+`
+
+// NewCmd...
+func NewCmd(clusteradmFlags *genericclioptionsclusteradm.ClusteradmFlags, streams genericclioptions.IOStreams) *cobra.Command {
+	o := newOptions(clusteradmFlags, streams)
+
+	cmd := &cobra.Command{
+		Use:          "work",
+		Short:        "create a work",
+		Example:      fmt.Sprintf(example, clusteradmhelpers.GetExampleHeader()),
+		SilenceUsage: true,
+		PreRunE: func(c *cobra.Command, args []string) error {
+			clusteradmhelpers.DryRunMessage(clusteradmFlags.DryRun)
+
+			return nil
+		},
+		RunE: func(c *cobra.Command, args []string) error {
+			if err := o.complete(c, args); err != nil {
+				return err
+			}
+			if err := o.validate(); err != nil {
+				return err
+			}
+			if err := o.run(); err != nil {
+				return err
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&o.Cluster, "cluster", "", "Names of the managed cluster to apply work")
+	cmd.Flags().BoolVar(&o.Overwrite, "overwrite", false, "Overwrite the existing work if it exists already")
+	o.FileNameFlags.AddFlags(cmd.Flags())
+
+	return cmd
+}

pkg/cmd/create/work/exec.go

@@ -0,0 +1,120 @@
+// Copyright Contributors to the Open Cluster Management project
+package work
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/cli-runtime/pkg/resource"
+	workclientset "open-cluster-management.io/api/client/work/clientset/versioned"
+	workapiv1 "open-cluster-management.io/api/work/v1"
+)
+
+func (o *Options) complete(cmd *cobra.Command, args []string) (err error) {
+	if len(args) == 0 {
+		return fmt.Errorf("work name must be specified")
+	}
+
+	if len(args) > 1 {
+		return fmt.Errorf("only one work name can be specified")
+	}
+
+	o.Workname = args[0]
+
+	return nil
+}
+
+func (o *Options) validate() (err error) {
+	if len(o.Cluster) == 0 {
+		return fmt.Errorf("the name of the cluster must be specified")
+	}
+	if len(*o.FileNameFlags.Filenames) == 0 {
+		return fmt.Errorf("manifest files must be specified")
+	}
+	return nil
+}
+
+func (o *Options) run() (err error) {
+	restConfig, err := o.ClusteradmFlags.KubectlFactory.ToRESTConfig()
+	if err != nil {
+		return err
+	}
+	workClient, err := workclientset.NewForConfig(restConfig)
+	if err != nil {
+		return err
+	}
+
+	manifests, err := o.readManifests()
+	if err != nil {
+		return err
+	}
+
+	err = o.applyWork(workClient, manifests)
+	if err != nil {
+		return err
+	}
+
+	fmt.Fprintf(o.Streams.Out, "work %s in cluster %s is created\n", o.Workname, o.Cluster)
+	return
+}
+
+func (o *Options) readManifests() ([]workapiv1.Manifest, error) {
+	opt := o.FileNameFlags.ToOptions()
+	builder := resource.NewLocalBuilder().
+		Unstructured().
+		FilenameParam(false, &opt).
+		Flatten().
+		ContinueOnError()
+	result := builder.Do()
+
+	if err := result.Err(); err != nil {
+		return nil, err
+	}
+
+	manifests := []workapiv1.Manifest{}
+
+	items, err := result.Infos()
+	if err != nil {
+		return nil, err
+	}
+	for _, item := range items {
+		manifests = append(manifests, workapiv1.Manifest{RawExtension: runtime.RawExtension{Object: item.Object}})
+	}
+
+	return manifests, nil
+}
+
+func (o *Options) applyWork(workClient workclientset.Interface, manifests []workapiv1.Manifest) error {
+	work, err := workClient.WorkV1().ManifestWorks(o.Cluster).Get(context.TODO(), o.Workname, metav1.GetOptions{})
+
+	switch {
+	case errors.IsNotFound(err):
+		work = &workapiv1.ManifestWork{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      o.Workname,
+				Namespace: o.Cluster,
+			},
+			Spec: workapiv1.ManifestWorkSpec{
+				Workload: workapiv1.ManifestsTemplate{
+					Manifests: manifests,
+				},
+			},
+		}
+		_, createErr := workClient.WorkV1().ManifestWorks(o.Cluster).Create(context.TODO(), work, metav1.CreateOptions{})
+		return createErr
+	case err != nil:
+		return err
+	}
+
+	if !o.Overwrite {
+		return fmt.Errorf("work %s in cluster %s already exists", o.Workname, o.Cluster)
+	}
+
+	work.Spec.Workload.Manifests = manifests
+	_, err = workClient.WorkV1().ManifestWorks(o.Cluster).Update(context.TODO(), work, metav1.UpdateOptions{})
+	return err
+}

pkg/cmd/create/work/options.go

@@ -0,0 +1,38 @@
+// Copyright Contributors to the Open Cluster Management project
+package work
+
+import (
+	"k8s.io/cli-runtime/pkg/genericclioptions"
+	genericclioptionsclusteradm "open-cluster-management.io/clusteradm/pkg/genericclioptions"
+)
+
+type Options struct {
+	//ClusteradmFlags: The generic optiosn from the clusteradm cli-runtime.
+	ClusteradmFlags *genericclioptionsclusteradm.ClusteradmFlags
+
+	Streams genericclioptions.IOStreams
+
+	Cluster string
+
+	Workname string
+
+	FileNameFlags genericclioptions.FileNameFlags
+
+	Overwrite bool
+}
+
+func newOptions(clusteradmFlags *genericclioptionsclusteradm.ClusteradmFlags, streams genericclioptions.IOStreams) *Options {
+	return &Options{
+		ClusteradmFlags: clusteradmFlags,
+		Streams:         streams,
+		Cluster:         "",
+		FileNameFlags: genericclioptions.FileNameFlags{
+			Filenames: &[]string{},
+			Recursive: boolPtr(true),
+		},
+	}
+}
+
+func boolPtr(val bool) *bool {
+	return &val
+}