From 51187256a19452ded163bbd0c31475b3b32214d6 Mon Sep 17 00:00:00 2001 From: Alexey Baranov Date: Fri, 10 Mar 2017 14:45:27 +0700 Subject: [PATCH] replace dash with empty character in import alias name --- gen/generator.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/gen/generator.go b/gen/generator.go index 630f938..1257b44 100644 --- a/gen/generator.go +++ b/gen/generator.go @@ -212,6 +212,11 @@ func fixPkgPathVendoring(pkgPath string) string { return pkgPath } +func fixAliasName(alias string) string { + alias = strings.Replace(alias, "-", "", -1) + return alias +} + // pkgAlias creates and returns and import alias for a given package. func (g *Generator) pkgAlias(pkgPath string) string { pkgPath = fixPkgPathVendoring(pkgPath) @@ -220,7 +225,7 @@ func (g *Generator) pkgAlias(pkgPath string) string { } for i := 0; ; i++ { - alias := path.Base(pkgPath) + alias := fixAliasName(path.Base(pkgPath)) if i > 0 { alias += fmt.Sprint(i) }