package controller

import (
	"context"

	schedulev1 "github.com/baschno/tdset-operator/api/v1"
	"k8s.io/apimachinery/pkg/api/meta"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/log"
)

// ConditionStatus defines TDSet condition status.
type ConditionStatus string

// Defines TDSet condition status.
const (
	TypeAvailable   ConditionStatus = "Available"
	TypeProgressing ConditionStatus = "Progressing"
	TypeDegraded    ConditionStatus = "Degraded"
)

// GetTDSet gets the TDSet from api server.
func (r *TDSetReconciler) GetTDSet(ctx context.Context, req ctrl.Request, tdSet *schedulev1.TDSet) error {

	err := r.Get(ctx, req.NamespacedName, tdSet)
	if err != nil {
		return err
	}
	return nil
}

// SetInitialCondition sets the status condition of the TDSet to available initially
// when no condition exists yet.
func (r *TDSetReconciler) SetInitialCondition(ctx context.Context, req ctrl.Request, tdSet *schedulev1.TDSet) error {
	if tdSet.Status.Conditions != nil || len(tdSet.Status.Conditions) != 0 {
		return nil
	}

	err := r.SetCondition(ctx, req, tdSet, TypeAvailable, "Starting reconciliation")

	return err
}

func (r *TDSetReconciler) SetCondition(ctx context.Context, req ctrl.Request,
	tdSet *schedulev1.TDSet, condition ConditionStatus, message string) error {

	log := log.FromContext(ctx)

	meta.SetStatusCondition(
		&tdSet.Status.Conditions,
		metav1.Condition{
			Type: string(condition),
			Status: metav1.ConditionUnknown, Reason: "Reconciling",
			Message: message,
		}
	)

	if err := r.Status().Update(ctx, tdSet); err != nil {
		log.Error(err, "failed to update TDSet status condition")
		return err
	}
	
	if err := r.Get(ctx, req.NamespacedName, tdSet); err != nil {
		log.Error(err, "failed to get updated TDSet after setting condition")
		return err
	}

	return nil
}