Create spaces for dashboard

Signed-off-by: Md. Ishtiaq Islam <[email protected]>
kubedb · Apr 18, 2024 · 33742df · 33742df
1 parent e01cb9e
commit 33742df
Show file tree

Hide file tree

Showing 5 changed files with 75 additions and 18 deletions.
diff --git a/elasticsearchdashboard/api.go b/elasticsearchdashboard/api.go
@@ -25,5 +25,6 @@ type EDClient interface {
 	GetStateFromHealthResponse(health *Health) (esapi.DashboardServerState, error)
 	ExportSavedObjects(spaceName string) (*Response, error)
 	ImportSavedObjects(spaceName, filepath string) (*Response, error)
-	ListSpaces() ([]string, error)
+	ListSpaces() ([]Space, error)
+	CreateSpace(space Space) error
 }
diff --git a/elasticsearchdashboard/client.go b/elasticsearchdashboard/client.go
@@ -37,7 +37,7 @@ const (
 "observability-visualization", "search"]}`
 	SavedObjectsExportURL = "/api/saved_objects/_export"
 	SavedObjectsImportURL = "/api/saved_objects/_import"
-	ListSpacesURL         = "/api/spaces/space"
+	SpacesURL             = "/api/spaces/space"
 )
 
 type Client struct {
@@ -88,3 +88,13 @@ type ResponseBody struct {
 	Status  map[string]interface{} `json:"status"`
 	Metrics map[string]interface{} `json:"metrics"`
 }
+
+type Space struct {
+	Id               string   `json:"id"`
+	Name             string   `json:"name"`
+	Description      string   `json:"description,omitempty"`
+	Color            string   `json:"color,omitempty"`
+	Initials         string   `json:"initials,omitempty"`
+	DisabledFeatures []string `json:"disabledFeatures,omitempty"`
+	ImageUrl         string   `json:"imageUrl,omitempty"`
+}
diff --git a/elasticsearchdashboard/ed_client_v7.go b/elasticsearchdashboard/ed_client_v7.go
@@ -147,14 +147,14 @@ func (h *EDClientV7) ImportSavedObjects(spaceName, filepath string) (*Response,
 	}, nil
 }
 
-func (h *EDClientV7) ListSpaces() ([]string, error) {
+func (h *EDClientV7) ListSpaces() ([]Space, error) {
 	req := h.Client.R().
 		SetDoNotParseResponse(true).
 		SetHeaders(map[string]string{
 			"Content-Type": "application/json",
 			"kbn-xsrf":     "true",
 		})
-	res, err := req.Get(ListSpacesURL)
+	res, err := req.Get(SpacesURL)
 	if err != nil {
 		klog.Error("Failed to send http request")
 		return nil, err
@@ -169,15 +169,36 @@ func (h *EDClientV7) ListSpaces() ([]string, error) {
 		return nil, fmt.Errorf("failed to list dashboard spaces %s", string(body))
 	}
 
-	var spaces []map[string]interface{}
+	var spaces []Space
 	if err = json.Unmarshal(body, &spaces); err != nil {
 		return nil, err
 	}
 
-	var spacesName []string
-	for _, space := range spaces {
-		spacesName = append(spacesName, space["id"].(string))
+	return spaces, nil
+}
+
+func (h *EDClientV7) CreateSpace(space Space) error {
+	req := h.Client.R().
+		SetDoNotParseResponse(true).
+		SetHeaders(map[string]string{
+			"Content-Type": "application/json",
+			"kbn-xsrf":     "true",
+		}).
+		SetBody(space)
+	res, err := req.Post(SpacesURL)
+	if err != nil {
+		klog.Error(err, "Failed to send http request")
+		return err
+	}
+
+	body, err := io.ReadAll(res.RawBody())
+	if err != nil {
+		return err
+	}
+
+	if res.StatusCode() != http.StatusOK {
+		return fmt.Errorf("failed to create dashboard space %s: %s", space.Name, string(body))
 	}
 
-	return spacesName, nil
+	return nil
 }
diff --git a/elasticsearchdashboard/ed_client_v8.go b/elasticsearchdashboard/ed_client_v8.go
@@ -145,14 +145,14 @@ func (h *EDClientV8) ImportSavedObjects(spaceName, filepath string) (*Response,
 	}, nil
 }
 
-func (h *EDClientV8) ListSpaces() ([]string, error) {
+func (h *EDClientV8) ListSpaces() ([]Space, error) {
 	req := h.Client.R().
 		SetDoNotParseResponse(true).
 		SetHeaders(map[string]string{
 			"Content-Type": "application/json",
 			"kbn-xsrf":     "true",
 		})
-	res, err := req.Get(ListSpacesURL)
+	res, err := req.Get(SpacesURL)
 	if err != nil {
 		klog.Error("Failed to send http request")
 		return nil, err
@@ -167,15 +167,36 @@ func (h *EDClientV8) ListSpaces() ([]string, error) {
 		return nil, fmt.Errorf("failed to list dashboard spaces %s", string(body))
 	}
 
-	var spaces []map[string]interface{}
+	var spaces []Space
 	if err = json.Unmarshal(body, &spaces); err != nil {
 		return nil, err
 	}
 
-	var spacesName []string
-	for _, space := range spaces {
-		spacesName = append(spacesName, space["id"].(string))
+	return spaces, nil
+}
+
+func (h *EDClientV8) CreateSpace(space Space) error {
+	req := h.Client.R().
+		SetDoNotParseResponse(true).
+		SetHeaders(map[string]string{
+			"Content-Type": "application/json",
+			"kbn-xsrf":     "true",
+		}).
+		SetBody(space)
+	res, err := req.Post(SpacesURL)
+	if err != nil {
+		klog.Error(err, "Failed to send http request")
+		return err
+	}
+
+	body, err := io.ReadAll(res.RawBody())
+	if err != nil {
+		return err
+	}
+
+	if res.StatusCode() != http.StatusOK {
+		return fmt.Errorf("failed to create dashboard space %s: %s", space.Name, string(body))
 	}
 
-	return spacesName, nil
+	return nil
 }
diff --git a/elasticsearchdashboard/os_client.go b/elasticsearchdashboard/os_client.go
@@ -145,6 +145,10 @@ func (h *OSClient) ImportSavedObjects(_, filepath string) (*Response, error) {
 	}, nil
 }
 
-func (h *OSClient) ListSpaces() ([]string, error) {
-	return []string{"default"}, nil
+func (h *OSClient) ListSpaces() ([]Space, error) {
+	return []Space{{Id: "default"}}, nil
+}
+
+func (h *OSClient) CreateSpace(_ Space) error {
+	return nil
 }