Refactor crd package to depend on transformerconfig

This commit is contained in:
Jingfang Liu
2018-09-27 10:39:29 -07:00
parent 20fa90a137
commit 7ab710889c
20 changed files with 204 additions and 217 deletions

View File

@@ -19,7 +19,6 @@ package crds
import (
"encoding/json"
"reflect"
"strings"
"github.com/ghodss/yaml"
@@ -27,61 +26,25 @@ import (
"k8s.io/kube-openapi/pkg/common"
"sigs.k8s.io/kustomize/pkg/gvk"
"sigs.k8s.io/kustomize/pkg/loader"
"sigs.k8s.io/kustomize/pkg/transformers"
"sigs.k8s.io/kustomize/pkg/transformerconfig"
)
type pathConfigs struct {
labelPathConfig transformers.PathConfig
annotationPathConfig transformers.PathConfig
prefixPathConfig transformers.PathConfig
namereferencePathConfigs []transformers.ReferencePathConfig
}
func (p *pathConfigs) addLabelPathConfig(config transformers.PathConfig) {
if *(p.labelPathConfig.GroupVersionKind) == *config.GroupVersionKind {
p.labelPathConfig.Path = append(p.labelPathConfig.Path, config.Path...)
} else {
p.labelPathConfig = config
}
}
func (p *pathConfigs) addAnnotationPathConfig(config transformers.PathConfig) {
if *(p.annotationPathConfig.GroupVersionKind) == *config.GroupVersionKind {
p.annotationPathConfig.Path = append(p.labelPathConfig.Path, config.Path...)
} else {
p.annotationPathConfig = config
}
}
func (p *pathConfigs) addNamereferencePathConfig(config transformers.ReferencePathConfig) {
p.namereferencePathConfigs = transformers.MergeNameReferencePathConfigs(p.namereferencePathConfigs, config)
}
func (p *pathConfigs) addPrefixPathConfig(config transformers.PathConfig) {
if *(p.prefixPathConfig.GroupVersionKind) == *config.GroupVersionKind {
p.prefixPathConfig.Path = append(p.prefixPathConfig.Path, config.Path...)
} else {
p.prefixPathConfig = config
}
}
// RegisterCRDs parse CRD schemas from paths and update various pathConfigs
func RegisterCRDs(loader loader.Loader, paths []string) error {
var pathConfigs []pathConfigs
func RegisterCRDs(loader loader.Loader, paths []string) (*transformerconfig.TransformerConfig, error) {
pathConfigs := transformerconfig.MakeEmptyTransformerConfig()
for _, path := range paths {
pathConfig, err := registerCRD(loader, path)
if err != nil {
return err
return nil, err
}
pathConfigs = append(pathConfigs, pathConfig...)
pathConfigs = pathConfigs.Merge(pathConfig)
}
addPathConfigs(pathConfigs)
return nil
return pathConfigs, nil
}
// register CRD from one path
func registerCRD(loader loader.Loader, path string) ([]pathConfigs, error) {
var result []pathConfigs
func registerCRD(loader loader.Loader, path string) (*transformerconfig.TransformerConfig, error) {
result := transformerconfig.MakeEmptyTransformerConfig()
content, err := loader.Load(path)
if err != nil {
return result, err
@@ -102,15 +65,13 @@ func registerCRD(loader loader.Loader, path string) ([]pathConfigs, error) {
crds := getCRDs(types)
for crd, k := range crds {
crdPathConfigs := pathConfigs{}
crdPathConfigs := transformerconfig.MakeEmptyTransformerConfig()
err = getCRDPathConfig(
types, crd, crd, gvk.FromSchemaGvk(k), []string{}, &crdPathConfigs)
types, crd, crd, gvk.FromSchemaGvk(k), []string{}, crdPathConfigs)
if err != nil {
return result, err
}
if !reflect.DeepEqual(crdPathConfigs, pathConfigs{}) {
result = append(result, crdPathConfigs)
}
result = result.Merge(crdPathConfigs)
}
return result, nil
@@ -138,7 +99,7 @@ func getCRDs(types map[string]common.OpenAPIDefinition) map[string]schema.GroupV
// getCRDPathConfig gets pathConfigs for one CRD recursively
func getCRDPathConfig(
types map[string]common.OpenAPIDefinition, atype string, crd string, in gvk.Gvk,
path []string, configs *pathConfigs) error {
path []string, configs *transformerconfig.TransformerConfig) error {
if _, ok := types[crd]; !ok {
return nil
}
@@ -146,31 +107,31 @@ func getCRDPathConfig(
for propname, property := range types[atype].Schema.SchemaProps.Properties {
_, annotate := property.Extensions.GetString(Annotation)
if annotate {
configs.addAnnotationPathConfig(
transformers.PathConfig{
configs.AddAnnotationPathConfig(
transformerconfig.PathConfig{
CreateIfNotPresent: false,
GroupVersionKind: &in,
Path: append(path, propname),
Gvk: in,
Path: strings.Join(append(path, propname), "/"),
},
)
}
_, label := property.Extensions.GetString(LabelSelector)
if label {
configs.addLabelPathConfig(
transformers.PathConfig{
configs.AddLabelPathConfig(
transformerconfig.PathConfig{
CreateIfNotPresent: false,
GroupVersionKind: &in,
Path: append(path, propname),
Gvk: in,
Path: strings.Join(append(path, propname), "/"),
},
)
}
_, identity := property.Extensions.GetString(Identity)
if identity {
configs.addPrefixPathConfig(
transformers.PathConfig{
configs.AddPrefixPathConfig(
transformerconfig.PathConfig{
CreateIfNotPresent: false,
GroupVersionKind: &in,
Path: append(path, propname),
Gvk: in,
Path: strings.Join(append(path, propname), "/"),
},
)
}
@@ -182,14 +143,16 @@ func getCRDPathConfig(
if !ok {
nameKey = "name"
}
configs.addNamereferencePathConfig(transformers.NewReferencePathConfig(
gvk.Gvk{Kind: kind, Version: version},
[]transformers.PathConfig{
{CreateIfNotPresent: false,
GroupVersionKind: &in,
Path: append(path, propname, nameKey),
}}))
configs.AddNamereferencePathConfig(transformerconfig.ReferencePathConfig{
Gvk: gvk.Gvk{Kind: kind, Version: version},
PathConfigs: []transformerconfig.PathConfig{
{
CreateIfNotPresent: false,
Gvk: in,
Path: strings.Join(append(path, propname, nameKey), "/"),
},
},
})
}
}
@@ -199,13 +162,3 @@ func getCRDPathConfig(
}
return nil
}
// addPathConfigs add extra path configs to the default ones
func addPathConfigs(p []pathConfigs) {
for _, pc := range p {
transformers.AddLabelsPathConfigs(pc.labelPathConfig)
transformers.AddAnnotationsPathConfigs(pc.annotationPathConfig)
transformers.AddNameReferencePathConfigs(pc.namereferencePathConfigs)
transformers.AddPrefixPathConfigs(pc.prefixPathConfig)
}
}

View File

@@ -24,7 +24,7 @@ import (
"sigs.k8s.io/kustomize/pkg/gvk"
"sigs.k8s.io/kustomize/pkg/internal/loadertest"
"sigs.k8s.io/kustomize/pkg/loader"
"sigs.k8s.io/kustomize/pkg/transformers"
"sigs.k8s.io/kustomize/pkg/transformerconfig"
)
const (
@@ -151,47 +151,41 @@ func makeLoader(t *testing.T) loader.Loader {
}
func TestRegisterCRD(t *testing.T) {
refpathconfigs := []transformers.ReferencePathConfig{
transformers.NewReferencePathConfig(
gvk.Gvk{Kind: "Bee", Version: "v1beta1"},
[]transformers.PathConfig{
refpathconfigs := []transformerconfig.ReferencePathConfig{
{
Gvk: gvk.Gvk{Kind: "Bee", Version: "v1beta1"},
PathConfigs: []transformerconfig.PathConfig{
{
CreateIfNotPresent: false,
GroupVersionKind: &gvk.Gvk{Kind: "MyKind"},
Path: []string{"spec", "beeRef", "name"},
Gvk: gvk.Gvk{Kind: "MyKind"},
Path: "spec/beeRef/name",
},
},
),
transformers.NewReferencePathConfig(
gvk.Gvk{Kind: "Secret", Version: "v1"},
[]transformers.PathConfig{
},
{
Gvk: gvk.Gvk{Kind: "Secret", Version: "v1"},
PathConfigs: []transformerconfig.PathConfig{
{
CreateIfNotPresent: false,
GroupVersionKind: &gvk.Gvk{Kind: "MyKind"},
Path: []string{"spec", "secretRef", "name"},
Gvk: gvk.Gvk{Kind: "MyKind"},
Path: "spec/secretRef/name",
},
},
),
},
}
sort.Slice(refpathconfigs, func(i, j int) bool {
return refpathconfigs[i].GVK() < refpathconfigs[j].GVK()
return refpathconfigs[i].Gvk.String() < refpathconfigs[j].Gvk.String()
})
expected := []pathConfigs{
{
namereferencePathConfigs: refpathconfigs,
},
expected := &transformerconfig.TransformerConfig{
NameReference: refpathconfigs,
}
ldr := makeLoader(t)
pathconfig, _ := registerCRD(ldr, "/testpath/crd.json")
sort.Slice(pathconfig[0].namereferencePathConfigs, func(i, j int) bool {
return pathconfig[0].namereferencePathConfigs[i].GVK() < pathconfig[0].namereferencePathConfigs[j].GVK()
})
if !reflect.DeepEqual(pathconfig, expected) {
t.Fatalf("expected\n %v\n but got\n %v\n", expected, pathconfig)
}