mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-06-10 16:42:51 +00:00
add goplugin for exec generators and transformers
This commit is contained in:
72
pkg/plugins/builtin/executable.go
Normal file
72
pkg/plugins/builtin/executable.go
Normal file
@@ -0,0 +1,72 @@
|
||||
// +build plugin
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/ghodss/yaml"
|
||||
"sigs.k8s.io/kustomize/pkg/ifc"
|
||||
"sigs.k8s.io/kustomize/pkg/pgmconfig"
|
||||
"sigs.k8s.io/kustomize/pkg/resmap"
|
||||
)
|
||||
|
||||
type plugin struct {
|
||||
name string
|
||||
input string
|
||||
rf *resmap.Factory
|
||||
}
|
||||
|
||||
var KustomizePlugin plugin
|
||||
|
||||
func (p *plugin) Config(
|
||||
ldr ifc.Loader, rf *resmap.Factory, k ifc.Kunstructured) error {
|
||||
dir := filepath.Join(pgmconfig.ConfigRoot(), "plugins")
|
||||
id := k.GetGvk()
|
||||
p.name = filepath.Join(dir, id.Group, id.Version, id.Kind)
|
||||
content, err := yaml.Marshal(k)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
p.input = string(content)
|
||||
p.rf = rf
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *plugin) Generate() (resmap.ResMap, error) {
|
||||
return p.run(nil)
|
||||
}
|
||||
|
||||
func (p *plugin) Transformer(rm resmap.ResMap) error {
|
||||
result, err := p.run(rm)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for id := range rm {
|
||||
delete(rm, id)
|
||||
}
|
||||
for id, r := range result {
|
||||
rm[id] = r
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *plugin) run(rm resmap.ResMap) (resmap.ResMap, error) {
|
||||
cmd := exec.Command(p.name, p.input)
|
||||
cmd.Env = os.Environ()
|
||||
if rm != nil {
|
||||
content, err := rm.EncodeAsYaml()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
cmd.Stdin = bytes.NewReader(content)
|
||||
}
|
||||
output, err := cmd.Output()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return p.rf.NewResMapFromBytes(output)
|
||||
}
|
||||
Reference in New Issue
Block a user