package controller

import (
	zitadelv1alpha1 "github.com/HaimKortovich/zitadel-resources-operator/api/v1alpha1"
	"github.com/HaimKortovich/zitadel-resources-operator/pkg/controller/core"

	"context"
	"fmt"

	clientv2 "github.com/zitadel/zitadel-go/v3/pkg/client"
	"github.com/zitadel/zitadel-go/v3/pkg/client/zitadel/project/v2"
	"sigs.k8s.io/controller-runtime/pkg/client"
	ctrlClient "sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
)

const (
	projectFinalizerName = "project.zitadel.github.com/project"
)

type wrappedProjectFinalizer struct {
	client.Client
	project     *zitadelv1alpha1.Project
	refresolver *zitadelv1alpha1.RefResolver
}

func newWrappedProjectFinalizer(client client.Client, project *zitadelv1alpha1.Project, refresolver *zitadelv1alpha1.RefResolver) core.WrappedCoreFinalizer {
	return &wrappedProjectFinalizer{
		Client:      client,
		project:     project,
		refresolver: refresolver,
	}
}

func (wf *wrappedProjectFinalizer) AddFinalizer(ctx context.Context) error {
	if wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.project, func(project *zitadelv1alpha1.Project) {
		controllerutil.AddFinalizer(project, projectFinalizerName)
	})
}

func (wf *wrappedProjectFinalizer) RemoveFinalizer(ctx context.Context) error {
	if !wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.project, func(project *zitadelv1alpha1.Project) {
		controllerutil.RemoveFinalizer(wf.project, projectFinalizerName)
	})
}

func (wr *wrappedProjectFinalizer) ContainsFinalizer() bool {
	return controllerutil.ContainsFinalizer(wr.project, projectFinalizerName)
}

func (wf *wrappedProjectFinalizer) Reconcile(ctx context.Context, ztdClient *clientv2.Client) error {
	if wf.project.Status.ProjectId != nil {
		_, err := ztdClient.ProjectServiceV2().DeleteProject(ctx, &project.DeleteProjectRequest{ProjectId: *wf.project.Status.ProjectId})
		if err != nil {
			return err
		}
	}
	return nil
}

func (wr *wrappedProjectFinalizer) patch(ctx context.Context, project *zitadelv1alpha1.Project,
	patchFn func(*zitadelv1alpha1.Project)) error {
	patch := ctrlClient.MergeFrom(project.DeepCopy())
	patchFn(project)

	if err := wr.Client.Patch(ctx, project, patch); err != nil {
		return fmt.Errorf("error patching Project finalizer: %v", err)
	}
	return nil
}