/* Copyright 2019 The Knative Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ package generators import ( "path" "path/filepath" "strings" "k8s.io/code-generator/cmd/client-gen/generators/util" clientgentypes "k8s.io/code-generator/cmd/client-gen/types" "k8s.io/gengo/args" "k8s.io/gengo/generator" "k8s.io/gengo/namer" "k8s.io/gengo/types" "k8s.io/klog/v2" informergenargs "knative.dev/pkg/codegen/cmd/injection-gen/args" ) // Packages makes the client package definition. func Packages(context *generator.Context, arguments *args.GeneratorArgs) generator.Packages { boilerplate, err := arguments.LoadGoBoilerplate() if err != nil { klog.Fatal("Failed loading boilerplate: ", err) } customArgs, ok := arguments.CustomArgs.(*informergenargs.CustomArgs) if !ok { klog.Fatalf("Wrong CustomArgs type: %T", arguments.CustomArgs) } versionPackagePath := filepath.Join(arguments.OutputPackagePath) var packageList generator.Packages groupVersions := make(map[string]clientgentypes.GroupVersions) groupGoNames := make(map[string]string) groupVersionTypes := make(map[string]map[clientgentypes.Version][]*types.Type) for _, inputDir := range arguments.InputDirs { p := context.Universe.Package(vendorless(inputDir)) var gv clientgentypes.GroupVersion var targetGroupVersions map[string]clientgentypes.GroupVersions parts := strings.Split(p.Path, "/") gv.Group = clientgentypes.Group(parts[len(parts)-2]) gv.Version = clientgentypes.Version(parts[len(parts)-1]) targetGroupVersions = groupVersions groupPackageName := gv.Group.NonEmpty() gvPackage := path.Clean(p.Path) // If there's a comment of the form "// +groupName=somegroup" or // "// +groupName=somegroup.foo.bar.io", use the first field (somegroup) as the name of the // group when generating. if override := types.ExtractCommentTags("+", p.Comments)["groupName"]; override != nil { gv.Group = clientgentypes.Group(override[0]) } // If there's a comment of the form "// +groupGoName=SomeUniqueShortName", use that as // the Go group identifier in CamelCase. It defaults groupGoNames[groupPackageName] = namer.IC(strings.SplitN(gv.Group.NonEmpty(), ".", 2)[0]) if override := types.ExtractCommentTags("+", p.Comments)["groupGoName"]; override != nil { groupGoNames[groupPackageName] = namer.IC(override[0]) } var typesWithInformers []*types.Type var duckTypes []*types.Type var reconcilerTypes []*types.Type var clientTypes []*types.Type for _, t := range p.Types { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) if tags.NeedsInformerInjection() { typesWithInformers = append(typesWithInformers, t) } if tags.NeedsDuckInjection() { duckTypes = append(duckTypes, t) } if tags.NeedsReconciler(t, customArgs) { reconcilerTypes = append(reconcilerTypes, t) } if tags.GenerateClient { clientTypes = append(clientTypes, t) } } groupVersionsEntry, ok := targetGroupVersions[groupPackageName] if !ok { groupVersionsEntry = clientgentypes.GroupVersions{ PackageName: groupPackageName, Group: gv.Group, } } groupVersionsEntry.Versions = append(groupVersionsEntry.Versions, clientgentypes.PackageVersion{Version: gv.Version, Package: gvPackage}) targetGroupVersions[groupPackageName] = groupVersionsEntry verTypes, ok := groupVersionTypes[groupPackageName] if !ok { verTypes = make(map[clientgentypes.Version][]*types.Type) } verTypes[gv.Version] = clientTypes groupVersionTypes[groupPackageName] = verTypes if len(typesWithInformers) != 0 { orderer := namer.Orderer{Namer: namer.NewPrivateNamer(0)} typesWithInformers = orderer.OrderTypes(typesWithInformers) // Generate the informer and fake, for each type. packageList = append(packageList, versionInformerPackages(versionPackagePath, groupPackageName, gv, groupGoNames[groupPackageName], boilerplate, typesWithInformers, customArgs)...) } if len(duckTypes) != 0 { orderer := namer.Orderer{Namer: namer.NewPrivateNamer(0)} duckTypes = orderer.OrderTypes(duckTypes) // Generate a duck-typed informer for each type. packageList = append(packageList, versionDuckPackages(versionPackagePath, groupPackageName, gv, groupGoNames[groupPackageName], boilerplate, duckTypes, customArgs)...) } if len(reconcilerTypes) != 0 { orderer := namer.Orderer{Namer: namer.NewPrivateNamer(0)} reconcilerTypes = orderer.OrderTypes(reconcilerTypes) // Generate a reconciler and controller for each type. packageList = append(packageList, reconcilerPackages(versionPackagePath, groupPackageName, gv, groupGoNames[groupPackageName], boilerplate, reconcilerTypes, customArgs)...) } } // Generate the client and fake. packageList = append(packageList, versionClientsPackages(versionPackagePath, boilerplate, customArgs, groupVersions, groupGoNames, groupVersionTypes)...) // Generate the informer factory and fake. packageList = append(packageList, versionFactoryPackages(versionPackagePath, boilerplate, customArgs)...) return packageList } // Tags represents a genclient configuration for a single type. type Tags struct { util.Tags GenerateDuck bool GenerateReconciler bool } func (t Tags) NeedsInformerInjection() bool { return t.GenerateClient && !t.NoVerbs && t.HasVerb("list") && t.HasVerb("watch") } func (t Tags) NeedsDuckInjection() bool { return t.GenerateDuck } func (t Tags) NeedsReconciler(kind *types.Type, args *informergenargs.CustomArgs) bool { // Overrides kinds := strings.Split(args.ForceKinds, ",") for _, k := range kinds { if kind.Name.Name == k { klog.V(5).Infof("Kind %s was forced to generate reconciler.", k) return true } } // Normal return t.GenerateReconciler } // MustParseClientGenTags calls ParseClientGenTags but instead of returning error it panics. func MustParseClientGenTags(lines []string) Tags { ret := Tags{ Tags: util.MustParseClientGenTags(lines), } values := ExtractCommentTags("+", lines) _, ret.GenerateDuck = values["genduck"] _, ret.GenerateReconciler = values["genreconciler"] return ret } func extractCommentTags(t *types.Type) CommentTags { comments := append(append([]string{}, t.SecondClosestCommentLines...), t.CommentLines...) return ExtractCommentTags("+", comments) } func extractReconcilerClassesTag(tags CommentTags) ([]string, bool) { vals, ok := tags["genreconciler"] if !ok { return nil, false } classnames, has := vals["class"] if has && len(classnames) == 0 { return nil, false } return classnames, has } func isKRShaped(tags CommentTags) bool { vals, has := tags["genreconciler"] if !has { return false } stringVals, has := vals["krshapedlogic"] if !has || len(vals) == 0 { return true // Default is true } return stringVals[0] != "false" } func isNonNamespaced(tags CommentTags) bool { vals, has := tags["genclient"] if !has { return false } _, has = vals["nonNamespaced"] return has } func stubs(tags CommentTags) bool { vals, has := tags["genreconciler"] if !has { return false } _, has = vals["stubs"] return has } func vendorless(p string) string { if pos := strings.LastIndex(p, "/vendor/"); pos != -1 { return p[pos+len("/vendor/"):] } return p } func typedInformerPackage(groupPkgName string, gv clientgentypes.GroupVersion, externalVersionsInformersPackage string) string { return filepath.Join(externalVersionsInformersPackage, groupPkgName, gv.Version.String()) } func versionClientsPackages(basePackage string, boilerplate []byte, customArgs *informergenargs.CustomArgs, groupVersions map[string]clientgentypes.GroupVersions, groupGoNames map[string]string, groupVersionTypes map[string]map[clientgentypes.Version][]*types.Type) []generator.Package { packagePath := filepath.Join(basePackage, "client") return []generator.Package{ // Impl &generator.DefaultPackage{ PackageName: "client", PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &clientGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "client", }, groupVersions: groupVersions, groupGoNames: groupGoNames, groupVersionTypes: groupVersionTypes, outputPackage: packagePath, imports: generator.NewImportTracker(), clientSetPackage: customArgs.VersionedClientSetPackage, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, // Fake &generator.DefaultPackage{ PackageName: "fake", PackagePath: filepath.Join(packagePath, "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeClientGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake", }, outputPackage: filepath.Join(packagePath, "fake"), imports: generator.NewImportTracker(), fakeClientPkg: filepath.Join(customArgs.VersionedClientSetPackage, "fake"), clientInjectionPkg: packagePath, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, } } func versionFactoryPackages(basePackage string, boilerplate []byte, customArgs *informergenargs.CustomArgs) []generator.Package { packagePath := filepath.Join(basePackage, "informers", "factory") return []generator.Package{ // Impl &generator.DefaultPackage{ PackageName: "factory", PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &factoryGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "factory", }, outputPackage: packagePath, cachingClientSetPackage: filepath.Join(basePackage, "client"), sharedInformerFactoryPackage: customArgs.ExternalVersionsInformersPackage, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, // Fake &generator.DefaultPackage{ PackageName: "fake", PackagePath: filepath.Join(packagePath, "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeFactoryGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake", }, outputPackage: filepath.Join(packagePath, "fake"), factoryInjectionPkg: packagePath, fakeClientInjectionPkg: filepath.Join(basePackage, "client", "fake"), sharedInformerFactoryPackage: customArgs.ExternalVersionsInformersPackage, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, // FilterFactoryImpl &generator.DefaultPackage{ PackageName: "filteredFactory", PackagePath: filepath.Join(packagePath, "filtered"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &filteredFactoryGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "filtered_factory", }, outputPackage: filepath.Join(packagePath, "filtered"), cachingClientSetPackage: filepath.Join(basePackage, "client"), sharedInformerFactoryPackage: customArgs.ExternalVersionsInformersPackage, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, // FakeFilterFactory &generator.DefaultPackage{ PackageName: "fakeFilteredFactory", PackagePath: filepath.Join(packagePath, "filtered", "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeFilteredFactoryGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake_filtered_factory", }, outputPackage: filepath.Join(packagePath, "filtered", "fake"), factoryInjectionPkg: filepath.Join(packagePath, "filtered"), fakeClientInjectionPkg: filepath.Join(basePackage, "client", "fake"), sharedInformerFactoryPackage: customArgs.ExternalVersionsInformersPackage, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }, } } func versionInformerPackages(basePackage string, groupPkgName string, gv clientgentypes.GroupVersion, groupGoName string, boilerplate []byte, typesToGenerate []*types.Type, customArgs *informergenargs.CustomArgs) []generator.Package { factoryPackagePath := filepath.Join(basePackage, "informers", "factory") filteredFactoryPackagePath := filepath.Join(basePackage, "informers", "factory", "filtered") packagePath := filepath.Join(basePackage, "informers", groupPkgName, strings.ToLower(gv.Version.NonEmpty())) listerPackagePath := filepath.Join(customArgs.ListersPackage, groupPkgName, strings.ToLower(gv.Version.NonEmpty())) vers := make([]generator.Package, 0, 2*len(typesToGenerate)) for _, t := range typesToGenerate { // Fix for golang iterator bug. t := t packagePath := packagePath + "/" + strings.ToLower(t.Name.Name) typedInformerPackage := typedInformerPackage(groupPkgName, gv, customArgs.ExternalVersionsInformersPackage) // Impl vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &injectionGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: strings.ToLower(t.Name.Name), }, outputPackage: packagePath, groupVersion: gv, groupGoName: groupGoName, typeToGenerate: t, imports: generator.NewImportTracker(), typedInformerPackage: typedInformerPackage, groupInformerFactoryPackage: factoryPackagePath, clientSetPackage: customArgs.VersionedClientSetPackage, injectionClientSetPackage: filepath.Join(basePackage, "client"), listerPkg: listerPackagePath, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }) // Fake vers = append(vers, &generator.DefaultPackage{ PackageName: "fake", PackagePath: filepath.Join(packagePath, "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeInformerGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake", }, outputPackage: filepath.Join(packagePath, "fake"), imports: generator.NewImportTracker(), typeToGenerate: t, groupVersion: gv, groupGoName: groupGoName, informerInjectionPkg: packagePath, fakeFactoryInjectionPkg: filepath.Join(factoryPackagePath, "fake"), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }) // FilteredInformer vers = append(vers, &generator.DefaultPackage{ PackageName: "filtered", PackagePath: filepath.Join(packagePath, "filtered"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &filteredInjectionGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: strings.ToLower(t.Name.Name), }, outputPackage: filepath.Join(packagePath, "filtered"), groupVersion: gv, groupGoName: groupGoName, typeToGenerate: t, imports: generator.NewImportTracker(), typedInformerPackage: typedInformerPackage, groupInformerFactoryPackage: filteredFactoryPackagePath, clientSetPackage: customArgs.VersionedClientSetPackage, injectionClientSetPackage: filepath.Join(basePackage, "client"), listerPkg: listerPackagePath, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }) // FakeFilteredInformer vers = append(vers, &generator.DefaultPackage{ PackageName: "fake", PackagePath: filepath.Join(packagePath, "filtered", "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeFilteredInformerGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake", }, outputPackage: filepath.Join(packagePath, "filtered", "fake"), imports: generator.NewImportTracker(), typeToGenerate: t, groupVersion: gv, groupGoName: groupGoName, informerInjectionPkg: filepath.Join(packagePath, "filtered"), fakeFactoryInjectionPkg: filteredFactoryPackagePath, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsInformerInjection() }, }) } return vers } func reconcilerPackages(basePackage string, groupPkgName string, gv clientgentypes.GroupVersion, groupGoName string, boilerplate []byte, typesToGenerate []*types.Type, customArgs *informergenargs.CustomArgs) []generator.Package { packagePath := filepath.Join(basePackage, "reconciler", groupPkgName, strings.ToLower(gv.Version.NonEmpty())) clientPackagePath := filepath.Join(basePackage, "client") vers := make([]generator.Package, 0, 4*len(typesToGenerate)) for _, t := range typesToGenerate { // Fix for golang iterator bug. t := t extracted := extractCommentTags(t) reconcilerClasses, hasReconcilerClass := extractReconcilerClassesTag(extracted) nonNamespaced := isNonNamespaced(extracted) isKRShaped := isKRShaped(extracted) stubs := stubs(extracted) packagePath := filepath.Join(packagePath, strings.ToLower(t.Name.Name)) informerPackagePath := filepath.Join(basePackage, "informers", groupPkgName, strings.ToLower(gv.Version.NonEmpty()), strings.ToLower(t.Name.Name)) listerPackagePath := filepath.Join(customArgs.ListersPackage, groupPkgName, strings.ToLower(gv.Version.NonEmpty())) // Controller vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &reconcilerControllerGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "controller", }, typeToGenerate: t, outputPackage: packagePath, imports: generator.NewImportTracker(), groupName: gv.Group.String(), clientPkg: clientPackagePath, informerPackagePath: informerPackagePath, schemePkg: filepath.Join(customArgs.VersionedClientSetPackage, "scheme"), reconcilerClasses: reconcilerClasses, hasReconcilerClass: hasReconcilerClass, hasStatus: hasStatus(t), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsReconciler(t, customArgs) }, }) if stubs { // Controller Stub vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: filepath.Join(packagePath, "stub"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &reconcilerControllerStubGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "controller", }, typeToGenerate: t, reconcilerPkg: packagePath, outputPackage: filepath.Join(packagePath, "stub"), imports: generator.NewImportTracker(), informerPackagePath: informerPackagePath, reconcilerClasses: reconcilerClasses, hasReconcilerClass: hasReconcilerClass, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsReconciler(t, customArgs) }, }) } // Reconciler vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &reconcilerReconcilerGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "reconciler", }, typeToGenerate: t, outputPackage: packagePath, imports: generator.NewImportTracker(), clientsetPkg: customArgs.VersionedClientSetPackage, listerName: t.Name.Name + "Lister", listerPkg: listerPackagePath, groupGoName: groupGoName, groupVersion: gv, reconcilerClasses: reconcilerClasses, hasReconcilerClass: hasReconcilerClass, nonNamespaced: nonNamespaced, isKRShaped: isKRShaped, hasStatus: hasStatus(t), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsReconciler(t, customArgs) }, }) if stubs { // Reconciler Stub vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: filepath.Join(packagePath, "stub"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &reconcilerReconcilerStubGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "reconciler", }, typeToGenerate: t, reconcilerPkg: packagePath, outputPackage: filepath.Join(packagePath, "stub"), imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsReconciler(t, customArgs) }, }) } // Reconciler State vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // state generators = append(generators, &reconcilerStateGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "state", }, typeToGenerate: t, outputPackage: packagePath, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsReconciler(t, customArgs) }, }) } return vers } func versionDuckPackages(basePackage string, groupPkgName string, gv clientgentypes.GroupVersion, groupGoName string, boilerplate []byte, typesToGenerate []*types.Type, _ *informergenargs.CustomArgs) []generator.Package { packagePath := filepath.Join(basePackage, "ducks", groupPkgName, strings.ToLower(gv.Version.NonEmpty())) vers := make([]generator.Package, 0, 2*len(typesToGenerate)) for _, t := range typesToGenerate { // Fix for golang iterator bug. t := t packagePath := filepath.Join(packagePath, strings.ToLower(t.Name.Name)) // Impl vers = append(vers, &generator.DefaultPackage{ PackageName: strings.ToLower(t.Name.Name), PackagePath: packagePath, HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &duckGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: strings.ToLower(t.Name.Name), }, outputPackage: packagePath, groupVersion: gv, groupGoName: groupGoName, typeToGenerate: t, imports: generator.NewImportTracker(), }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsDuckInjection() }, }) // Fake vers = append(vers, &generator.DefaultPackage{ PackageName: "fake", PackagePath: filepath.Join(packagePath, "fake"), HeaderText: boilerplate, GeneratorFunc: func(c *generator.Context) (generators []generator.Generator) { // Impl generators = append(generators, &fakeDuckGenerator{ DefaultGen: generator.DefaultGen{ OptionalName: "fake", }, outputPackage: filepath.Join(packagePath, "fake"), imports: generator.NewImportTracker(), typeToGenerate: t, groupVersion: gv, groupGoName: groupGoName, duckInjectionPkg: packagePath, }) return generators }, FilterFunc: func(c *generator.Context, t *types.Type) bool { tags := MustParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...)) return tags.NeedsDuckInjection() }, }) } return vers } func hasStatus(t *types.Type) bool { for _, member := range t.Members { if member.Name == "Status" { return true } } return false }