feat(cluster): 增加集群导入

internal/api/v1/cluster/cluster.go

@@ -0,0 +1,101 @@
+package cluster
+
+import (
+	kContext "context"
+	"fmt"
+	v1 "github.com/KubeOperator/ekko/internal/model/v1"
+	v1Cluster "github.com/KubeOperator/ekko/internal/model/v1/cluster"
+	"github.com/KubeOperator/ekko/internal/service/v1/cluster"
+	"github.com/KubeOperator/ekko/internal/util/kubernetes"
+	"github.com/asdine/storm/v3"
+	"github.com/google/uuid"
+	"github.com/kataras/iris/v12"
+	"github.com/kataras/iris/v12/context"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+type Handler struct {
+	clusterService cluster.Cluster
+}
+
+func NewHandler() *Handler {
+	return &Handler{
+		clusterService: cluster.NewClusterService(),
+	}
+}
+
+func (h *Handler) Create() iris.Handler {
+	return func(ctx *context.Context) {
+		var clusterImport Import
+		if err := ctx.ReadJSON(&clusterImport); err != nil {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", err.Error())
+			return
+		}
+		c, err := h.clusterService.Get(clusterImport.Name)
+		if err != nil && err != storm.ErrNotFound {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("error: %s", err.Error()))
+			return
+		}
+		if c != nil && c.Name != "" {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("cluster name %s is alerady exist", clusterImport.Name))
+			return
+		}
+		client, err := kubernetes.NewKubernetesClient(&kubernetes.Config{
+			Host:  clusterImport.ApiServer,
+			Token: clusterImport.Token,
+		})
+		if err != nil {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("import cluster filed: %s", err.Error()))
+			return
+		}
+		_, err = client.CoreV1().Namespaces().List(kContext.TODO(), metav1.ListOptions{})
+		if err != nil {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("import cluster failed: %s", err.Error()))
+			return
+		}
+		cs := v1Cluster.Cluster{
+			Metadata: v1.Metadata{
+				Name: clusterImport.Name,
+				UUID: uuid.New().String(),
+			},
+			Token:     clusterImport.Token,
+			Router:    clusterImport.Router,
+			ApiServer: clusterImport.ApiServer,
+			Status:    "Running",
+		}
+		err = h.clusterService.Create(&cs)
+		if err != nil {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("create cluster failed: %s", err.Error()))
+			return
+		}
+		ctx.StatusCode(iris.StatusOK)
+		ctx.Values().Set("data", cs)
+	}
+}
+
+func (h *Handler) ListAll() iris.Handler {
+	return func(ctx *context.Context) {
+		var clusters []v1Cluster.Cluster
+		clusters, err := h.clusterService.All()
+		if err != nil {
+			ctx.StatusCode(iris.StatusBadRequest)
+			ctx.Values().Set("message", fmt.Sprintf("get clusters failed: %s", err.Error()))
+			return
+		}
+		ctx.StatusCode(iris.StatusOK)
+		ctx.Values().Set("data", clusters)
+	}
+}
+
+func Install(parent iris.Party) {
+	handler := NewHandler()
+	sp := parent.Party("/clusters")
+	sp.Post("", handler.Create())
+	sp.Get("", handler.ListAll())
+}