mirror of
https://github.com/golang/go
synced 2024-11-19 02:04:42 -07:00
90c8b4f75b
This change makes sure to handle escaped characters, for instance, files in the module cache contain "@" in the path. Also, return errors when a package is not found for a file. Updates golang/go#30027 Change-Id: I6c4f6d0f30dde55701308b89785f02b671e81cc6 Reviewed-on: https://go-review.googlesource.com/c/161077 Reviewed-by: Ian Cottrell <iancottrell@google.com> Reviewed-by: Yasuhiro MATSUMOTO <mattn.jp@gmail.com>
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package lsp
|
|
|
|
import (
|
|
"context"
|
|
"go/token"
|
|
|
|
"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 protocol.DocumentURI, 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, err := f.GetToken()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var r source.Range
|
|
if rng == nil {
|
|
r.Start = tok.Pos(0)
|
|
r.End = tok.Pos(tok.Size())
|
|
} else {
|
|
r = fromProtocolRange(tok, *rng)
|
|
}
|
|
content, err := f.Read()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
edits, err := source.Format(ctx, f, r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return toProtocolEdits(tok, content, edits), nil
|
|
}
|
|
|
|
func toProtocolEdits(tok *token.File, content []byte, edits []source.TextEdit) []protocol.TextEdit {
|
|
if edits == nil {
|
|
return nil
|
|
}
|
|
// 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
|
|
}
|