+++ /dev/null
-// 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 main
-
-import (
- "fmt"
- "go/ast"
- "go/build"
- "go/parser"
- "go/token"
- pathpkg "path"
- "path/filepath"
- "strconv"
-
- "golang.org/x/tools/cmd/guru/serial"
- "golang.org/x/tools/go/buildutil"
- "golang.org/x/tools/go/loader"
-)
-
-// definition reports the location of the definition of an identifier.
-func definition(q *Query) error {
- // First try the simple resolution done by parser.
- // It only works for intra-file references but it is very fast.
- // (Extending this approach to all the files of the package,
- // resolved using ast.NewPackage, was not worth the effort.)
- {
- qpos, err := fastQueryPos(q.Build, q.Pos)
- if err != nil {
- return err
- }
-
- id, _ := qpos.path[0].(*ast.Ident)
- if id == nil {
- return fmt.Errorf("no identifier here")
- }
-
- // Did the parser resolve it to a local object?
- if obj := id.Obj; obj != nil && obj.Pos().IsValid() {
- q.Output(qpos.fset, &definitionResult{
- pos: obj.Pos(),
- descr: fmt.Sprintf("%s %s", obj.Kind, obj.Name),
- })
- return nil // success
- }
-
- // Qualified identifier?
- if pkg := packageForQualIdent(qpos.path, id); pkg != "" {
- srcdir := filepath.Dir(qpos.fset.File(qpos.start).Name())
- tok, pos, err := findPackageMember(q.Build, qpos.fset, srcdir, pkg, id.Name)
- if err != nil {
- return err
- }
- q.Output(qpos.fset, &definitionResult{
- pos: pos,
- descr: fmt.Sprintf("%s %s.%s", tok, pkg, id.Name),
- })
- return nil // success
- }
-
- // Fall back on the type checker.
- }
-
- // Run the type checker.
- lconf := loader.Config{Build: q.Build}
- allowErrors(&lconf)
-
- if _, err := importQueryPackage(q.Pos, &lconf); err != nil {
- return err
- }
-
- // Load/parse/type-check the program.
- lprog, err := lconf.Load()
- if err != nil {
- return err
- }
-
- qpos, err := parseQueryPos(lprog, q.Pos, false)
- if err != nil {
- return err
- }
-
- id, _ := qpos.path[0].(*ast.Ident)
- if id == nil {
- return fmt.Errorf("no identifier here")
- }
-
- // Look up the declaration of this identifier.
- // If id is an anonymous field declaration,
- // it is both a use of a type and a def of a field;
- // prefer the use in that case.
- obj := qpos.info.Uses[id]
- if obj == nil {
- obj = qpos.info.Defs[id]
- if obj == nil {
- // Happens for y in "switch y := x.(type)",
- // and the package declaration,
- // but I think that's all.
- return fmt.Errorf("no object for identifier")
- }
- }
-
- if !obj.Pos().IsValid() {
- return fmt.Errorf("%s is built in", obj.Name())
- }
-
- q.Output(lprog.Fset, &definitionResult{
- pos: obj.Pos(),
- descr: qpos.objectString(obj),
- })
- return nil
-}
-
-// packageForQualIdent returns the package p if id is X in a qualified
-// identifier p.X; it returns "" otherwise.
-//
-// Precondition: id is path[0], and the parser did not resolve id to a
-// local object. For speed, packageForQualIdent assumes that p is a
-// package iff it is the basename of an import path (and not, say, a
-// package-level decl in another file or a predeclared identifier).
-func packageForQualIdent(path []ast.Node, id *ast.Ident) string {
- if sel, ok := path[1].(*ast.SelectorExpr); ok && sel.Sel == id && ast.IsExported(id.Name) {
- if pkgid, ok := sel.X.(*ast.Ident); ok && pkgid.Obj == nil {
- f := path[len(path)-1].(*ast.File)
- for _, imp := range f.Imports {
- path, _ := strconv.Unquote(imp.Path.Value)
- if imp.Name != nil {
- if imp.Name.Name == pkgid.Name {
- return path // renaming import
- }
- } else if pathpkg.Base(path) == pkgid.Name {
- return path // ordinary import
- }
- }
- }
- }
- return ""
-}
-
-// findPackageMember returns the type and position of the declaration of
-// pkg.member by loading and parsing the files of that package.
-// srcdir is the directory in which the import appears.
-func findPackageMember(ctxt *build.Context, fset *token.FileSet, srcdir, pkg, member string) (token.Token, token.Pos, error) {
- bp, err := ctxt.Import(pkg, srcdir, 0)
- if err != nil {
- return 0, token.NoPos, err // no files for package
- }
-
- // TODO(adonovan): opt: parallelize.
- for _, fname := range bp.GoFiles {
- filename := filepath.Join(bp.Dir, fname)
-
- // Parse the file, opening it the file via the build.Context
- // so that we observe the effects of the -modified flag.
- f, _ := buildutil.ParseFile(fset, ctxt, nil, ".", filename, parser.Mode(0))
- if f == nil {
- continue
- }
-
- // Find a package-level decl called 'member'.
- for _, decl := range f.Decls {
- switch decl := decl.(type) {
- case *ast.GenDecl:
- for _, spec := range decl.Specs {
- switch spec := spec.(type) {
- case *ast.ValueSpec:
- // const or var
- for _, id := range spec.Names {
- if id.Name == member {
- return decl.Tok, id.Pos(), nil
- }
- }
- case *ast.TypeSpec:
- if spec.Name.Name == member {
- return token.TYPE, spec.Name.Pos(), nil
- }
- }
- }
- case *ast.FuncDecl:
- if decl.Recv == nil && decl.Name.Name == member {
- return token.FUNC, decl.Name.Pos(), nil
- }
- }
- }
- }
-
- return 0, token.NoPos, fmt.Errorf("couldn't find declaration of %s in %q", member, pkg)
-}
-
-type definitionResult struct {
- pos token.Pos // (nonzero) location of definition
- descr string // description of object it denotes
-}
-
-func (r *definitionResult) PrintPlain(printf printfFunc) {
- printf(r.pos, "defined here as %s", r.descr)
-}
-
-func (r *definitionResult) JSON(fset *token.FileSet) []byte {
- return toJSON(&serial.Definition{
- Desc: r.descr,
- ObjPos: fset.Position(r.pos).String(),
- })
-}