added func for PQA-1848

client/cluster.go

@@ -1,7 +1,11 @@
 package client
 
 import (
+	"bytes"
 	"fmt"
+	"io"
+	"os"
+	"path/filepath"
 	"strings"
 
 	clientV1 "github.com/spectrocloud/palette-api-go/client/v1"
@@ -247,3 +251,43 @@ func clusterNameEqFilter(name string) *models.V1SearchFilterItem {
 		Type:     models.V1SearchFilterPropertyTypeString,
 	}
 }
+
+func (h *V1Client) GetLogFetcherStatus(uid string, logFetcherUid *string) (*models.V1ClusterLogFetcher, error) {
+	params := clientV1.NewV1ClusterFeatureLogFetcherGetParamsWithContext(h.ctx).WithUID(uid).WithRequestID(logFetcherUid)
+	resp, err := h.Client.V1ClusterFeatureLogFetcherGet(params)
+	if err != nil {
+		return nil, err
+	}
+	return resp.Payload, nil
+}
+
+func (h *V1Client) InitiateDownloadOfClusterLogs(uid string, V1ClusterLogFetcherRequestObj *models.V1ClusterLogFetcherRequest) (*string, error) {
+	params := clientV1.NewV1ClusterFeatureLogFetcherCreateParamsWithContext(h.ctx).WithUID(uid).WithBody(V1ClusterLogFetcherRequestObj)
+	resp, err := h.Client.V1ClusterFeatureLogFetcherCreate(params)
+	if err != nil {
+		return nil, err
+	}
+	return resp.GetPayload().UID, nil
+}
+
+func (h *V1Client) DownloadLogs(uid string, logFetcherUid string) (io.Writer, error) {
+	filename := "logs-" + uid + ".zip"
+	params := clientV1.NewV1ClusterFeatureLogFetcherLogDownloadParamsWithContext(h.ctx).WithUID(logFetcherUid).WithFileName(&filename)
+	var buf bytes.Buffer
+	writer := io.Writer(&buf)
+	resp, err := h.Client.V1ClusterFeatureLogFetcherLogDownload(params, writer)
+	if err != nil {
+		return nil, err
+	}
+	logfile := resp.GetPayload()
+	file_location := "/tmp/" + filename
+	fo, err := os.Create(filepath.Clean(file_location))
+	if err != nil {
+		return nil, fmt.Errorf("error while creating a file %v", err)
+	}
+	_, err = buf.WriteTo(fo)
+	if err != nil {
+		return nil, fmt.Errorf("error while writing log content to a file %v", err)
+	}
+	return logfile, nil
+}