package controller

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

	"context"
	"fmt"

	clientv2 "github.com/zitadel/zitadel-go/v3/pkg/client"
	"github.com/zitadel/zitadel-go/v3/pkg/client/zitadel/user/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 (
	machineuserFinalizerName = "machineuser.zitadel.topmanage.com/machineuser"
)

type wrappedMachineUserFinalizer struct {
	client.Client
	machineuser *zitadelv1alpha1.MachineUser
}

func newWrappedMachineUserFinalizer(client client.Client, machineuser *zitadelv1alpha1.MachineUser) core.WrappedCoreFinalizer {
	return &wrappedMachineUserFinalizer{
		Client:      client,
		machineuser: machineuser,
	}
}

func (wf *wrappedMachineUserFinalizer) AddFinalizer(ctx context.Context) error {
	if wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.machineuser, func(machineuser *zitadelv1alpha1.MachineUser) {
		controllerutil.AddFinalizer(machineuser, machineuserFinalizerName)
	})
}

func (wf *wrappedMachineUserFinalizer) RemoveFinalizer(ctx context.Context) error {
	if !wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.machineuser, func(machineuser *zitadelv1alpha1.MachineUser) {
		controllerutil.RemoveFinalizer(wf.machineuser, machineuserFinalizerName)
	})
}

func (wr *wrappedMachineUserFinalizer) ContainsFinalizer() bool {
	return controllerutil.ContainsFinalizer(wr.machineuser, machineuserFinalizerName)
}

func (wf *wrappedMachineUserFinalizer) Reconcile(ctx context.Context, ztdClient *clientv2.Client) error {
	if wf.machineuser.Status.UserId != nil {
		if _, err := ztdClient.UserServiceV2().DeleteUser(ctx,
			&user.DeleteUserRequest{
				UserId: *wf.machineuser.Status.UserId,
			},
		); err != nil {
			return fmt.Errorf("Error deleting organization: %v", err)
		}
	}
	return nil
}

func (wr *wrappedMachineUserFinalizer) patch(ctx context.Context, machineuser *zitadelv1alpha1.MachineUser,
	patchFn func(*zitadelv1alpha1.MachineUser)) error {
	patch := ctrlClient.MergeFrom(machineuser.DeepCopy())
	patchFn(machineuser)

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