mirror of
https://github.com/golang/go
synced 2024-11-18 21:54:49 -07:00
2208e1677e
This change eliminates the extra step of calling GetFile on the view and getting the FileHandle from the snapshot. It also eliminiates the redundant source.File type. Follow up changes will clean up the file kind handling, since it still exists on the fileBase type. Change-Id: I635ab8632821b36e062be5151eaab425a5698f60 Reviewed-on: https://go-review.googlesource.com/c/tools/+/211778 Run-TryBot: Rebecca Stambler <rstambler@golang.org> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Heschi Kreinick <heschi@google.com>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
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) foldingRange(ctx context.Context, params *protocol.FoldingRangeParams) ([]protocol.FoldingRange, error) {
|
|
uri := span.NewURI(params.TextDocument.URI)
|
|
view, err := s.session.ViewOf(uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
snapshot := view.Snapshot()
|
|
fh, err := snapshot.GetFile(ctx, uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var ranges []*source.FoldingRangeInfo
|
|
switch fh.Identity().Kind {
|
|
case source.Go:
|
|
ranges, err = source.FoldingRange(ctx, snapshot, fh, view.Options().LineFoldingOnly)
|
|
case source.Mod:
|
|
ranges = nil
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toProtocolFoldingRanges(ranges)
|
|
}
|
|
|
|
func toProtocolFoldingRanges(ranges []*source.FoldingRangeInfo) ([]protocol.FoldingRange, error) {
|
|
result := make([]protocol.FoldingRange, 0, len(ranges))
|
|
for _, info := range ranges {
|
|
rng, err := info.Range()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result = append(result, protocol.FoldingRange{
|
|
StartLine: rng.Start.Line,
|
|
StartCharacter: rng.Start.Character,
|
|
EndLine: rng.End.Line,
|
|
EndCharacter: rng.End.Character,
|
|
Kind: string(info.Kind),
|
|
})
|
|
}
|
|
return result, nil
|
|
}
|