language_server: add context management, and don't terminate process on m_exit()

Author: i80and

snooty/language_server.py

@@ -300,9 +300,15 @@ def m_shutdown(self, **_kwargs: object) -> None:
 
     def m_exit(self, **_kwargs: object) -> None:
         self._endpoint.shutdown()
-        self._jsonrpc_stream_reader.close()
-        self._jsonrpc_stream_writer.close()
-        sys.exit(0)
+        if self.project:
+            self.project.stop_monitoring()
+
+    def __enter__(self) -> "LanguageServer":
+        return self
+
+    def __exit__(self, *args: object) -> None:
+        self.m_shutdown()
+        self.m_exit()
 
 
 def start() -> None: