mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-06-12 01:14:22 +00:00
add secret and configmap generator plugins
This commit is contained in:
@@ -24,6 +24,7 @@ import (
|
||||
"github.com/pkg/errors"
|
||||
"k8s.io/apimachinery/pkg/util/validation"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv/plugin"
|
||||
"sigs.k8s.io/kustomize/pkg/ifc"
|
||||
"sigs.k8s.io/kustomize/pkg/types"
|
||||
)
|
||||
@@ -32,11 +33,20 @@ import (
|
||||
type baseFactory struct {
|
||||
ldr ifc.Loader
|
||||
options *types.GeneratorOptions
|
||||
reg plugin.Registry
|
||||
}
|
||||
|
||||
func (bf baseFactory) loadKvPairs(
|
||||
args types.GeneratorArgs) (all []kv.Pair, err error) {
|
||||
pairs, err := bf.keyValuesFromEnvFile(args.EnvSource)
|
||||
pairs, err := bf.keyValuesFromPlugins(args.KVSources)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, fmt.Sprintf(
|
||||
"plugins: %s",
|
||||
args.EnvSource))
|
||||
}
|
||||
all = append(all, pairs...)
|
||||
|
||||
pairs, err = bf.keyValuesFromEnvFile(args.EnvSource)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, fmt.Sprintf(
|
||||
"env source file: %s",
|
||||
@@ -81,6 +91,22 @@ func keyValuesFromLiteralSources(sources []string) ([]kv.Pair, error) {
|
||||
return kvs, nil
|
||||
}
|
||||
|
||||
func (bf baseFactory) keyValuesFromPlugins(sources []types.KVSource) ([]kv.Pair, error) {
|
||||
var allKvs []kv.Pair
|
||||
for _, s := range sources {
|
||||
plug, err := bf.reg.Load(s.PluginType, s.Name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
kvs, err := plug.Get(bf.reg.Root(), s.Args)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
allKvs = append(allKvs, kvs...)
|
||||
}
|
||||
return allKvs, nil
|
||||
}
|
||||
|
||||
func (bf baseFactory) keyValuesFromFileSources(sources []string) ([]kv.Pair, error) {
|
||||
var kvs []kv.Pair
|
||||
for _, s := range sources {
|
||||
|
||||
@@ -21,8 +21,10 @@ import (
|
||||
"testing"
|
||||
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv/plugin"
|
||||
"sigs.k8s.io/kustomize/pkg/fs"
|
||||
"sigs.k8s.io/kustomize/pkg/loader"
|
||||
"sigs.k8s.io/kustomize/pkg/types"
|
||||
)
|
||||
|
||||
func TestKeyValuesFromFileSources(t *testing.T) {
|
||||
@@ -45,7 +47,9 @@ func TestKeyValuesFromFileSources(t *testing.T) {
|
||||
|
||||
fSys := fs.MakeFakeFS()
|
||||
fSys.WriteFile("/files/app-init.ini", []byte("FOO=bar"))
|
||||
bf := baseFactory{loader.NewFileLoaderAtRoot(fSys), nil}
|
||||
ldr := loader.NewFileLoaderAtRoot(fSys)
|
||||
reg := plugin.NewRegistry(ldr)
|
||||
bf := baseFactory{loader.NewFileLoaderAtRoot(fSys), nil, reg}
|
||||
for _, tc := range tests {
|
||||
kvs, err := bf.keyValuesFromFileSources(tc.sources)
|
||||
if err != nil {
|
||||
@@ -56,3 +60,51 @@ func TestKeyValuesFromFileSources(t *testing.T) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestKeyValuesFromPlugins(t *testing.T) {
|
||||
tests := []struct {
|
||||
description string
|
||||
sources []types.KVSource
|
||||
expected []kv.Pair
|
||||
}{
|
||||
{
|
||||
description: "Create kv.Pairs from plugin",
|
||||
sources: []types.KVSource{
|
||||
{
|
||||
PluginType: "testonly",
|
||||
Name: "testonly",
|
||||
Args: []string{"FOO", "BAR", "BAZ"},
|
||||
},
|
||||
},
|
||||
expected: []kv.Pair{
|
||||
{
|
||||
Key: "k_FOO",
|
||||
Value: "v_FOO",
|
||||
},
|
||||
{
|
||||
Key: "k_BAR",
|
||||
Value: "v_BAR",
|
||||
},
|
||||
{
|
||||
Key: "k_BAZ",
|
||||
Value: "v_BAZ",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
fSys := fs.MakeFakeFS()
|
||||
ldr := loader.NewFileLoaderAtRoot(fSys)
|
||||
reg := plugin.NewRegistry(ldr)
|
||||
bf := baseFactory{ldr, nil, reg}
|
||||
|
||||
for _, tc := range tests {
|
||||
kvs, err := bf.keyValuesFromPlugins(tc.sources)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if !reflect.DeepEqual(kvs, tc.expected) {
|
||||
t.Fatalf("in testcase: %q updated:\n%#v\ndoesn't match expected:\n%#v\n", tc.description, kvs, tc.expected)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -21,8 +21,9 @@ import (
|
||||
"fmt"
|
||||
"unicode/utf8"
|
||||
|
||||
"k8s.io/api/core/v1"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
v1 "k8s.io/api/core/v1"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv/plugin"
|
||||
"sigs.k8s.io/kustomize/pkg/ifc"
|
||||
"sigs.k8s.io/kustomize/pkg/types"
|
||||
)
|
||||
@@ -34,8 +35,8 @@ type Factory struct {
|
||||
|
||||
// NewFactory returns a new Factory.
|
||||
func NewFactory(
|
||||
l ifc.Loader, o *types.GeneratorOptions) *Factory {
|
||||
return &Factory{baseFactory{ldr: l, options: o}}
|
||||
l ifc.Loader, o *types.GeneratorOptions, reg plugin.Registry) *Factory {
|
||||
return &Factory{baseFactory{ldr: l, options: o, reg: reg}}
|
||||
}
|
||||
|
||||
func makeFreshConfigMap(
|
||||
|
||||
@@ -22,6 +22,7 @@ import (
|
||||
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv/plugin"
|
||||
"sigs.k8s.io/kustomize/pkg/fs"
|
||||
"sigs.k8s.io/kustomize/pkg/loader"
|
||||
"sigs.k8s.io/kustomize/pkg/types"
|
||||
@@ -141,8 +142,10 @@ func TestConstructConfigMap(t *testing.T) {
|
||||
fSys.WriteFile("/configmap/app.env", []byte("DB_USERNAME=admin\nDB_PASSWORD=somepw\n"))
|
||||
fSys.WriteFile("/configmap/app-init.ini", []byte("FOO=bar\nBAR=baz\n"))
|
||||
fSys.WriteFile("/configmap/app.bin", []byte{0xff, 0xfd})
|
||||
ldr := loader.NewFileLoaderAtRoot(fSys)
|
||||
reg := plugin.NewRegistry(ldr)
|
||||
for _, tc := range testCases {
|
||||
f := NewFactory(loader.NewFileLoaderAtRoot(fSys), tc.options)
|
||||
f := NewFactory(ldr, tc.options, reg)
|
||||
cm, err := f.MakeConfigMap(&tc.input)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
|
||||
@@ -22,6 +22,7 @@ import (
|
||||
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kv/plugin"
|
||||
"sigs.k8s.io/kustomize/pkg/fs"
|
||||
"sigs.k8s.io/kustomize/pkg/loader"
|
||||
"sigs.k8s.io/kustomize/pkg/types"
|
||||
@@ -138,8 +139,10 @@ func TestConstructSecret(t *testing.T) {
|
||||
fSys := fs.MakeFakeFS()
|
||||
fSys.WriteFile("/secret/app.env", []byte("DB_USERNAME=admin\nDB_PASSWORD=somepw\n"))
|
||||
fSys.WriteFile("/secret/app-init.ini", []byte("FOO=bar\nBAR=baz\n"))
|
||||
ldr := loader.NewFileLoaderAtRoot(fSys)
|
||||
reg := plugin.NewRegistry(ldr)
|
||||
for _, tc := range testCases {
|
||||
f := NewFactory(loader.NewFileLoaderAtRoot(fSys), tc.options)
|
||||
f := NewFactory(ldr, tc.options, reg)
|
||||
cm, err := f.MakeSecret(&tc.input)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
|
||||
Reference in New Issue
Block a user