package controller

import (
	"strings"

	zitadelv1alpha1 "gitea.corredorconect.com/software-engineering/zitadel-resources-operator/api/v1alpha1"
	"gitea.corredorconect.com/software-engineering/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/application/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 (
	APIAppFinalizerName = "apiapp.zitadel.github.com/apiapp"
)

type wrappedAPIAppFinalizer struct {
	client.Client
	APIApp      *zitadelv1alpha1.APIApp
	refresolver *zitadelv1alpha1.RefResolver
}

func newWrappedAPIAppFinalizer(client client.Client, APIApp *zitadelv1alpha1.APIApp, refresolver *zitadelv1alpha1.RefResolver) core.WrappedCoreFinalizer {
	return &wrappedAPIAppFinalizer{
		Client:      client,
		APIApp:      APIApp,
		refresolver: refresolver,
	}
}

func (wf *wrappedAPIAppFinalizer) AddFinalizer(ctx context.Context) error {
	if wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.APIApp, func(APIApp *zitadelv1alpha1.APIApp) {
		controllerutil.AddFinalizer(APIApp, APIAppFinalizerName)
	})
}

func (wf *wrappedAPIAppFinalizer) RemoveFinalizer(ctx context.Context) error {
	if !wf.ContainsFinalizer() {
		return nil
	}
	return wf.patch(ctx, wf.APIApp, func(APIApp *zitadelv1alpha1.APIApp) {
		controllerutil.RemoveFinalizer(wf.APIApp, APIAppFinalizerName)
	})
}

func (wr *wrappedAPIAppFinalizer) ContainsFinalizer() bool {
	return controllerutil.ContainsFinalizer(wr.APIApp, APIAppFinalizerName)
}

func (wf *wrappedAPIAppFinalizer) Reconcile(ctx context.Context, ztdClient *clientv2.Client) error {
	if wf.APIApp.Status.AppId != nil {
		projectRef, err := wf.APIApp.Project(ctx, wf.refresolver)
		if err != nil {
			return err
		}
		_, err = ztdClient.ApplicationServiceV2().DeleteApplication(ctx, &application.DeleteApplicationRequest{
			ApplicationId: *wf.APIApp.Status.AppId,
			ProjectId:     projectRef.ID,
		})
		if err != nil {
			if strings.Contains(err.Error(), "doesn't exist") {
				return nil
			}
			return err
		}
	}
	return nil
}

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

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