feat: Add extended capabilities to LSP clients

Add a broad set of capability mixins to Pyrefly, Pyright and Typescript clients (notifications, requests,
server_requests and additional server_notification handlers). Use --stdio for PyrightLocalServer and add explicit typing
for initialization options.

Author: observerw

src/lsp_client/clients/pyrefly.py

@@ -9,8 +9,32 @@
 from attrs import define
 from loguru import logger
 
-from lsp_client.capability.request import WithRequestDefinition, WithRequestReferences
-from lsp_client.capability.server_notification import WithReceiveLogMessage
+from lsp_client.capability.notification import (
+    WithNotifyDidChangeConfiguration,
+)
+from lsp_client.capability.request import (
+    WithRequestCallHierarchy,
+    WithRequestDeclaration,
+    WithRequestDefinition,
+    WithRequestDocumentSymbol,
+    WithRequestHover,
+    WithRequestImplementation,
+    WithRequestReferences,
+    WithRequestTypeDefinition,
+    WithRequestWorkspaceSymbol,
+)
+from lsp_client.capability.server_notification import (
+    WithReceiveLogMessage,
+    WithReceiveLogTrace,
+    WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
+)
+from lsp_client.capability.server_request import (
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
+)
 from lsp_client.client.abc import LSPClient
 from lsp_client.server.local import LocalServer
 from lsp_client.utils.types import lsp_type
@@ -21,9 +45,24 @@
 @define
 class PyreflyClient(
     LSPClient,
-    WithRequestReferences,
+    WithNotifyDidChangeConfiguration,
+    WithRequestCallHierarchy,
+    WithRequestDeclaration,
     WithRequestDefinition,
+    WithRequestDocumentSymbol,
+    WithRequestHover,
+    WithRequestImplementation,
+    WithRequestReferences,
+    WithRequestTypeDefinition,
+    WithRequestWorkspaceSymbol,
     WithReceiveLogMessage,
+    WithReceiveLogTrace,
+    WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
 ):
     """
     - Language: Python
@@ -45,7 +84,7 @@ def get_language_id(self) -> lsp_type.LanguageKind:
 
     @override
     def create_initialization_options(self) -> dict[str, Any]:
-        options = {}
+        options: dict[str, Any] = {}
 
         options["trace"] = {"server": self.trace_server}
         options["diagnostic_mode"] = self.diagnostic_mode

src/lsp_client/clients/pyright.py

@@ -9,15 +9,37 @@
 from attrs import define
 from loguru import logger
 
-from lsp_client.capability.request import WithRequestDefinition, WithRequestReferences
-from lsp_client.capability.server_notification import WithReceivePublishDiagnostics
-from lsp_client.capability.server_notification.log_message import WithReceiveLogMessage
+from lsp_client.capability.notification import (
+    WithNotifyDidChangeConfiguration,
+)
+from lsp_client.capability.request import (
+    WithRequestCallHierarchy,
+    WithRequestDeclaration,
+    WithRequestDefinition,
+    WithRequestDocumentSymbol,
+    WithRequestHover,
+    WithRequestReferences,
+    WithRequestTypeDefinition,
+    WithRequestWorkspaceSymbol,
+)
+from lsp_client.capability.server_notification import (
+    WithReceiveLogMessage,
+    WithReceiveLogTrace,
+    WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
+)
+from lsp_client.capability.server_request import (
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
+)
 from lsp_client.client.abc import LSPClient
 from lsp_client.server.docker import DockerServer
 from lsp_client.server.local import LocalServer
 from lsp_client.utils.types import lsp_type
 
-PyrightLocalServer = partial(LocalServer, command=["pyright-langserver"])
+PyrightLocalServer = partial(LocalServer, command=["pyright-langserver", "--stdio"])
 PyrightDockerServer = partial(
     DockerServer, image="docker.io/lspcontainers/pyright-langserver"
 )
@@ -26,10 +48,23 @@
 @define
 class PyrightClient(
     LSPClient,
-    WithRequestReferences,
+    WithNotifyDidChangeConfiguration,
+    WithRequestCallHierarchy,
+    WithRequestDeclaration,
     WithRequestDefinition,
+    WithRequestDocumentSymbol,
+    WithRequestHover,
+    WithRequestReferences,
+    WithRequestTypeDefinition,
+    WithRequestWorkspaceSymbol,
     WithReceiveLogMessage,
+    WithReceiveLogTrace,
     WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
 ):
     """
     - Language: Python

src/lsp_client/clients/typescript.py

@@ -9,6 +9,9 @@
 from attrs import define
 from loguru import logger
 
+from lsp_client.capability.notification import (
+    WithNotifyDidChangeConfiguration,
+)
 from lsp_client.capability.request import (
     WithRequestDefinition,
     WithRequestDocumentSymbol,
@@ -19,9 +22,17 @@
     WithRequestWorkspaceSymbol,
 )
 from lsp_client.capability.server_notification import (
+    WithReceiveLogTrace,
     WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
 )
 from lsp_client.capability.server_notification.log_message import WithReceiveLogMessage
+from lsp_client.capability.server_request import (
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
+)
 from lsp_client.client.abc import LSPClient
 from lsp_client.server.docker import DockerServer
 from lsp_client.server.local import LocalServer
@@ -36,6 +47,7 @@
 @define
 class TypescriptClient(
     LSPClient,
+    WithNotifyDidChangeConfiguration,
     WithRequestHover,
     WithRequestDefinition,
     WithRequestReferences,
@@ -44,7 +56,13 @@ class TypescriptClient(
     WithRequestDocumentSymbol,
     WithRequestWorkspaceSymbol,
     WithReceiveLogMessage,
+    WithReceiveLogTrace,
     WithReceivePublishDiagnostics,
+    WithReceiveShowMessage,
+    WithRespondConfigurationRequest,
+    WithRespondShowDocumentRequest,
+    WithRespondShowMessageRequest,
+    WithRespondWorkspaceFoldersRequest,
 ):
     """
     - Language: TypeScript, JavaScript