clusterprofile_controller.go

/*
Copyright 2024. projectsveltos.io. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package controller

import (
	"context"

	"github.com/go-logr/logr"
	"github.com/pkg/errors"
	corev1 "k8s.io/api/core/v1"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	"k8s.io/apimachinery/pkg/runtime"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/controller"
	"sigs.k8s.io/controller-runtime/pkg/reconcile"

	configv1beta1 "github.com/projectsveltos/addon-controller/api/v1beta1"
	logs "github.com/projectsveltos/libsveltos/lib/logsettings"
	libsveltosset "github.com/projectsveltos/libsveltos/lib/set"
	"github.com/projectsveltos/ui-backend/internal/server"
)

// ClusterProfileReconciler reconciles a ClusterProfile object
type ClusterProfileReconciler struct {
	client.Client
	Scheme               *runtime.Scheme
	ConcurrentReconciles int
}

//+kubebuilder:rbac:groups=config.projectsveltos.io,resources=clusterprofiles,verbs=get;list;watch
//+kubebuilder:rbac:groups=config.projectsveltos.io,resources=clusterprofiles/status,verbs=get;list;watch

func (r *ClusterProfileReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	logger := ctrl.LoggerFrom(ctx)
	logger.V(logs.LogInfo).Info("Reconciling")

	// Fecth the ClusterProfile instance
	cp := &configv1beta1.ClusterProfile{}
	if err := r.Get(ctx, req.NamespacedName, cp); err != nil {
		if apierrors.IsNotFound(err) {
			r.removeClusterProfile(req.Name, logger)
			return reconcile.Result{}, nil
		}
		logger.Error(err, "Failed to fetch ClusterProfile")
		return reconcile.Result{}, errors.Wrapf(
			err,
			"Failed to fetch ClusterProfile %s",
			req.NamespacedName,
		)
	}

	// Handle deleted clusterProfile
	if !cp.DeletionTimestamp.IsZero() {
		r.removeClusterProfile(cp.Name, logger)
	} else {
		// Handle non-deleted clusterProfile
		r.reconcileNormal(cp, logger)
	}

	return reconcile.Result{}, nil
}

func (r *ClusterProfileReconciler) removeClusterProfile(clusterProfileName string, logger logr.Logger) {
	logger.V(logs.LogInfo).Info("Reconciling ClusterProfile delete")

	manager := server.GetManagerInstance()
	if manager == nil {
		logger.V(logs.LogInfo).Info("Manager not initialized yet, skipping")
		return
	}

	profileRef := &corev1.ObjectReference{
		Kind:       configv1beta1.ClusterProfileKind,
		APIVersion: configv1beta1.GroupVersion.String(),
		Name:       clusterProfileName,
	}

	manager.RemoveProfile(profileRef)

	logger.V(logs.LogInfo).Info("Reconcile delete success")
}

func (r *ClusterProfileReconciler) reconcileNormal(clusterProfile *configv1beta1.ClusterProfile, logger logr.Logger) {
	logger.V(logs.LogInfo).Info("Reconciling ClusterProfile normal")

	manager := server.GetManagerInstance()
	if manager == nil {
		logger.V(logs.LogInfo).Info("Manager not initialized yet, skipping")
		return
	}

	profileRef := &corev1.ObjectReference{
		Kind:       configv1beta1.ClusterProfileKind,
		APIVersion: configv1beta1.GroupVersion.String(),
		Name:       clusterProfile.Name,
	}

	dependencies := &libsveltosset.Set{}
	for i := range clusterProfile.Spec.DependsOn {
		dependencies.Insert(&corev1.ObjectReference{
			Kind:       configv1beta1.ClusterProfileKind,
			APIVersion: configv1beta1.GroupVersion.String(),
			Name:       clusterProfile.Spec.DependsOn[i],
		})
	}

	manager.AddProfile(profileRef, clusterProfile.Spec.ClusterSelector, clusterProfile.Spec.Tier, dependencies)

	logger.V(logs.LogInfo).Info("Reconcile normal success")
}

// SetupWithManager sets up the controller with the Manager.
func (r *ClusterProfileReconciler) SetupWithManager(mgr ctrl.Manager) error {
	return ctrl.NewControllerManagedBy(mgr).
		For(&configv1beta1.ClusterProfile{}).
		WithOptions(controller.Options{
			MaxConcurrentReconciles: r.ConcurrentReconciles,
		}).
		Complete(r)
}