Improve formatId() robustness

This commit is contained in:
Simon Ser 2022-10-07 09:15:43 +02:00
parent 4677a978a1
commit c7b65e3ab4

View file

@ -5,6 +5,7 @@ import (
"log"
"os"
"strings"
"unicode"
"github.com/dave/jennifer/jen"
@ -12,11 +13,13 @@ import (
)
func formatId(s string) string {
s = strings.Title(s)
// TODO: improve robustness
s = strings.ReplaceAll(s, "-", "")
s = strings.ReplaceAll(s, "_", "")
return s
fields := strings.FieldsFunc(s, func(c rune) bool {
return !unicode.IsLetter(c) && !unicode.IsNumber(c)
})
for i, v := range fields {
fields[i] = strings.Title(v)
}
return strings.Join(fields, "")
}
func refName(ref string) string {