transformers.go

package secrets

import (
	"fmt"

	v1 "k8s.io/api/core/v1"
	"strings"
)

// SecretTransformer is a function that modifies the given secret.
type SecretTransformer = func(*v1.Secret)

// UniqueNameTransformer returns a secret transformer function that sets
// `metadata.generateName` to the original `metadata.name` with '-' appended as
// separator and removes `metadata.name`.
func UniqueNameTransformer() SecretTransformer {
	return func(secret *v1.Secret) {
		secret.SetGenerateName(fmt.Sprintf("%s-", secret.GetName()))
		secret.SetName("")
	}
}

// SetAnnotationTransformer returns a secret transformer function that sets the
// annotation with the given key to the given value.
func SetAnnotationTransformer(key string, value string) SecretTransformer {
	return func(secret *v1.Secret) {
		annotations := secret.GetAnnotations()
		if annotations == nil {
			annotations = make(map[string]string)
		}
		annotations[key] = value
		secret.SetAnnotations(annotations)
	}
}

// StripAnnotationsTransformer returns a secret transformer function that
// removes all annotations where the key starts with the given 'keyPrefix'.
func StripAnnotationsTransformer(keyPrefix string) SecretTransformer {
	return func(secret *v1.Secret) {
		annotations := secret.GetAnnotations()
		if annotations == nil {
			annotations = make(map[string]string)
		}
		for key := range annotations {
			if strings.HasPrefix(key, keyPrefix) {
				delete(annotations, key)
			}
		}
		secret.SetAnnotations(annotations)
	}
}

// SetLabelTransformer returns a secret transformer function that sets the
// label with the given key to the given value.
func SetLabelTransformer(key string, value string) SecretTransformer {
	return func(secret *v1.Secret) {
		labels := secret.GetLabels()
		if labels == nil {
			labels = make(map[string]string)
		}
		labels[key] = value
		secret.SetLabels(labels)
	}
}

// StripLabelsTransformer returns a secret transformer function that
// removes all labels where the key starts with the given 'keyPrefix'.
func StripLabelsTransformer(keyPrefix string) SecretTransformer {
	return func(secret *v1.Secret) {
		labels := secret.GetLabels()
		if labels == nil {
			labels = make(map[string]string)
		}
		for key := range labels {
			if strings.HasPrefix(key, keyPrefix) {
				delete(labels, key)
			}
		}
		secret.SetLabels(labels)
	}
}