111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
/*
|
|
Copyright 2017 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 args
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"k8s.io/code-generator/cmd/client-gen/types"
|
|
)
|
|
|
|
type gvTypeValue struct {
|
|
gvToTypes *map[types.GroupVersion][]string
|
|
changed bool
|
|
}
|
|
|
|
func NewGVTypesValue(gvToTypes *map[types.GroupVersion][]string, def []string) *gvTypeValue {
|
|
gvt := new(gvTypeValue)
|
|
gvt.gvToTypes = gvToTypes
|
|
if def != nil {
|
|
if err := gvt.set(def); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
return gvt
|
|
}
|
|
|
|
var _ flag.Value = &gvTypeValue{}
|
|
|
|
func (s *gvTypeValue) set(vs []string) error {
|
|
if !s.changed {
|
|
*s.gvToTypes = map[types.GroupVersion][]string{}
|
|
}
|
|
|
|
for _, input := range vs {
|
|
gvString, typeStr, err := parseGroupVersionType(input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
gv, err := types.ToGroupVersion(gvString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
types, ok := (*s.gvToTypes)[gv]
|
|
if !ok {
|
|
types = []string{}
|
|
}
|
|
types = append(types, typeStr)
|
|
(*s.gvToTypes)[gv] = types
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *gvTypeValue) Set(val string) error {
|
|
vs, err := readAsCSV(val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := s.set(vs); err != nil {
|
|
return err
|
|
}
|
|
s.changed = true
|
|
return nil
|
|
}
|
|
|
|
func (s *gvTypeValue) Type() string {
|
|
return "stringSlice"
|
|
}
|
|
|
|
func (s *gvTypeValue) String() string {
|
|
strs := make([]string, 0, len(*s.gvToTypes))
|
|
for gv, ts := range *s.gvToTypes {
|
|
for _, t := range ts {
|
|
strs = append(strs, gv.Group.String()+"/"+gv.Version.String()+"/"+t)
|
|
}
|
|
}
|
|
str, _ := writeAsCSV(strs)
|
|
return "[" + str + "]"
|
|
}
|
|
|
|
func parseGroupVersionType(gvtString string) (gvString string, typeStr string, err error) {
|
|
invalidFormatErr := fmt.Errorf("invalid value: %s, should be of the form group/version/type", gvtString)
|
|
subs := strings.Split(gvtString, "/")
|
|
length := len(subs)
|
|
switch length {
|
|
case 2:
|
|
// gvtString of the form group/type, e.g. api/Service,extensions/ReplicaSet
|
|
return subs[0] + "/", subs[1], nil
|
|
case 3:
|
|
return strings.Join(subs[:length-1], "/"), subs[length-1], nil
|
|
default:
|
|
return "", "", invalidFormatErr
|
|
}
|
|
}
|