mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-07-17 01:39:06 +00:00
Refactor crd package to depend on transformerconfig
This commit is contained in:
115
pkg/crds/crds.go
115
pkg/crds/crds.go
@@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user