qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-10-01 03:38:32 -06:00
Code Releases Activity

internal/lsp: organize imports as a quickfix

Browse Source 
This change adds the ability to organize all imports as quickfix on a
diagnostic, in addition to the standard "source.organizeImports" code
action kind. This change still requires a bit of work, since really, we
should only add one import per diagnostic and each edit should be the
edits necessary for the addition or removal of the individual import,
but this is a good start.

Change-Id: I57679446bec833d215c35533240102fb61c86f20
Reviewed-on: https://go-review.googlesource.com/c/tools/+/172397
Run-TryBot: Rebecca Stambler <rstambler@golang.org>
Reviewed-by: Ian Cottrell <iancottrell@google.com>
This commit is contained in:
Rebecca Stambler 2019-04-16 12:50:59 -04:00
parent 52f869c107
commit b8fc0e1722
1 changed files with 37 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
internal/lsp/code_action.go
View File

@ -28,12 +28,29 @@ func (s *Server) codeAction(ctx context.Context, params *protocol.CodeActionPara
var codeActions []protocol.CodeAction
// Determine what code actions we should take based on the diagnostics.
if findImportErrors(params.Context.Diagnostics) {
codeAction, err := organizeImports(ctx, view, spn)
edits, err := organizeImports(ctx, view, spn)
if err != nil {
return nil, err
}
if codeAction != nil {
codeActions = append(codeActions, *codeAction)
if len(edits) > 0 {
// TODO(rstambler): Handle params.Context.Only when VSCode-Go uses a
// version of vscode-languageclient that fixes
// https://github.com/Microsoft/vscode-languageserver-node/issues/442.
codeActions = append(codeActions, protocol.CodeAction{
Title: "Organize Imports",
Kind: protocol.SourceOrganizeImports,
Edit: &protocol.WorkspaceEdit{
Changes: &map[string][]protocol.TextEdit{
string(spn.URI()): edits,
},
},
})
// Add any quick fixes for each import-related diagnostic that we see.
fixes, err := quickFixes(spn.URI(), params.Context.Diagnostics, edits)
if err != nil {
return nil, err
}
codeActions = append(codeActions, fixes...)
}
}
return codeActions, nil
@ -60,7 +77,7 @@ func findImportErrors(diagnostics []protocol.Diagnostic) bool {
return false
}
func organizeImports(ctx context.Context, v source.View, s span.Span) (*protocol.CodeAction, error) {
func organizeImports(ctx context.Context, v source.View, s span.Span) ([]protocol.TextEdit, error) {
f, m, err := newColumnMap(ctx, v, s.URI())
if err != nil {
return nil, err
@ -81,21 +98,22 @@ func organizeImports(ctx context.Context, v source.View, s span.Span) (*protocol
if err != nil {
return nil, err
}
protocolEdits, err := ToProtocolEdits(m, edits)
if err != nil {
return nil, err
}
if len(protocolEdits) == 0 {
return nil, nil
}
codeAction := protocol.CodeAction{
Title: "Organize Imports",
Kind: protocol.SourceOrganizeImports,
return ToProtocolEdits(m, edits)
}
// TODO(rstambler): Separate this into a set of codeActions per diagnostic,
// where each action is the addition or removal of one import.
// This can only be done when https://golang.org/issue/31493 is resolved.
func quickFixes(uri span.URI, diagnostics []protocol.Diagnostic, edits []protocol.TextEdit) ([]protocol.CodeAction, error) {
return []protocol.CodeAction{
{
Title: "Organize All Imports",
Kind: protocol.QuickFix,
Edit: &protocol.WorkspaceEdit{
Changes: &map[string][]protocol.TextEdit{
string(s.URI()): protocolEdits,
string(uri): edits,
},
},
}
return &codeAction, nil
},
}, nil
}