| // Copyright 2020 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 server |
| |
| import ( |
| "context" |
| |
| "golang.org/x/tools/gopls/internal/file" |
| "golang.org/x/tools/gopls/internal/golang" |
| "golang.org/x/tools/gopls/internal/label" |
| "golang.org/x/tools/gopls/internal/protocol" |
| "golang.org/x/tools/gopls/internal/template" |
| "golang.org/x/tools/internal/event" |
| ) |
| |
| func (s *server) SemanticTokensFull(ctx context.Context, params *protocol.SemanticTokensParams) (*protocol.SemanticTokens, error) { |
| return s.semanticTokens(ctx, params.TextDocument, nil) |
| } |
| |
| func (s *server) SemanticTokensRange(ctx context.Context, params *protocol.SemanticTokensRangeParams) (*protocol.SemanticTokens, error) { |
| return s.semanticTokens(ctx, params.TextDocument, ¶ms.Range) |
| } |
| |
| func (s *server) semanticTokens(ctx context.Context, td protocol.TextDocumentIdentifier, rng *protocol.Range) (*protocol.SemanticTokens, error) { |
| ctx, done := event.Start(ctx, "lsp.Server.semanticTokens", label.URI.Of(td.URI)) |
| defer done() |
| |
| fh, snapshot, release, err := s.fileOf(ctx, td.URI) |
| if err != nil { |
| return nil, err |
| } |
| defer release() |
| if !snapshot.Options().SemanticTokens { |
| // Note: returning new(protocol.SemanticTokens) is necessary here to |
| // invalidate semantic tokens in VS Code (and perhaps other editors). |
| // Previously, an error was returned here to achieve the same effect, but |
| // that had the side effect of very noisy "semantictokens are disabled" |
| // logs on every keystroke. |
| return new(protocol.SemanticTokens), nil |
| } |
| |
| switch snapshot.FileKind(fh) { |
| case file.Tmpl: |
| return template.SemanticTokens(ctx, snapshot, fh.URI()) |
| |
| case file.Go: |
| return golang.SemanticTokens(ctx, snapshot, fh, rng) |
| |
| default: |
| // TODO(adonovan): should return an error! |
| return nil, nil // empty result |
| } |
| } |