github.com/xeptore/docker-cli@v20.10.14+incompatible/cli/command/stack/kubernetes/stackclient.go

package kubernetes

import (
	"fmt"

	composev1alpha3 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1alpha3"
	composev1beta1 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1beta1"
	composev1beta2 "github.com/docker/compose-on-kubernetes/api/client/clientset/typed/compose/v1beta2"
	"github.com/docker/compose-on-kubernetes/api/compose/v1alpha3"
	"github.com/docker/compose-on-kubernetes/api/compose/v1beta1"
	"github.com/docker/compose-on-kubernetes/api/compose/v1beta2"
	"github.com/docker/compose-on-kubernetes/api/labels"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
	"k8s.io/client-go/rest"
)

// StackClient talks to a kubernetes compose component.
type StackClient interface {
	StackConverter
	CreateOrUpdate(s Stack, childResources []childResource) error
	Delete(name string) error
	Get(name string) (Stack, error)
	List(opts metav1.ListOptions) ([]Stack, error)
	IsColliding(servicesClient corev1.ServiceInterface, s Stack) error
}

// stackV1Beta1 implements stackClient interface and talks to compose component v1beta1.
type stackV1Beta1 struct {
	stackV1Beta1Converter
	stacks composev1beta1.StackInterface
}

func newStackV1Beta1(config *rest.Config, namespace string) (*stackV1Beta1, error) {
	client, err := composev1beta1.NewForConfig(config)
	if err != nil {
		return nil, err
	}
	return &stackV1Beta1{stacks: client.Stacks(namespace)}, nil
}

func (s *stackV1Beta1) CreateOrUpdate(internalStack Stack, childResources []childResource) error {
	// If it already exists, update the stack
	var (
		stack *v1beta1.Stack
		err   error
	)
	if stack, err = s.stacks.Get(internalStack.Name, metav1.GetOptions{}); err == nil {
		stack.Spec.ComposeFile = internalStack.ComposeFile
		stack, err = s.stacks.Update(stack)
	} else {
		// Or create it
		stack, err = s.stacks.Create(stackToV1beta1(internalStack))
	}
	if err != nil {
		deleteChildResources(childResources)
		return err
	}
	blockOwnerDeletion := true
	isController := true
	return setChildResourcesOwner(childResources, metav1.OwnerReference{
		APIVersion:         v1beta1.SchemeGroupVersion.String(),
		Kind:               "Stack",
		Name:               stack.Name,
		UID:                stack.UID,
		BlockOwnerDeletion: &blockOwnerDeletion,
		Controller:         &isController,
	})
}

func (s *stackV1Beta1) Delete(name string) error {
	return s.stacks.Delete(name, &metav1.DeleteOptions{})
}

func (s *stackV1Beta1) Get(name string) (Stack, error) {
	stackBeta1, err := s.stacks.Get(name, metav1.GetOptions{})
	if err != nil {
		return Stack{}, err
	}
	return stackFromV1beta1(stackBeta1)
}

func (s *stackV1Beta1) List(opts metav1.ListOptions) ([]Stack, error) {
	list, err := s.stacks.List(opts)
	if err != nil {
		return nil, err
	}
	stacks := make([]Stack, len(list.Items))
	for i := range list.Items {
		stack, err := stackFromV1beta1(&list.Items[i])
		if err != nil {
			return nil, err
		}
		stacks[i] = stack
	}
	return stacks, nil
}

// IsColliding verifies that services defined in the stack collides with already deployed services
func (s *stackV1Beta1) IsColliding(servicesClient corev1.ServiceInterface, st Stack) error {
	for _, srv := range st.getServices() {
		if err := verify(servicesClient, st.Name, srv); err != nil {
			return err
		}
	}
	return nil
}

// verify checks whether the service is already present in kubernetes.
// If we find the service by name but it doesn't have our label or it has a different value
// than the stack name for the label, we fail (i.e. it will collide)
func verify(services corev1.ServiceInterface, stackName string, service string) error {
	svc, err := services.Get(service, metav1.GetOptions{})
	if err == nil {
		if key, ok := svc.ObjectMeta.Labels[labels.ForStackName]; ok {
			if key != stackName {
				return fmt.Errorf("service %s already present in stack named %s", service, key)
			}
			return nil
		}
		return fmt.Errorf("service %s already present in the cluster", service)
	}
	return nil
}

// stackV1Beta2 implements stackClient interface and talks to compose component v1beta2.
type stackV1Beta2 struct {
	stackV1Beta2Converter
	stacks composev1beta2.StackInterface
}

