capsule/pkg/webhook/ingress/utils.go

// Copyright 2020-2021 Clastix Labs
// SPDX-License-Identifier: Apache-2.0

package ingress

import (
	"context"
	"fmt"

	extensionsv1beta1 "k8s.io/api/extensions/v1beta1"
	networkingv1 "k8s.io/api/networking/v1"
	networkingv1beta1 "k8s.io/api/networking/v1beta1"
	"k8s.io/apimachinery/pkg/fields"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/webhook/admission"

	capsulev1beta1 "github.com/clastix/capsule/api/v1beta1"
)

func tenantFromIngress(ctx context.Context, c client.Client, ingress Ingress) (*capsulev1beta1.Tenant, error) {
	tenantList := &capsulev1beta1.TenantList{}
	if err := c.List(ctx, tenantList, client.MatchingFieldsSelector{
		Selector: fields.OneTermEqualSelector(".status.namespaces", ingress.Namespace()),
	}); err != nil {
		return nil, err
	}

	if len(tenantList.Items) == 0 {
		return nil, nil
	}

	return &tenantList.Items[0], nil
}

func ingressFromRequest(req admission.Request, decoder *admission.Decoder) (ingress Ingress, err error) {
	switch req.Kind.Group {
	case "networking.k8s.io":
		if req.Kind.Version == "v1" {
			ingressObj := &networkingv1.Ingress{}
			if err = decoder.Decode(req, ingressObj); err != nil {
				return
			}
			ingress = NetworkingV1{Ingress: ingressObj}
			break
		}
		ingressObj := &networkingv1beta1.Ingress{}
		if err = decoder.Decode(req, ingressObj); err != nil {
			return
		}
		ingress = NetworkingV1Beta1{Ingress: ingressObj}
	case "extensions":
		ingressObj := &extensionsv1beta1.Ingress{}
		if err = decoder.Decode(req, ingressObj); err != nil {
			return
		}
		ingress = Extension{Ingress: ingressObj}
	default:
		err = fmt.Errorf("cannot recognize type %s", req.Kind.Group)
	}
	return
}