1
0
mirror of https://github.com/beego/bee.git synced 2024-12-22 15:10:49 +00:00

Using local package name for renamed packages.

This commit is contained in:
Lei Cao 2015-05-14 15:56:19 +08:00
parent e23253bc23
commit a1fe5bf010

View File

@ -149,7 +149,11 @@ func generateDocs(curpath string) {
}
}
for _, im := range f.Imports {
analisyscontrollerPkg(im.Path.Value)
localName := ""
if im.Name != nil {
localName = im.Name.Name
}
analisyscontrollerPkg(localName, im.Path.Value)
}
for _, d := range f.Decls {
switch specDecl := d.(type) {
@ -256,13 +260,17 @@ func analisysNSInclude(baseurl string, ce *ast.CallExpr) string {
return cname
}
func analisyscontrollerPkg(pkgpath string) {
func analisyscontrollerPkg(localName, pkgpath string) {
pkgpath = strings.Trim(pkgpath, "\"")
if isSystemPackage(pkgpath) {
return
}
pps := strings.Split(pkgpath, "/")
importlist[pps[len(pps)-1]] = pkgpath
if localName != "" {
importlist[localName] = pkgpath
} else {
pps := strings.Split(pkgpath, "/")
importlist[pps[len(pps)-1]] = pkgpath
}
if pkgpath == "github.com/astaxie/beego" {
return
}