Add VSCode extension configuration and tests for selective language service disabling

Co-authored-by: kv9898 <[email protected]>

Author: Copilot

lsp/README.md

@@ -24,13 +24,25 @@ to ensure your project is set up properly, see
 The following configuration options are IDE-specific and exposed as VSCode
 settings:
 
-- `python.pyrefly.disableTypeErrors` [enum: default, force-on, force-off]: by
+- `python.pyrefly.displayTypeErrors` [enum: default, force-on, force-off]: by
   default, Pyrefly will only provide type errors in your project if a
   `pyrefly.toml` is present. Modify this setting to override the IDE
   diagnostics.
 - `python.pyrefly.disableLanguageServices` [boolean: false]: by default, Pyrefly
   will provide both type errors and other language features like go-to
   definition, intellisense, hover, etc. Enable this option to keep type errors
   from Pyrefly unchanged but use VSCode's Python extension for everything else.
+- `python.pyrefly.disabledLanguageServices` [object]: selectively disable
+  individual language services. This is an object with boolean properties for
+  each service: `definition`, `typeDefinition`, `codeAction`, `completion`,
+  `documentHighlight`, `references`, `rename`, `signatureHelp`, `hover`,
+  `inlayHint`, `documentSymbol`, `workspaceSymbol`, and `semanticTokens`.
+  For example, to disable hover and document symbols:
+  ```json
+  "python.pyrefly.disabledLanguageServices": {
+    "hover": true,
+    "documentSymbol": true
+  }
+  ```
 - `pyrefly.lspPath` [string: '']: if your platform is not supported, you can
   build pyrefly from source and specify the binary here.

lsp/package.json

@@ -77,6 +77,78 @@
                     "default": false,
                     "description": "If true, pyrefly will not provide other IDE services like completions, hover, definition, etc. To control type errors, see `python.pyrefly.displayTypeErrors`"
                 },
+                "python.pyrefly.disabledLanguageServices": {
+                    "type": "object",
+                    "default": {},
+                    "description": "Selectively disable individual language services",
+                    "properties": {
+                        "definition": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable go-to-definition"
+                        },
+                        "typeDefinition": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable go-to-type-definition"
+                        },
+                        "codeAction": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable code actions (quick fixes)"
+                        },
+                        "completion": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable completion (autocomplete)"
+                        },
+                        "documentHighlight": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable document highlight"
+                        },
+                        "references": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable find references"
+                        },
+                        "rename": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable rename"
+                        },
+                        "signatureHelp": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable signature help (parameter hints)"
+                        },
+                        "hover": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable hover (tooltips)"
+                        },
+                        "inlayHint": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable inlay hints"
+                        },
+                        "documentSymbol": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable document symbols (outline)"
+                        },
+                        "workspaceSymbol": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable workspace symbols"
+                        },
+                        "semanticTokens": {
+                            "type": "boolean",
+                            "default": false,
+                            "description": "Disable semantic tokens"
+                        }
+                    }
+                },
                 "python.pyrefly.displayTypeErrors": {
                     "type": "string",
                     "description": "If 'default', Pyrefly will only provide type check squiggles in the IDE if your file is covered by a Pyrefly configuration. If 'force-off', Pyrefly will never provide type check squiggles in the IDE. If 'force-on', Pyrefly will always provide type check squiggles in the IDE.",

pyrefly/lib/test/lsp/lsp_interaction/configuration.rs

@@ -825,3 +825,47 @@ fn test_diagnostics_file_in_excludes() {
 
     interaction.shutdown();
 }
+
+#[test]
+fn test_selective_disable_hover() {
+    let test_files_root = get_test_files_root();
+    let scope_uri = Url::from_file_path(test_files_root.path()).unwrap();
+    let mut interaction = LspInteraction::new();
+    interaction.set_root(test_files_root.path().to_path_buf());
+    interaction.initialize(InitializeSettings {
+        workspace_folders: Some(vec![("test".to_owned(), scope_uri.clone())]),
+        configuration: Some(None),
+        ..Default::default()
+    });
+
+    interaction.server.did_open("foo.py");
+    
+    // Now configure to disable hover
+    interaction.server.did_change_configuration();
+
+    interaction
+        .client
+        .expect_configuration_request(2, Some(vec![&scope_uri]));
+    interaction.server.send_configuration_response(
+        2, 
+        serde_json::json!([{
+            "pyrefly": {
+                "disabledLanguageServices": {
+                    "hover": true
+                }
+            }
+        }])
+    );
+
+    // Test that hover is now disabled (returns empty response)
+    interaction.server.hover("foo.py", 6, 16);
+    interaction.client.expect_response(Response {
+        id: RequestId::from(2),
+        result: Some(serde_json::json!({
+            "contents": []
+        })),
+        error: None,
+    });
+
+    interaction.shutdown();
+}