a3bf13711c
I had to run `make generate`. Some API functions got additional parameters `Options` and `Context`. I used empty options and `context.TODO()` for now. Signed-off-by: leonnicolas <leonloechner@gmx.de>
247 lines
7.4 KiB
Go
247 lines
7.4 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes 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 (
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"k8s.io/gengo/generator"
|
|
"k8s.io/gengo/namer"
|
|
"k8s.io/gengo/types"
|
|
|
|
"k8s.io/code-generator/cmd/client-gen/generators/util"
|
|
"k8s.io/code-generator/cmd/client-gen/path"
|
|
)
|
|
|
|
// genGroup produces a file for a group client, e.g. ExtensionsClient for the extension group.
|
|
type genGroup struct {
|
|
generator.DefaultGen
|
|
outputPackage string
|
|
group string
|
|
version string
|
|
groupGoName string
|
|
apiPath string
|
|
// types in this group
|
|
types []*types.Type
|
|
imports namer.ImportTracker
|
|
inputPackage string
|
|
clientsetPackage string
|
|
// If the genGroup has been called. This generator should only execute once.
|
|
called bool
|
|
}
|
|
|
|
var _ generator.Generator = &genGroup{}
|
|
|
|
// We only want to call GenerateType() once per group.
|
|
func (g *genGroup) Filter(c *generator.Context, t *types.Type) bool {
|
|
if !g.called {
|
|
g.called = true
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (g *genGroup) Namers(c *generator.Context) namer.NameSystems {
|
|
return namer.NameSystems{
|
|
"raw": namer.NewRawNamer(g.outputPackage, g.imports),
|
|
}
|
|
}
|
|
|
|
func (g *genGroup) Imports(c *generator.Context) (imports []string) {
|
|
imports = append(imports, g.imports.ImportLines()...)
|
|
imports = append(imports, filepath.Join(g.clientsetPackage, "scheme"))
|
|
return
|
|
}
|
|
|
|
func (g *genGroup) GenerateType(c *generator.Context, t *types.Type, w io.Writer) error {
|
|
sw := generator.NewSnippetWriter(w, c, "$", "$")
|
|
|
|
apiPath := func(group string) string {
|
|
if group == "core" {
|
|
return `"/api"`
|
|
}
|
|
return `"` + g.apiPath + `"`
|
|
}
|
|
|
|
groupName := g.group
|
|
if g.group == "core" {
|
|
groupName = ""
|
|
}
|
|
// allow user to define a group name that's different from the one parsed from the directory.
|
|
p := c.Universe.Package(path.Vendorless(g.inputPackage))
|
|
if override := types.ExtractCommentTags("+", p.Comments)["groupName"]; override != nil {
|
|
groupName = override[0]
|
|
}
|
|
|
|
m := map[string]interface{}{
|
|
"group": g.group,
|
|
"version": g.version,
|
|
"groupName": groupName,
|
|
"GroupGoName": g.groupGoName,
|
|
"Version": namer.IC(g.version),
|
|
"types": g.types,
|
|
"apiPath": apiPath(g.group),
|
|
"schemaGroupVersion": c.Universe.Type(types.Name{Package: "k8s.io/apimachinery/pkg/runtime/schema", Name: "GroupVersion"}),
|
|
"runtimeAPIVersionInternal": c.Universe.Variable(types.Name{Package: "k8s.io/apimachinery/pkg/runtime", Name: "APIVersionInternal"}),
|
|
"restConfig": c.Universe.Type(types.Name{Package: "k8s.io/client-go/rest", Name: "Config"}),
|
|
"restDefaultKubernetesUserAgent": c.Universe.Function(types.Name{Package: "k8s.io/client-go/rest", Name: "DefaultKubernetesUserAgent"}),
|
|
"restRESTClientInterface": c.Universe.Type(types.Name{Package: "k8s.io/client-go/rest", Name: "Interface"}),
|
|
"restRESTClientFor": c.Universe.Function(types.Name{Package: "k8s.io/client-go/rest", Name: "RESTClientFor"}),
|
|
"SchemeGroupVersion": c.Universe.Variable(types.Name{Package: path.Vendorless(g.inputPackage), Name: "SchemeGroupVersion"}),
|
|
}
|
|
sw.Do(groupInterfaceTemplate, m)
|
|
sw.Do(groupClientTemplate, m)
|
|
for _, t := range g.types {
|
|
tags, err := util.ParseClientGenTags(append(t.SecondClosestCommentLines, t.CommentLines...))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
wrapper := map[string]interface{}{
|
|
"type": t,
|
|
"GroupGoName": g.groupGoName,
|
|
"Version": namer.IC(g.version),
|
|
}
|
|
if tags.NonNamespaced {
|
|
sw.Do(getterImplNonNamespaced, wrapper)
|
|
} else {
|
|
sw.Do(getterImplNamespaced, wrapper)
|
|
}
|
|
}
|
|
sw.Do(newClientForConfigTemplate, m)
|
|
sw.Do(newClientForConfigOrDieTemplate, m)
|
|
sw.Do(newClientForRESTClientTemplate, m)
|
|
if g.version == "" {
|
|
sw.Do(setInternalVersionClientDefaultsTemplate, m)
|
|
} else {
|
|
sw.Do(setClientDefaultsTemplate, m)
|
|
}
|
|
sw.Do(getRESTClient, m)
|
|
|
|
return sw.Error()
|
|
}
|
|
|
|
var groupInterfaceTemplate = `
|
|
type $.GroupGoName$$.Version$Interface interface {
|
|
RESTClient() $.restRESTClientInterface|raw$
|
|
$range .types$ $.|publicPlural$Getter
|
|
$end$
|
|
}
|
|
`
|
|
|
|
var groupClientTemplate = `
|
|
// $.GroupGoName$$.Version$Client is used to interact with features provided by the $.groupName$ group.
|
|
type $.GroupGoName$$.Version$Client struct {
|
|
restClient $.restRESTClientInterface|raw$
|
|
}
|
|
`
|
|
|
|
var getterImplNamespaced = `
|
|
func (c *$.GroupGoName$$.Version$Client) $.type|publicPlural$(namespace string) $.type|public$Interface {
|
|
return new$.type|publicPlural$(c, namespace)
|
|
}
|
|
`
|
|
|
|
var getterImplNonNamespaced = `
|
|
func (c *$.GroupGoName$$.Version$Client) $.type|publicPlural$() $.type|public$Interface {
|
|
return new$.type|publicPlural$(c)
|
|
}
|
|
`
|
|
|
|
var newClientForConfigTemplate = `
|
|
// NewForConfig creates a new $.GroupGoName$$.Version$Client for the given config.
|
|
func NewForConfig(c *$.restConfig|raw$) (*$.GroupGoName$$.Version$Client, error) {
|
|
config := *c
|
|
if err := setConfigDefaults(&config); err != nil {
|
|
return nil, err
|
|
}
|
|
client, err := $.restRESTClientFor|raw$(&config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &$.GroupGoName$$.Version$Client{client}, nil
|
|
}
|
|
`
|
|
|
|
var newClientForConfigOrDieTemplate = `
|
|
// NewForConfigOrDie creates a new $.GroupGoName$$.Version$Client for the given config and
|
|
// panics if there is an error in the config.
|
|
func NewForConfigOrDie(c *$.restConfig|raw$) *$.GroupGoName$$.Version$Client {
|
|
client, err := NewForConfig(c)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return client
|
|
}
|
|
`
|
|
|
|
var getRESTClient = `
|
|
// RESTClient returns a RESTClient that is used to communicate
|
|
// with API server by this client implementation.
|
|
func (c *$.GroupGoName$$.Version$Client) RESTClient() $.restRESTClientInterface|raw$ {
|
|
if c == nil {
|
|
return nil
|
|
}
|
|
return c.restClient
|
|
}
|
|
`
|
|
|
|
var newClientForRESTClientTemplate = `
|
|
// New creates a new $.GroupGoName$$.Version$Client for the given RESTClient.
|
|
func New(c $.restRESTClientInterface|raw$) *$.GroupGoName$$.Version$Client {
|
|
return &$.GroupGoName$$.Version$Client{c}
|
|
}
|
|
`
|
|
|
|
var setInternalVersionClientDefaultsTemplate = `
|
|
func setConfigDefaults(config *$.restConfig|raw$) error {
|
|
config.APIPath = $.apiPath$
|
|
if config.UserAgent == "" {
|
|
config.UserAgent = $.restDefaultKubernetesUserAgent|raw$()
|
|
}
|
|
if config.GroupVersion == nil || config.GroupVersion.Group != scheme.Scheme.PrioritizedVersionsForGroup("$.groupName$")[0].Group {
|
|
gv := scheme.Scheme.PrioritizedVersionsForGroup("$.groupName$")[0]
|
|
config.GroupVersion = &gv
|
|
}
|
|
config.NegotiatedSerializer = scheme.Codecs
|
|
|
|
if config.QPS == 0 {
|
|
config.QPS = 5
|
|
}
|
|
if config.Burst == 0 {
|
|
config.Burst = 10
|
|
}
|
|
|
|
return nil
|
|
}
|
|
`
|
|
|
|
var setClientDefaultsTemplate = `
|
|
func setConfigDefaults(config *$.restConfig|raw$) error {
|
|
gv := $.SchemeGroupVersion|raw$
|
|
config.GroupVersion = &gv
|
|
config.APIPath = $.apiPath$
|
|
config.NegotiatedSerializer = scheme.Codecs.WithoutConversion()
|
|
|
|
if config.UserAgent == "" {
|
|
config.UserAgent = $.restDefaultKubernetesUserAgent|raw$()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
`
|