prelim changes to clear out errors in all previous files

Author: m0mosenpai

client/commands/v2/pysa_server.py

@@ -58,6 +58,7 @@ def __init__(
         pyre_arguments: start.Arguments,
         binary_location: str,
         server_identifier: str,
+        file_tracker: dict,
     ) -> None:
         self.input_channel = input_channel
         self.output_channel = output_channel
@@ -68,6 +69,7 @@ def __init__(
         self.pyre_connection = api_connection.PyreConnection(
             Path(self.pyre_arguments.global_root)
         )
+        self.file_tracker = file_tracker
 
     def invalid_model_to_diagnostic(
         self,
@@ -103,9 +105,12 @@ def invalid_models_to_diagnostics(
                 result[Path(model.path)].append(self.invalid_model_to_diagnostic(model))
         return result
 
-    async def update_errors(self, document_path: Path) -> None:
-        # Publishing empty diagnostics to clear errors in VSCode
-        await _publish_diagnostics(self.output_channel, document_path, [])
+    async def update_errors(self) -> None:
+        # Publishing empty diagnostics to clear errors in VSCode and reset self.file_tracker
+        for document_path in self.file_tracker.keys():
+            await _publish_diagnostics(self.output_channel, Path(document_path), [])
+        self.file_tracker = {}
+        # await self.log_and_show_message_to_client(f"All files cleared: {self.file_tracker}")
 
         try:
             model_errors = query.get_invalid_taint_models(self.pyre_connection)
@@ -164,16 +169,22 @@ async def process_open_request(
         self, parameters: lsp.DidOpenTextDocumentParameters
     ) -> None:
         document_path = parameters.text_document.document_uri().to_file_path()
+        self.file_tracker[str(document_path)] = True
+        # await self.log_and_show_message_to_client(f"Tracking Files: {self.file_tracker}")
+
         if document_path is None:
             raise json_rpc.InvalidRequestError(
                 f"Document URI is not a file: {parameters.text_document.uri}"
             )
-        await self.update_errors(document_path)
+        await self.update_errors()
 
     async def process_close_request(
         self, parameters: lsp.DidCloseTextDocumentParameters
     ) -> None:
         document_path = parameters.text_document.document_uri().to_file_path()
+        # self.file_tracker.pop(str(document_path))
+        # await self.log_and_show_message_to_client(f"Closed a file. Tracking: {self.file_tracker}")
+
         if document_path is None:
             raise json_rpc.InvalidRequestError(
                 f"Document URI is not a file: {parameters.text_document.uri}"
@@ -192,7 +203,7 @@ async def process_did_save_request(
             raise json_rpc.InvalidRequestError(
                 f"Document URI is not a file: {parameters.text_document.uri}"
             )
-        await self.update_errors(document_path)
+        await self.update_errors()
 
     async def run(self) -> int:
         while True:
@@ -279,6 +290,7 @@ async def run_persistent(
                 binary_location=binary_location,
                 server_identifier=server_identifier,
                 pyre_arguments=pysa_arguments,
+                file_tracker={},
             )
             return await server.run()
         elif isinstance(initialize_result, InitializationFailure):