package lattice

import (
	"github.com/aws/aws-application-networking-k8s/pkg/model/core"
)

const (
	K8SServiceNameKey        = "K8SServiceName"
	K8SServiceNamespaceKey   = "K8SServiceNamespace"
	K8SParentRefTypeKey      = "K8SParentRefTypeKey"
	K8SHTTPRouteNameKey      = "K8SHTTPRouteName"
	K8SHTTPRouteNamespaceKey = "K8SHTTPRouteNamespace"
	K8SServiceExportType     = "K8SServiceExportType"
	K8SHTTPRouteType         = "K8SHTTPRouteType"
)

type TargetGroup struct {
	core.ResourceMeta `json:"-"`
	Spec              TargetGroupSpec    `json:"spec"`
	Status            *TargetGroupStatus `json:"status,omitempty"`
}

type TargetGroupSpec struct {
	Name      string
	Config    TargetGroupConfig `json:"config"`
	Type      TargetGroupType
	IsDeleted bool
	LatticeID string
}

type TargetGroupConfig struct {
	Port            int32  `json:"port"`
	Protocol        string `json:"protocol"`
	ProtocolVersion string `json:"protocolversion"`
	VpcID           string `json:"vpcid"`
	EKSClusterName  string `json:"eksclustername"`
	IsServiceImport bool   `json:"serviceimport"`
	// the following fields are used for AWS resource tagging
	IsServiceExport       bool   `json:"serviceexport"`
	K8SServiceName        string `json:"k8sservice"`
	K8SServiceNamespace   string `json:"k8sservicenamespace"`
	K8SHTTPRouteName      string `json:"k8shttproutename"`
	K8SHTTPRouteNamespace string `json:"k8shttproutenamespace"`
}

type TargetGroupStatus struct {
	TargetGroupARN string `json:"latticeServiceARN"`
	TargetGroupID  string `json:"latticeServiceID"`
}

type TargetGroupType string

const (
	TargetGroupTypeIP TargetGroupType = "IP"
)

func NewTargetGroup(stack core.Stack, id string, spec TargetGroupSpec) *TargetGroup {
	tg := &TargetGroup{
		ResourceMeta: core.NewResourceMeta(stack, "AWS:VPCServiceNetwork::TargetGroup", id),
		Spec:         spec,
		Status:       nil,
	}

	stack.AddResource(tg)

	return tg
}