Allow attaching to URI schemes other than the 'file' scheme

plugin/core/configurations.py

@@ -1,8 +1,9 @@
 from .logging import debug
 from .types import ClientConfig
-from .typing import Any, Generator, List, Set, Dict
+from .typing import Generator, List, Set, Dict
 from .workspace import enable_in_project, disable_in_project
 import sublime
+import urllib.parse
 
 
 class ConfigManager(object):
@@ -36,18 +37,22 @@ def get_configs(self) -> List[ClientConfig]:
 
     def match_view(self, view: sublime.View, include_disabled: bool = False) -> Generator[ClientConfig, None, None]:
         """
-        Yields configurations matching with the language's document_selector
+        Yields configurations where:
+
+        - the configuration's "selector" matches with the view's base scope, and
+        - the view's URI scheme is an element of the configuration's "schemes".
         """
         try:
+            uri = view.settings().get("lsp_uri")
+            if not isinstance(uri, str):
+                return
+            scheme = urllib.parse.urlparse(uri).scheme
             for config in self.all.values():
-                if config.match_view(view) and (config.enabled or include_disabled):
+                if config.match_view(view, scheme) and (config.enabled or include_disabled):
                     yield config
         except (IndexError, RuntimeError):
             pass
 
-    def is_supported(self, view: Any) -> bool:
-        return any(self.match_view(view))
-
     def update(self) -> None:
         project_settings = (self._window.project_data() or {}).get("settings", {}).get("LSP", {})
         self.all.clear()

plugin/core/sessions.py

@@ -35,7 +35,6 @@
 from .types import method_to_capability
 from .types import SettingsRegistration
 from .typing import Callable, cast, Dict, Any, Optional, List, Tuple, Generator, Type, Protocol, Mapping, Union
-from .url import uri_to_filename
 from .version import __version__
 from .views import COMPLETION_KINDS
 from .views import extract_variables
@@ -316,6 +315,12 @@ class SessionViewProtocol(Protocol):
     listener = None  # type: Any
     session_buffer = None  # type: Any
 
+    def get_uri(self) -> Optional[str]:
+        ...
+
+    def get_language_id(self) -> Optional[str]:
+        ...
+
     def on_capability_added_async(self, registration_id: str, capability_path: str, options: Dict[str, Any]) -> None:
         ...
 
@@ -354,8 +359,12 @@ class SessionBufferProtocol(Protocol):
 
     session = None  # type: Session
     session_views = None  # type: WeakSet[SessionViewProtocol]
