zitadel-k8s-operator/pkg/builder/service_builder.go

package builder

import (
	"fmt"

	zitadelv1alpha1 "gitea.corredorconect.com/software-engineering/zitadel-k8s-operator/api/v1alpha1"
	labels "gitea.corredorconect.com/software-engineering/zitadel-k8s-operator/pkg/builder/labels"
	metadata "gitea.corredorconect.com/software-engineering/zitadel-k8s-operator/pkg/builder/metadata"
	corev1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/types"
	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
)

type ServiceOpts struct {
	Ports []corev1.ServicePort
}

func (b *Builder) BuildService(zitadel *zitadelv1alpha1.Cluster, key types.NamespacedName,
	opts ServiceOpts) (*corev1.Service, error) {
	objMeta :=
		metadata.NewMetadataBuilder(key).
			WithZitadel(zitadel).
			Build()

	svc := &corev1.Service{
		ObjectMeta: objMeta,
		Spec: corev1.ServiceSpec{
			Type:     corev1.ServiceTypeClusterIP,
			Ports:    opts.Ports,
			Selector: serviceSelectorLabels(opts, zitadel),
		},
	}
	if err := controllerutil.SetControllerReference(zitadel, svc, b.scheme); err != nil {
		return nil, fmt.Errorf("error setting controller reference to Service: %v", err)
	}
	return svc, nil
}

func (b *Builder) BuildLoginService(instance *zitadelv1alpha1.Instance, key types.NamespacedName,
	opts ServiceOpts) (*corev1.Service, error) {
	objMeta :=
		metadata.NewMetadataBuilder(key).
			Build()

	svc := &corev1.Service{
		ObjectMeta: objMeta,
		Spec: corev1.ServiceSpec{
			Type:     corev1.ServiceTypeClusterIP,
			Ports:    opts.Ports,
			Selector: serviceLoginSelectorLabels(opts, instance),
		},
	}
	if err := controllerutil.SetControllerReference(instance, svc, b.scheme); err != nil {
		return nil, fmt.Errorf("error setting controller reference to Service: %v", err)
	}
	return svc, nil
}

func serviceLoginSelectorLabels(opts ServiceOpts, instance *zitadelv1alpha1.Instance) map[string]string {
	return labels.NewLabelsBuilder().
		WithLoginUISelectorLabels(instance).
		Build()
}

func serviceSelectorLabels(opts ServiceOpts, cluster *zitadelv1alpha1.Cluster) map[string]string {
	return labels.NewLabelsBuilder().
		WithZitadelSelectorLabels(cluster).
		Build()
}