porter/internal/helm/driver.go

package helm

// Helm contains support for several different storage drivers.
//
// This includes (as of October 2020):
// - configmap
// - secret
// - memory
// - postgres
//
// This file implements first-class support for the first three driver types,
// and integrates with the logger.
//
// TODO -- include support for SQL storage...

import (
	"github.com/porter-dev/porter/internal/logger"

	"helm.sh/helm/v3/pkg/storage"
	"helm.sh/helm/v3/pkg/storage/driver"
	"k8s.io/client-go/kubernetes"
)

// NewStorageDriver is a function type for returning a new storage driver
type NewStorageDriver func(l *logger.Logger, namespace string, clientset *kubernetes.Clientset) *storage.Storage

// StorageMap is a map from storage configuration env variables to a function
// that initializes that Helm storage driver.
var StorageMap map[string]NewStorageDriver = map[string]NewStorageDriver{
	"secret":    newSecretStorageDriver,
	"configmap": newConfigMapsStorageDriver,
	"memory":    newMemoryStorageDriver,
}

// NewSecretStorageDriver returns a storage using the Secret driver.
func newSecretStorageDriver(
	l *logger.Logger,
	namespace string,
	clientset *kubernetes.Clientset,
) *storage.Storage {
	d := driver.NewSecrets(clientset.CoreV1().Secrets(namespace))
	d.Log = l.Printf
	return storage.Init(d)
}

// NewConfigMapsStorageDriver returns a storage using the ConfigMap driver.
func newConfigMapsStorageDriver(
	l *logger.Logger,
	namespace string,
	clientset *kubernetes.Clientset,
) *storage.Storage {
	d := driver.NewConfigMaps(clientset.CoreV1().ConfigMaps(namespace))
	d.Log = l.Printf
	return storage.Init(d)
}

// NewMemoryStorageDriver returns a storage using the In-Memory driver.
func newMemoryStorageDriver(
	_ *logger.Logger,
	namespace string,
	_ *kubernetes.Clientset,
) *storage.Storage {
	d := driver.NewMemory()
	return storage.Init(d)
}