package utils

import (
	"errors"

	"edge-infra.dev/pkg/edge/api/graph/model"
)

var ErrRegistryURLCannotBeEmpty = errors.New("registry URL cannot be empty")

func CreateArtifactRegistryModel(registryEdgeID, bannerEdgeID, url string, description *string) *model.ArtifactRegistry {
	return &model.ArtifactRegistry{
		RegistryEdgeID: registryEdgeID,
		BannerEdgeID:   bannerEdgeID,
		Description:    description,
		URL:            url,
	}
}

func ValidateArtifactRegistryCreateInput(createArtifactRegistry *model.ArtifactRegistryCreateInput) error {
	if createArtifactRegistry.URL == "" {
		return ErrRegistryURLCannotBeEmpty
	}
	return nil
}

func UpdateArtifactRegistry(currentArtifactRegistry *model.ArtifactRegistry, updateArtifactRegistry *model.ArtifactRegistryUpdateInput) *model.ArtifactRegistry {
	currentArtifactRegistry.Description = updateArtifactRegistry.Description
	currentArtifactRegistry.URL = updateArtifactRegistry.URL
	return currentArtifactRegistry
}

func ValidateArtifactRegistryUpdateInput(updateArtifactRegistry *model.ArtifactRegistryUpdateInput) error {
	if updateArtifactRegistry.URL == "" {
		return ErrRegistryURLCannotBeEmpty
	}
	return nil
}

func ArtifactRegistryNeedsUpdating(currentArtifactRegistry *model.ArtifactRegistry, updateArtifactRegistry *model.ArtifactRegistryUpdateInput) bool {
	if currentArtifactRegistry.URL != updateArtifactRegistry.URL {
		return true
	}

	if currentArtifactRegistry.Description != nil && updateArtifactRegistry.Description != nil {
		return *currentArtifactRegistry.Description != *updateArtifactRegistry.Description
	}

	return currentArtifactRegistry.Description == updateArtifactRegistry.Description
}

func CreateClusterArtifactRegistryModel(clusterRegistryEdgeID, clusterEdgeID, registryEdgeID string) *model.ClusterArtifactRegistry {
	return &model.ClusterArtifactRegistry{
		ClusterRegistryEdgeID: clusterRegistryEdgeID,
		ClusterEdgeID:         clusterEdgeID,
		RegistryEdgeID:        registryEdgeID,
	}
}