package controller

import (
	"context"

	"k8s.io/apimachinery/pkg/runtime"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/log"

	forgebotv1alpha1 "git.unkin.net/unkin/forgebot/api/v1alpha1"
)

type AgentPoolReconciler struct {
	client.Client
	Scheme *runtime.Scheme
}

// +kubebuilder:rbac:groups=forgebot.unkin.net,resources=agentpools,verbs=get;list;watch;create;update;patch;delete
// +kubebuilder:rbac:groups=forgebot.unkin.net,resources=agentpools/status,verbs=get;update;patch

func (r *AgentPoolReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	logger := log.FromContext(ctx)

	var pool forgebotv1alpha1.AgentPool
	if err := r.Get(ctx, req.NamespacedName, &pool); err != nil {
		return ctrl.Result{}, client.IgnoreNotFound(err)
	}

	var taskList forgebotv1alpha1.AgentTaskList
	if err := r.List(ctx, &taskList, client.InNamespace(req.Namespace)); err != nil {
		return ctrl.Result{}, err
	}

	active := 0
	for _, task := range taskList.Items {
		if task.Spec.PoolRef == pool.Name && task.Status.Phase == forgebotv1alpha1.TaskRunning {
			active++
		}
	}

	if pool.Status.ActiveJobs != active {
		pool.Status.ActiveJobs = active
		if err := r.Status().Update(ctx, &pool); err != nil {
			return ctrl.Result{}, err
		}
		logger.Info("updated pool status", "pool", pool.Name, "active", active)
	}

	return ctrl.Result{}, nil
}

func (r *AgentPoolReconciler) SetupWithManager(mgr ctrl.Manager) error {
	return ctrl.NewControllerManagedBy(mgr).
		For(&forgebotv1alpha1.AgentPool{}).
		Complete(r)
}