Merge pull request #618 from replicatedhq/divolgin/sc-51321/kots-does-not-automatically-re-write-private

Adding some utility interfaces for collectors

Author: divolgin

pkg/apis/troubleshoot/v1beta2/collector_interfaces.go

@@ -0,0 +1,105 @@
+package v1beta2
+
+import (
+	corev1 "k8s.io/api/core/v1"
+)
+
+func (c *Run) GetImage() string {
+	return c.Image
+}
+
+func (c *Run) SetImage(image string) {
+	c.Image = image
+}
+
+func (c *Run) GetImagePullSecret() *ImagePullSecrets {
+	return c.ImagePullSecret
+}
+
+func (c *Run) SetImagePullSecret(secrets *ImagePullSecrets) {
+	c.ImagePullSecret = secrets
+}
+
+func (c *Run) GetNamespace() string {
+	return c.Namespace
+}
+
+func (c *CopyFromHost) GetImage() string {
+	return c.Image
+}
+
+func (c *CopyFromHost) SetImage(image string) {
+	c.Image = image
+}
+
+func (c *CopyFromHost) GetImagePullSecret() *ImagePullSecrets {
+	return c.ImagePullSecret
+}
+
+func (c *CopyFromHost) SetImagePullSecret(secrets *ImagePullSecrets) {
+	c.ImagePullSecret = secrets
+}
+
+func (c *CopyFromHost) GetNamespace() string {
+	return c.Namespace
+}
+
+func (c *Sysctl) GetImage() string {
+	return c.Image
+}
+
+func (c *Sysctl) SetImage(image string) {
+	c.Image = image
+}
+
+func (c *Sysctl) GetImagePullSecret() *ImagePullSecrets {
+	return c.ImagePullSecret
+}
+
+func (c *Sysctl) SetImagePullSecret(secrets *ImagePullSecrets) {
+	c.ImagePullSecret = secrets
+}
+
+func (c *Sysctl) GetNamespace() string {
+	return c.Namespace
+}
+
+func (c *Collectd) GetImage() string {
+	return c.Image
+}
+
+func (c *Collectd) SetImage(image string) {
+	c.Image = image
+}
+
+func (c *Collectd) GetImagePullSecret() *ImagePullSecrets {
+	return c.ImagePullSecret
+}
+
+func (c *Collectd) SetImagePullSecret(secrets *ImagePullSecrets) {
+	c.ImagePullSecret = secrets
+}
+
+func (c *Collectd) GetNamespace() string {
+	return c.Namespace
+}
+
+func (c *RunPod) GetPodSpec() corev1.PodSpec {
+	return c.PodSpec
+}
+
+func (c *RunPod) SetPodSpec(podSpec corev1.PodSpec) {
+	c.PodSpec = podSpec
+}
+
+func (c *RunPod) GetImagePullSecret() *ImagePullSecrets {
+	return c.ImagePullSecret
+}
+
+func (c *RunPod) SetImagePullSecret(secrets *ImagePullSecrets) {
+	c.ImagePullSecret = secrets
+}
+
+func (c *RunPod) GetNamespace() string {
+	return c.Namespace
+}

pkg/apis/troubleshoot/v1beta2/collector_shared.go

@@ -2,6 +2,7 @@ package v1beta2
 
 import (
 	"fmt"
+	"reflect"
 	"strings"
 
 	"github.com/replicatedhq/troubleshoot/pkg/multitype"
@@ -538,3 +539,20 @@ func pickNamespaceOrDefault(collectorNS string, overrideNS string) string {
 	}
 	return "default"
 }
+
+func GetCollector(collector *Collect) interface{} {
+	if collector == nil {
+		return nil
+	}
+
+	reflected := reflect.ValueOf(collector).Elem()
+	for i := 0; i < reflected.NumField(); i++ {
+		if reflected.Field(i).IsNil() {
+			continue
+		}
+
+		return reflect.Indirect(reflected.Field(i)).Addr().Interface()
+	}
+
+	return nil
+}

pkg/apis/troubleshoot/v1beta2/zz_generated.deepcopy.go

@@ -0,0 +1,33 @@
+package collect
+
+import (
+	"github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta2"
+	corev1 "k8s.io/api/core/v1"
+)
+
+type ImageRunner interface {
+	GetImage() string
+	SetImage(string)
+
+	GetImagePullSecret() *v1beta2.ImagePullSecrets
+	SetImagePullSecret(*v1beta2.ImagePullSecrets)
+
+	GetNamespace() string
+}
+
+var _ ImageRunner = &v1beta2.Run{}
+var _ ImageRunner = &v1beta2.CopyFromHost{}
+var _ ImageRunner = &v1beta2.Sysctl{}
+var _ ImageRunner = &v1beta2.Collectd{}
+
+type PodSpecRunner interface {
+	GetPodSpec() corev1.PodSpec
+	SetPodSpec(corev1.PodSpec)
+
+	GetImagePullSecret() *v1beta2.ImagePullSecrets
+	SetImagePullSecret(*v1beta2.ImagePullSecrets)
+
+	GetNamespace() string
+}
+
+var _ PodSpecRunner = &v1beta2.RunPod{}

pkg/collect/interfaces.go

@@ -0,0 +1,57 @@
+package collect
+
+import (
+	"testing"
+
+	"github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta2"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_interfaces(t *testing.T) {
+	runCollector := &v1beta2.Run{}
+	clusterInfoCollector := &v1beta2.ClusterInfo{}
+
+	tests := []struct {
+		name       string
+		collect    *v1beta2.Collect
+		want       interface{}
+		wantRunner bool
+	}{
+		{
+			name: "image runner collector",
+			collect: &v1beta2.Collect{
+				Run: runCollector,
+			},
+			want:       runCollector,
+			wantRunner: true,
+		},
+		{
+			name: "not image runner collector",
+			collect: &v1beta2.Collect{
+				ClusterInfo: clusterInfoCollector,
+			},
+			want:       clusterInfoCollector,
+			wantRunner: false,
+		},
+		{
+			name:       "no collector",
+			collect:    &v1beta2.Collect{},
+			want:       nil,
+			wantRunner: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			req := require.New(t)
+
+			got := v1beta2.GetCollector(tt.collect)
+			req.EqualValues(tt.want, got)
+
+			runner, ok := got.(ImageRunner)
+			req.EqualValues(tt.wantRunner, ok)
+			if tt.wantRunner {
+				req.EqualValues(tt.want, runner)
+			}
+		})
+	}
+}