| // 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/lsp/telemetry" |
| "golang.org/x/tools/internal/span" |
| "golang.org/x/tools/internal/telemetry/log" |
| ) |
| |
| func (s *Server) documentHighlight(ctx context.Context, params *protocol.DocumentHighlightParams) ([]protocol.DocumentHighlight, error) { |
| uri := span.NewURI(params.TextDocument.URI) |
| view, err := s.session.ViewOf(uri) |
| if err != nil { |
| return nil, err |
| } |
| snapshot := view.Snapshot() |
| f, err := view.GetFile(ctx, uri) |
| if err != nil { |
| return nil, err |
| } |
| |
| var rngs []protocol.Range |
| switch f.Kind() { |
| case source.Go: |
| rngs, err = source.Highlight(ctx, snapshot, f, params.Position) |
| case source.Mod: |
| return nil, nil |
| } |
| |
| if err != nil { |
| log.Error(ctx, "no highlight", err, telemetry.URI.Of(uri)) |
| } |
| return toProtocolHighlight(rngs), nil |
| } |
| |
| func toProtocolHighlight(rngs []protocol.Range) []protocol.DocumentHighlight { |
| result := make([]protocol.DocumentHighlight, 0, len(rngs)) |
| kind := protocol.Text |
| for _, rng := range rngs { |
| result = append(result, protocol.DocumentHighlight{ |
| Kind: kind, |
| Range: rng, |
| }) |
| } |
| return result |
| } |