Changing the logic for adding clusterrolelabel

Signed-off-by: Feny Mehta <[email protected]>

controllers/toolchaincluster/toolchaincluster_controller.go

@@ -16,13 +16,14 @@ import (
 )
 
 // NewReconciler returns a new Reconciler
-func NewReconciler(mgr manager.Manager, namespace string, timeout time.Duration) *Reconciler {
+func NewReconciler(mgr manager.Manager, namespace string, timeout time.Duration, useClusterRL bool) *Reconciler {
 	cacheLog := log.Log.WithName("toolchaincluster_cache")
 	clusterCacheService := cluster.NewToolchainClusterService(mgr.GetClient(), cacheLog, namespace, timeout)
 	return &Reconciler{
 		client:              mgr.GetClient(),
 		scheme:              mgr.GetScheme(),
 		clusterCacheService: clusterCacheService,
+		useClusterRL:        useClusterRL,
 	}
 }
 
@@ -38,6 +39,7 @@ type Reconciler struct {
 	client              client.Client
 	scheme              *runtime.Scheme
 	clusterCacheService cluster.ToolchainClusterService
+	useClusterRL        bool
 }
 
 // Reconcile reads that state of the cluster for a ToolchainCluster object and makes changes based on the state read
@@ -61,5 +63,29 @@ func (r *Reconciler) Reconcile(ctx context.Context, request ctrl.Request) (ctrl.
 		return reconcile.Result{}, err
 	}
 
+	// add toolchaincluster role label if not present
+	reqLogger.Info("adding cluster role label based on type")
+	if err := r.addToolchainClusterRoleLabelFromType(ctx, toolchainCluster); err != nil {
+		return reconcile.Result{}, err
+	}
 	return reconcile.Result{}, r.clusterCacheService.AddOrUpdateToolchainCluster(toolchainCluster)
 }
+
+func (r *Reconciler) addToolchainClusterRoleLabelFromType(ctx context.Context, toolchainCluster *toolchainv1alpha1.ToolchainCluster) error {
+	logger := log.FromContext(ctx)
+
+	if !r.useClusterRL {
+		logger.Info("Skiping the cluster role label setting as not running in host cluster")
+		return nil
+	}
+	clusterRoleLabel := cluster.RoleLabel(cluster.Tenant)
+	if _, exists := toolchainCluster.Labels[clusterRoleLabel]; !exists {
+		logger.Info("setting cluster role label for toolchaincluster", clusterRoleLabel, toolchainCluster.Name)
+		// We use only the label key, the value can remain empty.
+		toolchainCluster.Labels[clusterRoleLabel] = ""
+		if err := r.client.Update(ctx, toolchainCluster); err != nil {
+			return err
+		}
+	}
+	return nil
+}