Capi self filtering

api/v1alpha1/automatedclusterdiscovery_types.go

@@ -28,6 +28,12 @@ type AKS struct {
 	SubscriptionID string `json:"subscriptionID"`
 }
 
+type Cluster struct {
+	// Name is the name of the cluster
+	// +required
+	Name string `json:"name"`
+}
+
 // AutomatedClusterDiscoverySpec defines the desired state of AutomatedClusterDiscovery
 type AutomatedClusterDiscoverySpec struct {
 	// Name is the name of the cluster
@@ -58,6 +64,10 @@ type AutomatedClusterDiscoverySpec struct {
 	CommonLabels map[string]string `json:"commonLabels,omitempty"`
 	// Annotations to add to all generated resources.
 	CommonAnnotations map[string]string `json:"commonAnnotations,omitempty"`
+
+	// Current Cluster name indicating the management cluster
+	// used to avoid choosing the cluster the controller is running in
+	CurrentClusterRef Cluster `json:"currentClusterRef,omitempty"`
 }
 
 // AutomatedClusterDiscoveryStatus defines the observed state of AutomatedClusterDiscovery

config/crd/bases/clusters.weave.works_automatedclusterdiscoveries.yaml

@@ -44,7 +44,7 @@ spec:
               AutomatedClusterDiscovery
             properties:
               aks:
-                description: AKS defines the desired state of AKS
+                description: AKS configures discovery of AKS clusters from Azure.
                 properties:
                   subscriptionID:
                     description: SubscriptionID is the Azure subscription ID
@@ -62,6 +62,16 @@ spec:
                   type: string
                 description: Labels to add to all generated resources.
                 type: object
+              currentClusterRef:
+                description: Current Cluster name indicating the management cluster
+                  used to avoid choosing the cluster the controller is running in
+                properties:
+                  name:
+                    description: Name is the name of the cluster
+                    type: string
+                required:
+                - name
+                type: object
               disableTags:
                 description: If DisableTags is true, labels will not be applied to
                   the generated Clusters from the tags on the upstream Clusters.

config/samples/capi_discovery.yaml

@@ -6,3 +6,5 @@ metadata:
 spec:
   type: capi
   interval: 10m
+  currentClusterRef:
+    name: management-cluster

internal/controller/automatedclusterdiscovery_controller.go

@@ -55,7 +55,7 @@ type AutomatedClusterDiscoveryReconciler struct {
 	EventRecorder eventRecorder
 
 	AKSProvider  func(string) providers.Provider
-	CAPIProvider func(client.Client, string) providers.Provider
+	CAPIProvider func(client.Client, string, *clustersv1alpha1.Cluster) providers.Provider
 }
 
 // event emits a Kubernetes event and forwards the event to the event recorder
@@ -179,7 +179,8 @@ func (r *AutomatedClusterDiscoveryReconciler) reconcileResources(ctx context.Con
 			"name", cd.Spec.Name,
 		)
 
-		capiProvider := r.CAPIProvider(r.Client, cd.Namespace)
+		capiProvider := r.CAPIProvider(r.Client, cd.Namespace, &cd.Spec.CurrentClusterRef)
+		clusterID = cd.Spec.CurrentClusterRef.Name
 
 		clusters, err = capiProvider.ListClusters(ctx)
 		if err != nil {

internal/controller/automatedclusterdiscovery_controller_test.go

@@ -942,7 +942,7 @@ func TestAutomatedClusterDiscoveryReconciler(t *testing.T) {
 		reconciler := &AutomatedClusterDiscoveryReconciler{
 			Client: k8sClient,
 			Scheme: scheme,
-			CAPIProvider: func(capiclient client.Client, namespace string) providers.Provider {
+			CAPIProvider: func(capiclient client.Client, namespace string, managementClusterRef *clustersv1alpha1.Cluster) providers.Provider {
 				return &testProvider
 			},
 			EventRecorder: &mockEventRecorder{},

main.go

@@ -114,8 +114,8 @@ func main() {
 		AKSProvider: func(subscriptionID string) providers.Provider {
 			return azure.NewAzureProvider(subscriptionID)
 		},
-		CAPIProvider: func(kubeclient client.Client, namespace string) providers.Provider {
-			return capi.NewCAPIProvider(kubeclient, namespace)
+		CAPIProvider: func(kubeclient client.Client, namespace string, managementClusterRef *clustersv1alpha1.Cluster) providers.Provider {
+			return capi.NewCAPIProvider(kubeclient, namespace, managementClusterRef)
 		},
 	}).SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "AutomatedClusterDiscovery")

pkg/providers/capi/capi.go

@@ -3,23 +3,26 @@ package capi
 import (
 	"context"
 
+	clustersv1alpha1 "github.com/weaveworks/cluster-reflector-controller/api/v1alpha1"
 	"github.com/weaveworks/cluster-reflector-controller/pkg/providers"
 	capiclusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
 type CAPIProvider struct {
-	Kubeclient client.Client
-	Namespace  string
+	Kubeclient           client.Client
+	Namespace            string
+	ManagementClusterRef *clustersv1alpha1.Cluster
 }
 
 var _ providers.Provider = (*CAPIProvider)(nil)
 
 // NewCAPIProvider creates and returns a CAPIProvider ready for use
-func NewCAPIProvider(client client.Client, namespace string) *CAPIProvider {
+func NewCAPIProvider(client client.Client, namespace string, managementClusterRef *clustersv1alpha1.Cluster) *CAPIProvider {
 	provider := &CAPIProvider{
-		Kubeclient: client,
-		Namespace:  namespace,
+		Kubeclient:           client,
+		Namespace:            namespace,
+		ManagementClusterRef: managementClusterRef,
 	}
 	return provider
 }
@@ -37,7 +40,7 @@ func (p *CAPIProvider) ListClusters(ctx context.Context) ([]*providers.ProviderC
 	for _, capiCluster := range capiClusters.Items {
 		clusters = append(clusters, &providers.ProviderCluster{
 			Name:       capiCluster.Name,
-			ID:         string(capiCluster.GetObjectMeta().GetUID()),
+			ID:         capiCluster.Name,
 			KubeConfig: nil,
 			Labels:     capiCluster.Labels,
 		})
@@ -49,5 +52,5 @@ func (p *CAPIProvider) ListClusters(ctx context.Context) ([]*providers.ProviderC
 // ProviderCluster has an ID to identify the cluster, but capi cluster doesn't have a Cluster ID
 // therefore wont't match in the case of CAPI
 func (p *CAPIProvider) ClusterID(ctx context.Context, kubeClient client.Reader) (string, error) {
-	return "", nil
+	return p.ManagementClusterRef.Name, nil
 }

pkg/providers/capi/capi_test.go

@@ -11,6 +11,7 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/client/fake"
 
+	clustersv1alpha1 "github.com/weaveworks/cluster-reflector-controller/api/v1alpha1"
 	"github.com/weaveworks/cluster-reflector-controller/pkg/providers"
 )
 
@@ -29,7 +30,7 @@ func TestClusterProvider_ListClusters(t *testing.T) {
 	}
 
 	client := fake.NewClientBuilder().WithScheme(scheme).WithObjects(clusters...).Build()
-	provider := NewCAPIProvider(client, "default")
+	provider := NewCAPIProvider(client, "default", &clustersv1alpha1.Cluster{Name: "management-cluster"})
 
 	provided, err := provider.ListClusters(context.TODO())
 	if err != nil {
@@ -39,6 +40,7 @@ func TestClusterProvider_ListClusters(t *testing.T) {
 	expected := []*providers.ProviderCluster{
 		{
 			Name:       "cluster-1",
+			ID:         "cluster-1",
 			KubeConfig: nil,
 		},
 	}