mirror of https://github.com/k3s-io/k3s
124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes Authors All rights reserved.
|
|
|
|
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 main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
_ "k8s.io/kubernetes/pkg/api/v1"
|
|
_ "k8s.io/kubernetes/pkg/apis/experimental"
|
|
_ "k8s.io/kubernetes/pkg/apis/experimental/v1"
|
|
pkg_runtime "k8s.io/kubernetes/pkg/runtime"
|
|
"k8s.io/kubernetes/pkg/util/sets"
|
|
|
|
"github.com/golang/glog"
|
|
flag "github.com/spf13/pflag"
|
|
"golang.org/x/tools/imports"
|
|
)
|
|
|
|
const pkgBase = "k8s.io/kubernetes/pkg"
|
|
|
|
var (
|
|
functionDest = flag.StringP("funcDest", "f", "-", "Output for conversion functions; '-' means stdout")
|
|
groupVersion = flag.StringP("version", "v", "api/v1", "groupPath/version for conversion.")
|
|
)
|
|
|
|
// We're moving to pkg/apis/group/version. This handles new and legacy packages.
|
|
func pkgPath(group, version string) string {
|
|
if group == "" {
|
|
group = "api"
|
|
}
|
|
gv := group
|
|
if version != "" {
|
|
gv = path.Join(group, version)
|
|
}
|
|
switch {
|
|
case group == "api":
|
|
// TODO(lavalamp): remove this special case when we move api to apis/api
|
|
return path.Join(pkgBase, gv)
|
|
default:
|
|
return path.Join(pkgBase, "apis", gv)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
flag.Parse()
|
|
|
|
var funcOut io.Writer
|
|
if *functionDest == "-" {
|
|
funcOut = os.Stdout
|
|
} else {
|
|
file, err := os.Create(*functionDest)
|
|
if err != nil {
|
|
glog.Fatalf("Couldn't open %v: %v", *functionDest, err)
|
|
}
|
|
defer file.Close()
|
|
funcOut = file
|
|
}
|
|
|
|
data := new(bytes.Buffer)
|
|
|
|
group, version := path.Split(*groupVersion)
|
|
group = strings.TrimRight(group, "/")
|
|
|
|
_, err := data.WriteString(fmt.Sprintf("package %v\n", version))
|
|
if err != nil {
|
|
glog.Fatalf("error writing package line: %v", err)
|
|
}
|
|
|
|
versionPath := pkgPath(group, version)
|
|
generator := pkg_runtime.NewConversionGenerator(api.Scheme.Raw(), versionPath)
|
|
apiShort := generator.AddImport(path.Join(pkgBase, "api"))
|
|
generator.AddImport(path.Join(pkgBase, "api/resource"))
|
|
// TODO(wojtek-t): Change the overwrites to a flag.
|
|
generator.OverwritePackage(version, "")
|
|
for _, knownType := range api.Scheme.KnownTypes(version) {
|
|
if knownType.PkgPath() != versionPath {
|
|
continue
|
|
}
|
|
if err := generator.GenerateConversionsForType(version, knownType); err != nil {
|
|
glog.Errorf("error while generating conversion functions for %v: %v", knownType, err)
|
|
}
|
|
}
|
|
generator.RepackImports(sets.NewString())
|
|
if err := generator.WriteImports(data); err != nil {
|
|
glog.Fatalf("error while writing imports: %v", err)
|
|
}
|
|
if err := generator.WriteConversionFunctions(data); err != nil {
|
|
glog.Fatalf("Error while writing conversion functions: %v", err)
|
|
}
|
|
if err := generator.RegisterConversionFunctions(data, fmt.Sprintf("%s.Scheme", apiShort)); err != nil {
|
|
glog.Fatalf("Error while writing conversion functions: %v", err)
|
|
}
|
|
|
|
b, err := imports.Process("", data.Bytes(), nil)
|
|
if err != nil {
|
|
glog.Fatalf("error while update imports: %v", err)
|
|
}
|
|
if _, err := funcOut.Write(b); err != nil {
|
|
glog.Fatalf("error while writing out the resulting file: %v", err)
|
|
}
|
|
}
|