mirror of
https://github.com/golang/go
synced 2024-11-19 01:44:40 -07:00
73ad5c96a1
Change-Id: I899b3cd89901b99a5c65a5be79ac561289506623 Reviewed-on: https://go-review.googlesource.com/c/tools/+/193724 Run-TryBot: Rebecca Stambler <rstambler@golang.org> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Ian Cottrell <iancottrell@google.com>
42 lines
1.1 KiB
Go
42 lines
1.1 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 := s.session.ViewOf(uri)
|
|
f, err := getGoFile(ctx, view, uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ranges, err := source.FoldingRange(ctx, view, f, s.session.Options().LineFoldingOnly)
|
|
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
|
|
}
|