Experimental capabilities hook (python-lsp#225)

gatesn · web-flow · commit 85907d6f7af5 · 2018-01-10T15:30:10.000Z
* Custom method dispatchers

* Capabilities hook

* Capabilities hook
diff --git a/pyls/hookspecs.py b/pyls/hookspecs.py
@@ -57,6 +57,11 @@ def pyls_execute_command(config, workspace, command, arguments):
     pass
 
 
+@hookspec
+def pyls_experimental_capabilities(config, workspace):
+    pass
+
+
 @hookspec(firstresult=True)
 def pyls_format_document(config, workspace, document):
     pass
diff --git a/pyls/python_ls.py b/pyls/python_ls.py
@@ -44,7 +44,7 @@ def _hook(self, hook_name, doc_uri=None, **kwargs):
         return self._hook_caller(hook_name)(config=self.config, workspace=self.workspace, document=doc, **kwargs)
 
     def capabilities(self):
-        return {
+        server_capabilities = {
             'codeActionProvider': True,
             'codeLensProvider': {
                 'resolveProvider': False,  # We may need to make this configurable
@@ -66,8 +66,11 @@ def capabilities(self):
             'signatureHelpProvider': {
                 'triggerCharacters': ['(', ',']
             },
-            'textDocumentSync': lsp.TextDocumentSyncKind.INCREMENTAL
+            'textDocumentSync': lsp.TextDocumentSyncKind.INCREMENTAL,
+            'experimental': merge(self._hook('pyls_experimental_capabilities'))
         }
+        log.info("Server capabilities: %s", server_capabilities)
+        return server_capabilities
 
     def initialize(self, root_uri, init_opts, _process_id):
         self.workspace = Workspace(root_uri, lang_server=self)
@@ -200,3 +203,7 @@ def m_workspace__execute_command(self, command=None, arguments=None):
 
 def flatten(list_of_lists):
     return [item for lst in list_of_lists for item in lst]
+
+
+def merge(list_of_dicts):
+    return {k: v for dictionary in list_of_dicts for k, v in dictionary.items()}
diff --git a/test/test_pyls.py b/test/test_pyls.py
