Merge pull request #19 from serverscom/k8s-clusters

k8s clusters cmd

Author: olegy89

Makefile

@@ -16,6 +16,7 @@ generate: deps
 	mockgen --destination ./internal/mocks/invoices_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/invoices.go
 	mockgen --destination ./internal/mocks/account_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/accounts.go
 	mockgen --destination ./internal/mocks/locations_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/locations.go
+	mockgen --destination ./internal/mocks/kubernetes_clusters_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/kubernetes_clusters.go
 	sed -i '' 's|github.com/serverscom/srvctl/vendor/github.com/serverscom/serverscom-go-client/pkg|github.com/serverscom/serverscom-go-client/pkg|g' \
 	./internal/mocks/ssh_service.go \
 	./internal/mocks/hosts_service.go \
@@ -25,5 +26,6 @@ generate: deps
 	./internal/mocks/invoices_service.go \
 	./internal/mocks/account_service.go \
 	./internal/mocks/collection.go \
-	./internal/mocks/locations_service.go
+	./internal/mocks/locations_service.go \
+	./internal/mocks/kubernetes_clusters_service.go
 

cmd/entities/k8s/get.go

@@ -0,0 +1,80 @@
+package k8s
+
+import (
+	"log"
+
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newGetCmd(cmdContext *base.CmdContext) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get <id>",
+		Short: "Get a kubernetes cluster",
+		Long:  "Get a kubernetes cluster by id",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			id := args[0]
+			k8s, err := scClient.KubernetesClusters.Get(ctx, id)
+			if err != nil {
+				return err
+			}
+
+			if k8s != nil {
+				formatter := cmdContext.GetOrCreateFormatter(cmd)
+				return formatter.Format(k8s)
+			}
+			return nil
+		},
+	}
+
+	return cmd
+}
+
+func newGetNodeCmd(cmdContext *base.CmdContext) *cobra.Command {
+	var clusterID string
+	cmd := &cobra.Command{
+		Use:   "get-node <node-id> --cluster-id <string>",
+		Short: "Get a kubernetes cluster node",
+		Long:  "Get a kubernetes cluster node by ID",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			nodeID := args[0]
+			k8s, err := scClient.KubernetesClusters.GetNode(ctx, clusterID, nodeID)
+			if err != nil {
+				return err
+			}
+
+			if k8s != nil {
+				formatter := cmdContext.GetOrCreateFormatter(cmd)
+				return formatter.Format(k8s)
+			}
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&clusterID, "cluster-id", "", "cluster id")
+	if err := cmd.MarkFlagRequired("cluster-id"); err != nil {
+		log.Fatal(err)
+	}
+
+	return cmd
+}

cmd/entities/k8s/k8s.go

@@ -0,0 +1,59 @@
+package k8s
+
+import (
+	"log"
+
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/serverscom/srvctl/internal/output/entities"
+	"github.com/spf13/cobra"
+)
+
+func NewCmd(cmdContext *base.CmdContext) *cobra.Command {
+	entitiesMap, err := getK8sEntities()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	cmd := &cobra.Command{
+		Use:   "k8s",
+		Short: "Manage kubernetes clusters",
+		PersistentPreRunE: base.CombinePreRunE(
+			base.CheckFormatterFlags(cmdContext, entitiesMap),
+			base.CheckEmptyContexts(cmdContext),
+		),
+		Args: base.NoArgs,
+		Run:  base.UsageRun,
+	}
+
+	cmd.AddCommand(
+		newListCmd(cmdContext),
+		newListNodesCmd(cmdContext),
+		newGetCmd(cmdContext),
+		newGetNodeCmd(cmdContext),
+		newUpdateCmd(cmdContext),
+	)
+
+	base.AddFormatFlags(cmd)
+
+	return cmd
+}
+
+func getK8sEntities() (map[string]entities.EntityInterface, error) {
+	result := make(map[string]entities.EntityInterface)
+
+	k8sEntity, err := entities.Registry.GetEntityFromValue(serverscom.KubernetesCluster{})
+	if err != nil {
+		log.Fatal(err)
+	}
+	result["k8s"] = k8sEntity
+
+	k8sNodeEntity, err := entities.Registry.GetEntityFromValue(serverscom.KubernetesClusterNode{})
+	if err != nil {
+		log.Fatal(err)
+	}
+	result["list-nodes"] = k8sNodeEntity
+	result["get-node"] = k8sNodeEntity
+
+	return result, nil
+}