Use generated types for python 38

plugin/completion.py

@@ -3,7 +3,7 @@
 from .core.edit import apply_text_edits
 from .core.logging import debug
 from .core.promise import Promise
-from .core.protocol import CompletionEditRange
+from .core.protocol import EditRangeWithInsertReplace
 from .core.protocol import CompletionItem
 from .core.protocol import CompletionItemDefaults
 from .core.protocol import CompletionItemKind
@@ -127,7 +127,7 @@ def is_range(val: Any) -> TypeGuard[Range]:
     return isinstance(val, dict) and 'start' in val and 'end' in val
 
 
-def is_edit_range(val: Any) -> TypeGuard[CompletionEditRange]:
+def is_edit_range(val: Any) -> TypeGuard[EditRangeWithInsertReplace]:
     return isinstance(val, dict) and 'insert' in val and 'replace' in val
 
 

plugin/core/edit.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 from .logging import debug
+from .protocol import is_text_edit
 from .protocol import Position
 from .protocol import TextEdit
 from .protocol import UINT_MAX
@@ -24,7 +25,11 @@ def parse_workspace_edit(workspace_edit: WorkspaceEdit) -> WorkspaceChanges:
             uri = text_document['uri']
             version = text_document.get('version')
             edits = document_change.get('edits')
-            changes.setdefault(uri, ([], version))[0].extend(edits)
+            changes.setdefault(uri, ([], version))[0].extend(
+                # currently LSP code supprots only `TextEdit`s
+                # TODO: Extend WorkspaceEdit to support `AnnotatedTextEdit` and `SnippetTextEdit`
+                [edit for edit in edits if is_text_edit(edit)]
+            )
     else:
         raw_changes = workspace_edit.get('changes')
         if isinstance(raw_changes, dict):