package utils

import (
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	bannerApi "edge-infra.dev/pkg/edge/apis/banner/v1alpha1"
	"edge-infra.dev/pkg/edge/constants/api/banner"
)

const (
	// BannerNS is the namespace for the banner k8s resource.
	BannerNS = "banners"
	// BannerKind is the kind for the banner k8s resource.
	BannerKind = "Banner"
)

// NewBanner returns a new banner resource.
func NewBanner(uuid, name, organizationName, enterpriseUnitID, projectID string, enablements []string) *bannerApi.Banner {
	return &bannerApi.Banner{
		TypeMeta: metav1.TypeMeta{
			Kind:       BannerKind,
			APIVersion: bannerApi.GroupVersion.String(),
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      uuid,
			Namespace: BannerNS,
		},
		Spec: bannerApi.BannerSpec{
			DisplayName: name,
			BSL: bannerApi.BSLConfig{
				EnterpriseUnit: bannerApi.BSLEnterpriseUnit{
					ID: enterpriseUnitID,
				},
				Organization: bannerApi.BSLOrganization{
					Name: organizationName,
				},
			},
			GCP: bannerApi.GCPConfig{
				ProjectID: projectID,
			},
			Enablements: enablements,
		},
	}
}

func GetBannerType(isOrgBanner *bool) banner.Type {
	if isOrgBanner != nil && *isOrgBanner {
		return banner.Org
	}
	return banner.EU
}