Resolve completions completion item before applying additional text edits

boot.py

@@ -5,8 +5,7 @@
 
 # Please keep this list sorted (Edit -> Sort Lines)
 from .plugin.code_actions import LspCodeActionsCommand
-from .plugin.completion import LspCompleteInsertTextCommand
-from .plugin.completion import LspCompleteTextEditCommand
+from .plugin.completion import LspSelectCompletionItemCommand
 from .plugin.completion import LspResolveDocsCommand
 from .plugin.configuration import LspDisableLanguageServerGloballyCommand
 from .plugin.configuration import LspDisableLanguageServerInProjectCommand

plugin/completion.py

@@ -1,11 +1,10 @@
 import sublime
-import sublime_plugin
 import webbrowser
 from .core.logging import debug
 from .core.edit import parse_text_edit
 from .core.protocol import Request, InsertTextFormat, Range, CompletionItem
 from .core.registry import LspTextCommand
-from .core.typing import Any, List, Dict, Optional, Generator, Union
+from .core.typing import List, Dict, Optional, Generator, Union
 from .core.views import FORMAT_STRING, FORMAT_MARKUP_CONTENT, minihtml
 from .core.views import range_to_region
 from .core.views import show_lsp_popup
@@ -67,52 +66,27 @@ def _on_navigate(self, url: str) -> None:
         webbrowser.open(url)
 
 
-class LspCompleteCommand(sublime_plugin.TextCommand):
-
-    def epilogue(self, item: CompletionItem, session_name: Optional[str] = None) -> None:
-        additional_edits = item.get('additionalTextEdits')
-        if additional_edits:
-            edits = [parse_text_edit(additional_edit) for additional_edit in additional_edits]
-            self.view.run_command("lsp_apply_document_edit", {'changes': edits})
-        command = item.get("command")
-        if command:
-            debug('Running server command "{}" for view {}'.format(command, self.view.id()))
-            args = {
-                "command_name": command["command"],
-                "command_args": command.get("arguments"),
-                "session_name": session_name
-            }
-            self.view.run_command("lsp_execute", args)
-
-
-class LspCompleteInsertTextCommand(LspCompleteCommand):
-
-    def run(self, edit: sublime.Edit, item: Any, session_name: Optional[str] = None) -> None:
-        insert_text = item.get("insertText") or item["label"]
-        if item.get("insertTextFormat", InsertTextFormat.PlainText) == InsertTextFormat.Snippet:
-            self.view.run_command("insert_snippet", {"contents": insert_text})
-        else:
-            self.view.run_command("insert", {"characters": insert_text})
-        self.epilogue(item, session_name)
-
-
-class LspCompleteTextEditCommand(LspCompleteCommand):
-
-    def run(self, edit: sublime.Edit, item: CompletionItem, session_name: Optional[str] = None) -> None:
+class LspSelectCompletionItemCommand(LspTextCommand):
+    def run(self, edit: sublime.Edit, item: CompletionItem, session_name: str) -> None:
         text_edit = item.get("textEdit")
-        if not text_edit:
-            return
-        new_text = text_edit["newText"]
-        edit_region = range_to_region(Range.from_lsp(text_edit['range']), self.view)
-        if item.get("insertTextFormat", InsertTextFormat.PlainText) == InsertTextFormat.Snippet:
-            for region in self.translated_regions(edit_region):
-                self.view.erase(edit, region)
-            self.view.run_command("insert_snippet", {"contents": new_text})
+        if text_edit:
+            new_text = text_edit["newText"]
+            edit_region = range_to_region(Range.from_lsp(text_edit['range']), self.view)
+            if item.get("insertTextFormat", InsertTextFormat.PlainText) == InsertTextFormat.Snippet:
+                for region in self.translated_regions(edit_region):
+                    self.view.erase(edit, region)
+                self.view.run_command("insert_snippet", {"contents": new_text})
+            else:
+                for region in self.translated_regions(edit_region):
+                    # NOTE: Cannot do .replace, because ST will select the replacement.
+                    self.view.erase(edit, region)
+                    self.view.insert(edit, region.a, new_text)
         else:
