-
Notifications
You must be signed in to change notification settings - Fork 181
/
execute_command.py
99 lines (87 loc) · 4.66 KB
/
execute_command.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from .core.protocol import Error
from .core.protocol import ExecuteCommandParams
from .core.registry import LspTextCommand
from .core.registry import windows
from .core.typing import List, Optional, Any
from .core.views import first_selection_region
from .core.views import uri_from_view, offset_to_point, region_to_range, text_document_identifier, text_document_position_params # noqa: E501
import sublime
class LspExecuteCommand(LspTextCommand):
"""
Helper command for triggering workspace/executeCommand requests.
"""
def run(self,
edit: sublime.Edit,
command_name: Optional[str] = None,
command_args: Optional[List[Any]] = None,
session_name: Optional[str] = None,
event: Optional[dict] = None) -> None:
# Handle VSCode-specific command for triggering AC/sighelp
if command_name == "editor.action.triggerSuggest":
# Triggered from set_timeout as suggestions popup doesn't trigger otherwise.
return sublime.set_timeout(lambda: self.view.run_command("auto_complete"))
if command_name == "editor.action.triggerParameterHints":
def run_async() -> None:
listener = windows.listener_for_view(self.view)
if listener:
listener.do_signature_help_async(manual=False)
return sublime.set_timeout_async(run_async)
session = self.session_by_name(session_name if session_name else self.session_name)
if session and command_name:
params = {"command": command_name} # type: ExecuteCommandParams
if command_args:
params["arguments"] = self._expand_variables(command_args)
def handle_response(response: Any) -> None:
assert command_name
if isinstance(response, Error):
self.handle_error_async(response, command_name)
return
self.handle_success_async(response, command_name)
session.execute_command(params, progress=True).then(handle_response)
def handle_success_async(self, result: Any, command_name: str) -> None:
"""
Override this method to handle successful response to workspace/executeCommand.
:param result: The result returned from the server.
:param command_name: The name of the command that was executed.
"""
msg = "command {} completed".format(command_name)
window = self.view.window()
if window:
window.status_message(msg)
def handle_error_async(self, error: Error, command_name: str) -> None:
"""
Override this method to handle failed response to workspace/executeCommand.
:param error: The Error object.
:param command_name: The name of the command that was executed.
"""
sublime.message_dialog("command {} failed. Reason: {}".format(command_name, str(error)))
def _expand_variables(self, command_args: List[Any]) -> List[Any]:
view = self.view # type: sublime.View
region = first_selection_region(view)
for i, arg in enumerate(command_args):
if arg in ["$document_id", "${document_id}"]:
command_args[i] = text_document_identifier(view)
elif arg in ["$file_uri", "${file_uri}"]:
command_args[i] = uri_from_view(view)
elif region is not None:
if arg in ["$selection", "${selection}"]:
command_args[i] = view.substr(region)
elif arg in ["$offset", "${offset}"]:
command_args[i] = region.b
elif arg in ["$selection_begin", "${selection_begin}"]:
command_args[i] = region.begin()
elif arg in ["$selection_end", "${selection_end}"]:
command_args[i] = region.end()
elif arg in ["$position", "${position}"]:
command_args[i] = offset_to_point(view, region.b).to_lsp()
elif arg in ["$line", "${line}"]:
command_args[i] = offset_to_point(view, region.b).row
elif arg in ["$character", "${character}"]:
command_args[i] = offset_to_point(view, region.b).col
elif arg in ["$range", "${range}"]:
command_args[i] = region_to_range(view, region)
elif arg in ["$text_document_position", "${text_document_position}"]:
command_args[i] = text_document_position_params(view, region.b)
window = view.window()
window_variables = window.extract_variables() if window else {}
return sublime.expand_variables(command_args, window_variables)