1 // Copyright (c) 2020 Cisco and/or its affiliates.
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at:
7 // http://www.apache.org/licenses/LICENSE-2.0
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
22 // commonInitialisms is a set of common initialisms that need to stay in upper case.
23 var commonInitialisms = map[string]bool{
26 // NOTE: There are only two occurences of the word 'ascii' and
27 // these already have initialism before and after ASCII part,
28 // thus disabling initialism for this case.
71 // specialInitialisms is a set of special initialisms that need part to stay in upper case.
72 var specialInitialisms = map[string]string{
76 func usesInitialism(s string) string {
77 if u := strings.ToUpper(s); commonInitialisms[u] {
79 } else if su, ok := specialInitialisms[u]; ok {
85 // camelCaseName returns correct name identifier (camelCase).
86 func camelCaseName(name string) (should string) {
87 name = strings.Title(name)
89 // Fast path for simple cases: "_" and all lowercase.
94 for _, r := range name {
95 if !unicode.IsLower(r) {
104 // Split camelCase at any lower->upper transition, and split on underscores.
105 // Check each word for common initialisms.
106 runes := []rune(name)
107 w, i := 0, 0 // index of start of word, scan
108 for i+1 <= len(runes) {
109 eow := false // whether we hit the end of a word
110 if i+1 == len(runes) {
112 } else if runes[i+1] == '_' {
113 // underscore; shift the remainder forward over any run of underscores
116 for i+n+1 < len(runes) && runes[i+n+1] == '_' {
120 // Leave at most one underscore if the underscore is between two digits
121 if i+n+1 < len(runes) && unicode.IsDigit(runes[i]) && unicode.IsDigit(runes[i+n+1]) {
125 copy(runes[i+1:], runes[i+n+1:])
126 runes = runes[:len(runes)-n]
127 } else if unicode.IsLower(runes[i]) && !unicode.IsLower(runes[i+1]) {
137 word := string(runes[w:i])
138 if u := usesInitialism(word); u != "" {
139 // Keep consistent case, which is lowercase only at the start.
140 if w == 0 && unicode.IsLower(runes[w]) {
141 u = strings.ToLower(u)
143 // All the common initialisms are ASCII,
144 // so we can replace the bytes exactly.
145 copy(runes[w:], []rune(u))
146 } else if w > 0 && strings.ToLower(word) == word {
147 // already all lowercase, and not the first word, so uppercase the first character.
148 runes[w] = unicode.ToUpper(runes[w])