add support for ruff extension option

pylsp_ruff/plugin.py

@@ -556,6 +556,11 @@ def build_arguments(
             if not PurePath(document_path).match(path):
                 continue
             args.append(f"--ignore={','.join(errors)}")
+    if settings.extension:
+        extension_pairs = ",".join(
+            f"{ext}:{lang}" for ext, lang in settings.extension.items()
+        )
+        args.append(f"--extension={extension_pairs}")
 
     if extra_arguments:
         args.extend(extra_arguments)

pylsp_ruff/settings.py

@@ -27,6 +27,8 @@ class PluginSettings:
 
     severities: Optional[Dict[str, str]] = None
 
+    extension: Optional[Dict[str, str]] = None
+
 
 def to_camel_case(snake_str: str) -> str:
     components = snake_str.split("_")

pyproject.toml

@@ -13,7 +13,7 @@ readme = "README.md"
 requires-python = ">=3.7"
 license = {text = "MIT"}
 dependencies = [
-  "ruff>=0.1.0, <0.2.0",
+  "ruff>=0.1.5, <0.2.0",
   "python-lsp-server",
   "lsprotocol>=2022.0.0a1",
   "tomli>=1.1.0; python_version < '3.11'",

tests/test_ruff_lint.py

@@ -85,6 +85,7 @@ def test_ruff_config_param(workspace):
                         "config": ruff_conf,
                         "extendSelect": ["D", "F"],
                         "extendIgnore": ["E"],
+                        "extension": {"ipynb": "python"},
                     }
                 }
             }
@@ -96,6 +97,7 @@ def test_ruff_config_param(workspace):
         assert f"--config={ruff_conf}" in call_args
         assert "--extend-select=D,F" in call_args
         assert "--extend-ignore=E" in call_args
+        assert "--extension=ipynb:python" in call_args
 
 
 def test_ruff_executable_param(workspace):
@@ -242,3 +244,37 @@ def f():
         assert diag["code"] != "F401"
 
     os.unlink(os.path.join(workspace.root_path, "pyproject.toml"))
+
+
+def test_notebook_input(workspace):
+    doc_str = r"""
+print('hi')
+import os
+def f():
+    a = 2
+"""
+    # attribute the python code to a notebook file name per jupyterlab-lsp
+    doc_uri = uris.from_fs_path(os.path.join(workspace.root_path, "Untitled.ipynb"))
+    workspace.put_document(doc_uri, doc_str)
+    doc = workspace.get_document(doc_uri)
+
+    diags = ruff_lint.pylsp_lint(workspace, doc)
+    # without the extension option, we get a syntax error because ruff expects JSON
+    assert len(diags)
+    assert diags[0]["code"] == "E999"
+
+    workspace._config.update(
+        {
+            "plugins": {
+                "ruff": {
+                    "extension": {"ipynb": "python"},
+                }
+            }
+        }
+    )
+    diags = ruff_lint.pylsp_lint(workspace, doc)
+    diag_codes = [diag["code"] for diag in diags]
+    assert "E999" not in diag_codes
+    assert "E402" in diag_codes
+    assert "F401" in diag_codes
+    assert "F841" in diag_codes