completion: Prompt picking type of provider/data/resource automatically

Author: radeksimko

commands/completion_command.go

@@ -127,7 +127,7 @@ func (c *CompletionCommand) Run(args []string) int {
 	}
 
 	cc := &lsp.ClientCapabilities{}
-	items := ilsp.CompletionList(candidates, cc.TextDocument)
+	items := ilsp.ToCompletionList(candidates, cc.TextDocument)
 
 	c.Ui.Output(fmt.Sprintf("%#v", items))
 	return 0

internal/lsp/completion.go

@@ -6,9 +6,19 @@ import (
 	lsp "github.com/sourcegraph/go-lsp"
 )
 
-func CompletionList(candidates lang.Candidates, caps lsp.TextDocumentClientCapabilities) lsp.CompletionList {
-	list := lsp.CompletionList{
-		Items:        make([]lsp.CompletionItem, len(candidates.List)),
+type CompletionList struct {
+	IsIncomplete bool             `json:"isIncomplete"`
+	Items        []CompletionItem `json:"items"`
+}
+
+type CompletionItem struct {
+	lsp.CompletionItem
+	Command *lsp.Command `json:"command,omitempty"`
+}
+
+func ToCompletionList(candidates lang.Candidates, caps lsp.TextDocumentClientCapabilities) CompletionList {
+	list := CompletionList{
+		Items:        make([]CompletionItem, len(candidates.List)),
 		IsIncomplete: !candidates.IsComplete,
 	}
 
@@ -21,13 +31,13 @@ func CompletionList(candidates lang.Candidates, caps lsp.TextDocumentClientCapab
 	}
 
 	for i, c := range candidates.List {
-		list.Items[i] = CompletionItem(c, snippetSupport, markdown)
+		list.Items[i] = toCompletionItem(c, snippetSupport, markdown)
 	}
 
 	return list
 }
 
-func CompletionItem(candidate lang.Candidate, snippet, markdown bool) lsp.CompletionItem {
+func toCompletionItem(candidate lang.Candidate, snippet, markdown bool) CompletionItem {
 	doc := candidate.Description.Value
 
 	// TODO: revisit once go-lsp supports markdown in CompletionItem
@@ -44,14 +54,24 @@ func CompletionItem(candidate lang.Candidate, snippet, markdown bool) lsp.Comple
 	}
 
 	te, format := textEdit(candidate.TextEdit, snippet)
+	var cmd *lsp.Command
+	if candidate.TriggerSuggest {
+		cmd = &lsp.Command{
+			Command: "editor.action.triggerSuggest",
+			Title:   "Suggest",
+		}
+	}
 
-	return lsp.CompletionItem{
-		Label:            candidate.Label,
-		Kind:             kind,
-		InsertTextFormat: format,
-		Detail:           candidate.Detail,
-		Documentation:    doc,
-		TextEdit:         te,
+	return CompletionItem{
+		CompletionItem: lsp.CompletionItem{
+			Label:            candidate.Label,
+			Kind:             kind,
+			InsertTextFormat: format,
+			Detail:           candidate.Detail,
+			Documentation:    doc,
+			TextEdit:         te,
+		},
+		Command: cmd,
 	}
 }
 

langserver/handlers/complete.go

@@ -8,8 +8,8 @@ import (
 	lsp "github.com/sourcegraph/go-lsp"
 )
 
-func (h *logHandler) TextDocumentComplete(ctx context.Context, params lsp.CompletionParams) (lsp.CompletionList, error) {
-	var list lsp.CompletionList
+func (h *logHandler) TextDocumentComplete(ctx context.Context, params lsp.CompletionParams) (ilsp.CompletionList, error) {
+	var list ilsp.CompletionList
 
 	fs, err := lsctx.DocumentStorage(ctx)
 	if err != nil {
@@ -54,5 +54,5 @@ func (h *logHandler) TextDocumentComplete(ctx context.Context, params lsp.Comple
 	h.logger.Printf("Looking for candidates at %q -> %#v", file.Filename(), fPos.Position())
 	candidates, err := d.CandidatesAtPos(file.Filename(), fPos.Position())
 	h.logger.Printf("received candidates: %#v", candidates)
-	return ilsp.CompletionList(candidates, cc.TextDocument), err
+	return ilsp.ToCompletionList(candidates, cc.TextDocument), err
 }