resources.go

package openshift

import (
	"context"

	"github.com/appuio/seiso/pkg/kubernetes"
	imagev1 "github.com/openshift/api/image/v1"
	log "github.com/sirupsen/logrus"
	"github.com/thoas/go-funk"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime/schema"
)

var (
	PredefinedResources = []schema.GroupVersionResource{
		{Version: "v1", Resource: "pods"},
		{Group: "apps", Version: "v1", Resource: "statefulsets"},
		{Group: "apps", Version: "v1", Resource: "deployments"},
		{Group: "apps", Version: "v1", Resource: "daemonsets"},
		{Group: "apps", Version: "v1", Resource: "replicasets"},
		{Group: "apps.openshift.io", Version: "v1", Resource: "deploymentconfigs"},
		{Group: "batch", Version: "v1", Resource: "cronjobs"},
	}
	helper = kubernetes.New()
)

// GetActiveImageStreamTags retrieves the image streams tags referenced in some Kubernetes resources
func GetActiveImageStreamTags(ctx context.Context, namespace, imageStream string, imageStreamTags []string) (activeImageStreamTags []string, funcError error) {
	log.WithFields(log.Fields{
		"namespace": namespace,
		"imageName": imageStream,
		"imageTags": imageStreamTags,
	}).Debug("Looking for active images")
	if len(imageStreamTags) == 0 {
		return []string{}, nil
	}
	funk.ForEach(PredefinedResources, func(predefinedResource schema.GroupVersionResource) {
		funk.ForEach(imageStreamTags, func(imageStreamTag string) {
			if funk.ContainsString(activeImageStreamTags, imageStreamTag) {
				// already marked as existing, skip this
				return
			}
			image := BuildImageStreamTagName(imageStream, imageStreamTag)
			contains, err := helper.ResourceContains(ctx, namespace, image, predefinedResource)
			if err != nil {
				funcError = err
				return
			}

			if contains {
				activeImageStreamTags = append(activeImageStreamTags, imageStreamTag)
			}
		})
	})
	return activeImageStreamTags, funcError
}

// GetImageStreamTags returns the tags of an image stream older than the specified time
func GetImageStreamTags(ctx context.Context, namespace, imageStreamName string) ([]imagev1.NamedTagEventList, error) {

	imageClient, err := NewImageV1Client()
	if err != nil {
		return nil, err
	}

	imageStream, err := imageClient.ImageStreams(namespace).Get(ctx, imageStreamName, metav1.GetOptions{})
	if err != nil {
		return nil, err
	}

	return imageStream.Status.Tags, nil
}

// DeleteImageStreamTag deletes the image stream tag
func DeleteImageStreamTag(ctx context.Context, namespace, name string) error {
	imageclient, err := NewImageV1Client()
	if err != nil {
		return err
	}

	return imageclient.ImageStreamTags(namespace).Delete(ctx, name, metav1.DeleteOptions{})
}

// BuildImageStreamTagName combines a name of an image stream and a tag
func BuildImageStreamTagName(imageStream string, imageStreamTag string) string {
	return imageStream + ":" + imageStreamTag
}

// ListImageStreams lists all available image streams in a namespace
func ListImageStreams(ctx context.Context, namespace string) ([]imagev1.ImageStream, error) {
	imageClient, err := NewImageV1Client()
	if err != nil {
		return nil, err
	}

	imageStreams, err := imageClient.ImageStreams(namespace).List(ctx, metav1.ListOptions{})
	if err != nil {
		return nil, err
	}
	return imageStreams.Items, nil
}