package v1

import (
	"context"
	"fmt"

	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/types"
	"k8s.io/apimachinery/pkg/util/validation/field"
	ctrl "sigs.k8s.io/controller-runtime"
	logf "sigs.k8s.io/controller-runtime/pkg/log"
	"sigs.k8s.io/controller-runtime/pkg/webhook"
	"sigs.k8s.io/controller-runtime/pkg/webhook/admission"
)

// log is for logging in this package.
var dwworkerlog = logf.Log.WithName("dwworker-resource")

func (r *DWWorker) setupWebhookWithManager(mgr ctrl.Manager) error {
	return ctrl.NewWebhookManagedBy(mgr).
		For(r).
		Complete()
}

// TODO(user): EDIT THIS FILE!  THIS IS SCAFFOLDING FOR YOU TO OWN!

//+kubebuilder:webhook:path=/mutate-cloud-pingcap-com-v1-dwworker,mutating=true,failurePolicy=fail,sideEffects=None,groups=cloud.pingcap.com,resources=dwworkers,verbs=create;update,versions=v1,name=mdwworker.kb.io,admissionReviewVersions={v1,v1beta1}

var _ webhook.Defaulter = &DWWorker{}

// Default implements webhook.Defaulter so a webhook will be registered for the type
func (r *DWWorker) Default() {
	if r.Labels == nil {
		r.Labels = map[string]string{}
	}

	var c Cluster
	err := webhookClient.Get(context.TODO(), types.NamespacedName{
		Namespace: r.Namespace,
		Name:      r.Spec.ClusterName,
	}, &c)
	if err != nil {
		dwworkerlog.Error(err, "error get cluster of dwworker, forward to validation")
		return
	}
	r.Labels[ProviderTypeLabel] = c.GetProviderType()
	r.Labels[RegionLabel] = c.Labels[RegionLabel]

	network := c.GetNetworkRef()
	r.Labels[DWWorkerLabelNetworkName] = network.Name
	r.Labels[DWWorkerLabelNetworkNamespace] = network.Namespace
}

// TODO(user): change verbs to "verbs=create;update;delete" if you want to enable deletion validation.
//+kubebuilder:webhook:path=/validate-cloud-pingcap-com-v1-dwworker,mutating=false,failurePolicy=fail,sideEffects=None,groups=cloud.pingcap.com,resources=dwworkers,verbs=create;update,versions=v1,name=vdwworker.kb.io,admissionReviewVersions={v1,v1beta1}

var _ webhook.Validator = &DWWorker{}

// ValidateCreate implements webhook.Validator so a webhook will be registered for the type
func (r *DWWorker) ValidateCreate() (admission.Warnings, error) {
	dwworkerlog.Info("validate create", "name", r.Name)

	var allErrs field.ErrorList
	if errs := r.validateSpec(); errs != nil {
		allErrs = append(allErrs, errs...)
	}

	return nil, invalidOrNil(allErrs, r)
}

func (r *DWWorker) validateSpec() field.ErrorList {
	var allErrs field.ErrorList
	fld := field.NewPath("spec")
	var c Cluster

	err := webhookClient.Get(context.TODO(), types.NamespacedName{
		Namespace: r.Namespace,
		Name:      r.Spec.ClusterName,
	}, &c)
	if err != nil {
		allErrs = append(allErrs, field.Invalid(fld.Child("clusterName"), r.Spec.ClusterName,
			fmt.Sprintf("cluster not found, err: %s", err.Error())))
		return allErrs
	}

	return allErrs
}

// ValidateUpdate implements webhook.Validator so a webhook will be registered for the type
func (r *DWWorker) ValidateUpdate(old runtime.Object) (admission.Warnings, error) {
	dwworkerlog.Info("validate update", "name", r.Name)

	var allErrs field.ErrorList
	oldR := old.(*DWWorker)
	if r.Spec.ClusterName != oldR.Spec.ClusterName {
		allErrs = append(allErrs, field.Invalid(field.NewPath("spec").Child("clusterName"), r.Spec.ClusterName, "field is immutable"))
	}
	if r.GetLabels()[ProviderTypeLabel] != oldR.GetLabels()[ProviderTypeLabel] {
		allErrs = append(allErrs, field.Invalid(field.NewPath("metadata").Child("labels"), r.GetLabels()[ProviderTypeLabel],
			fmt.Sprintf("%s label is immutable", ProviderTypeLabel)))
	}

	return nil, invalidOrNil(allErrs, r)
}

// ValidateDelete implements webhook.Validator so a webhook will be registered for the type
func (r *DWWorker) ValidateDelete() (admission.Warnings, error) {
	dwworkerlog.Info("validate delete", "name", r.Name)

	// TODO(user): fill in your validation logic upon object deletion.
	return nil, nil
}