Add new subcommand for the kubectl-hns plugin

Add a subcommand to list resources by their parent namespace:

  kubectl hns get -n <parent ns> <resource>

Author: cmurphy

internal/kubectl/get.go

@@ -0,0 +1,115 @@
+package kubectl
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/cli-runtime/pkg/printers"
+	"k8s.io/client-go/dynamic"
+	"k8s.io/client-go/kubernetes/scheme"
+)
+
+var (
+	watch         bool
+	namespace     string
+	allNamespaces bool
+	GroupVersion  = schema.GroupVersion{Group: "resources.hnc.x-k8s.io", Version: "v1alpha2"}
+)
+
+var getCmd = &cobra.Command{
+	Use:   "get",
+	Short: "list resources",
+	RunE: func(c *cobra.Command, args []string) error {
+		if len(args) == 0 {
+			return fmt.Errorf("not enough arguments to 'get'")
+		}
+		if len(args) > 1 {
+			return fmt.Errorf("too many arguments to 'get'")
+		}
+		d := newDoer(args[0])
+		rv, err := d.get(args[0])
+		if err != nil {
+			return err
+		}
+		if watch {
+			return d.watch(args[0], rv)
+		}
+		return nil
+	},
+}
+
+func newGetCmd() *cobra.Command {
+	return getCmd
+}
+
+func init() {
+	getCmd.Flags().BoolVarP(&watch, "watch", "w", false, "watch for changes")
+	getCmd.PersistentFlags().StringVarP(&namespace, "namespace", "n", "default", "parent namespace to scope request to")
+	getCmd.PersistentFlags().BoolVarP(&allNamespaces, "all-namespaces", "A", false, "list across all namespaces")
+}
+
+type doer struct {
+	client  dynamic.ResourceInterface
+	printer printers.ResourcePrinter
+}
+
+func newDoer(resource string) doer {
+	client := getClient(resource)
+	printOpts := printers.PrintOptions{WithNamespace: true}
+	return doer{
+		client:  client,
+		printer: printers.NewTypeSetter(scheme.Scheme).ToPrinter(printers.NewTablePrinter(printOpts)),
+	}
+}
+
+func getClient(resource string) dynamic.ResourceInterface {
+	gvr := GroupVersion.WithResource(resource)
+	gvr, _ = mapper.ResourceFor(gvr)
+	resourceClient := dynamicClient.Resource(gvr)
+	if allNamespaces {
+		return resourceClient
+	}
+	return resourceClient.Namespace(namespace)
+}
+
+func (d doer) get(resource string) (string, error) {
+	unst, err := d.client.List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		return "", err
+	}
+	rv := unst.GetResourceVersion()
+	if len(unst.Items) == 0 {
+		fmt.Printf("No resources found in %s namespace\n", namespace)
+		return rv, nil
+	}
+	unst.SetAPIVersion("resources.hnc.x-k8s.io/v1alpha2")
+	kind := unst.Items[0].GetKind()
+	unst.SetKind(kind + "List")
+	if err = d.printer.PrintObj(unst, os.Stdout); err != nil {
+		return rv, err
+	}
+	return rv, nil
+}
+
+func (d doer) watch(resource, rv string) error {
+	if rv == "" {
+		rv = "0"
+	}
+	opts := metav1.ListOptions{
+		ResourceVersion: rv,
+	}
+	watcher, err := d.client.Watch(context.TODO(), opts)
+	if err != nil {
+		return err
+	}
+	for event := range watcher.ResultChan() {
+		if err = d.printer.PrintObj(event.Object, os.Stdout); err != nil {
+			return err
+		}
+	}
+	return nil
+}

internal/kubectl/root.go

@@ -24,10 +24,12 @@ import (
 
 	"github.com/spf13/cobra"
 	"k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime/serializer"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/cli-runtime/pkg/genericclioptions"
+	"k8s.io/client-go/dynamic"
 	"k8s.io/client-go/kubernetes"
 	"k8s.io/client-go/kubernetes/scheme"
 	_ "k8s.io/client-go/plugin/pkg/client/auth"
@@ -41,6 +43,8 @@ var k8sClient *kubernetes.Clientset
 var hncClient *rest.RESTClient
 var rootCmd *cobra.Command
 var client Client
+var dynamicClient dynamic.Interface
+var mapper meta.RESTMapper
 
 type realClient struct{}
 type anchorStatus map[string]string
@@ -93,6 +97,16 @@ func init() {
 				return err
 			}
 
+			mapper, err = kubecfgFlags.ToRESTMapper()
+			if err != nil {
+				return err
+			}
+
+			dynamicClient, err = dynamic.NewForConfig(config)
+			if err != nil {
+				return err
+			}
+
 			return nil
 		},
 	}
@@ -105,6 +119,7 @@ func init() {
 	rootCmd.AddCommand(newConfigCmd())
 	rootCmd.AddCommand(newVersionCmd())
 	rootCmd.AddCommand(newHrqCmd())
+	rootCmd.AddCommand(newGetCmd())
 }
 
 func Execute() {