func newStackV1Beta2(config *rest.Config, namespace string) (*stackV1Beta2, error) {
	client, err := composev1beta2.NewForConfig(config)
	if err != nil {
		return nil, err
	}
	return &stackV1Beta2{stacks: client.Stacks(namespace)}, nil
}

func (s *stackV1Beta2) CreateOrUpdate(internalStack Stack, childResources []childResource) error {
	var (
		stack *v1beta2.Stack
		err   error
	)
	resolved, err := stackToV1beta2(internalStack)
	if err != nil {
		deleteChildResources(childResources)
		return err
	}
	if stack, err = s.stacks.Get(internalStack.Name, metav1.GetOptions{}); err == nil {
		stack.Spec = resolved.Spec
		stack, err = s.stacks.Update(stack)
	} else {
		// Or create it
		stack, err = s.stacks.Create(resolved)
	}
	if err != nil {
		deleteChildResources(childResources)
		return err
	}
	blockOwnerDeletion := true
	isController := true
	return setChildResourcesOwner(childResources, metav1.OwnerReference{
		APIVersion:         v1beta2.SchemeGroupVersion.String(),
		Kind:               "Stack",
		Name:               stack.Name,
		UID:                stack.UID,
		BlockOwnerDeletion: &blockOwnerDeletion,
		Controller:         &isController,
	})
}

func (s *stackV1Beta2) Delete(name string) error {
	return s.stacks.Delete(name, &metav1.DeleteOptions{})
}

func (s *stackV1Beta2) Get(name string) (Stack, error) {
	stackBeta2, err := s.stacks.Get(name, metav1.GetOptions{})
	if err != nil {
		return Stack{}, err
	}
	return stackFromV1beta2(stackBeta2)
}

func (s *stackV1Beta2) List(opts metav1.ListOptions) ([]Stack, error) {
	list, err := s.stacks.List(opts)
	if err != nil {
		return nil, err
	}
	stacks := make([]Stack, len(list.Items))
	for i := range list.Items {
		if stacks[i], err = stackFromV1beta2(&list.Items[i]); err != nil {
			return nil, err
		}
	}
	return stacks, nil
}

// IsColliding is handle server side with the compose api v1beta2, so nothing to do here
func (s *stackV1Beta2) IsColliding(servicesClient corev1.ServiceInterface, st Stack) error {
	return nil
}

// stackV1Beta2 implements stackClient interface and talks to compose component v1beta2.
type stackV1Alpha3 struct {
	stackV1Alpha3Converter
	stacks composev1alpha3.StackInterface
}

func newStackV1Alpha3(config *rest.Config, namespace string) (*stackV1Alpha3, error) {
	client, err := composev1alpha3.NewForConfig(config)
	if err != nil {
		return nil, err
	}
	return &stackV1Alpha3{stacks: client.Stacks(namespace)}, nil
}

func (s *stackV1Alpha3) CreateOrUpdate(internalStack Stack, childResources []childResource) error {
	var (
		stack *v1alpha3.Stack
		err   error
	)
	resolved := stackToV1alpha3(internalStack)
	if stack, err = s.stacks.Get(internalStack.Name, metav1.GetOptions{}); err == nil {
		stack.Spec = resolved.Spec
		stack, err = s.stacks.Update(stack)
	} else {
		// Or create it
		stack, err = s.stacks.Create(resolved)
	}
	if err != nil {
		deleteChildResources(childResources)
		return err
	}
	blockOwnerDeletion := true
	isController := true
	return setChildResourcesOwner(childResources, metav1.OwnerReference{
		APIVersion:         v1alpha3.SchemeGroupVersion.String(),
		Kind:               "Stack",
		Name:               stack.Name,
		UID:                stack.UID,
		BlockOwnerDeletion: &blockOwnerDeletion,
		Controller:         &isController,
	})
}

func (s *stackV1Alpha3) Delete(name string) error {
	return s.stacks.Delete(name, &metav1.DeleteOptions{})
}

func (s *stackV1Alpha3) Get(name string) (Stack, error) {
	stackAlpha3, err := s.stacks.Get(name, metav1.GetOptions{})
	if err != nil {
		return Stack{}, err
	}
	return stackFromV1alpha3(stackAlpha3), nil
}

func (s *stackV1Alpha3) List(opts metav1.ListOptions) ([]Stack, error) {
	list, err := s.stacks.List(opts)
	if err != nil {
		return nil, err
	}
	stacks := make([]Stack, len(list.Items))
	for i := range list.Items {
		stacks[i] = stackFromV1alpha3(&list.Items[i])
	}
	return stacks, nil
}

// IsColliding is handle server side with the compose api v1beta2, so nothing to do here
func (s *stackV1Alpha3) IsColliding(servicesClient corev1.ServiceInterface, st Stack) error {
	return nil
}