feat(kots): add embedded cluster distribution (#4605)

replicatedhq · May 9, 2024 · 6462a77 · 6462a77
1 parent de315f2
commit 6462a77
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/pkg/reporting/distribution.go b/pkg/reporting/distribution.go
@@ -93,6 +93,9 @@ func distributionFromLabels(clientset kubernetes.Interface) Distribution {
 				// Based on: https://docs.oracle.com/en-us/iaas/Content/ContEng/Reference/contengsupportedlabelsusecases.htm
 				return OKE
 			}
+			if k == "kots.io/embedded-cluster-role" {
+				return EmbeddedCluster
+			}
 		}
 	}
 	return UnknownDistribution

diff --git a/pkg/reporting/distribution_test.go b/pkg/reporting/distribution_test.go
@@ -162,6 +162,23 @@ func TestGetDistribution(t *testing.T) {
 			},
 			want: Minikube,
 		},
+		{
+			name: "EmbeddedCluster from labels",
+			args: args{
+				clientset: mockClientsetForDistribution(&mockClientsetForDistributionOpts{
+					objects: []runtime.Object{
+						&corev1.Node{
+							ObjectMeta: metav1.ObjectMeta{
+								Labels: map[string]string{
+									"kots.io/embedded-cluster-role": "foo",
+								},
+							},
+						},
+					},
+				}),
+			},
+			want: EmbeddedCluster,
+		},
 		{
 			name: "OKE from labels",
 			args: args{

diff --git a/pkg/reporting/types.go b/pkg/reporting/types.go
@@ -26,6 +26,7 @@ const (
 	OpenShift
 	RKE2
 	Tanzu
+	EmbeddedCluster
 )
 
 type Reporter interface {
@@ -79,6 +80,8 @@ func (d Distribution) String() string {
 		return "rke2"
 	case Tanzu:
 		return "tanzu"
+	case EmbeddedCluster:
+		return "embedded-cluster"
 	}
 	return "unknown"
 }