package polardbxparameter

import (
	"time"

	polardbxv1polardbx "github.com/alibaba/polardbx-operator/api/v1/polardbx"

	polardbxv1 "github.com/alibaba/polardbx-operator/api/v1"
	"github.com/alibaba/polardbx-operator/test/framework"
	"github.com/alibaba/polardbx-operator/test/framework/log"
	pxmframework "github.com/alibaba/polardbx-operator/test/framework/polardbxparameter"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

func CreatePolarDBXParameterAndWaitUntilRunningOrFail(f *framework.Framework, polardbxparameter *polardbxv1.PolarDBXParameter,
	timeout time.Duration) {
	log.Logf("Starting to create a PolarDBXParameter for test: %s ...", polardbxparameter.Name)

	// Create the given polardbx cluster.
	err := f.Client.Create(f.Ctx, polardbxparameter)
	framework.ExpectNoError(err)

	// Wait until in modifying status.
	pxmframework.WaitPolarDBXParameterInPhase(f.Client,
		polardbxparameter.Name, polardbxparameter.Namespace,
		[]polardbxv1polardbx.ParameterPhase{
			polardbxv1polardbx.ParameterStatusFinished,
		},
		timeout)
	framework.ExpectNoError(err)
}

func CreatePolarDBXParameterTemplate(f *framework.Framework, polardbxparametertemplate *polardbxv1.PolarDBXParameterTemplate,
	timeout time.Duration) {
	log.Logf("Starting to create a PolarDBXParameterTemplate for test: %s ...", polardbxparametertemplate.Name)

	// Create the given polardbx cluster.
	err := f.Client.Create(f.Ctx, polardbxparametertemplate)
	framework.ExpectNoError(err)
}

func DeletePolarDBXParameterAndWaitUntilItDisappear(f *framework.Framework, polardbxparameter *polardbxv1.PolarDBXParameter, timeout time.Duration) {
	log.Logf("Cleaning PolarDBXParameter: %s, should complete in 1 minute...", polardbxparameter.Name)

	err := f.Client.Delete(f.Ctx, polardbxparameter, client.PropagationPolicy(metav1.DeletePropagationBackground))
	if !apierrors.IsNotFound(err) {
		framework.ExpectNoError(err, "cleanup failed")
	}
	err = pxmframework.WaitForPolarDBXParameterToDisappear(f.Client, polardbxparameter.Name, polardbxparameter.Namespace, timeout)

	framework.ExpectNoError(err, "failed to wait for PolarDBXParameter object to disappear")
}

func DeletePolarDBXParameterTemplateAndWaitUntilItDisappear(f *framework.Framework, polardbxparametertemplate *polardbxv1.PolarDBXParameterTemplate, timeout time.Duration) {
	log.Logf("Cleaning PolarDBXParameter: %s, should complete in 1 minute...", polardbxparametertemplate.Name)

	err := f.Client.Delete(f.Ctx, polardbxparametertemplate, client.PropagationPolicy(metav1.DeletePropagationBackground))
	if !apierrors.IsNotFound(err) {
		framework.ExpectNoError(err, "cleanup failed")
	}
	err = pxmframework.WaitForPolarDBXParameterTemplateToDisappear(f.Client, polardbxparametertemplate.Name, polardbxparametertemplate.Namespace, timeout)

	framework.ExpectNoError(err, "failed to wait for PolarDBXParameterTemplate object to disappear")
}