invalidate references if a file change

Author: d-biehl

robotcode/language_server/robotframework/parts/codelens.py

@@ -28,9 +28,13 @@ def __init__(self, parent: RobotLanguageServerProtocol) -> None:
 
         parent.code_lens.collect.add(self.collect)
         parent.code_lens.resolve.add(self.resolve)
+        parent.robot_references.cache_cleared.add(self.robot_references_cache_cleared)
 
         self._running_task: Set[Tuple[TextDocument, KeywordDoc]] = set()
 
+    async def robot_references_cache_cleared(self, sender: Any) -> None:  # NOSONAR
+        await self.parent.code_lens.refresh()
+
     @language_id("robotframework")
     @threaded()
     async def collect(self, sender: Any, document: TextDocument) -> Optional[List[CodeLens]]:

robotcode/language_server/robotframework/parts/references.py

@@ -15,7 +15,7 @@
 )
 
 from ....utils.async_cache import AsyncSimpleLRUCache
-from ....utils.async_tools import threaded
+from ....utils.async_tools import async_tasking_event, threaded
 from ....utils.logging import LoggingDescriptor
 from ....utils.uri import Uri
 from ...common.decorators import language_id
@@ -48,9 +48,17 @@ def __init__(self, parent: RobotLanguageServerProtocol) -> None:
         super().__init__(parent)
 
         parent.references.collect.add(self.collect)
-
+        parent.documents.did_change.add(self.document_did_change)
         self._keyword_reference_cache = AsyncSimpleLRUCache(max_items=128)
 
+    @async_tasking_event
+    async def cache_cleared(sender) -> None:  # NOSONAR
+        ...
+
+    async def document_did_change(self, sender: Any, document: TextDocument) -> None:
+        await self._keyword_reference_cache.clear()
+        await self.cache_cleared(self)
+
     def _find_method(self, cls: Type[Any]) -> Optional[_ReferencesMethod]:
         if cls is ast.AST:
             return None