Ben Kraft
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package generate
|
|
|
|
import (
|
|
"strings"
|
|
"unicode"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
func reverse(slice []string) {
|
|
for left, right := 0, len(slice)-1; left < right; left, right = left+1, right-1 {
|
|
slice[left], slice[right] = slice[right], slice[left]
|
|
}
|
|
}
|
|
|
|
func changeFirst(s string, f func(rune) rune) string {
|
|
c, n := utf8.DecodeRuneInString(s)
|
|
if c == utf8.RuneError { // empty or invalid
|
|
return s
|
|
}
|
|
return string(f(c)) + s[n:]
|
|
}
|
|
|
|
func lowerFirst(s string) string {
|
|
return changeFirst(strings.TrimLeft(s, "_"), unicode.ToLower)
|
|
}
|
|
|
|
func upperFirst(s string) string {
|
|
// TODO: initialisms
|
|
return changeFirst(strings.TrimLeft(s, "_"), unicode.ToUpper)
|
|
}
|
|
|
|
func matchFirst(s, tmpl string) string {
|
|
c, n := utf8.DecodeRuneInString(s)
|
|
t, _ := utf8.DecodeRuneInString(tmpl)
|
|
if c == utf8.RuneError || n == utf8.RuneError { // empty or invalid
|
|
return s
|
|
}
|
|
|
|
if unicode.IsUpper(t) {
|
|
c = unicode.ToUpper(c)
|
|
} else {
|
|
c = unicode.ToLower(c)
|
|
}
|
|
return string(c) + s[n:]
|
|
}
|
|
|
|
func goConstName(s string) string {
|
|
if strings.TrimLeft(s, "_") == "" {
|
|
return s
|
|
}
|
|
var prev rune
|
|
return strings.Map(func(r rune) rune {
|
|
var ret rune
|
|
if r == '_' {
|
|
ret = -1
|
|
} else if prev == '_' || prev == 0 {
|
|
ret = unicode.ToUpper(r)
|
|
} else {
|
|
ret = unicode.ToLower(r)
|
|
}
|
|
prev = r
|
|
return ret
|
|
}, s)
|
|
}
|