-    file_name = None  # type: str
-    language_id = None  # type: str
+
+    def get_uri(self) -> Optional[str]:
+        ...
+
+    def get_language_id(self) -> Optional[str]:
+        ...
 
     def register_capability_async(
         self,
@@ -658,6 +667,12 @@ def on_session_end_async(self) -> None:
 
 
 _plugins = {}  # type: Dict[str, Tuple[Type[AbstractPlugin], SettingsRegistration]]
+_pending_uris_to_set = {}  # type: Dict[str, DocumentUri]
+
+
+def get_possible_uri_to_set(title: str) -> Optional[str]:
+    global _pending_uris_to_set
+    return _pending_uris_to_set.get(title)
 
 
 def _register_plugin_impl(plugin: Type[AbstractPlugin], notify_listener: bool) -> None:
@@ -920,23 +935,25 @@ def session_buffers_async(self) -> Generator[SessionBufferProtocol, None, None]:
         """
         yield from self._session_buffers
 
-    def get_session_buffer_for_uri_async(self, uri: str) -> Optional[SessionBufferProtocol]:
-        file_name = uri_to_filename(uri)
+    def get_session_buffer_for_uri_async(self, uri: DocumentUri) -> Optional[SessionBufferProtocol]:
         for sb in self.session_buffers_async():
-            try:
-                if sb.file_name == file_name or os.path.samefile(file_name, sb.file_name):
-                    return sb
-            except FileNotFoundError:
-                pass
+            if sb.get_uri() == uri:
+                return sb
         return None
 
     # --- capability observers -----------------------------------------------------------------------------------------
 
-    def can_handle(self, view: sublime.View, capability: Optional[str], inside_workspace: bool) -> bool:
-        file_name = view.file_name() or ''
-        if (self.config.match_view(view)
-                and self.state == ClientStates.READY
-                and self.handles_path(file_name, inside_workspace)):
+    def can_handle(self, view: sublime.View, scheme: str, capability: Optional[str], inside_workspace: bool) -> bool:
+        if not self.state == ClientStates.READY:
+            return False
+        if scheme == "file":
+            file_name = view.file_name()
+            if not file_name:
+                # We're closing down
+                return False
+            elif not self.handles_path(file_name, inside_workspace):
+                return False
+        if self.config.match_view(view, scheme):
             # If there's no capability requirement then this session can handle the view
             if capability is None:
                 return True
@@ -980,10 +997,11 @@ def handles_path(self, file_path: Optional[str], inside_workspace: bool) -> bool
         if self._supports_workspace_folders():
             # A workspace-aware language server handles any path, both inside and outside the workspaces.
             return True
+        # buffer views or URI views
+        if not file_path:
+            return True
         # If we end up here then the language server is workspace-unaware. This means there can be more than one
         # language server with the same config name. So we have to actually do the subpath checks.
-        if not file_path:
-            return False
         if not self._workspace_folders or not inside_workspace:
             return True
         for folder in self._workspace_folders:
@@ -1120,6 +1138,8 @@ def open_uri_async(
                 result = Promise.packaged_task()  # type: PackagedTask[bool]
 
                 def open_scratch_buffer(title: str, content: str, syntax: str) -> None:
+                    global _pending_uris_to_set
+                    _pending_uris_to_set[title] = uri
                     v = self.window.new_file(syntax=syntax, flags=flags)
                     v.set_scratch(True)
                     v.set_name(title)

plugin/core/types.py

@@ -13,6 +13,8 @@
 import socket
 import sublime
 import time
+import urllib.parse
+
 
 TCP_CONNECT_TIMEOUT = 5  # seconds
 FEATURES_TIMEOUT = 300  # milliseconds
@@ -279,8 +281,9 @@ def __call__(self, view: sublime.View) -> bool:
             if not syntax or basescope2languageid(syntax.scope) != self.language:
                 return False
         if self.scheme:
-            # Can be "file" or "untitled"?
-            pass
+            uri = view.settings().get("lsp_uri")
+            if isinstance(uri, str) and urllib.parse.urlparse(uri).scheme != self.scheme:
+                return False
         if self.pattern:
             if not globmatch(view.file_name() or "", self.pattern, flags=GLOBSTAR | BRACE):
                 return False
@@ -532,6 +535,7 @@ def __init__(self,
                  name: str,
                  selector: str,
                  priority_selector: Optional[str] = None,
+                 schemes: Optional[List[str]] = None,
                  command: Optional[List[str]] = None,
                  binary_args: Optional[List[str]] = None,  # DEPRECATED
                  tcp_port: Optional[int] = None,
@@ -546,6 +550,10 @@ def __init__(self,
         self.name = name
         self.selector = selector
         self.priority_selector = priority_selector if priority_selector else self.selector
+        if isinstance(schemes, list):
+            self.schemes = schemes  # type: List[str]
+        else:
+            self.schemes = ["file"]
         if isinstance(command, list):
             self.command = command
         else:
@@ -578,6 +586,7 @@ def from_sublime_settings(cls, name: str, s: sublime.Settings, file: str) -> "Cl
             name=name,
             selector=_read_selector(s),
             priority_selector=_read_priority_selector(s),
+            schemes=s.get("schemes"),
             command=read_list_setting(s, "command", []),
             tcp_port=s.get("tcp_port"),
             auto_complete_selector=s.get("auto_complete_selector"),
@@ -598,10 +607,14 @@ def from_dict(cls, name: str, d: Dict[str, Any]) -> "ClientConfig":
             disabled_capabilities = DottedDict(disabled_capabilities)
         else:
             disabled_capabilities = DottedDict()
+        schemes = d.get("schemes")
+        if not isinstance(schemes, list):
+            schemes = ["file"]
         return ClientConfig(
             name=name,
             selector=_read_selector(d),
             priority_selector=_read_priority_selector(d),
+            schemes=schemes,
             command=d.get("command", []),
             tcp_port=d.get("tcp_port"),
             auto_complete_selector=d.get("auto_complete_selector"),
@@ -626,6 +639,7 @@ def from_config(cls, src_config: "ClientConfig", override: Dict[str, Any]) -> "C
             name=src_config.name,
             selector=_read_selector(override) or src_config.selector,
             priority_selector=_read_priority_selector(override) or src_config.priority_selector,
+            schemes=override.get("schemes", src_config.schemes),
             command=override.get("command", src_config.command),
             tcp_port=override.get("tcp_port", src_config.tcp_port),
             auto_complete_selector=override.get("auto_complete_selector", src_config.auto_complete_selector),
@@ -677,15 +691,15 @@ def set_view_status(self, view: sublime.View, message: str) -> None:
     def erase_view_status(self, view: sublime.View) -> None:
         view.erase_status(self.status_key)
 
-    def match_view(self, view: sublime.View) -> bool:
+    def match_view(self, view: sublime.View, scheme: str) -> bool:
         syntax = view.syntax()
-        if syntax:
-            # Every part of a x.y.z scope seems to contribute 8.
-            # An empty selector result in a score of 1.
-            # A non-matching non-empty selector results in a score of 0.
-            # We want to match at least one part of an x.y.z, and we don't want to match on empty selectors.
-            return sublime.score_selector(syntax.scope, self.selector) >= 8
-        return False
+        if not syntax:
+            return False
+        # Every part of a x.y.z scope seems to contribute 8.
+        # An empty selector result in a score of 1.
+        # A non-matching non-empty selector results in a score of 0.
+        # We want to match at least one part of an x.y.z, and we don't want to match on empty selectors.
+        return scheme in self.schemes and sublime.score_selector(syntax.scope, self.selector) >= 8
 
     def map_client_path_to_server_uri(self, path: str) -> str:
         if self.path_maps:

plugin/core/url.py

@@ -1,19 +1,65 @@
+from .typing import Any, Optional
+from urllib.parse import quote
 from urllib.parse import urljoin
 from urllib.parse import urlparse
 from urllib.request import pathname2url
 from urllib.request import url2pathname
 import os
+import re
 
+import sublime
 
-def filename_to_uri(path: str) -> str:
-    return urljoin('file:', pathname2url(path))
+
+def filename_to_uri(file_name: str) -> str:
+    """
+    Convert a file name obtained from view.file_name() into an URI
+    """
+    prefix = sublime.installed_packages_path()
+    if file_name.startswith(prefix):
+        return _to_resource_uri(file_name, prefix)
+    prefix = sublime.packages_path()
+    if file_name.startswith(prefix) and not os.path.exists(file_name):
+        return _to_resource_uri(file_name, prefix)
+    path = pathname2url(file_name)
+    re.sub(r"^([A-Z]):/", _lowercase_driveletter, path)
+    return urljoin("file:", path)
+
+
+def view_to_uri(view: sublime.View, foreign_uri: Optional[str] = None) -> str:
+    if isinstance(foreign_uri, str):
+        return foreign_uri
+    file_name = view.file_name()
+    if not file_name:
+        return "buffer://sublime/{}".format(view.buffer_id())
+    return filename_to_uri(file_name)
 
 
 def uri_to_filename(uri: str) -> str:
+    """
+    DEPRECATED: An URI associated to a view does not necessarily have a "file:" scheme.
+    Use urllib.parse.urlparse to determine the scheme and go from there.
+    Use urllib.parse.unquote to unquote the path.
+    """
     parsed = urlparse(uri)
     assert parsed.scheme == "file"
     if os.name == 'nt':
         # url2pathname does not understand %3A (VS Code's encoding forced on all servers :/)
         return url2pathname(parsed.path).strip('\\')
     else:
         return url2pathname(parsed.path)
+
+
+def _to_resource_uri(path: str, prefix: str) -> str:
+    """
+    Terrible hacks from ST core leak into packages as well.
+
+    See: https://github.com/sublimehq/sublime_text/issues/3742
+    """
+    return "res://Packages{}".format(quote(path[len(prefix):]))
+
+
+def _lowercase_driveletter(match: Any) -> str:
+    """
+    For compatibility with certain other language clients.
+    """
+    return "{}:/".format(match.group(1).lower())