feat: add backstage support

api/v1alpha1/k8sgpt_types.go

@@ -35,13 +35,14 @@ type K8sGPTSpec struct {
 	Backend `json:"backend"`
 	BaseUrl string `json:"baseUrl,omitempty"`
 	// +kubebuilder:default:=gpt-3.5-turbo
-	Model    string     `json:"model,omitempty"`
-	Engine   string     `json:"engine,omitempty"`
-	Secret   *SecretRef `json:"secret,omitempty"`
-	Version  string     `json:"version,omitempty"`
-	EnableAI bool       `json:"enableAI,omitempty"`
-	NoCache  bool       `json:"noCache,omitempty"`
-	Filters  []string   `json:"filters,omitempty"`
+	Model     string     `json:"model,omitempty"`
+	Engine    string     `json:"engine,omitempty"`
+	Secret    *SecretRef `json:"secret,omitempty"`
+	Version   string     `json:"version,omitempty"`
+	EnableAI  bool       `json:"enableAI,omitempty"`
+	NoCache   bool       `json:"noCache,omitempty"`
+	Filters   []string   `json:"filters,omitempty"`
+	Backstage bool       `json:"backstage,omitempty"`
 }
 
 type Backend string

config/crd/bases/core.k8sgpt.ai_k8sgpts.yaml

@@ -42,6 +42,8 @@ spec:
                 - localai
                 - azureopenai
                 type: string
+              backstage:
+                type: boolean
               baseUrl:
                 type: string
               enableAI:

controllers/k8sgpt_controller.go

