fix on redeclared import path issue

hexdigest · Aug 25, 2024 · 24844ac · 24844ac
1 parent 3dec19e
commit 24844ac
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/generator/generator.go b/generator/generator.go
@@ -266,13 +266,15 @@ func getLastImportPart(srcPackageImport string) string {
 func getSrcPackageAlias(imports []*ast.ImportSpec, srcPackageImport string) string {
 	var srcPackageImportAlias string
 	for _, imp := range imports {
+		var partToCompare string
 		if imp.Name != nil {
-			continue
+			partToCompare = imp.Name.Name
+		} else {
+			partToCompare = getLastImportPart(imp.Path.Value)
 		}
 
 		lastSrcPackagePart := getLastImportPart(srcPackageImport)
-		lastImpPart := getLastImportPart(imp.Path.Value)
-		if lastImpPart == lastSrcPackagePart {
+		if partToCompare == lastSrcPackagePart {
 			srcPackageImportAlias = "__" + lastSrcPackagePart
 		}
 	}