Add notebook document sync infrastructure for Jupyter support (#664)

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: edvilme

bundled/tool/lsp_server.py

@@ -77,8 +77,28 @@ def update_environ_path() -> None:
 RUNNER = pathlib.Path(__file__).parent / "lsp_runner.py"
 
 MAX_WORKERS = 5
+NOTEBOOK_SYNC_OPTIONS = lsp.NotebookDocumentSyncOptions(
+    notebook_selector=[
+        lsp.NotebookDocumentFilterWithNotebook(
+            notebook="jupyter-notebook",
+            cells=[
+                lsp.NotebookCellLanguage(language="python"),
+            ],
+        ),
+        lsp.NotebookDocumentFilterWithNotebook(
+            notebook="interactive",
+            cells=[
+                lsp.NotebookCellLanguage(language="python"),
+            ],
+        ),
+    ],
+    save=True,
+)
 LSP_SERVER = LanguageServer(
-    name="black-server", version="v0.1.0", max_workers=MAX_WORKERS
+    name="black-server",
+    version="v0.1.0",
+    max_workers=MAX_WORKERS,
+    notebook_document_sync=NOTEBOOK_SYNC_OPTIONS,
 )
 
 

package.json

@@ -48,7 +48,9 @@
         "onLanguage:python",
         "workspaceContains:pyproject.toml",
         "workspaceContains:.black",
-        "workspaceContains:*.py"
+        "workspaceContains:*.py",
+        "onNotebook:jupyter-notebook",
+        "onNotebook:interactive"
     ],
     "main": "./dist/extension.js",
     "l10n": "./l10n",

src/test/python_tests/test_get_cwd.py

@@ -23,10 +23,10 @@ def feature(self, *args, **kwargs):
         def command(self, *args, **kwargs):
             return lambda f: f
 
-        def show_message_log(self, *args, **kwargs):
+        def window_log_message(self, *args, **kwargs):
             pass
 
-        def show_message(self, *args, **kwargs):
+        def window_show_message(self, *args, **kwargs):
             pass
 
     mock_lsp_server = types.ModuleType("pygls.lsp.server")
@@ -48,6 +48,10 @@ def show_message(self, *args, **kwargs):
         "INITIALIZE",
         "EXIT",
         "SHUTDOWN",
+        "NOTEBOOK_DOCUMENT_DID_OPEN",
+        "NOTEBOOK_DOCUMENT_DID_CHANGE",
+        "NOTEBOOK_DOCUMENT_DID_SAVE",
+        "NOTEBOOK_DOCUMENT_DID_CLOSE",
     ]:
         setattr(mock_lsp, _name, _name)
     for _name in [
@@ -56,13 +60,21 @@ def show_message(self, *args, **kwargs):
         "DidCloseTextDocumentParams",
         "DidOpenTextDocumentParams",
         "DidSaveTextDocumentParams",
+        "DidChangeNotebookDocumentParams",
+        "DidCloseNotebookDocumentParams",
+        "DidOpenNotebookDocumentParams",
+        "DidSaveNotebookDocumentParams",
         "DocumentFormattingParams",
         "InitializeParams",
+        "NotebookCellKind",
+        "NotebookCellLanguage",
+        "NotebookDocumentFilterWithNotebook",
+        "NotebookDocumentSyncOptions",
         "Position",
         "Range",
         "TextEdit",
     ]:
-        setattr(mock_lsp, _name, type(_name, (), {}))
+        setattr(mock_lsp, _name, type(_name, (), {"__init__": lambda self, **kw: None}))
     mock_lsp.MessageType = type(
         "MessageType", (), {"Log": 4, "Error": 1, "Warning": 2, "Info": 3}
     )