package controller

import (
	"context"

	databasev1alpha1 "git.ddupan.top/panxiao81/ayatori/api/database/v1alpha1"
	"git.ddupan.top/panxiao81/ayatori/internal/database/adapter/kubernetes"
	"git.ddupan.top/panxiao81/ayatori/internal/database/application"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/handler"
)

// CredentialReconciler 只连接事件、用例和重试，不在控制器中编排凭据写入。
type CredentialReconciler struct {
	Client client.Client
	Reader client.Reader
	Store  application.CredentialStore
}

func (r *CredentialReconciler) Reconcile(ctx context.Context, request ctrl.Request) (ctrl.Result, error) {
	service := application.CredentialPreparation{
		Resources: &kubernetes.CredentialResources{Client: r.Client, Reader: r.Reader}, Store: r.Store,
	}
	if err := service.Reconcile(ctx, request.Name); err != nil {
		return ctrl.Result{}, err
	}
	// Bao 的可用性和版本变化没有 Kubernetes watch；与已有依赖重查保持一致。
	return ctrl.Result{RequeueAfter: dependencyRetry}, nil
}

func (r *CredentialReconciler) SetupWithManager(manager ctrl.Manager) error {
	if r.Client == nil {
		r.Client = manager.GetClient()
	}
	if r.Reader == nil {
		r.Reader = manager.GetAPIReader()
	}
	return ctrl.NewControllerManagedBy(manager).
		Named("database-credentials").For(&databasev1alpha1.PostgreSQLDatabase{}).
		Watches(&databasev1alpha1.PostgreSQLTenant{}, handler.EnqueueRequestsFromMapFunc(r.requestsForTenant)).
		Watches(&databasev1alpha1.PostgreSQLInstance{}, handler.EnqueueRequestsFromMapFunc(r.requestsForInstance)).
		Complete(r)
}

func (r *CredentialReconciler) requestsForTenant(_ context.Context, object client.Object) []ctrl.Request {
	tenant := object.(*databasev1alpha1.PostgreSQLTenant)
	if tenant.Status.DatabaseRef == nil {
		return nil
	}
	return []ctrl.Request{{Name: string(tenant.Status.DatabaseRef.Name)}}
}

func (r *CredentialReconciler) requestsForInstance(ctx context.Context, object client.Object) []ctrl.Request {
	databases := &databasev1alpha1.PostgreSQLDatabaseList{}
	if err := r.Client.List(ctx, databases); err != nil {
		ctrl.LoggerFrom(ctx).Error(err, "无法映射 Instance 凭据准备事件；等待低频重试")
		return nil
	}
	var requests []ctrl.Request
	for _, database := range databases.Items {
		if string(database.Spec.InstanceRef.Name) == object.GetName() {
			requests = append(requests, ctrl.Request{NamespacedName: client.ObjectKeyFromObject(&database)})
		}
	}
	return requests
}