Initial commit

[ZITADOPER-1]

src/api/v1alpha1/refresolver.go

@@ -0,0 +1,123 @@
+package v1alpha1
+
+import (
+	"context"
+	"fmt"
+	crdbv1alpha1 "github.com/cockroachdb/cockroach-operator/apis/v1alpha1"
+	corev1 "k8s.io/api/core/v1"
+	v1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+// +kubebuilder:object:generate=false
+type RefResolver struct {
+	client client.Client
+}
+
+func NewRefResolver(client client.Client) *RefResolver {
+	return &RefResolver{
+		client: client,
+	}
+}
+
+func (r *RefResolver) ZitadelCluster(ctx context.Context, ref *ZitadelClusterRef,
+	namespace string) (*ZitadelCluster, error) {
+	if ref.Kind != "" && ref.Kind != "ZitadelCluster" {
+		return nil, fmt.Errorf("Unsupported reference kind: '%s'", ref.Kind)
+	}
+
+	key := types.NamespacedName{
+		Name:      ref.Name,
+		Namespace: namespace,
+	}
+	if ref.Namespace != "" {
+		key.Namespace = ref.Namespace
+	}
+
+	var zitadel ZitadelCluster
+	if err := r.client.Get(ctx, key, &zitadel); err != nil {
+		return nil, err
+	}
+	return &zitadel, nil
+}
+
+func (r *RefResolver) ProjectRef(ctx context.Context, ref *ProjectRef,
+	namespace string) (*Project, error) {
+	if ref.Kind != "" && ref.Kind != "Project" {
+		return nil, fmt.Errorf("Unsupported reference kind: '%s'", ref.Kind)
+	}
+
+	key := types.NamespacedName{
+		Name:      ref.Name,
+		Namespace: namespace,
+	}
+	if ref.Namespace != "" {
+		key.Namespace = ref.Namespace
+	}
+
+	var zitadel Project
+	if err := r.client.Get(ctx, key, &zitadel); err != nil {
+		return nil, err
+	}
+	return &zitadel, nil
+}
+
+func (r *RefResolver) OrganizationRef(ctx context.Context, ref *OrganizationRef,
+	namespace string) (*Organization, error) {
+	if ref.Kind != "" && ref.Kind != "Organization" {
+		return nil, fmt.Errorf("Unsupported reference kind: '%s'", ref.Kind)
+	}
+
+	key := types.NamespacedName{
+		Name:      ref.Name,
+		Namespace: namespace,
+	}
+	if ref.Namespace != "" {
+		key.Namespace = ref.Namespace
+	}
+
+	var zitadel Organization
+	if err := r.client.Get(ctx, key, &zitadel); err != nil {
+		return nil, err
+	}
+	return &zitadel, nil
+}
+
+func (r *RefResolver) CrdbClusterRef(ctx context.Context, ref *CrdbClusterRef, namespace string) (*crdbv1alpha1.CrdbCluster, error) {
+	if ref.Kind != "" && ref.Kind != "CrdbCluster" {
+		return nil, fmt.Errorf("Unsupported reference kind: '%s'", ref.Kind)
+	}
+	key := types.NamespacedName{
+		Name:      ref.Name,
+		Namespace: namespace,
+	}
+	if ref.Namespace != "" {
+		key.Namespace = ref.Namespace
+	}
+
+	var crdb crdbv1alpha1.CrdbCluster
+	if err := r.client.Get(ctx, key, &crdb); err != nil {
+		return nil, err
+	}
+	return &crdb, nil
+}
+
+func (r *RefResolver) SecretKeyRef(ctx context.Context, selector corev1.SecretKeySelector,
+	namespace string) (string, error) {
+	nn := types.NamespacedName{
+		Name:      selector.Name,
+		Namespace: namespace,
+	}
+	var secret v1.Secret
+	if err := r.client.Get(ctx, nn, &secret); err != nil {
+		return "", fmt.Errorf("error getting secret: %v", err)
+	}
+
+	data, ok := secret.Data[selector.Key]
+	if !ok {
+		return "", fmt.Errorf("secret key \"%s\" not found", selector.Key)
+	}
+
+	return string(data), nil
+}