fix: start diagnostics only when the language server is fully initialized

Author: d-biehl

robotcode/language_server/common/parts/diagnostics.py

@@ -27,6 +27,7 @@
 from ..lsp_types import (
     Diagnostic,
     DiagnosticOptions,
+    DiagnosticServerCancellationData,
     DocumentDiagnosticParams,
     DocumentDiagnosticReport,
     ErrorCodes,
@@ -487,6 +488,11 @@ async def _text_document_diagnostic(
         **kwargs: Any,
     ) -> DocumentDiagnosticReport:
         try:
+            if not self.parent.is_initialized:
+                raise JsonRPCErrorException(
+                    ErrorCodes.SERVER_CANCELLED, "Server not initialized.", DiagnosticServerCancellationData(True)
+                )
+
             document = await self.parent.documents.get(text_document.uri)
             if document is None:
                 raise JsonRPCErrorException(ErrorCodes.SERVER_CANCELLED, f"Document {text_document!r} not found.")

robotcode/language_server/common/protocol.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import asyncio
-from typing import Any, List, Optional, Set, Union, cast
+from typing import Any, List, NamedTuple, Optional, Set, Union, cast
 
 from ...jsonrpc2.protocol import (
     JsonRPCErrorException,
@@ -70,6 +70,12 @@ class LanguageServerException(JsonRPCException):
     pass
 
 
+class LanguageDefinition(NamedTuple):
+    id: str
+    extensions: List[str]
+    aliases: Optional[List[str]] = None
+
+
 class LanguageServerProtocol(JsonRPCProtocol):
 
     _logger = LoggingDescriptor()
@@ -103,6 +109,7 @@ class LanguageServerProtocol(JsonRPCProtocol):
     version: Optional[str] = None
 
     file_extensions: Set[str] = set()
+    languages: List[LanguageDefinition] = []
 
     def __init__(self, server: JsonRPCServer[Any]):
         super().__init__()
@@ -125,6 +132,7 @@ def __init__(self, server: JsonRPCServer[Any]):
         )
 
         self._trace = TraceValue.OFF
+        self._is_initialized = False
 
     @async_event
     async def on_shutdown(sender) -> None:  # pragma: no cover, NOSONAR
@@ -236,6 +244,12 @@ async def on_initialize(sender, initialization_options: Optional[Any] = None) ->
     async def _initialized(self, params: InitializedParams, *args: Any, **kwargs: Any) -> None:
         await self.on_initialized(self)
 
+        self._is_initialized = True
+
+    @property
+    def is_initialized(self) -> bool:
+        return self._is_initialized
+
     @async_event
     async def on_initialized(sender) -> None:  # pragma: no cover, NOSONAR
         ...