mirror of
https://github.com/golang/go.git
synced 2025-05-07 00:23:03 +00:00
6a988fd03a
Gopls presently uses hand-coded data types (in internal/lsp/protocol) for communicating with LSP clients. Instead, modify it to use the automatically generated file (internal/lsp/protocol/tsprotocol.go). Replaced files have been put (temporarily) in a directory 'preserve' so readers can compare the old data types with the new ones. Change-Id: Idfa53a5783e2d6a47e03b20641dd76fbc2c32677 Reviewed-on: https://go-review.googlesource.com/c/tools/+/166757 Run-TryBot: Peter Weinberger <pjw@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Rebecca Stambler <rstambler@golang.org>
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package lsp
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/tools/internal/lsp/protocol"
|
|
"golang.org/x/tools/internal/lsp/source"
|
|
)
|
|
|
|
// formatRange formats a document with a given range.
|
|
func formatRange(ctx context.Context, v source.View, uri string, rng *protocol.Range) ([]protocol.TextEdit, error) {
|
|
sourceURI, err := fromProtocolURI(uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
f, err := v.GetFile(ctx, sourceURI)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
tok := f.GetToken(ctx)
|
|
var r source.Range
|
|
if rng == nil {
|
|
r.Start = tok.Pos(0)
|
|
r.End = tok.Pos(tok.Size())
|
|
} else {
|
|
r = fromProtocolRange(tok, *rng)
|
|
}
|
|
edits, err := source.Format(ctx, f, r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toProtocolEdits(ctx, f, edits), nil
|
|
}
|
|
|
|
func toProtocolEdits(ctx context.Context, f source.File, edits []source.TextEdit) []protocol.TextEdit {
|
|
if edits == nil {
|
|
return nil
|
|
}
|
|
tok := f.GetToken(ctx)
|
|
content := f.GetContent(ctx)
|
|
// When a file ends with an empty line, the newline character is counted
|
|
// as part of the previous line. This causes the formatter to insert
|
|
// another unnecessary newline on each formatting. We handle this case by
|
|
// checking if the file already ends with a newline character.
|
|
hasExtraNewline := content[len(content)-1] == '\n'
|
|
result := make([]protocol.TextEdit, len(edits))
|
|
for i, edit := range edits {
|
|
rng := toProtocolRange(tok, edit.Range)
|
|
// If the edit ends at the end of the file, add the extra line.
|
|
if hasExtraNewline && tok.Offset(edit.Range.End) == len(content) {
|
|
rng.End.Line++
|
|
rng.End.Character = 0
|
|
}
|
|
result[i] = protocol.TextEdit{
|
|
Range: rng,
|
|
NewText: edit.NewText,
|
|
}
|
|
}
|
|
return result
|
|
}
|