-            for region in self.translated_regions(edit_region):
-                # NOTE: Cannot do .replace, because ST will select the replacement.
-                self.view.erase(edit, region)
-                self.view.insert(edit, region.a, new_text)
+            insert_text = item.get("insertText") or item.get("label")
+            if item.get("insertTextFormat", InsertTextFormat.PlainText) == InsertTextFormat.Snippet:
+                self.view.run_command("insert_snippet", {"contents": insert_text})
+            else:
+                self.view.run_command("insert", {"characters": insert_text})
         self.epilogue(item, session_name)
 
     def translated_regions(self, edit_region: sublime.Region) -> Generator[sublime.Region, None, None]:
@@ -125,3 +99,26 @@ def translated_regions(self, edit_region: sublime.Region) -> Generator[sublime.R
             translation = region.b - primary_cursor_position
             translated_edit_region = sublime.Region(edit_region.a + translation, edit_region.b + translation)
             yield translated_edit_region
+
+    def epilogue(self, item: CompletionItem, session_name: str) -> None:
+        session = self.session_by_name(session_name, 'completionProvider.resolveProvider')
+        if session:
+            request = Request.resolveCompletionItem(item, self.view)
+            session.send_request_async(request, lambda response: self.on_resolved(response, session_name))
+        else:
+            self.on_resolved(item, session_name)
+
+    def on_resolved(self, item: CompletionItem, session_name: str) -> None:
+        additional_edits = item.get('additionalTextEdits')
+        if additional_edits:
+            edits = [parse_text_edit(additional_edit) for additional_edit in additional_edits]
+            self.view.run_command("lsp_apply_document_edit", {'changes': edits})
+        command = item.get("command")
+        if command:
+            debug('Running server command "{}" for view {}'.format(command, self.view.id()))
+            args = {
+                "command_name": command["command"],
+                "command_args": command.get("arguments"),
+                "session_name": session_name
+            }
+            self.view.run_command("lsp_execute", args)

plugin/core/sessions.py

@@ -155,6 +155,9 @@ def get_initialize_params(variables: Dict[str, str], workspace_folders: List[Wor
                     "documentationFormat": ["markdown", "plaintext"],
                     "tagSupport": {
                         "valueSet": completion_tag_value_set
+                    },
+                    "resolveSupport": {
+                        "properties": ["detail", "documentation", "additionalTextEdits", "command"]
                     }
                 },
                 "completionItemKind": {

plugin/core/views.py

@@ -4,7 +4,6 @@
 from .protocol import Diagnostic
 from .protocol import DiagnosticRelatedInformation
 from .protocol import DiagnosticSeverity
-from .protocol import InsertTextFormat
 from .protocol import Location
 from .protocol import LocationLink
 from .protocol import Notification
@@ -718,34 +717,17 @@ def format_completion(
     else:
         st_trigger = lsp_label
 
-    # NOTE: Some servers return "textEdit": null. We have to check if it's truthy.
-    if item.get("textEdit") or item.get("additionalTextEdits") or item.get("command"):
-        command = "lsp_complete_text_edit" if item.get("textEdit") else "lsp_complete_insert_text"
-        args = {"item": item}  # type: Dict[str, Any]
-        if item.get("command"):
-            # If there is a "command", we'll have to run the command on the same session so store the name.
-            args["session_name"] = session_name
-        completion = sublime.CompletionItem.command_completion(
-            trigger=st_trigger,
-            command=command,
-            args=args,
-            annotation=st_annotation,
-            kind=kind,
-            details=st_details)
-        if item.get("textEdit"):
-            completion.flags = sublime.COMPLETION_FLAG_KEEP_PREFIX
-    else:
-        # A plain old completion suffices for insertText with no additionalTextEdits and no command.
-        if item.get("insertTextFormat", InsertTextFormat.PlainText) == InsertTextFormat.PlainText:
-            st_format = sublime.COMPLETION_FORMAT_TEXT
-        else:
-            st_format = sublime.COMPLETION_FORMAT_SNIPPET
-        completion = sublime.CompletionItem(
-            trigger=st_trigger,
-            annotation=st_annotation,
-            completion=item.get("insertText") or item["label"],
-            completion_format=st_format,
-            kind=kind,
-            details=st_details)
-
+    args = {
+        "item": item,
+        "session_name": session_name
+    }  # type: Dict[str, Any]
+    completion = sublime.CompletionItem.command_completion(
+        trigger=st_trigger,
+        command="lsp_select_completion_item",
+        args=args,
+        annotation=st_annotation,
+        kind=kind,
+        details=st_details)
+    if item.get("textEdit"):
+        completion.flags = sublime.COMPLETION_FLAG_KEEP_PREFIX
     return completion