mirror of
https://github.com/golang/go
synced 2024-11-18 14:14:46 -07:00
75aaabac35
also change the return type to be and end function and not an incomplete span Change-Id: Icd99d93ac98a0f8088f33e905cf1ee3fe410c024 Reviewed-on: https://go-review.googlesource.com/c/tools/+/185349 Run-TryBot: Ian Cottrell <iancottrell@google.com> Reviewed-by: Rebecca Stambler <rstambler@golang.org>
80 lines
2.1 KiB
Go
80 lines
2.1 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/lsp/telemetry/trace"
|
|
"golang.org/x/tools/internal/span"
|
|
)
|
|
|
|
func (s *Server) documentSymbol(ctx context.Context, params *protocol.DocumentSymbolParams) ([]protocol.DocumentSymbol, error) {
|
|
ctx, done := trace.StartSpan(ctx, "lsp.Server.documentSymbol")
|
|
defer done()
|
|
uri := span.NewURI(params.TextDocument.URI)
|
|
view := s.session.ViewOf(uri)
|
|
f, m, err := getGoFile(ctx, view, uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
symbols, err := source.DocumentSymbols(ctx, f)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toProtocolDocumentSymbols(m, symbols), nil
|
|
}
|
|
|
|
func toProtocolDocumentSymbols(m *protocol.ColumnMapper, symbols []source.Symbol) []protocol.DocumentSymbol {
|
|
result := make([]protocol.DocumentSymbol, 0, len(symbols))
|
|
for _, s := range symbols {
|
|
ps := protocol.DocumentSymbol{
|
|
Name: s.Name,
|
|
Kind: toProtocolSymbolKind(s.Kind),
|
|
Detail: s.Detail,
|
|
Children: toProtocolDocumentSymbols(m, s.Children),
|
|
}
|
|
if r, err := m.Range(s.Span); err == nil {
|
|
ps.Range = r
|
|
}
|
|
if r, err := m.Range(s.SelectionSpan); err == nil {
|
|
ps.SelectionRange = r
|
|
}
|
|
result = append(result, ps)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func toProtocolSymbolKind(kind source.SymbolKind) protocol.SymbolKind {
|
|
switch kind {
|
|
case source.StructSymbol:
|
|
return protocol.Struct
|
|
case source.PackageSymbol:
|
|
return protocol.Package
|
|
case source.VariableSymbol:
|
|
return protocol.Variable
|
|
case source.ConstantSymbol:
|
|
return protocol.Constant
|
|
case source.FunctionSymbol:
|
|
return protocol.Function
|
|
case source.MethodSymbol:
|
|
return protocol.Method
|
|
case source.InterfaceSymbol:
|
|
return protocol.Interface
|
|
case source.NumberSymbol:
|
|
return protocol.Number
|
|
case source.StringSymbol:
|
|
return protocol.String
|
|
case source.BooleanSymbol:
|
|
return protocol.Boolean
|
|
case source.FieldSymbol:
|
|
return protocol.Field
|
|
default:
|
|
return 0
|
|
}
|
|
}
|