go/types, types2: implement Info.PkgNameOf

[gostls13.git] / src / cmd / compile / internal / types2 / api.go

diff --git a/src/cmd/compile/internal/types2/api.go b/src/cmd/compile/internal/types2/api.go
index 675882c49f1b8d874e6355d8b9b6cd322e33be8c..2093ceb817a01bd0532ed2be61c14353501ea9b0 100644 (file)
--- a/src/cmd/compile/internal/types2/api.go
+++ b/src/cmd/compile/internal/types2/api.go
@@ -342,6 +342,23 @@ func (info *Info) ObjectOf(id *syntax.Name) Object {
        return info.Uses[id]
 }
 
+// PkgNameOf returns the local package name defined by the import,
+// or nil if not found.
+//
+// For dot-imports, the package name is ".".
+//
+// Precondition: the Defs and Implicts maps are populated.
+func (info *Info) PkgNameOf(imp *syntax.ImportDecl) *PkgName {
+       var obj Object
+       if imp.LocalPkgName != nil {
+               obj = info.Defs[imp.LocalPkgName]
+       } else {
+               obj = info.Implicits[imp]
+       }
+       pkgname, _ := obj.(*PkgName)
+       return pkgname
+}
+
 // TypeAndValue reports the type and value (for constants)
 // of the corresponding expression.
 type TypeAndValue struct {