mirror of
https://github.com/golang/go
synced 2024-11-18 17:04:41 -07:00
58bb5bbe30
Find references to identifiers in both a package and its test package. Change-Id: I9d9da4aa37c36c448336aed044df79cfd1c903f1 Reviewed-on: https://go-review.googlesource.com/c/tools/+/183990 Run-TryBot: Suzy Mueller <suzmue@golang.org> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Rebecca Stambler <rstambler@golang.org>
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
// Copyright 2019 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 lsp
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/tools/internal/lsp/protocol"
|
|
"golang.org/x/tools/internal/lsp/source"
|
|
"golang.org/x/tools/internal/span"
|
|
)
|
|
|
|
func (s *Server) references(ctx context.Context, params *protocol.ReferenceParams) ([]protocol.Location, error) {
|
|
uri := span.NewURI(params.TextDocument.URI)
|
|
view := s.session.ViewOf(uri)
|
|
f, m, err := getGoFile(ctx, view, uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
spn, err := m.PointSpan(params.Position)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rng, err := spn.Range(m.Converter)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Find all references to the identifier at the position.
|
|
ident, err := source.Identifier(ctx, view, f, rng.Start)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
references, err := ident.References(ctx)
|
|
if err != nil {
|
|
view.Session().Logger().Errorf(ctx, "no references for %s: %v", ident.Name, err)
|
|
}
|
|
// Get the location of each reference to return as the result.
|
|
locations := make([]protocol.Location, 0, len(references))
|
|
seen := make(map[span.Span]bool)
|
|
for _, ref := range references {
|
|
refSpan, err := ref.Range.Span()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if seen[refSpan] {
|
|
continue // already added this location
|
|
}
|
|
seen[refSpan] = true
|
|
|
|
_, refM, err := getSourceFile(ctx, view, refSpan.URI())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
loc, err := refM.Location(refSpan)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
locations = append(locations, loc)
|
|
}
|
|
return locations, nil
|
|
}
|