fix: subscription close before disconnect

nostr/client/client.py

@@ -25,7 +25,7 @@ def reconnect(self, relays):
 
     def close(self):
         try:
-            self.relay_manager.close_subscriptions()
+            self.relay_manager.close_all_subscriptions()
             self.relay_manager.close_connections()
 
             self.running = False

nostr/relay_manager.py

@@ -81,11 +81,14 @@ def close_subscription(self, id: str):
         except Exception as e:
             logger.debug(e)
 
-    def close_subscriptions(self):
-        all_subscriptions = list(self._cached_subscriptions.keys())
-        for id in all_subscriptions:
+    def close_subscriptions(self, subscriptions: List[str]):
+        for id in subscriptions:
             self.close_subscription(id)
 
+    def close_all_subscriptions(self):
+        all_subscriptions = list(self._cached_subscriptions.keys())
+        self.close_subscriptions(all_subscriptions)
+
     def check_and_restart_relays(self):
         stopped_relays = [r for r in self.relays.values() if r.shutdown]
         for relay in stopped_relays:

router.py

@@ -31,7 +31,7 @@ def start(self):
         self.tasks.append(asyncio.create_task(self._client_to_nostr()))
         self.tasks.append(asyncio.create_task(self._nostr_to_client()))
 
-    def stop(self):
+    async def stop(self):
         nostr_client.relay_manager.close_subscriptions(self.subscriptions)
         self.connected = False
 
@@ -42,7 +42,7 @@ def stop(self):
                 pass
 
         try:
-            self.websocket.close()
+            await self.websocket.close()
         except Exception as _:
             pass
 
@@ -53,8 +53,9 @@ async def _client_to_nostr(self):
         while self.connected:
             try:
                 json_str = await self.websocket.receive_text()
-            except WebSocketDisconnect:
-                self.stop()
+            except WebSocketDisconnect as e:
+                logger.debug(e)
+                await self.stop()
                 break
 
             try:

views_api.py

@@ -117,7 +117,7 @@ async def api_test_endpoint(data: TestMessage) -> TestMessageResponse:
 async def api_stop():
     for router in all_routers:
         try:
-            router.stop()
+            await router.stop()
             all_routers.remove(router)
         except Exception as e:
             logger.error(e)
@@ -146,6 +146,6 @@ async def ws_relay(websocket: WebSocket) -> None:
     while router.connected:
         await asyncio.sleep(10)
 
-    router.stop()
+    await router.stop()
     all_routers.remove(router)