mirror of https://github.com/k3s-io/k3s
add a plural namer
parent
4a22f2f5f5
commit
e2a30aba7a
|
@ -0,0 +1,56 @@
|
|||
/*
|
||||
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 namer
|
||||
|
||||
import "k8s.io/kubernetes/cmd/libs/go2idl/types"
|
||||
|
||||
type pluralNamer struct {
|
||||
exceptions map[string]string
|
||||
finalize func(string) string
|
||||
}
|
||||
|
||||
// NewPluralNamer returns a namer that returns the plural form of the input
|
||||
// type's name.
|
||||
func NewPluralNamer(exceptions map[string]string, public bool) *pluralNamer {
|
||||
var finalize func(string) string
|
||||
if public {
|
||||
finalize = IC
|
||||
} else {
|
||||
finalize = IL
|
||||
}
|
||||
return &pluralNamer{exceptions, finalize}
|
||||
}
|
||||
|
||||
// Name returns the plural form of the type's name. If the type's name is found
|
||||
// in the exceptions map, the map value is returned.
|
||||
func (r *pluralNamer) Name(t *types.Type) string {
|
||||
singular := t.Name.Name
|
||||
var plural string
|
||||
var ok bool
|
||||
if plural, ok = r.exceptions[singular]; ok {
|
||||
return r.finalize(plural)
|
||||
}
|
||||
switch string(singular[len(singular)-1]) {
|
||||
case "s", "x":
|
||||
plural = singular + "es"
|
||||
case "y":
|
||||
plural = singular[:len(singular)-1] + "ies"
|
||||
default:
|
||||
plural = singular + "s"
|
||||
}
|
||||
return r.finalize(plural)
|
||||
}
|
|
@ -0,0 +1,68 @@
|
|||
/*
|
||||
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 namer
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"k8s.io/kubernetes/cmd/libs/go2idl/types"
|
||||
)
|
||||
|
||||
func TestPluralNamer(t *testing.T) {
|
||||
exceptions := map[string]string{
|
||||
// The type name is already in the plural form
|
||||
"Endpoints": "endpoints",
|
||||
}
|
||||
public := NewPluralNamer(exceptions, true)
|
||||
private := NewPluralNamer(exceptions, false)
|
||||
|
||||
cases := []struct {
|
||||
typeName string
|
||||
expectedPrivate string
|
||||
expectedPublic string
|
||||
}{
|
||||
{
|
||||
"Pod",
|
||||
"pods",
|
||||
"Pods",
|
||||
},
|
||||
{
|
||||
"Entry",
|
||||
"entries",
|
||||
"Entries",
|
||||
},
|
||||
{
|
||||
"Endpoints",
|
||||
"endpoints",
|
||||
"Endpoints",
|
||||
},
|
||||
{
|
||||
"Bus",
|
||||
"buses",
|
||||
"Buses",
|
||||
},
|
||||
}
|
||||
for _, c := range cases {
|
||||
testType := &types.Type{Name: types.Name{Name: c.typeName}}
|
||||
if e, a := c.expectedPrivate, private.Name(testType); e != a {
|
||||
t.Errorf("Unexpected result from private plural namer. Expected: %s, Got: %s", e, a)
|
||||
}
|
||||
if e, a := c.expectedPublic, public.Name(testType); e != a {
|
||||
t.Errorf("Unexpected result from public plural namer. Expected: %s, Got: %s", e, a)
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue