v0.0.6: Implement basic persist

Author: subnut

.binary_version

@@ -1 +1 @@
-v0.0.4
+v0.0.6

binary.py

@@ -16,7 +16,7 @@
 from simple_websocket_server import WebSocket
 from simple_websocket_server import WebSocketServer
 
-BUILD_VERSION = "v0.0.4"
+BUILD_VERSION = "v0.0.6"
 TEMP_FILEPATH = os.path.join(tempfile.gettempdir(), "nvim-ghost.nvim.port")
 WINDOWS = os.name == "nt"
 LOCALHOST = "127.0.0.1" if WINDOWS else "localhost"
@@ -108,6 +108,8 @@ def __init__(self):
         }
         self.argument_handlers_nodata = {
             "--start-server": self._start,
+            "--nopersist": self._nopersist,
+            "--persist": self._persist,
             "--version": self._version,
             "--focus": self._focus,
             "--help": self._help,
@@ -146,6 +148,16 @@ def _start(self):
         global START_SERVER
         START_SERVER = True
 
+    def _persist(self):
+        global PERSIST
+        PERSIST = True
+        self.server_requests.append("/persist")
+
+    def _nopersist(self):
+        global PERSIST
+        PERSIST = False
+        self.server_requests.append("/nopersist")
+
     def _port(self, port: str):
         if not port.isdigit():
             sys.exit("Invalid port")
@@ -184,6 +196,8 @@ def do_GET(self):
             "/version": self._version_responder,
             "/exit": self._exit_responder,
             "/kill": self._exit_responder,
+            "/persist": self._persist_responder,
+            "/nopersist": self._nopersist_responder,
             "/is_ghost_binary": self._sanityCheck_responder,
         }
 
@@ -224,6 +238,22 @@ def _exit_responder(self):
         global RUNNING
         RUNNING = False
 
+    def _persist_responder(self):
+        global PERSIST
+        PERSIST = True
+        self.send_response(200)
+        self.send_header("Content-Type", "text/plain")
+        self.end_headers()
+        self.wfile.write(f"PERSIST={PERSIST}".encode("utf-8"))
+
+    def _nopersist_responder(self):
+        global PERSIST
+        PERSIST = False
+        self.send_response(200)
+        self.send_header("Content-Type", "text/plain")
+        self.end_headers()
+        self.wfile.write(f"PERSIST={PERSIST}".encode("utf-8"))
+
     def _sanityCheck_responder(self):
         self.send_response(200)
         self.send_header("Content-Type", "text/plain")
@@ -255,9 +285,10 @@ def _session_closed_responder(self, query_string):
         self.end_headers()
         self.wfile.write(address.encode("utf-8"))
         global WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS
-        for item in WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS[address]:
-            item.close()
-        del WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS[address]
+        if WEBSOCKET_PER_BUFFER_PER_NEOVIM_ADDRESS.__contains__(address):
+            for item in WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS[address]:
+                item.close()
+            del WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS[address]
         if not PERSIST and len(WEBSOCKETS_PER_NEOVIM_SOCKET_ADDRESS) == 0:
             global RUNNING
             RUNNING = False