package task

import (
	"context"
	"time"

	"github.com/go-logr/logr"
	"github.com/tidbcloud/infra-provider/pkg/deploy"
	"github.com/tidbcloud/infra-provider/pkg/namer"
	"github.com/tidbcloud/infra-provider/regional-server/api/v1alpha1"
	"github.com/tidbcloud/infra-provider/regional-server/pkg/config"
	"github.com/tidbcloud/infra-provider/regional-server/pkg/resourcemanager/appbox"
	"github.com/tidbcloud/infra-provider/regional-server/pkg/resourcemanager/approute"
	"github.com/tidbcloud/infra-provider/regional-server/pkg/resourcemanager/internet"
	"github.com/tidbcloud/infra-provider/regional-server/pkg/resourcemanager/vpc"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

type Context struct {
	Local      client.Client
	Logger     logr.Logger
	BeginAt    *time.Time
	Namer      *namer.Namer
	Internet   *v1alpha1.Internet
	VPC        *v1alpha1.VPC
	InternetRW internet.InternetReaderWriter
	VPCR       vpc.VPCReader
	APPBoxR    appbox.APPBoxReader
	APPRouteRW approute.APPRouteReaderWriter
	Config     config.RootConfig
}

var _ deploy.Task = &BaseTask{}

func (c *Context) Default() (*Context, error) {
	// Use vpc name and namespace as the network name and namespace
	// to ensure compatibility
	c.Namer = namer.NewNamer().
		LegacyWithNetworkName(c.VPC.Name).
		LegacyWithNetworkNamespace(c.VPC.Namespace)
	return c, nil
}

func (c *Context) ProviderConfigName() string {
	if c.VPC.Spec.AccountRef != nil {
		return c.VPC.Spec.AccountRef.ProviderConfigName
	}
	return ""
}

func (c *Context) S3GatewayPrefix() string {
	return c.InternetName()
}

func (c *Context) InternetName() string {
	if name := c.Internet.Annotations[v1alpha1.AnnotationKeyOwnCrossplaneNameVPC]; name != "" {
		return name
	}
	return namer.UniqueIdentifier(c.Internet.ObjectMeta)
}

func (c *Context) IsHibernated() bool {
	return c.Internet.Annotations[v1alpha1.AnnotationKeyIsHiberated] == "true"
}

func (c *Context) VPCIsLegacy() bool {
	if c.VPC != nil {
		return c.VPC.Labels[v1alpha1.IsLegacyKey] == "true"
	}
	return false
}

func (c *Context) InternetIsLegacy() bool {
	if c.Internet != nil {
		return c.Internet.Labels[v1alpha1.IsLegacyKey] == "true"
	}
	return false
}

func (c *Context) VPCName() string {
	if name := c.Internet.Annotations[v1alpha1.AnnotationKeyOwnCrossplaneNameVPC]; name != "" {
		return name
	}
	if c.VPC.Status.VPC.Name != "" {
		return c.VPC.Status.VPC.Name
	}
	return namer.UniqueIdentifier(c.VPC.ObjectMeta)
}

func (c *Context) GetZones() []string {
	if len(c.VPC.Spec.Zones) > 0 {
		return c.VPC.Spec.Zones
	}
	return c.Config.RegionConfig.Zones
}

type BaseTask struct {
	*Context
}

func (t *BaseTask) Name() string {
	return ""
}

func (t *BaseTask) Category() deploy.TaskCategory {
	return deploy.ClusterTask
}

func (t *BaseTask) Create(ctx context.Context) deploy.Result {
	return deploy.Done()
}

func (t *BaseTask) PostCreate(ctx context.Context) deploy.Result {
	return deploy.Done()
}

func (t *BaseTask) Delete(ctx context.Context) deploy.Result {
	return deploy.Done()
}

func (t *BaseTask) PostDelete(ctx context.Context) deploy.Result {
	return deploy.Done()
}

func (t *BaseTask) MetricsStartTime() *time.Time {
	return t.BeginAt
}