@@ -23,6 +23,10 @@ import (
 	"time"
 
 	corev1alpha1 "github.com/k8sgpt-ai/k8sgpt-operator/api/v1alpha1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/cli-runtime/pkg/genericclioptions"
+	cmdutil "k8s.io/kubectl/pkg/cmd/util"
+
 	kclient "github.com/k8sgpt-ai/k8sgpt-operator/pkg/client"
 	"github.com/k8sgpt-ai/k8sgpt-operator/pkg/resources"
 	"github.com/k8sgpt-ai/k8sgpt-operator/pkg/utils"
@@ -31,6 +35,7 @@ import (
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 	"k8s.io/apimachinery/pkg/runtime"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -159,7 +164,7 @@ func (r *K8sGPTReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctr
 			// Create a new client
 			var address string
 			if os.Getenv("LOCAL_MODE") != "" {
-				address = "localhost:8080"
+				address = "localhost:8082"
 			} else {
 				// Get k8sgpt-deployment service pod ip
 				podList := &corev1.PodList{}
@@ -176,7 +181,7 @@ func (r *K8sGPTReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctr
 					k8sgptReconcileErrorCount.Inc()
 					return r.finishReconcile(fmt.Errorf("no pods found for k8sgpt-deployment"), false)
 				}
-				address = fmt.Sprintf("%s:8080", podList.Items[0].Status.PodIP)
+				address = fmt.Sprintf("%s:8082", podList.Items[0].Status.PodIP)
 			}
 
 			fmt.Printf("Creating new client for %s\n", address)
@@ -218,6 +223,52 @@ func (r *K8sGPTReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctr
 					Namespace: k8sgptConfig.Namespace,
 				},
 			}
+			if k8sgptConfig.Spec.Backstage {
+				labelKey := "backstage.io/kubernetes-id"
+				namespace, resourceName, _ := strings.Cut(resultSpec.Name, "/")
+				m, err := cmdutil.NewFactory(genericclioptions.NewConfigFlags(true)).ToRESTMapper()
+				if err != nil {
+					k8sgptReconcileErrorCount.Inc()
+					return r.finishReconcile(err, false)
+				}
+
+				gvr, err := m.ResourceFor(schema.GroupVersionResource{
+					Resource: resultSpec.Kind,
+				})
+				if err != nil {
+					k8sgptReconcileErrorCount.Inc()
+					return r.finishReconcile(err, false)
+				}
+
+				obj := &unstructured.Unstructured{}
+				gvk := schema.GroupVersionKind{
+					Group:   gvr.Group,
+					Kind:    resultSpec.Kind,
+					Version: gvr.Version,
+				}
+
+				obj.SetGroupVersionKind(gvk)
+
+				// Retrieve the resource using the client
+				err = r.Client.Get(ctx, client.ObjectKey{Name: resourceName, Namespace: namespace}, obj)
+				if err != nil {
+					if errors.IsNotFound(err) {
+						k8sgptReconcileErrorCount.Inc()
+						return r.finishReconcile(err, false)
+					} else {
+						k8sgptReconcileErrorCount.Inc()
+						return r.finishReconcile(err, false)
+					}
+				}
+				labels := obj.GetLabels()
+				if value, exists := labels[labelKey]; exists {
+					// Assign the same label key/value to result CR
+					result.ObjectMeta.Labels = map[string]string{labelKey: value}
+				} else {
+					// too verbose?
+					fmt.Printf("Label key '%s' does not exist in %s resource: %s\n", labelKey, resultSpec.Kind, obj.GetName())
+				}
+			}
 			rawResults[name] = result
 		}
 
@@ -256,7 +307,7 @@ func (r *K8sGPTReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctr
 			err = r.Get(ctx, client.ObjectKey{Namespace: k8sgptConfig.Namespace,
 				Name: result.Name}, &existingResult)
 			if err != nil {
-				// if the result already exists, we will update it
+				// if the result doesn't exist, we will create it
 				if errors.IsNotFound(err) {
 					err = r.Create(ctx, &result)
 					if err != nil {
@@ -275,6 +326,7 @@ func (r *K8sGPTReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctr
 			} else {
 				// If the result already exists we will update it
 				existingResult.Spec = result.Spec
+				existingResult.Labels = result.Labels
 				err = r.Update(ctx, &existingResult)
 				if err != nil {
 					k8sgptReconcileErrorCount.Inc()

go.mod

@@ -11,7 +11,9 @@ require (
 	google.golang.org/grpc v1.55.0
 	k8s.io/api v0.27.2
 	k8s.io/apimachinery v0.27.2
+	k8s.io/cli-runtime v0.27.2
 	k8s.io/client-go v0.27.2
+	k8s.io/kubectl v0.27.2
 	sigs.k8s.io/controller-runtime v0.15.0
 )
 
@@ -23,13 +25,18 @@ require (
 )
 
 require (
+	github.com/Azure/go-ansiterm v0.0.0-20210617225240-d185dfc1b5a1 // indirect
+	github.com/MakeNowJust/heredoc v1.0.0 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.2.0 // indirect
+	github.com/chai2010/gettext-go v1.0.2 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/emicklei/go-restful/v3 v3.10.2 // indirect
 	github.com/evanphx/json-patch v5.6.0+incompatible // indirect
 	github.com/evanphx/json-patch/v5 v5.6.0 // indirect
+	github.com/exponent-io/jsonpath v0.0.0-20151013193312-d6023ce2651d // indirect
 	github.com/fsnotify/fsnotify v1.6.0 // indirect
+	github.com/go-errors/errors v1.4.2 // indirect
 	github.com/go-logr/logr v1.2.4 // indirect
 	github.com/go-logr/zapr v1.2.4 // indirect
 	github.com/go-openapi/jsonpointer v0.19.6 // indirect
@@ -38,23 +45,37 @@ require (
 	github.com/gogo/protobuf v1.3.2 // indirect
 	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
 	github.com/golang/protobuf v1.5.3 // indirect
+	github.com/google/btree v1.0.1 // indirect
 	github.com/google/gnostic v0.6.9 // indirect
 	github.com/google/go-cmp v0.5.9 // indirect
 	github.com/google/gofuzz v1.2.0 // indirect
+	github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 // indirect
 	github.com/google/uuid v1.3.0 // indirect
+	github.com/gregjones/httpcache v0.0.0-20180305231024-9cad4c3443a7 // indirect
 	github.com/imdario/mergo v0.3.15 // indirect
+	github.com/inconshreveable/mousetrap v1.0.1 // indirect
 	github.com/josharian/intern v1.0.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/liggitt/tabwriter v0.0.0-20181228230101-89fcab3d43de // indirect
 	github.com/mailru/easyjson v0.7.7 // indirect
 	github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
+	github.com/mitchellh/go-wordwrap v1.0.0 // indirect
+	github.com/moby/spdystream v0.2.0 // indirect
+	github.com/moby/term v0.0.0-20221205130635-1aeaba878587 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/monochromegane/go-gitignore v0.0.0-20200626010858-205db1a8cc00 // indirect
 	github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
+	github.com/peterbourgon/diskv v2.0.1+incompatible // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/prometheus/client_model v0.4.0 // indirect
 	github.com/prometheus/common v0.42.0 // indirect
 	github.com/prometheus/procfs v0.9.0 // indirect
+	github.com/russross/blackfriday/v2 v2.1.0 // indirect
+	github.com/spf13/cobra v1.6.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/xlab/treeprint v1.1.0 // indirect
+	go.starlark.net v0.0.0-20200306205701-8dd3e2ee1dd5 // indirect
 	go.uber.org/atomic v1.10.0 // indirect
 	go.uber.org/multierr v1.10.0 // indirect
 	go.uber.org/zap v1.24.0 // indirect
@@ -77,6 +98,8 @@ require (
 	k8s.io/kube-openapi v0.0.0-20230501164219-8b0f38b5fd1f // indirect
 	k8s.io/utils v0.0.0-20230313181309-38a27ef9d749 // indirect
 	sigs.k8s.io/json v0.0.0-20221116044647-bc3834ca7abd // indirect
+	sigs.k8s.io/kustomize/api v0.13.2 // indirect
+	sigs.k8s.io/kustomize/kyaml v0.14.1 // indirect
 	sigs.k8s.io/structured-merge-diff/v4 v4.2.3 // indirect
 	sigs.k8s.io/yaml v1.3.0 // indirect
 )