Convert plugins to accept bytes instead of unstruct.

This commit is contained in:
Jeffrey Regan
2019-05-07 17:54:02 -07:00
committed by jregan
parent 06acd3caa9
commit 2e71a3b862
14 changed files with 109 additions and 96 deletions

View File

@@ -33,8 +33,6 @@ import (
)
const (
ArgsOneLiner = "argsOneLiner"
ArgsFromFile = "argsFromFile"
idAnnotation = "kustomize.config.k8s.io/id"
)
@@ -75,30 +73,26 @@ func (p *ExecPlugin) isAvailable() bool {
}
func (p *ExecPlugin) Config(
ldr ifc.Loader, rf *resmap.Factory, k ifc.Kunstructured) error {
ldr ifc.Loader, rf *resmap.Factory, config []byte) error {
p.rf = rf
p.ldr = ldr
var err error
p.cfg, err = yaml.Marshal(k)
if err != nil {
return err
}
err = p.processOptionalArgsFields(k)
if err != nil {
return err
}
return nil
p.cfg = config
return p.processOptionalArgsFields()
}
func (p *ExecPlugin) processOptionalArgsFields(k ifc.Kunstructured) error {
args, err := k.GetFieldValue(ArgsOneLiner)
if err == nil && args != "" {
p.args = strings.Split(args, " ")
type argsConfig struct {
ArgsOneLiner string `json:"argsOneLiner,omitempty" yaml:"argsOneLiner,omitempty"`
ArgsFromFile string `json:"argsFromFile,omitempty" yaml:"argsFromFile,omitempty"`
}
func (p *ExecPlugin) processOptionalArgsFields() error {
var c argsConfig
yaml.Unmarshal(p.cfg, &c)
if c.ArgsOneLiner != "" {
p.args = strings.Split(c.ArgsOneLiner, " ")
}
fileName, err := k.GetFieldValue(ArgsFromFile)
if err == nil && fileName != "" {
content, err := p.ldr.Load(fileName)
if c.ArgsFromFile != "" {
content, err := p.ldr.Load(c.ArgsFromFile)
if err != nil {
return err
}

View File

@@ -40,8 +40,8 @@ func TestExecPluginConfig(t *testing.T) {
"metadata": map[string]interface{}{
"name": "some-random-name",
},
ArgsOneLiner: "one two",
ArgsFromFile: "sed-input.txt",
"argsOneLiner": "one two",
"argsFromFile": "sed-input.txt",
})
ldr.AddFile("/app/sed-input.txt", []byte(`
@@ -54,7 +54,11 @@ s/$BAR/bar/g
plugin.DefaultPluginConfig().DirectoryPath,
pluginConfig.Id())
p.Config(ldr, rf, pluginConfig)
yaml, err := pluginConfig.AsYAML()
if err != nil {
t.Fatalf("unexpected err: %v", err)
}
p.Config(ldr, rf, yaml)
expected := "/kustomize/plugin/someteam.example.com/v1/SedTransformer"
if !strings.HasSuffix(p.name, expected) {

View File

@@ -31,7 +31,7 @@ import (
)
type Configurable interface {
Config(ldr ifc.Loader, rf *resmap.Factory, k ifc.Kunstructured) error
Config(ldr ifc.Loader, rf *resmap.Factory, config []byte) error
}
type Loader struct {
@@ -114,7 +114,11 @@ func (l *Loader) loadAndConfigurePlugin(
return nil, err
}
}
err = c.Config(ldr, l.rf, res)
yaml, err := res.AsYAML()
if err != nil {
return nil, errors.Wrapf(err, "marshalling yaml from res %s", res.Id())
}
err = c.Config(ldr, l.rf, yaml)
if err != nil {
return nil, errors.Wrapf(
err, "plugin %s fails configuration", res.Id())

View File

@@ -18,7 +18,6 @@ package resmap_test
import (
"encoding/base64"
"fmt"
"reflect"
"testing"
@@ -124,7 +123,6 @@ metadata:
}),
}
m, err := rmF.NewResMapFromBytes(encoded)
fmt.Printf("%v\n", m)
if err != nil {
t.Fatalf("unexpected error: %v", err)
}

View File

@@ -23,6 +23,7 @@ import (
"sigs.k8s.io/kustomize/pkg/ifc"
"sigs.k8s.io/kustomize/pkg/resid"
"sigs.k8s.io/kustomize/pkg/types"
"sigs.k8s.io/yaml"
)
// Resource is map representation of a Kubernetes API resource object
@@ -56,6 +57,16 @@ func (r *Resource) DeepCopy() *Resource {
return rc
}
// AsYAML returns the resource in Yaml form.
// Easier to read than JSON.
func (r *Resource) AsYAML() ([]byte, error) {
json, err := r.MarshalJSON()
if err != nil {
return nil, err
}
return yaml.JSONToYAML(json)
}
// Behavior returns the behavior for the resource.
func (r *Resource) Behavior() types.GenerationBehavior {
return r.options.Behavior()

View File

@@ -54,6 +54,21 @@ var testDeployment = factory.FromMap(
const deploymentAsString = `{"apiVersion":"apps/v1","kind":"Deployment","metadata":{"name":"pooh"}}`
func TestAsYAML(t *testing.T) {
expected := `apiVersion: apps/v1
kind: Deployment
metadata:
name: pooh
`
yaml, err := testDeployment.AsYAML()
if err != nil {
t.Fatal(err)
}
if string(yaml) != expected {
t.Fatalf("--- expected\n%s\n--- got\n%s\n", expected, string(yaml))
}
}
func TestResourceString(t *testing.T) {
tests := []struct {
in *Resource

View File

@@ -31,7 +31,7 @@ apiVersion: someteam.example.com/v1
kind: ServiceGenerator
metadata:
name: myServiceGenerator
service: my-service
name: my-service
port: "12345"
`)
}
@@ -74,12 +74,13 @@ func writeSecretGeneratorConfig(th *kusttest_test.KustTestHarness, root string)
apiVersion: builtin
kind: SecretGenerator
metadata:
name: mySecret
name: mySecGen
name: mySecret
behavior: merge
envFiles:
envs:
- a.env
- b.env
valueFiles:
files:
- longsecret.txt
literals:
- FRUIT=apple