change on_initialize/on_initialized methods to async_events

Author: d-biehl

robotcode/language_server/common/parts/workspace.py

@@ -211,7 +211,7 @@ async def _workspace_did_change_configuration(self, settings: Dict[str, Any], *a
         await self.did_change_configuration(self, settings)
 
     @async_event
-    async def will_create_files(sender, files: List[str]) -> Mapping[str, TextEdit]:
+    async def will_create_files(sender, files: List[str]) -> Optional[Mapping[str, List[TextEdit]]]:
         ...
 
     @async_event

robotcode/language_server/common/protocol.py

@@ -158,7 +158,7 @@ async def _initialize(
 
         self.initialization_options = initialization_options
         try:
-            await self.on_initialize(initialization_options)
+            await self.on_initialize(self, initialization_options)
         except (asyncio.CancelledError, SystemExit, KeyboardInterrupt):
             raise
         except JsonRPCErrorException:
@@ -177,15 +177,17 @@ async def _initialize(
             ),
         )
 
-    async def on_initialize(self, initialization_options: Optional[Any] = None) -> None:
-        pass
+    @async_event
+    async def on_initialize(sender, initialization_options: Optional[Any] = None) -> None:
+        ...
 
     @rpc_method(name="initialized", param_type=InitializedParams)
     async def _initialized(self, params: InitializedParams) -> None:
-        await self.on_initialized()
+        await self.on_initialized(self)
 
-    async def on_initialized(self) -> None:
-        pass
+    @async_event
+    async def on_initialized(sender) -> None:
+        ...
 
     @rpc_method(name="shutdown")
     @_logger.call

robotcode/language_server/robotframework/protocol.py

@@ -69,9 +69,11 @@ class RobotLanguageServerProtocol(LanguageServerProtocol):
     def __init__(self, server: "RobotLanguageServer"):
         super().__init__(server)
         self.options = Options()
+        super().on_initialize.add(self._on_initialize)
+        super().on_initialized.add(self._on_initialized)
 
     @_logger.call
-    async def on_initialize(self, initialization_options: Optional[Any] = None) -> None:
+    async def _on_initialize(self, sender: Any, initialization_options: Optional[Any] = None) -> None:
         check_robotframework()
 
         if initialization_options is not None:
@@ -80,7 +82,7 @@ async def on_initialize(self, initialization_options: Optional[Any] = None) -> N
         self._logger.info(f"initialized with {repr(self.options)}")
 
     @_logger.call
-    async def on_initialized(self) -> None:
+    async def _on_initialized(self, sender: Any) -> None:
         document_selector = [DocumentFilter(language="python")]
         await self.register_capability(
             str(uuid.uuid4()),