Support package-names with dashes in them (#232)

Support package-names with dashes in them

We were smart about aliasing if you have name-collisions, but not if
your package name is something that's not a valid identifier, like
`"path/to/my-package"`, which Go for better or worse allows. Now we
remove all the invalid characters (in practice mainly dashes, dots, and
leading digits).

Fixes #231.

Test plan: make check
This commit is contained in:
Ben Kraft
2022-11-09 09:43:04 -08:00
committed by GitHub
parent 1d71fffcb6
commit 80687e7336
6 changed files with 197 additions and 2 deletions
+30 -1
View File
@@ -2,14 +2,43 @@ package generate
import (
"fmt"
"go/token"
"go/types"
"regexp"
"strconv"
"strings"
"unicode"
)
// makeIdentifier takes a string and returns a valid go identifier like it.
//
// If the string is an identifier, return the input. Otherwise, munge it to
// make a valid identifier, which at worst (if the input is entirely emoji,
// say) means coming up with one out of whole cloth. This identifier need not
// be particularly unique; the caller may add a suffix.
func makeIdentifier(candidateIdentifier string) string {
if token.IsIdentifier(candidateIdentifier) {
return candidateIdentifier
}
var goodChars strings.Builder
for _, c := range candidateIdentifier {
// modified from token.IsIdentifier
if unicode.IsLetter(c) || c == '_' ||
// digits only valid after first char
goodChars.Len() > 0 && unicode.IsDigit(c) {
goodChars.WriteRune(c)
}
}
if goodChars.Len() > 0 {
return goodChars.String()
}
return "alias"
}
func (g *generator) addImportFor(pkgPath string) (alias string) {
pkgName := pkgPath[strings.LastIndex(pkgPath, "/")+1:]
pkgName := makeIdentifier(pkgPath[strings.LastIndex(pkgPath, "/")+1:])
alias = pkgName
suffix := 2
for g.usedAliases[alias] {