feat: enables remote custom analyzers (#906)

* feat: enables remote custom analyzers Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: fixed test that was broken Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: hiding custom analysis behind a flag Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: resolved govet issue Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: updated Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: updated Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: updated deps Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: updated deps Signed-off-by: Alex Jones <alexsimonjones@gmail.com> --------- Signed-off-by: Alex Jones <alexsimonjones@gmail.com>
2025-09-02 07:46:15 +00:00 · 2024-02-15 07:28:00 +00:00
parent 070aa7fdd0
commit c8c9dbfadc
5 changed files with 127 additions and 2 deletions
--- a/pkg/custom/client.go
+++ b/pkg/custom/client.go
@@ -0,0 +1,57 @@
+package custom
+
+import (
+	rpc "buf.build/gen/go/k8sgpt-ai/k8sgpt/grpc/go/schema/v1/schemav1grpc"
+	schemav1 "buf.build/gen/go/k8sgpt-ai/k8sgpt/protocolbuffers/go/schema/v1"
+	"context"
+	"fmt"
+	"github.com/k8sgpt-ai/k8sgpt/pkg/common"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+type Client struct {
+	c              *grpc.ClientConn
+	analyzerClient rpc.AnalyzerServiceClient
+}
+
+func NewClient(c Connection) (*Client, error) {
+
+	conn, err := grpc.Dial(fmt.Sprintf("%s:%s", c.Url, c.Port), grpc.WithTransportCredentials(insecure.NewCredentials()))
+
+	if err != nil {
+		return nil, err
+	}
+	client := rpc.NewAnalyzerServiceClient(conn)
+	return &Client{
+		c:              conn,
+		analyzerClient: client,
+	}, nil
+}
+
+func (cli *Client) Run() (common.Result, error) {
+	var result common.Result
+	req := &schemav1.AnalyzerRunRequest{}
+	res, err := cli.analyzerClient.Run(context.Background(), req)
+	if err != nil {
+		return result, err
+	}
+	if res.Result != nil {
+
+		// We should refactor this, because Error and Failure do not map 1:1 from K8sGPT/schema
+		var errorsFound []common.Failure
+		for _, e := range res.Result.Error {
+			errorsFound = append(errorsFound, common.Failure{
+				Text: e.Text,
+				// TODO: Support sensitive data
+			})
+		}
+
+		result.Name = res.Result.Name
+		result.Kind = res.Result.Kind
+		result.Details = res.Result.Details
+		result.ParentObject = res.Result.ParentObject
+		result.Error = errorsFound
+	}
+	return result, nil
+}