package annotator

import (
	cloudv1 "github.com/tidbcloud/infra-api/api/v1"

	"github.com/tidbcloud/infra-provider/pkg/config"
	"github.com/tidbcloud/infra-provider/pkg/gitops-sdk/consts"
)

func AnnotateCluster(cfg config.OverrideConfig, c *cloudv1.Cluster) {
	if c == nil {
		return
	}
	if c.Annotations == nil {
		c.Annotations = map[string]string{}
	}
	// Annotate only if annotations not set
	if c.Annotations[consts.OverrideHelperImageKey] == "" {
		c.Annotations[consts.OverrideHelperImageKey] = cfg.OverrideHelperImage
	}
	if c.Annotations[consts.OverrideRegistryKey] == "" {
		c.Annotations[consts.OverrideRegistryKey] = cfg.OverrideRegistry
	}
	if c.Annotations[consts.OverrideRepoPrefixKey] == "" {
		c.Annotations[consts.OverrideRepoPrefixKey] = cfg.OverrideRepoPrefix
	}

}