122 lines
3.4 KiB
Go
122 lines
3.4 KiB
Go
// Copyright 2013 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package loader
|
|
|
|
import (
|
|
"fmt"
|
|
"go/ast"
|
|
|
|
"code.google.com/p/go.tools/go/exact"
|
|
"code.google.com/p/go.tools/go/types"
|
|
)
|
|
|
|
// PackageInfo holds the ASTs and facts derived by the type-checker
|
|
// for a single package.
|
|
//
|
|
// Not mutated once exposed via the API.
|
|
//
|
|
type PackageInfo struct {
|
|
Pkg *types.Package
|
|
Importable bool // true if 'import "Pkg.Path()"' would resolve to this
|
|
TransitivelyErrorFree bool // true if Pkg and all its dependencies are free of errors
|
|
Files []*ast.File // abstract syntax for the package's files
|
|
Errors []error // non-nil if the package had errors
|
|
types.Info // type-checker deductions.
|
|
|
|
checker *types.Checker // transient type-checker state
|
|
}
|
|
|
|
func (info *PackageInfo) String() string {
|
|
return fmt.Sprintf("PackageInfo(%s)", info.Pkg.Path())
|
|
}
|
|
|
|
// TypeOf returns the type of expression e.
|
|
// Precondition: e belongs to the package's ASTs.
|
|
//
|
|
func (info *PackageInfo) TypeOf(e ast.Expr) types.Type {
|
|
if t, ok := info.Types[e]; ok {
|
|
return t.Type
|
|
}
|
|
// Defining ast.Idents (id := expr) get only Ident callbacks
|
|
// but not Expr callbacks.
|
|
if id, ok := e.(*ast.Ident); ok {
|
|
return info.ObjectOf(id).Type()
|
|
}
|
|
panic("no type for expression")
|
|
}
|
|
|
|
// ValueOf returns the value of expression e if it is a constant, nil
|
|
// otherwise.
|
|
// Precondition: e belongs to the package's ASTs.
|
|
//
|
|
func (info *PackageInfo) ValueOf(e ast.Expr) exact.Value {
|
|
return info.Types[e].Value
|
|
}
|
|
|
|
// ObjectOf returns the typechecker object denoted by the specified id.
|
|
//
|
|
// If id is an anonymous struct field, the field (*types.Var) is
|
|
// returned, not the type (*types.TypeName).
|
|
//
|
|
// Precondition: id belongs to the package's ASTs.
|
|
//
|
|
func (info *PackageInfo) ObjectOf(id *ast.Ident) types.Object {
|
|
obj, ok := info.Defs[id]
|
|
if ok {
|
|
return obj
|
|
}
|
|
return info.Uses[id]
|
|
}
|
|
|
|
// IsType returns true iff expression e denotes a type.
|
|
// Precondition: e belongs to the package's ASTs.
|
|
//
|
|
// TODO(gri): move this into go/types.
|
|
//
|
|
func (info *PackageInfo) IsType(e ast.Expr) bool {
|
|
switch e := e.(type) {
|
|
case *ast.SelectorExpr: // pkg.Type
|
|
if _, ok := info.Selections[e]; !ok {
|
|
// qualified identifier
|
|
_, isType := info.Uses[e.Sel].(*types.TypeName)
|
|
return isType
|
|
}
|
|
case *ast.StarExpr: // *T
|
|
return info.IsType(e.X)
|
|
case *ast.Ident:
|
|
_, isType := info.ObjectOf(e).(*types.TypeName)
|
|
return isType
|
|
case *ast.ArrayType, *ast.StructType, *ast.FuncType, *ast.InterfaceType, *ast.MapType, *ast.ChanType:
|
|
return true
|
|
case *ast.ParenExpr:
|
|
return info.IsType(e.X)
|
|
}
|
|
return false
|
|
}
|
|
|
|
// TypeCaseVar returns the implicit variable created by a single-type
|
|
// case clause in a type switch, or nil if not found.
|
|
//
|
|
func (info *PackageInfo) TypeCaseVar(cc *ast.CaseClause) *types.Var {
|
|
if v := info.Implicits[cc]; v != nil {
|
|
return v.(*types.Var)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ImportSpecPkg returns the PkgName for a given ImportSpec, possibly
|
|
// an implicit one for a dot-import or an import-without-rename.
|
|
// It returns nil if not found.
|
|
//
|
|
func (info *PackageInfo) ImportSpecPkg(spec *ast.ImportSpec) *types.PkgName {
|
|
if spec.Name != nil {
|
|
return info.ObjectOf(spec.Name).(*types.PkgName)
|
|
}
|
|
if p := info.Implicits[spec]; p != nil {
|
|
return p.(*types.PkgName)
|
|
}
|
|
return nil
|
|
}
|